AutoCython-zhang 2.2.1__tar.gz → 2.3.1__tar.gz

autocython_zhang-2.3.1/AutoCython/AutoCython.py

@@ -0,0 +1,44 @@
+import os
+import sys
+from .run_tasks import run_tasks
+from .compile import compile_to_binary
+from .tools import parse_arguments, find_python_files
+from .tools import show_no_compilable_files, show_file_not_found, show_path_not_found
+
+def compile():
+    try:
+        args = parse_arguments()
+        obfuscate_seed = args.seed
+
+        if args.file:
+            if os.path.isfile(args.file):
+                compile_file = args.file
+                del_source = args.del_source
+                tasks = [
+                    # 函数, 位置参数, 关键字参数
+                    (compile_to_binary, compile_file, (compile_file, del_source, True, obfuscate_seed), {}),
+                ]
+                run_tasks(tasks, max_workers=1, raise_on_failure=True) # 执行编译
+            else:
+                show_file_not_found(args.file)
+        elif args.path:
+            if os.path.exists(args.path) and not os.path.isfile(args.path):
+                compile_file_list = find_python_files(args.path)
+                if compile_file_list:
+                    del_source = args.del_source
+                    tasks = []
+                    for compile_file in compile_file_list:
+                        tasks.append(
+                            # 函数, 位置参数, 关键字参数
+                            (compile_to_binary, compile_file, (compile_file, del_source, True, obfuscate_seed), {}),
+                        )
+                    run_tasks(tasks, max_workers=args.conc, raise_on_failure=True) # 执行编译
+                else:
+                    show_no_compilable_files(args.path)
+            else:
+                show_path_not_found(args.path)
+    except KeyboardInterrupt:
+        sys.exit(130)
+    except Exception as e:
+        print(f"Error: {e}", file=sys.stderr)
+        sys.exit(1)

{autocython_zhang-2.2.1 → autocython_zhang-2.3.1}/AutoCython/_version.py

@@ -2,4 +2,4 @@
 # -*- coding: utf-8 -*-
 """版本号模块 - 单一版本号来源"""
 
-__version__ = "2.2.1"
+__version__ = "2.3.1"

{autocython_zhang-2.2.1 → autocython_zhang-2.3.1}/AutoCython/compile.py

@@ -2,21 +2,37 @@ import os
 import sys
 import glob
 import shutil
+import platform
 import tempfile
 import subprocess
 
+from .obfuscate import obfuscate_source
+
+
+def _strip_binary(path):
+    """strip 符号表，失败静默"""
+    try:
+        if sys.platform.startswith('win'):
+            return
+        cmd = ['strip', '-x' if sys.platform == 'darwin' else '--strip-all', path]
+        subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    except Exception:
+        pass
+
 def get_platform_extension() -> str:
     """返回当前平台的扩展名"""
     if sys.platform.startswith('win'):
         return '.pyd'
     return '.so'
 
-def compile_to_binary(file_path: str, del_source=False):
+def compile_to_binary(file_path: str, del_source=False, obfuscate=True, obfuscate_seed=None):
     """
     将指定的 Python 文件（.py）通过 Cython 编译为二进制扩展文件
 
     :param file_path: Python 文件路径（可以是相对路径或绝对路径）
     :param del_source: 是否删除源代码
+    :param obfuscate: 是否在编译前混淆源码（默认True）
+    :param obfuscate_seed: 混淆随机种子（None 表示不固定）
     :return: 生成的二进制文件路径（与输入保持相同的路径类型）
     """
     # 保存原始路径类型（相对/绝对）
@@ -36,6 +52,10 @@ def compile_to_binary(file_path: str, del_source=False):
     if ext != ".py":
         raise ValueError(f"ValueError: The file {file_path} is not a valid Python file (.py)!")
 
+    # 连字符文件名转下划线（Cython 模块名不允许连字符）
+    safe_module_name = module_name.replace('-', '_')
+    safe_file_name = safe_module_name + ext
+
     # 获取平台特定扩展名
     target_ext = get_platform_extension()
 
