fspack 0.1.0__py3-none-any.whl

fspack/__init__.py

@@ -0,0 +1,7 @@
+"""fspack - 极速 Python 项目打包器。."""
+
+from __future__ import annotations
+
+__all__ = ["__version__"]
+
+__version__ = "0.1.0"

fspack/analyzer.py

@@ -0,0 +1,296 @@
+"""AST 依赖分析：扫描 import，分类标准库/本地/第三方。."""
+
+from __future__ import annotations
+
+import ast
+import sys
+from pathlib import Path
+
+from fspack.config import DependencyReport
+
+__all__ = ["STDLIB_FALLBACK", "analyze_dependencies", "collect_imports"]
+
+# Python 3.8/3.9 没有 sys.stdlib_module_names，用 curate 的集合回退
+STDLIB_FALLBACK: frozenset[str] = frozenset(
+    {
+        "abc",
+        "aifc",
+        "argparse",
+        "array",
+        "ast",
+        "asynchat",
+        "asyncio",
+        "asyncore",
+        "atexit",
+        "audioop",
+        "base64",
+        "bdb",
+        "binascii",
+        "binhex",
+        "bisect",
+        "builtins",
+        "bz2",
+        "calendar",
+        "cgi",
+        "cgitb",
+        "chunk",
+        "cmath",
+        "cmd",
+        "code",
+        "codecs",
+        "codeop",
+        "collections",
+        "colorsys",
+        "compileall",
+        "concurrent",
+        "configparser",
+        "contextlib",
+        "contextvars",
+        "copy",
+        "copyreg",
+        "crypt",
+        "csv",
+        "ctypes",
+        "curses",
+        "dataclasses",
+        "datetime",
+        "decimal",
+        "difflib",
+        "dis",
+        "distutils",
+        "doctest",
+        "email",
+        "encodings",
+        "enum",
+        "errno",
+        "faulthandler",
+        "fcntl",
+        "filecmp",
+        "fileinput",
+        "fnmatch",
+        "formatter",
+        "fractions",
+        "ftplib",
+        "functools",
+        "gc",
+        "genericpath",
+        "getopt",
+        "getpass",
+        "gettext",
+        "glob",
+        "graphlib",
+        "grp",
+        "gzip",
+        "hashlib",
+        "heapq",
+        "hmac",
+        "html",
+        "http",
+        "idlelib",
+        "imaplib",
+        "imghdr",
+        "imp",
+        "importlib",
+        "inspect",
+        "io",
+        "ipaddress",
+        "itertools",
+        "json",
+        "keyword",
+        "lib2to3",
+        "linecache",
+        "locale",
+        "logging",
+        "lzma",
+        "mailbox",
+        "mailcap",
+        "marshal",
+        "math",
+        "mimetypes",
+        "mmap",
+        "modulefinder",
+        "msilib",
+        "msvcrt",
+        "multiprocessing",
+        "netrc",
+        "nis",
+        "nntplib",
+        "ntpath",
+        "numbers",
+        "opcode",
+        "operator",
+        "optparse",
+        "os",
+        "ossaudiodev",
+        "parser",
+        "pathlib",
+        "pdb",
+        "pickle",
+        "pickletools",
+        "pipes",
+        "pkgutil",
+        "platform",
+        "plistlib",
+        "poplib",
+        "posix",
+        "posixpath",
+        "pprint",
+        "profile",
+        "pstats",
+        "pty",
+        "pwd",
+        "py_compile",
+        "pyclbr",
+        "pydoc",
+        "pydoc_data",
+        "pyexpat",
+        "queue",
+        "quopri",
+        "random",
+        "re",
+        "readline",
+        "reprlib",
+        "resource",
+        "rlcompleter",
+        "runpy",
+        "sched",
+        "secrets",
+        "select",
+        "selectors",
+        "shelve",
+        "shlex",
+        "shutil",
+        "signal",
+        "site",
+        "smtpd",
+        "smtplib",
+        "sndhdr",
+        "socket",
+        "socketserver",
+        "spwd",
+        "sqlite3",
+        "sre_compile",
+        "sre_constants",
+        "sre_parse",
+        "ssl",
+        "stat",
+        "statistics",
+        "string",
+        "stringprep",
+        "struct",
+        "subprocess",
+        "sunau",
+        "symbol",
+        "symtable",
+        "sys",
+        "sysconfig",
+        "syslog",
+        "tabnanny",
+        "tarfile",
+        "telnetlib",
+        "tempfile",
+        "termios",
+        "test",
+        "textwrap",
+        "threading",
+        "time",
+        "timeit",
+        "tkinter",
+        "token",
+        "tokenize",
+        "trace",
+        "traceback",
+        "tracemalloc",
+        "tty",
+        "turtle",
+        "types",
+        "typing",
+        "unicodedata",
+        "unittest",
+        "urllib",
+        "uu",
+        "uuid",
+        "venv",
+        "warnings",
+        "wave",
+        "weakref",
+        "webbrowser",
+        "winreg",
+        "winsound",
+        "wsgiref",
+        "xdrlib",
+        "xml",
+        "xmlrpc",
+        "zipapp",
+        "zipfile",
+        "zipimport",
+        "zlib",
+        "_thread",
+        "__future__",
+    }
+)
+
+_STDLIB: frozenset[str] = getattr(sys, "stdlib_module_names", STDLIB_FALLBACK)
+
+
+def collect_imports(tree: ast.AST) -> list[str]:
+    """收集 AST 中所有 import 的顶层模块名，去重保序。."""
+    result: list[str] = []
+    seen: set[str] = set()
+    for node in ast.walk(tree):
+        if isinstance(node, ast.Import):
+            for alias in node.names:
+                _push(alias.name.split(".")[0], result, seen)
+        elif isinstance(node, ast.ImportFrom) and node.module and node.level == 0:
+            _push(node.module.split(".")[0], result, seen)
+    return result
+
+
+def _push(top: str, result: list[str], seen: set[str]) -> None:
+    if top and top not in seen:
+        seen.add(top)
+        result.append(top)
+
+
+def _local_packages(src_dir: Path, project_name: str) -> set[str]:
+    """识别项目本地包/模块名（顶层 .py 与含 __init__.py 的目录）。."""
+    local: set[str] = {project_name}
+    for entry in src_dir.iterdir():
+        if entry.is_file() and entry.suffix == ".py":
+            local.add(entry.stem)
+        elif entry.is_dir() and (entry / "__init__.py").is_file():
+            local.add(entry.name)
+    return local
+
+
+def analyze_dependencies(src_dir: Path, project_name: str, declared: tuple[str, ...]) -> DependencyReport:
+    """扫描 src_dir 下所有 .py，分类 import 为标准库/本地/第三方。."""
+    all_imports: list[str] = []
+    for py in src_dir.rglob("*.py"):
+        try:
+            tree = ast.parse(py.read_text(encoding="utf-8"))
+        except (SyntaxError, OSError):
+            continue
+        all_imports.extend(collect_imports(tree))
+
+    local = _local_packages(src_dir, project_name)
+    stdlib: list[str] = []
+    third: list[str] = []
+    local_imports: list[str] = []
+    seen: set[str] = set()
+    for imp in all_imports:
+        if imp in seen:
+            continue
+        seen.add(imp)
+        if imp in local:
+            local_imports.append(imp)
+        elif imp in _STDLIB:
+            stdlib.append(imp)
+        else:
+            third.append(imp)
+    return DependencyReport(
+        declared=declared,
+        ast_third_party=tuple(third),
+        ast_stdlib=tuple(stdlib),
+        ast_local=tuple(local_imports),
+    )

