From 281e5df55561f4a6f48f6e1d498a47220a01044d Mon Sep 17 00:00:00 2001
From: shenjack <3695888@qq.com>
Date: Mon, 23 Jan 2023 17:17:39 +0800
Subject: [PATCH] update setup [page skip]

---
 libs/Difficult_Rocket_rs/src/setup.py        | 18 ++++++++++++++++--
 libs/Difficult_Rocket_rs/src/src/__init__.py | 11 +++++++++++
 2 files changed, 27 insertions(+), 2 deletions(-)
 create mode 100644 libs/Difficult_Rocket_rs/src/src/__init__.py

diff --git a/libs/Difficult_Rocket_rs/src/setup.py b/libs/Difficult_Rocket_rs/src/setup.py
index b42a229..270289b 100644
--- a/libs/Difficult_Rocket_rs/src/setup.py
+++ b/libs/Difficult_Rocket_rs/src/setup.py
@@ -7,6 +7,7 @@ import os
 import sys
 import shutil
 import warnings
+import traceback
 from setuptools import setup
 from setuptools_rust import Binding, RustExtension
 
@@ -28,7 +29,7 @@ setup(
 )
 
 lib_path = '../lib'
-build_path = './build'
+build_path = 'build'
 
 if 'clean' in sys.argv:
     shutil.rmtree(lib_path, ignore_errors=True)
@@ -42,6 +43,11 @@ if not os.path.exists(lib_path):
 builds = os.listdir(build_path)
 print(os.path.abspath('.'))
 
+try:
+    shutil.copy('src/__init__.py', os.path.join(lib_path, '__init__.py'))
+except shutil.SameFileError:
+    traceback.print_exc()
+
 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}')
@@ -50,6 +56,14 @@ for build_dir in builds:
         # file_name = os.path.join(lib_path, file.replace(package_path, f'{package_path}.{DR_runtime.DR_Rust_version}'))
         file_name = os.path.join(lib_path, file)
         shutil.rmtree(file_name, ignore_errors=True)
-        shutil.copy(os.path.join(build_path, build_dir, package_path, file), file_name)
+        try:
+            shutil.copy(os.path.join(build_path, build_dir, package_path, file), file_name)
+        except (shutil.SameFileError, PermissionError):
+            # print(os.path.exists(os.path))
+            print(os.listdir(lib_path))
+            traceback.print_exc()
+            continue
+            os.remove(file_name)
+            shutil.copy(os.path.join(build_path, build_dir, package_path, file), file_name)
     # shutil.rmtree(os.path.join(build_path, build_dir))
     # print(os.path.join(build_path, build_dir))
diff --git a/libs/Difficult_Rocket_rs/src/src/__init__.py b/libs/Difficult_Rocket_rs/src/src/__init__.py
new file mode 100644
index 0000000..fbed5c0
--- /dev/null
+++ b/libs/Difficult_Rocket_rs/src/src/__init__.py
@@ -0,0 +1,11 @@
+#  -------------------------------
+#  Difficult Rocket
+#  Copyright © 2021-2022 by shenjackyuanjie 3695888@qq.com
+#  All rights reserved
+#  -------------------------------
+
+
+from typing import TYPE_CHECKING
+
+if not TYPE_CHECKING:
+    from .Difficult_Rocket_rs import *