setup.py 5.7 KB
Newer Older
Z
zhunaipan 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
#!/usr/bin/env python3
# encoding: utf-8
# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
L
leonwanghui 已提交
17
"""setup package."""
Z
zhunaipan 已提交
18 19
import os
import stat
20
import platform
L
leonwanghui 已提交
21

Z
zhunaipan 已提交
22 23 24 25
from setuptools import setup, find_packages
from setuptools.command.egg_info import egg_info
from setuptools.command.build_py import build_py

26
version = '0.7.0'
Z
zhunaipan 已提交
27 28 29

backend_policy = os.getenv('BACKEND_POLICY')
commit_id = os.getenv('COMMIT_ID').replace("\n", "")
30
package_name = os.getenv('MS_PACKAGE_NAME').replace("\n", "")
Z
zhunaipan 已提交
31 32 33 34

pwd = os.path.dirname(os.path.realpath(__file__))
pkg_dir = os.path.join(pwd, 'build/package')

L
leonwanghui 已提交
35 36

def _read_file(filename):
37
    with open(os.path.join(pwd, filename), encoding='UTF-8') as f:
L
leonwanghui 已提交
38 39 40 41 42 43 44 45
        return f.read()


readme = _read_file('README.md')
release = _read_file('RELEASE.md')


def _write_version(file):
Z
zhunaipan 已提交
46 47
    file.write("__version__ = '{}'\n".format(version))

L
leonwanghui 已提交
48 49

def _write_config(file):
Z
zhunaipan 已提交
50 51
    file.write("__backend__ = '{}'\n".format(backend_policy))

L
leonwanghui 已提交
52 53

def _write_commit_file(file):
Z
zhunaipan 已提交
54 55
    file.write("__commit_id__ = '{}'\n".format(commit_id))

L
leonwanghui 已提交
56 57

def build_dependencies():
Z
zhunaipan 已提交
58
    """generate python file"""
L
leonwanghui 已提交
59
    version_file = os.path.join(pkg_dir, 'mindspore', 'version.py')
Z
zhunaipan 已提交
60
    with open(version_file, 'w') as f:
L
leonwanghui 已提交
61
        _write_version(f)
Z
zhunaipan 已提交
62

L
leonwanghui 已提交
63
    version_file = os.path.join(pwd, 'mindspore', 'version.py')
Z
zhunaipan 已提交
64
    with open(version_file, 'w') as f:
L
leonwanghui 已提交
65
        _write_version(f)
Z
zhunaipan 已提交
66

L
leonwanghui 已提交
67
    config_file = os.path.join(pkg_dir, 'mindspore', 'default_config.py')
Z
zhunaipan 已提交
68
    with open(config_file, 'w') as f:
L
leonwanghui 已提交
69
        _write_config(f)
Z
zhunaipan 已提交
70

L
leonwanghui 已提交
71
    config_file = os.path.join(pwd, 'mindspore', 'default_config.py')
Z
zhunaipan 已提交
72
    with open(config_file, 'w') as f:
L
leonwanghui 已提交
73
        _write_config(f)
Z
zhunaipan 已提交
74

L
leonwanghui 已提交
75
    commit_file = os.path.join(pkg_dir, 'mindspore', '.commit_id')
Z
zhunaipan 已提交
76
    with open(commit_file, 'w') as f:
L
leonwanghui 已提交
77
        _write_commit_file(f)
Z
zhunaipan 已提交
78

L
leonwanghui 已提交
79
    commit_file = os.path.join(pwd, 'mindspore', '.commit_id')
Z
zhunaipan 已提交
80
    with open(commit_file, 'w') as f:
L
leonwanghui 已提交
81 82
        _write_commit_file(f)

Z
zhunaipan 已提交
83

L
leonwanghui 已提交
84 85 86 87 88 89 90 91 92 93 94
build_dependencies()

required_package = [
    'numpy >= 1.17.0',
    'protobuf >= 3.8.0',
    'asttokens >= 1.1.13',
    'pillow >= 6.2.0',
    'scipy == 1.3.3',
    'easydict >= 1.9',
    'sympy >= 1.4',
    'cffi >= 1.13.2',
95 96
    'decorator >= 4.4.0',
    'astunparse >= 1.6.3'
L
leonwanghui 已提交
97 98 99
]