fspack/builder.py

@@ -0,0 +1,156 @@
+"""构建流水线编排：解析 → embed → 依赖 → 源码 → loader。."""
+
+from __future__ import annotations
+
+import logging
+import shutil
+import subprocess
+import sys
+import zipfile
+from pathlib import Path
+
+from fspack.analyzer import analyze_dependencies
+from fspack.config import BuildConfig, MirrorConfig, ProjectInfo
+from fspack.embed import ensure_embed, write_pth
+from fspack.exceptions import DependencyError
+from fspack.loader import compile_loader, generate_loader_source
+from fspack.platform import Platform, detect_platform, wheel_platform_tag
+from fspack.project import DEFAULT_PY_VERSION, parse_project
+from fspack.standalone import STANDALONE_RELEASE_TAG, ensure_standalone
+
+__all__ = ["DEFAULT_PY_VERSION", "build", "copy_source", "download_wheels", "unpack_wheels"]
+
+_logger = logging.getLogger(__name__)
+
+_EXCLUDE = shutil.ignore_patterns(
+    "dist",
+    "build",
+    ".git",
+    "__pycache__",
+    "*.egg-info",
+    ".venv",
+    ".tox",
+    ".fspack",
+    ".trae",
+    ".pytest_cache",
+    ".ruff_cache",
+    ".mypy_cache",
+)
+
+
+def build(  # noqa: PLR0913
+    project_dir: Path,
+    mirror: MirrorConfig,
+    py_version: str = DEFAULT_PY_VERSION,
+    dist_dir: Path | None = None,
+    embed_cache: Path | None = None,
+    target: Platform | None = None,
+) -> ProjectInfo:
+    """执行完整构建流水线，返回项目信息。."""
+    project_dir = Path(project_dir).resolve()
+    target = target or detect_platform()
+    dist = dist_dir or project_dir / "dist"
+    cache = embed_cache or Path.home() / ".fspack" / "cache" / "embed"
+    cfg = BuildConfig(project_dir=project_dir, dist_dir=dist, embed_cache_dir=cache, mirror=mirror, target=target)
+    info = parse_project(project_dir, py_version)
+    _logger.info("项目: %s %s (%s) 目标: %s", info.name, info.version, info.app_type.value, target.value)
+
+    runtime_dir = cfg.dist_dir / "runtime"
+    if target is Platform.LINUX:
+        standalone_cache = Path.home() / ".fspack" / "cache" / "standalone"
+        ensure_standalone(info.py_version, STANDALONE_RELEASE_TAG, standalone_cache, runtime_dir)
+        major, minor = info.py_version.split(".")[:2]
+        site_packages = runtime_dir / "python" / "lib" / f"python{major}.{minor}" / "site-packages"
+    else:
+        ensure_embed(info.py_version, cfg.mirror, cfg.embed_cache_dir, runtime_dir)
+        site_packages = runtime_dir / "Lib" / "site-packages"
+
+    report = analyze_dependencies(project_dir, info.name, info.dependencies)
+    if report.missing:
+        _logger.info("AST 发现未声明依赖: %s", ", ".join(report.missing))
+    if report.ast_third_party:
+        wheelhouse = cfg.dist_dir / "wheelhouse"
+        download_wheels(
+            report.ast_third_party,
+            info.py_version,
+            cfg.mirror.pypi_index,
+            wheelhouse,
+            platform_tag=wheel_platform_tag(target),
+        )
+        unpack_wheels(wheelhouse, site_packages)
+    else:
+        _logger.info("无第三方依赖，跳过 wheel 下载")
+
+    if target is Platform.WINDOWS:
+        write_pth(cfg.dist_dir, info.py_version)
+    src_dst = cfg.dist_dir / "src"
+    copy_source(project_dir, src_dst)
+
+    entry_rel = info.entry_file.relative_to(info.src_dir).as_posix()
+    source = generate_loader_source(f"src/{entry_rel}", info.py_xy, target)
+    exe_name = info.exe_name if target is Platform.WINDOWS else info.name
+    exe = cfg.dist_dir / exe_name
+    compile_loader(source, exe, info.app_type, cfg.dist_dir / "build", target)
+    _logger.info("构建完成: %s", exe)
+    return info
+
+
+def copy_source(project_dir: Path, src_dst: Path) -> None:
+    """将项目源码复制到 dist/src，排除构建产物与缓存。."""
+    if src_dst.exists():
+        shutil.rmtree(src_dst)
+    shutil.copytree(project_dir, src_dst, ignore=_EXCLUDE)
+
+
+def download_wheels(
+    packages: tuple[str, ...] | list[str],
+    py_version: str,
+    pypi_index: str,
+    wheelhouse_dir: Path,
+    platform_tag: str = "win_amd64",
+) -> list[Path]:
+    """用 dev python 的 pip 下载指定平台 wheel 到 wheelhouse 目录。."""
+    wheelhouse_dir.mkdir(parents=True, exist_ok=True)
+    major, minor = py_version.split(".")[:2]
+    cmd: list[str] = [
+        sys.executable,
+        "-m",
+        "pip",
+        "download",
+        "-d",
+        str(wheelhouse_dir),
+        "--platform",
+        platform_tag,
+        "--python-version",
+        f"{major}.{minor}",
+        "--abi",
+        f"cp{major}{minor}",
+        "--implementation",
+        "cp",
+        "--only-binary=:all:",
+        "-i",
+        pypi_index,
+        *packages,
+    ]
+    _logger.info("下载依赖 wheel: %s", " ".join(packages))
+    try:
+        subprocess.run(cmd, check=True, capture_output=True, text=True)
+    except FileNotFoundError as e:
+        raise DependencyError(f"未找到 pip: {sys.executable}") from e
+    except subprocess.CalledProcessError as e:
+        raise DependencyError(f"依赖下载失败:\n{e.stderr}") from e
+    return sorted(wheelhouse_dir.glob("*.whl"))
+
+
+def unpack_wheels(wheelhouse_dir: Path, site_packages_dir: Path) -> int:
+    """将 wheelhouse 内所有 .whl 解包到 site-packages 目录，返回解包数量。."""
+    site_packages_dir.mkdir(parents=True, exist_ok=True)
+    count = 0
+    for whl in wheelhouse_dir.glob("*.whl"):
+        try:
+            with zipfile.ZipFile(whl) as zf:
+                zf.extractall(site_packages_dir)
+        except zipfile.BadZipFile as e:
+            raise DependencyError(f"wheel 损坏: {whl}") from e
+        count += 1
+    return count