@@ -43,10 +63,21 @@ def compile_to_binary(file_path: str, del_source=False):
     temp_dir = tempfile.mkdtemp()
 
     try:
-        # 将目标文件复制到临时目录
-        temp_file_path = os.path.join(temp_dir, file_name)
+        # 将目标文件复制到临时目录（使用安全文件名）
+        temp_file_path = os.path.join(temp_dir, safe_file_name)
         shutil.copy2(abs_file_path, temp_file_path)
 
+        # 混淆源码（失败则显式抛错，避免静默降级）
+        if obfuscate:
+            try:
+                with open(temp_file_path, 'r', encoding='utf-8') as f:
+                    original_source = f.read()
+                obfuscated = obfuscate_source(original_source, seed=obfuscate_seed)
+                with open(temp_file_path, 'w', encoding='utf-8') as f:
+                    f.write(obfuscated)
+            except Exception as exc:
+                raise RuntimeError(f"Obfuscation failed for {file_path}: {exc}") from exc
+
         # 创建临时的 setup.py 文件
         setup_code = f"""
 from setuptools import setup
@@ -58,13 +89,13 @@ compiler_directives = {{
     'annotation_typing': False,
     'always_allow_keywords': True,
     'binding': True,
-    'embedsignature': True,
+    'embedsignature': False,
     'wraparound': False,
 }}
 
 setup(
     ext_modules=cythonize(
-        {repr(file_name)},
+        {repr(safe_file_name)},
         compiler_directives=compiler_directives,
         force=True
     )
@@ -76,30 +107,34 @@ setup(
 
         # 执行编译命令
         command = [sys.executable, "setup.py", "build_ext", "--inplace"]
-        result = subprocess.run(
-            command,
-            cwd=temp_dir,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE
-        )
+        try:
+            result = subprocess.run(
+                command,
+                cwd=temp_dir,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                timeout=300
+            )
+        except subprocess.TimeoutExpired:
+            raise RuntimeError(f"Compilation timed out after 300s: {file_path}")
 
         if result.returncode != 0:
             error_msg = result.stderr.decode('utf-8', errors='replace')
             raise RuntimeError(f"Compilation failed: {error_msg}")
 
         # 查找生成的二进制文件
-        pattern = os.path.join(temp_dir, f"{module_name}*{target_ext}")
+        pattern = os.path.join(temp_dir, f"{safe_module_name}*{target_ext}")
         matches = glob.glob(pattern)
 
         if not matches:
-            pattern = os.path.join(temp_dir, f"*{module_name}*{target_ext}")
+            pattern = os.path.join(temp_dir, f"*{safe_module_name}*{target_ext}")
             matches = glob.glob(pattern)
 
         if not matches:
             raise FileNotFoundError(f"FileNotFoundError: The file {file_path} is not a valid Python file (.py)! Generated file {target_ext} not found, in {temp_dir} possible file: {os.listdir(temp_dir)}")
 
         # 取最新生成的文件
-        generated_file = max(matches, key=os.path.getctime)
+        generated_file = max(matches, key=os.path.getmtime)
 
         # 获取源文件的目录（使用原始路径类型）
         if is_absolute:
@@ -116,9 +151,14 @@ setup(
         output_file_name = os.path.basename(generated_file)
         output_path = os.path.join(output_dir, output_file_name)
 
-        # 移动文件
+        # 移动文件（先删除已存在的目标）
+        if os.path.exists(output_path):
+            os.remove(output_path)
         shutil.move(generated_file, output_path)
 
+        # strip 符号表
+        _strip_binary(output_path)
+
         if del_source:
             os.remove(file_path)
 
@@ -129,7 +169,7 @@ setup(
         shutil.rmtree(temp_dir, ignore_errors=True)
 
 # 测试函数
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     # 替换为你的 Python 文件路径
     target_file = "test/example.py"  # 请确保文件路径正确
 
@@ -137,4 +177,4 @@ if __name__ == "__main__":
         output_file = compile_to_binary(target_file)
         print(output_file)
     except Exception as e:
-        print(e)
+        print(e)