AutoCython-zhang 2.3.3__tar.gz → 2.3.5__tar.gz

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/AutoCython/_version.py

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

autocython_zhang-2.3.5/AutoCython/compile.py

@@ -0,0 +1,263 @@
+import glob
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+
+from .obfuscate import obfuscate_source
+
+_SUPPORTED_CYTHON_MAJOR = 3
+
+
+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_cython_major_version():
+    """返回当前运行环境的 Cython 主版本号。"""
+    try:
+        import Cython
+    except ImportError as exc:  # pragma: no cover
+        raise RuntimeError('编译 Python 源码需要先安装 Cython。') from exc
+
+    version = getattr(Cython, '__version__', '0')
+    major = str(version).split('.', 1)[0]
+    try:
+        return int(major)
+    except ValueError as exc:  # pragma: no cover
+        raise RuntimeError(f'无法解析 Cython 版本: {version}') from exc
+
+
+def _ensure_supported_cython():
+    """阻断已知会破坏运行时注解的旧版 Cython。"""
+    major = _get_cython_major_version()
+    if major != _SUPPORTED_CYTHON_MAJOR:
+        raise RuntimeError(
+            '检测到不受支持的 Cython 版本。'
+            'AutoCython 要求 Cython>=3,<4；'
+            '因为 Cython<3 可能剥离编译产物中的运行时注解，'
+            '导致 Pydantic/FastAPI/dataclass 的字段发现失效。'
+        )
+
+
+def get_platform_extension() -> str:
+    """返回当前平台的扩展名"""
+    if sys.platform.startswith('win'):
+        return '.pyd'
+    return '.so'
+
+
+def _infer_compile_root(abs_file_path: str) -> str:
+    """推导编译根目录。
+
+    对包内模块，返回最外层 package 目录的父目录，以便保留完整 qualified module name。
+    对普通脚本，返回源码所在目录，保持现有行为。
+    """
+    source_dir = os.path.dirname(abs_file_path)
+    current_dir = source_dir
+    top_level_package_dir = None
+
+    while os.path.isfile(os.path.join(current_dir, '__init__.py')):
+        top_level_package_dir = current_dir
+        parent_dir = os.path.dirname(current_dir)
+        if parent_dir == current_dir:
+            break
+        current_dir = parent_dir
+
+    if top_level_package_dir is None:
+        return source_dir
+
+    return os.path.dirname(top_level_package_dir)
+
+
+def _resolve_module_layout(abs_file_path: str):
+    """解析编译布局，保留 package 相对路径与 qualified module name。"""
+    compile_root = _infer_compile_root(abs_file_path)
+    rel_path = os.path.relpath(abs_file_path, compile_root)
+    rel_dir = os.path.dirname(rel_path)
+    file_name = os.path.basename(rel_path)
+    module_name, ext = os.path.splitext(file_name)
+    safe_module_name = module_name.replace('-', '_')
+
+    safe_rel_parts = [part for part in rel_dir.split(os.sep) if part and part != '.']
+    safe_rel_parts.append(safe_module_name + ext)
+    safe_rel_path = os.path.join(*safe_rel_parts) if safe_rel_parts else safe_module_name + ext
+
+    module_parts = [part for part in rel_dir.split(os.sep) if part and part != '.']
+    module_parts.append(safe_module_name)
+    qualified_module_name = '.'.join(module_parts)
+
+    return {
+        'compile_root': compile_root,
+        'rel_path': rel_path,
+        'safe_rel_path': safe_rel_path,
+        'safe_module_name': safe_module_name,
+        'qualified_module_name': qualified_module_name,
+    }
+
+
+def _copy_package_inits(abs_file_path: str, compile_root: str, temp_dir: str) -> None:
+    """复制包链上的 __init__.py，确保 build_ext --inplace 输出落到正确包路径。"""
+    current_dir = os.path.dirname(abs_file_path)
+
+    while True:
+        init_path = os.path.join(current_dir, '__init__.py')
+        if not os.path.isfile(init_path):
+            break
+
+        rel_init_path = os.path.relpath(init_path, compile_root)
+        temp_init_path = os.path.join(temp_dir, rel_init_path)
+        os.makedirs(os.path.dirname(temp_init_path), exist_ok=True)
+        shutil.copy2(init_path, temp_init_path)
+
+        if current_dir == compile_root:
+            break
+
+        parent_dir = os.path.dirname(current_dir)
+        if parent_dir == current_dir:
+            break
+        current_dir = parent_dir
+
+
+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: 生成的二进制文件路径（与输入保持相同的路径类型）
+    """
+    _ensure_supported_cython()
+
+    is_absolute = os.path.isabs(file_path)
+    abs_file_path = os.path.abspath(file_path)
+
+    if not os.path.isfile(abs_file_path):
+        raise FileNotFoundError(f"FileNotFoundError: {file_path}.")
+
+    file_name = os.path.basename(abs_file_path)
+    _, ext = os.path.splitext(file_name)
+    source_dir = os.path.dirname(abs_file_path)
+
+    if ext != '.py':
+        raise ValueError(f"ValueError: The file {file_path} is not a valid Python file (.py)!")
+
+    module_layout = _resolve_module_layout(abs_file_path)
+    safe_module_name = module_layout['safe_module_name']
+    target_ext = get_platform_extension()
+    temp_dir = tempfile.mkdtemp()
+
+    try:
+        temp_file_path = os.path.join(temp_dir, module_layout['safe_rel_path'])
+        os.makedirs(os.path.dirname(temp_file_path), exist_ok=True)
+        _copy_package_inits(abs_file_path, module_layout['compile_root'], temp_dir)
+        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_code = f"""
+from setuptools import setup
+from setuptools import Extension
+from Cython.Build import cythonize
+
+compiler_directives = {{
+    'language_level': '3',
+    'annotation_typing': False,
+    'always_allow_keywords': True,
+    'binding': True,
+    'embedsignature': False,
+    'wraparound': False,
+}}
+
+setup(
+    ext_modules=cythonize(
+        [
+            Extension(
+                {repr(module_layout['qualified_module_name'])},
+                [{repr(module_layout['safe_rel_path'])}],
+            )
+        ],
+        compiler_directives=compiler_directives,
+        force=True
+    )
+)
+"""
+        setup_path = os.path.join(temp_dir, 'setup.py')
+        with open(setup_path, 'w', encoding='utf-8') as f:
+            f.write(setup_code)
+
+        command = [sys.executable, 'setup.py', 'build_ext', '--inplace']
+        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}")
+
+        temp_output_dir = os.path.dirname(temp_file_path) or temp_dir
+        pattern = os.path.join(temp_output_dir, f"{safe_module_name}*{target_ext}")
+        matches = glob.glob(pattern)
+        if not matches:
+            pattern = os.path.join(temp_output_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)! "
+                f"Generated file {target_ext} not found, in {temp_dir} possible file: {os.listdir(temp_dir)}"
+            )
+
+        generated_file = max(matches, key=os.path.getmtime)
+        output_dir = source_dir if is_absolute else (os.path.dirname(file_path) or '.')
+        if output_dir and not os.path.exists(output_dir):
+            os.makedirs(output_dir, exist_ok=True)
+
+        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_binary(output_path)
+
+        if del_source:
+            os.remove(file_path)
+
+        return output_path
+    finally:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+
+
+if __name__ == '__main__':  # pragma: no cover
+    target_file = 'test/example.py'
+    try:
+        output_file = compile_to_binary(target_file)
+        print(output_file)
+    except Exception as e:
+        print(e)

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/AutoCython_zhang.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AutoCython-zhang
-Version: 2.3.3
+Version: 2.3.5
 Summary: 自动Cython，使用Cython批量编译.py文件为.pyd文件！
 Author-email: zhang_gavin <qq814608@163.com>
 License-Expression: MIT
@@ -14,7 +14,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: setuptools
-Requires-Dist: cython
+Requires-Dist: cython<4,>=3
 Requires-Dist: rich
 Dynamic: license-file
 
@@ -37,6 +37,7 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 - **并发编译** — 基于 `ThreadPoolExecutor` 的多线程并行编译，可配置并发数
 - **实时进度面板** — 基于 Rich 的实时任务状态表格、进度条、耗时统计
 - **跨平台** — 支持 Linux、macOS、Windows，自动适配 `.so`/`.pyd` 扩展名
+- **包路径保真** — 编译包内模块时保留 qualified module name，兼容 Pydantic / FastAPI 等依赖运行时模块命名空间的框架
 - **可复现构建** — 通过 `--seed` 参数固定混淆随机种子
 - **智能排除** — 自动跳过 `__init__.py`、虚拟环境、构建目录；支持 `# AutoCython No Compile` 标记豁免
 - **中英双语** — CLI 帮助信息和进度面板自动适配系统语言
@@ -47,11 +48,13 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 pip install AutoCython-zhang
 ```
 
+> 当前编译链要求 `Cython>=3,<4`。该约束用于避免旧版 Cython 在编译后破坏 `Pydantic` / `FastAPI` / `dataclass` 等依赖运行时注解的框架行为。
+
 ### 依赖
 
 | 包 | 用途 |
 |---|------|
-| `cython` | Python → C 编译核心 |
+| `cython>=3,<4` | Python → C 编译核心；确保运行时注解不被旧版编译链破坏 |
 | `setuptools` | 构建扩展模块 |
 | `rich` | 终端实时进度面板 |
 

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/AutoCython_zhang.egg-info/requires.txt

@@ -1,3 +1,3 @@
 setuptools
-cython
+cython<4,>=3
 rich

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/AutoCython_zhang.egg-info/top_level.txt

@@ -1,2 +1,3 @@
 AutoCython
 dist
+release_dist

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AutoCython-zhang
-Version: 2.3.3
+Version: 2.3.5
 Summary: 自动Cython，使用Cython批量编译.py文件为.pyd文件！
 Author-email: zhang_gavin <qq814608@163.com>
 License-Expression: MIT
@@ -14,7 +14,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: setuptools
-Requires-Dist: cython
+Requires-Dist: cython<4,>=3
 Requires-Dist: rich
 Dynamic: license-file
 
@@ -37,6 +37,7 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 - **并发编译** — 基于 `ThreadPoolExecutor` 的多线程并行编译，可配置并发数
 - **实时进度面板** — 基于 Rich 的实时任务状态表格、进度条、耗时统计
 - **跨平台** — 支持 Linux、macOS、Windows，自动适配 `.so`/`.pyd` 扩展名
+- **包路径保真** — 编译包内模块时保留 qualified module name，兼容 Pydantic / FastAPI 等依赖运行时模块命名空间的框架
 - **可复现构建** — 通过 `--seed` 参数固定混淆随机种子
 - **智能排除** — 自动跳过 `__init__.py`、虚拟环境、构建目录；支持 `# AutoCython No Compile` 标记豁免
 - **中英双语** — CLI 帮助信息和进度面板自动适配系统语言
@@ -47,11 +48,13 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 pip install AutoCython-zhang
 ```
 
+> 当前编译链要求 `Cython>=3,<4`。该约束用于避免旧版 Cython 在编译后破坏 `Pydantic` / `FastAPI` / `dataclass` 等依赖运行时注解的框架行为。
+
 ### 依赖
 
 | 包 | 用途 |
 |---|------|
-| `cython` | Python → C 编译核心 |
+| `cython>=3,<4` | Python → C 编译核心；确保运行时注解不被旧版编译链破坏 |
 | `setuptools` | 构建扩展模块 |
 | `rich` | 终端实时进度面板 |
 

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/README.md

@@ -17,6 +17,7 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 - **并发编译** — 基于 `ThreadPoolExecutor` 的多线程并行编译，可配置并发数
 - **实时进度面板** — 基于 Rich 的实时任务状态表格、进度条、耗时统计
 - **跨平台** — 支持 Linux、macOS、Windows，自动适配 `.so`/`.pyd` 扩展名
+- **包路径保真** — 编译包内模块时保留 qualified module name，兼容 Pydantic / FastAPI 等依赖运行时模块命名空间的框架
 - **可复现构建** — 通过 `--seed` 参数固定混淆随机种子
 - **智能排除** — 自动跳过 `__init__.py`、虚拟环境、构建目录；支持 `# AutoCython No Compile` 标记豁免
 - **中英双语** — CLI 帮助信息和进度面板自动适配系统语言
@@ -27,11 +28,13 @@ AutoCython 是一个 Python 源码保护工具，通过 Cython 编译 + AST 混
 pip install AutoCython-zhang
 ```
 
+> 当前编译链要求 `Cython>=3,<4`。该约束用于避免旧版 Cython 在编译后破坏 `Pydantic` / `FastAPI` / `dataclass` 等依赖运行时注解的框架行为。
+
 ### 依赖
 
 | 包 | 用途 |
 |---|------|
-| `cython` | Python → C 编译核心 |
+| `cython>=3,<4` | Python → C 编译核心；确保运行时注解不被旧版编译链破坏 |
 | `setuptools` | 构建扩展模块 |
 | `rich` | 终端实时进度面板 |
 

{autocython_zhang-2.3.3 → autocython_zhang-2.3.5}/pyproject.toml

@@ -18,7 +18,7 @@ classifiers = [
 ]
 dependencies = [
     "setuptools",
-    "cython",
+    "cython>=3,<4",
     "rich",
 ]
 

autocython_zhang-2.3.3/AutoCython/compile.py

@@ -1,180 +0,0 @@
-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, obfuscate=True, obfuscate_seed=None):
-    """
-    将指定的 Python 文件（.py）通过 Cython 编译为二进制扩展文件
-
-    :param file_path: Python 文件路径（可以是相对路径或绝对路径）
-    :param del_source: 是否删除源代码
-    :param obfuscate: 是否在编译前混淆源码（默认True）
-    :param obfuscate_seed: 混淆随机种子（None 表示不固定）
-    :return: 生成的二进制文件路径（与输入保持相同的路径类型）
-    """
-    # 保存原始路径类型（相对/绝对）
-    is_absolute = os.path.isabs(file_path)
-
-    # 获取绝对路径用于内部操作
-    abs_file_path = os.path.abspath(file_path)
-
-    if not os.path.isfile(abs_file_path):
-        raise FileNotFoundError(f"FileNotFoundError: {file_path}.")
-
-    # 获取文件名和目录
-    file_name = os.path.basename(abs_file_path)
-    module_name, ext = os.path.splitext(file_name)
-    source_dir = os.path.dirname(abs_file_path)  # 源文件所在目录（绝对路径）
-
-    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()
-
-    # 创建临时工作目录
-    temp_dir = tempfile.mkdtemp()
-
-    try:
-        # 将目标文件复制到临时目录（使用安全文件名）
-        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
-from Cython.Build import cythonize
-
-# 编译器指令
-compiler_directives = {{
-    'language_level': '3',
-    'annotation_typing': False,
-    'always_allow_keywords': True,
-    'binding': True,
-    'embedsignature': False,
-    'wraparound': False,
-}}
-
-setup(
-    ext_modules=cythonize(
-        {repr(safe_file_name)},
-        compiler_directives=compiler_directives,
-        force=True
-    )
-)
-"""
-        setup_path = os.path.join(temp_dir, "setup.py")
-        with open(setup_path, "w", encoding='utf-8') as f:
-            f.write(setup_code)
-
-        # 执行编译命令
-        command = [sys.executable, "setup.py", "build_ext", "--inplace"]
-        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"{safe_module_name}*{target_ext}")
-        matches = glob.glob(pattern)
-
-        if not matches:
-            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.getmtime)
-
-        # 获取源文件的目录（使用原始路径类型）
-        if is_absolute:
-            output_dir = source_dir
-        else:
-            # 如果输入是相对路径，保持相对路径
-            output_dir = os.path.dirname(file_path) or '.'
-
-        # 创建目标目录（如果不存在）
-        if output_dir and not os.path.exists(output_dir):
-            os.makedirs(output_dir, exist_ok=True)
-
-        # 目标文件路径（保持原始路径类型）
-        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)
-
-        # 返回与输入相同类型的路径
-        return output_path
-    finally:
-        # 清理临时目录
-        shutil.rmtree(temp_dir, ignore_errors=True)
-
-# 测试函数
-if __name__ == "__main__":  # pragma: no cover
-    # 替换为你的 Python 文件路径
-    target_file = "test/example.py"  # 请确保文件路径正确
-
-    try:
-        output_file = compile_to_binary(target_file)
-        print(output_file)
-    except Exception as e:
-        print(e)