fspack/cli.py

@@ -0,0 +1,87 @@
+"""fspack CLI 入口 —— cargo 风格短命令（fsp b/c/r）。."""
+
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+
+from fspack import __version__
+from fspack.commands import build as build_cmd
+from fspack.commands import clean as clean_cmd
+from fspack.commands import package as package_cmd
+from fspack.commands import run as run_cmd
+from fspack.mirror import MIRRORS
+from fspack.platform import Platform
+
+__all__ = ["build_parser", "main"]
+
+
+def build_parser() -> argparse.ArgumentParser:
+    """构建参数解析器。."""
+    parser = argparse.ArgumentParser(
+        prog="fspack",
+        description="极速 Python 打包器（cargo 风格短命令）。",
+    )
+    parser.add_argument("-V", "--version", action="version", version=f"%(prog)s {__version__}")
+
+    sub = parser.add_subparsers(dest="command", metavar="<command>")
+
+    p_build = sub.add_parser("build", aliases=["b"], help="打包项目")
+    p_build.add_argument("project", nargs="?", default=".", help="项目目录（默认当前目录）")
+    p_build.add_argument("--mirror", default=None, choices=list(MIRRORS), help="镜像源")
+    p_build.add_argument("--py-version", default=None, help="embed python 版本，如 3.11.9")
+    p_build.add_argument("--target", default=None, choices=["windows", "linux"], help="目标平台（默认当前平台）")
+
+    p_run = sub.add_parser("run", aliases=["r"], help="运行已打包项目")
+    p_run.add_argument("project", nargs="?", default=".", help="项目目录")
+    p_run.add_argument("rest", nargs=argparse.REMAINDER, default=[], help="透传给目标程序的参数（以 -- 分隔）")
+
+    p_clean = sub.add_parser("clean", aliases=["c"], help="清理 dist/")
+    p_clean.add_argument("project", nargs="?", default=".", help="项目目录")
+
+    p_pkg = sub.add_parser("package", aliases=["p"], help="生成 NSIS 安装包")
+    p_pkg.add_argument("project", nargs="?", default=".", help="项目目录")
+    p_pkg.add_argument("--mirror", default=None, choices=list(MIRRORS), help="镜像源")
+    p_pkg.add_argument("--py-version", default=None, help="embed python 版本，如 3.11.9")
+    p_pkg.add_argument("--no-build", action="store_true", help="跳过重建，直接打包已有 dist")
+    return parser
+
+
+def main(argv: list[str] | None = None) -> None:
+    """主入口，解析参数并分发到子命令。."""
+    parser = build_parser()
+    ns = parser.parse_args(argv)
+    command = ns.command
+    if command is None:
+        parser.print_help()
+        return
+
+    project = Path(ns.project).resolve()
+    if command in ("build", "b"):
+        build_cmd.run(project, mirror=ns.mirror, py_version=ns.py_version, target=_parse_target(ns.target))
+    elif command in ("run", "r"):
+        run_cmd.run(project, rest_args=_drop_separator(ns.rest))
+    elif command in ("clean", "c"):
+        clean_cmd.run(project)
+    elif command in ("package", "p"):
+        package_cmd.run(project, mirror=ns.mirror, py_version=ns.py_version, no_build=ns.no_build)
+
+
+def _drop_separator(rest: list[str]) -> list[str]:
+    """剔除 argparse REMAINDER 捕获的首个 -- 分隔符。."""
+    if rest and rest[0] == "--":
+        return rest[1:]
+    return rest
+
+
+def _parse_target(value: str | None) -> Platform | None:
+    """将 CLI 字符串转为 Platform 枚举，None 表示用当前平台。."""
+    if value is None:
+        return None
+    if value == "windows":
+        return Platform.WINDOWS
+    return Platform.LINUX
+
+
+if __name__ == "__main__":
+    main()

