1 Star 0 Fork 40

zheng-huanhuan / mindarmour

forked from MindSpore / mindarmour 
Create your Gitee Account
Explore and code with more than 6 million developers,Free private repositories !:)
Sign up
Clone or download
setup.py 3.24 KB
Copy Edit Web IDE Raw Blame History
pkuliuliu authored 2020-08-20 17:21 . 0.7.0 release
# Copyright 2019 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.
import os
import stat
from setuptools import find_packages
from setuptools import setup
from setuptools.command.egg_info import egg_info
from setuptools.command.build_py import build_py
version = '0.7.0'
cur_dir = os.path.dirname(os.path.realpath(__file__))
pkg_dir = os.path.join(cur_dir, 'build')
try:
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
class bdist_wheel(_bdist_wheel):
def finalize_options(self):
_bdist_wheel.finalize_options(self)
self.root_is_pure = False
except ImportError:
bdist_wheel = None
def write_version(file):
file.write("__version__ = '{}'\n".format(version))
def build_depends():
"""generate python file"""
version_file = os.path.join(cur_dir, 'mindarmour/', 'version.py')
with open(version_file, 'w') as f:
write_version(f)
build_depends()
def update_permissions(path):
"""
Update permissions.
Args:
path (str): Target directory path.
"""
for dirpath, dirnames, filenames in os.walk(path):
for dirname in dirnames:
dir_fullpath = os.path.join(dirpath, dirname)
os.chmod(dir_fullpath, stat.S_IREAD | stat.S_IWRITE | stat.S_IEXEC | stat.S_IRGRP | stat.S_IXGRP)
for filename in filenames:
file_fullpath = os.path.join(dirpath, filename)
os.chmod(file_fullpath, stat.S_IREAD)
class EggInfo(egg_info):
"""Egg info."""
def run(self):
super().run()
egg_info_dir = os.path.join(cur_dir, 'mindarmour.egg-info')
update_permissions(egg_info_dir)
class BuildPy(build_py):
"""BuildPy."""
def run(self):
super().run()
mindarmour_dir = os.path.join(pkg_dir, 'lib', 'mindarmour')
update_permissions(mindarmour_dir)
setup(
name='mindarmour',
version=version,
author='The MindSpore Authors',
author_email='contact@mindspore.cn',
url='https://www.mindspore.cn/',
download_url='https://gitee.com/mindspore/mindarmour/tags',
project_urls={
'Sources': 'https://gitee.com/mindspore/mindarmour',
'Issue Tracker': 'https://gitee.com/mindspore/mindarmour/issues',
},
description="A smart AI security and trustworthy tool box.",
license='Apache 2.0',
packages=find_packages(),
include_package_data=True,
cmdclass={
'egg_info': EggInfo,
'build_py': BuildPy,
'bdist_wheel': bdist_wheel
},
install_requires=[
'scipy >= 1.3.3',
'numpy >= 1.17.0',
'matplotlib >= 3.2.1',
'Pillow >= 2.0.0',
'scikit-learn >= 0.21.2'
],
classifiers=[
'License :: OSI Approved :: Apache Software License'
]
)
print(find_packages())

Comment ( 0 )

Sign in for post a comment

Python
1
https://git.oschina.net/Tinrry/mindarmour.git
git@git.oschina.net:Tinrry/mindarmour.git
Tinrry
mindarmour
mindarmour
master

Search

103611 48b8ff67 1899542 103622 4d02230c 1899542