diff --git a/src/mindspore2022/setup.py b/src/mindspore2022/setup.py index 63d69b00..092a4f28 100644 --- a/src/mindspore2022/setup.py +++ b/src/mindspore2022/setup.py @@ -23,42 +23,53 @@ from setuptools import setup, find_packages from setuptools.command.egg_info import egg_info from setuptools.command.build_py import build_py +# 获取环境变量 backend_policy = os.getenv('BACKEND_POLICY') device_target = os.getenv('BACKEND_TARGET') commit_id = os.getenv('COMMIT_ID').replace("\n", "") package_name = os.getenv('MS_PACKAGE_NAME').replace("\n", "") build_path = os.getenv('BUILD_PATH') +# 获取当前文件路径 pwd = os.path.dirname(os.path.realpath(__file__)) +# 获取包目录路径 pkg_dir = os.path.join(build_path, 'package') def _read_file(filename): + """读取文件内容""" with open(os.path.join(pwd, filename), encoding='UTF-8') as f: return f.read() +# 读取版本号 version = _read_file('version.txt').replace("\n", "") +# 读取README.md文件内容 readme = _read_file('README.md') def _write_version(file): + """写入版本号""" file.write("__version__ = '{}'\n".format(version)) def _write_config(file): + """写入后端策略""" file.write("__backend__ = '{}'\n".format(backend_policy)) def _write_commit_file(file): + """写入commit_id""" file.write("__commit_id__ = '{}'\n".format(commit_id)) def _write_package_name(file): + """写入包名""" file.write("__package_name__ = '{}'\n".format(package_name)) def _write_device_target(file): + """写入设备目标""" file.write("__device_target__ = '{}'\n".format(device_target))