fspack/commands/__init__.py

@@ -0,0 +1 @@
+"""fspack 子命令实现：build / clean / run。."""

fspack/commands/build.py

@@ -0,0 +1,25 @@
+"""fsp b —— 打包项目。."""
+
+from __future__ import annotations
+
+import logging
+from pathlib import Path
+
+from fspack.builder import DEFAULT_PY_VERSION, build
+from fspack.mirror import get_mirror
+from fspack.platform import Platform
+
+__all__ = ["run"]
+
+_logger = logging.getLogger(__name__)
+
+
+def run(
+    project: Path,
+    mirror: str | None = None,
+    py_version: str | None = None,
+    target: Platform | None = None,
+) -> None:
+    """执行项目构建。."""
+    mirror_cfg = get_mirror(mirror)
+    build(project, mirror_cfg, py_version or DEFAULT_PY_VERSION, target=target)

fspack/commands/clean.py

@@ -0,0 +1,21 @@
+"""fsp c —— 清理 dist/。."""
+
+from __future__ import annotations
+
+import logging
+import shutil
+from pathlib import Path
+
+__all__ = ["run"]
+
+_logger = logging.getLogger(__name__)
+
+
+def run(project: Path) -> None:
+    """清理项目下的 dist 目录。."""
+    dist = Path(project) / "dist"
+    if dist.is_dir():
+        shutil.rmtree(dist)
+        _logger.info("已清理: %s", dist)
+    else:
+        _logger.info("无 dist 目录可清理: %s", dist)

