Difficult-Rocket/libs/Difficult_Rocket_rs/setup.py
2023-01-20 20:20:35 +08:00

57 lines
1.7 KiB
Python

# -------------------------------
# Difficult Rocket
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
# All rights reserved
# -------------------------------
import os
import sys
import shutil
import warnings
from setuptools import setup
from setuptools_rust import Binding, RustExtension
sys.path.append('../../')
if '../../' in sys.path:
from Difficult_Rocket import DR_option
package_path = 'Difficult_Rocket_rs'
setup(
name='Difficult_Rocket_rs',
version='0.0.0.1',
author='shenjackyuanjie',
author_email='3695888@qq.com',
license='CC BY-NC-SA 4.0',
rust_extensions=[RustExtension(target="Difficult_Rocket_rs.Difficult_Rocket_rs",
# rust_version='2021',
binding=Binding.PyO3)],
packages=[package_path],
zip_safe=False
)
lib_path = './lib'
build_path = './build'
if 'clean' in sys.argv:
shutil.rmtree(lib_path, ignore_errors=True)
shutil.rmtree(build_path, ignore_errors=True)
sys.exit(0)
if not os.path.exists(lib_path):
os.mkdir(lib_path)
builds = os.listdir(build_path)
for build_dir in builds:
if not os.path.exists(os.path.join(build_path, build_dir, package_path)):
warnings.warn(f'package not found at {build_path}/{build_dir}')
continue
for file in os.listdir(os.path.join(build_path, build_dir, package_path)):
shutil.copy(os.path.join(build_path, build_dir, package_path, file), os.path.join(lib_path, file))
shutil.rmtree(os.path.join(build_path, build_dir))
# print(os.path.join(build_path, build_dir))
for package_file in os.listdir(package_path):
shutil.copy(os.path.join(package_path, package_file), os.path.join(lib_path, package_file))