chore: fix cython compile script (#396)

* chore: fix cython compile script

* chore: fix cython compile script

core/src/ten_runtime/binding/python/tools/cython_compiler.py

@@ -10,16 +10,21 @@
 # step1. replace the 'source_file_list' with the actual source files.
 # step2. run the following command:
 #
-#        python3 cython_compiler.py build_ext --inplace
+#        python3 cython_compiler.py -r <compile_root_dir>
 #
-# step3. the compiled dynamic library will be generated in __current__
-#        directory.
+# step3. the compiled dynamic library will be generated in <compile_root_dir>.
 
+import argparse
 import os
 import glob
 from setuptools import Extension, setup
 
 
+class ArgumentInfo(argparse.Namespace):
+    def __init__(self):
+        self.compile_root_dir: str
+
+
 def install_cython_if_needed():
     try:
         from Cython.Build import cythonize
@@ -36,6 +41,18 @@ def install_cython_if_needed():
 
 
 if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Cython compiler")
+    parser.add_argument(
+        "-r",
+        "--compile-root-dir",
+        type=str,
+        required=False,
+        default=os.path.dirname(os.path.abspath(__file__)),
+    )
+
+    arg_info = ArgumentInfo()
+    args = parser.parse_args(namespace=arg_info)
+
     install_rc = install_cython_if_needed()
     if not install_rc:
         print("Failed to install Cython")
@@ -44,13 +61,12 @@ def install_cython_if_needed():
     from Cython.Build import cythonize
 
     # Get the directory name of the directory.
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    dir_name = os.path.basename(current_dir)
+    dir_name = os.path.basename(args.compile_root_dir)
 
     # Recursively find all .pyx files in the current directory and its
     # subdirectories.
     pyx_files = glob.glob(
-        os.path.join(current_dir, "**", "*.pyx"), recursive=True
+        os.path.join(args.compile_root_dir, "**", "*.pyx"), recursive=True
     )
 
     if not pyx_files:
@@ -61,7 +77,7 @@ def install_cython_if_needed():
     for pyx_file in pyx_files:
         # Calculate the relative module name by removing the base directory and
         # extension.
-        rel_path = os.path.relpath(pyx_file, current_dir)
+        rel_path = os.path.relpath(pyx_file, args.compile_root_dir)
         module_name = rel_path.replace(os.path.sep, ".").rsplit(".", 1)[0]
 
         # Create an Extension object for each .pyx file.
@@ -74,5 +90,5 @@ def install_cython_if_needed():
         # The package_dir parameter is used to where the packed package will be
         # placed. In this case, the package will be placed in the current
         # directory.
-        package_dir={dir_name: ""},
+        package_dir={dir_name: args.compile_root_dir},
     )