fspack/commands/package.py

@@ -0,0 +1,26 @@
+"""fsp p —— 生成 NSIS 安装包。."""
+
+from __future__ import annotations
+
+import logging
+from pathlib import Path
+
+from fspack.installer import build_installer
+from fspack.mirror import get_mirror
+from fspack.project import DEFAULT_PY_VERSION
+
+__all__ = ["run"]
+
+_logger = logging.getLogger(__name__)
+
+
+def run(
+    project: Path,
+    mirror: str | None = None,
+    py_version: str | None = None,
+    no_build: bool = False,
+) -> None:
+    """生成 Windows 安装包到 dist/release/。."""
+    mirror_cfg = get_mirror(mirror)
+    out = build_installer(project, mirror_cfg, py_version or DEFAULT_PY_VERSION, no_build=no_build)
+    _logger.info("安装包已生成: %s", out)

fspack/commands/run.py

@@ -0,0 +1,36 @@
+"""fsp r —— 运行已打包项目（Linux 用 wine，Windows 直跑）。."""
+
+from __future__ import annotations
+
+import logging
+import platform
+import shutil
+import subprocess
+from pathlib import Path
+
+from fspack.exceptions import FspackError
+from fspack.project import parse_project
+
+__all__ = ["run"]
+
+_logger = logging.getLogger(__name__)
+
+
+def run(project: Path, rest_args: list[str] | None = None) -> None:
+    """运行 dist 下的可执行文件。."""
+    info = parse_project(project)
+    exe = Path(project) / "dist" / info.exe_name
+    if not exe.is_file():
+        raise FspackError(f"未找到已构建的可执行文件: {exe}（请先执行 fsp b）")
+    cmd = _build_cmd(exe) + (rest_args or [])
+    _logger.info("运行: %s", " ".join(cmd))
+    completed = subprocess.run(cmd, check=False)
+    if completed.returncode != 0:
+        raise FspackError(f"程序退出码非零: {completed.returncode}")
+
+
+def _build_cmd(exe: Path) -> list[str]:
+    if platform.system() == "Linux":
+        wine = shutil.which("wine") or "wine"
+        return [wine, str(exe)]
+    return [str(exe)]