package_data = {
Z
zhunaipan 已提交
100 101
    '': [
        '*.so*',
102 103
        '*.pyd',
        '*.dll',
Z
zhunaipan 已提交
104 105 106
        'lib/*.so*',
        'lib/*.a',
        '.commit_id',
107
        'ms_serving'
Z
zhunaipan 已提交
108 109 110 111 112 113 114 115 116 117 118
    ]
}


def update_permissions(path):
    """
    Update permissions.

    Args:
        path (str): Target directory path.
    """
119 120 121
    if platform.system() == "Windows":
        return

Z
zhunaipan 已提交
122 123 124
    for dirpath, dirnames, filenames in os.walk(path):
        for dirname in dirnames:
            dir_fullpath = os.path.join(dirpath, dirname)
L
leonwanghui 已提交
125 126
            os.chmod(dir_fullpath, stat.S_IREAD | stat.S_IWRITE |
                     stat.S_IEXEC | stat.S_IRGRP | stat.S_IXGRP)
Z
zhunaipan 已提交
127 128 129
        for filename in filenames:
            file_fullpath = os.path.join(dirpath, filename)
            os.chmod(file_fullpath, stat.S_IREAD)
130 131
            if filename == "ms_serving":
                os.chmod(file_fullpath, stat.S_IREAD | stat.S_IEXEC)
Z
zhunaipan 已提交
132

L
leonwanghui 已提交
133

Z
zhunaipan 已提交
134 135
class EggInfo(egg_info):
    """Egg info."""
L
leonwanghui 已提交
136

Z
zhunaipan 已提交
137 138 139 140 141
    def run(self):
        super().run()
        egg_info_dir = os.path.join(pkg_dir, 'mindspore.egg-info')
        update_permissions(egg_info_dir)

L
leonwanghui 已提交
142

Z
zhunaipan 已提交
143 144
class BuildPy(build_py):
    """BuildPy."""
L
leonwanghui 已提交
145

Z
zhunaipan 已提交
146 147 148 149
    def run(self):
        super().run()
        mindspore_dir = os.path.join(pkg_dir, 'build', 'lib', 'mindspore')
        update_permissions(mindspore_dir)
150
        mindspore_dir = os.path.join(pkg_dir, 'build', 'lib', 'akg')
Z
zhunaipan 已提交
151 152
        update_permissions(mindspore_dir)

L
leonwanghui 已提交
153

Z
zhunaipan 已提交
154 155 156
setup(
    name=package_name,
    version=version,
L
leonwanghui 已提交
157 158 159 160 161 162 163 164 165 166 167
    author='The MindSpore Authors',
    author_email='contact@mindspore.cn',
    url='https://www.mindspore.cn',
    download_url='https://gitee.com/mindspore/mindspore/tags',
    project_urls={
        'Sources': 'https://gitee.com/mindspore/mindspore',
        'Issue Tracker': 'https://gitee.com/mindspore/mindspore/issues',
    },
    description='MindSpore is a new open source deep learning training/inference '
    'framework that could be used for mobile, edge and cloud scenarios.',
    long_description="\n\n".join([readme, release]),
168
    long_description_content_type="text/markdown",
Z
zhunaipan 已提交
169
    packages=find_packages(),
L
leonwanghui 已提交
170
    package_data=package_data,
Z
zhunaipan 已提交
171 172 173 174 175
    include_package_data=True,
    cmdclass={
        'egg_info': EggInfo,
        'build_py': BuildPy,
    },
L
leonwanghui 已提交
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193
    python_requires='>=3.7',
    install_requires=required_package,
    classifiers=[
        'Development Status :: 4 - Beta',
        'Environment :: Console',
        'Intended Audience :: Science/Research',
        'Intended Audience :: Developers',
        'License :: OSI Approved :: Apache Software License',
        'Programming Language :: Python :: 3 :: Only',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: C++',
        'Topic :: Scientific/Engineering',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'Topic :: Software Development',
        'Topic :: Software Development :: Libraries',
        'Topic :: Software Development :: Libraries :: Python Modules',
    ],
Z
zhunaipan 已提交
194
    license='Apache 2.0',
L
leonwanghui 已提交
195
    keywords='mindspore machine learning',
Z
zhunaipan 已提交
196
)
新手
引导
客服 返回
顶部