PyInstallerEx 0.1.6__py3-none-any.whl

PyInstallerEx/__init__.py

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+"""
+PyInstallerEx - Enhanced PyInstaller packaging tool
+扩展PyInstaller使其拥有单文件安装功能
+"""
+
+__version__ = "0.1.6"
+__author__ = "iiixxxiii"
+
+from .core.packager import PackageBuilder
+from .core.config import PackageConfig
+
+__all__ = ["PackageBuilder", "PackageConfig"]

PyInstallerEx/__main__.py

@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+PyInstallerEx命令行入口点
+"""
+
+from __future__ import print_function
+import argparse
+import sys
+import os
+
+from .core.packager import PackageBuilder
+from .core.config import PackageConfig
+from .utils.logging import setup_logging
+from .utils.file_utils import FileUtils
+
+
+def main():
+    """主函数 - 命令行接口"""
+    parser = argparse.ArgumentParser(
+        description="PyInstallerEx - Enhanced PyInstaller packaging tool",
+        epilog="Example: python -m PyInstallerEx main.py --cfg config.json"
+    )
+    
+    parser.add_argument("script", nargs="?", default=None,
+                       help="Python script to package (mutually exclusive with --folder)")
+    parser.add_argument("--cfg", "--config", dest="config", 
+                       help="Custom configuration file path")
+    parser.add_argument("--output", "-o", help="Output directory")
+    parser.add_argument("--verbose", "-v", action="store_true", 
+                       help="Verbose output")
+    parser.add_argument("--name", "-n", help="Output filename (default: script name)")
+    parser.add_argument("--extract-mode", dest="extract_mode", choices=["temp", "local"],
+                       default="temp", help="Extraction mode: 'temp' (system temp dir, default) or 'local' (beside exe)")
+    parser.add_argument("--engine-mode", dest="engine_mode", choices=["pyinstaller", "nuitka", "pyoxidizer"],
+                       default="pyinstaller", help="Packaging engine: 'pyinstaller' (default), 'nuitka', or 'pyoxidizer'")
+    parser.add_argument("--folder", dest="folder",
+                       help="Package a local folder directly (skip engine build, mutually exclusive with script)")
+    parser.add_argument("--run", dest="run",
+                       help="Main entry file or relative path to run after extraction (used with --folder)")
+    
+    # 计算默认图标路径（包内的logo.ico）
+    _default_icon = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logo.ico")
+    parser.add_argument("--icon", default=_default_icon,
+                       help="Icon file path for the executable (default: package logo.ico)")
+    
+    args = parser.parse_args()
+    
+    # 设置日志
+    setup_logging(args.verbose)
+    
+    try:
+        # 参数互斥校验：script 和 --folder 不能同时使用
+        if args.script and args.folder:
+            print("[FAIL] Error: 'script' and '--folder' are mutually exclusive. Please specify only one.", file=sys.stderr)
+            sys.exit(1)
+        
+        # 必须提供 script 或 --folder 之一
+        if not args.script and not args.folder:
+            print("[FAIL] Error: either 'script' or '--folder' must be specified.", file=sys.stderr)
+            sys.exit(1)
+        
+        # --folder 模式校验：目录必须存在
+        if args.folder:
+            if not os.path.isdir(args.folder):
+                print("[FAIL] Error: --folder path does not exist or is not a directory: {0}".format(args.folder), file=sys.stderr)
+                sys.exit(1)
+        
+        # 加载配置
+        if args.config:
+            config = PackageConfig.from_file(args.config)
+        else:
+            # 使用默认配置
+            config = PackageConfig()
+        
+        # 设置解压模式、引擎模式和图标
+        config.extract_mode = args.extract_mode
+        config.engine_mode = args.engine_mode
+        
+        # 设置 --run 参数（main 字段）
+        if args.run:
+            config.main = args.run
+        
+        # 验证图标参数
+        icon_path = args.icon
+        if icon_path and icon_path != _default_icon:  # 用户指定了自定义图标
+            # 检查文件是否存在
+            if not os.path.exists(icon_path):
+                print("[WARNING] Icon file not found: {0}".format(icon_path), file=sys.stderr)
+                config.icon = None
+            # 检查文件格式是否为.ico
+            elif not icon_path.lower().endswith('.ico'):
+                print("[WARNING] Icon file must be in .ico format, got: {0}".format(icon_path), file=sys.stderr)
+                print("[WARNING] Please convert your image to .ico format before using --icon parameter", file=sys.stderr)
+                config.icon = None
+            else:
+                config.icon = icon_path
+        else:
+            # 使用默认图标（如果存在）
+            if os.path.exists(_default_icon):
+                config.icon = _default_icon
+            else:
+                config.icon = None
+        
+        # 创建构建器
+        builder = PackageBuilder(config, verbose=args.verbose)
+        
+        output_dir = args.output if args.output else None
+        
+        # --folder 模式：直接打包文件夹
+        if args.folder:
+            folder_path = os.path.abspath(args.folder)
+            
+            # 确定输出文件名：优先用 -n，否则用文件夹名
+            if args.name:
+                config.filename = args.name
+            elif config.filename == "{filename}":
+                config.filename = os.path.basename(folder_path)
+            
+            result_path = builder.build_folder(folder_path, output_dir)
+        else:
+            # 常规脚本打包模式
+            # 如果通过-n指定了名称，则覆盖配置中的filename
+            if args.name:
+                config.filename = args.name
+            # 如果配置中filename仍然是默认值"{filename}"，则使用脚本名
+            elif config.filename == "{filename}":
+                script_name = FileUtils.get_script_name(args.script)
+                config.filename = script_name
+            
+            # 执行打包
+            script_path = args.script
+            result_path = builder.build(script_path, output_dir)
+        
+        print("\n[OK] Packaging completed successfully!")
+        print("[Output] File: {0}".format(result_path))
+        
+    except Exception as e:
+        print("\n[FAIL] Packaging failed: {0}".format(e), file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

PyInstallerEx/core/__init__.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+"""
+PyInstallerEx核心功能模块
+"""
+
+from .packager import PackageBuilder
+from .config import PackageConfig
+
+__all__ = ["PackageBuilder", "PackageConfig"]

PyInstallerEx/core/config.py

@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+"""
+配置管理模块
+"""
+
+from __future__ import print_function
+import json
+import os
+import hashlib
+
+
+class PackageConfig(object):
+    """打包配置数据类"""
+    
+    def __init__(self, filename="{filename}", version="1.0.0", installer=None, 
+                 tmp_dir="ex_{filename}_{md5}", description="Application packaged with PyInstallerEx", 
+                 author="Unknown", extract_mode="temp", engine_mode="pyinstaller",
+                 icon=None, main=None):
+        self.filename = filename
+        self.version = version
+        self.installer = installer
+        self.tmp_dir = tmp_dir
+        self.description = description
+        self.author = author
+        self.extract_mode = extract_mode
+        self.engine_mode = engine_mode
+        self.icon = icon
+        self.main = main
+    
+    @classmethod
+    def from_file(cls, config_path):
+        """从配置文件加载配置"""
+        try:
+            with open(config_path, 'r') as f:
+                data = json.load(f)
+            return cls(**data)
+        except (ValueError, IOError) as e:
+            raise ValueError("Invalid configuration file {0}: {1}".format(config_path, e))
+    
+    def to_dict(self):
+        """转换为字典"""
+        return {
+            'filename': self.filename,
+            'version': self.version,
+            'installer': self.installer,
+            'tmp_dir': self.tmp_dir,
+            'description': self.description,
+            'author': self.author,
+            'extract_mode': self.extract_mode,
+            'engine_mode': self.engine_mode,
+            'icon': self.icon,
+            'main': self.main
+        }
+    
+    def validate(self):
+        """验证配置有效性"""
+        # 不再强制要求filename必须指定，因为会在运行时根据脚本名设置
+        if not self.version:
+            raise ValueError("Version must be specified")
+        return True
+    
+    def render_template(self, script_name, script_md5):
+        """渲染配置模板，替换占位符"""
+        filename = self.filename.replace("{filename}", script_name)
+        tmp_dir = self.tmp_dir.replace("{filename}", script_name).replace("{md5}", script_md5)
+        
+        # 处理installer默认值
+        installer = self.installer
+        if not installer:
+            if os.name == 'nt':  # Windows
+                installer = os.environ.get('TEMP', '/tmp')
+            else:  # Unix/Linux
+                installer = '/tmp'
+        
+        return PackageConfig(
+            filename=filename,
+            version=self.version,
+            installer=installer,
+            tmp_dir=tmp_dir,
+            description=self.description,
+            author=self.author,
+            extract_mode=self.extract_mode,
+            engine_mode=self.engine_mode,
+            icon=self.icon,
+            main=self.main
+        )
+    
+    def save_to_file(self, output_path):
+        """保存配置到文件"""
+        with open(output_path, 'w') as f:
+            json.dump(self.to_dict(), f, indent=2)