PyInstallerEx 0.1.6__py3-none-any.whl

PyInstallerEx/utils/__init__.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+"""
+PyInstallerEx工具函数模块
+"""
+
+from .logging import setup_logging, PackageError, handle_error
+from .system_utils import SystemInfo
+from .file_utils import FileUtils
+from .compression import CompressionUtils
+
+__all__ = [
+    "setup_logging", "PackageError", "handle_error",
+    "SystemInfo", "FileUtils", "CompressionUtils"
+]

PyInstallerEx/utils/compression.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+"""
+压缩和解压工具模块
+"""
+
+from __future__ import print_function
+import zipfile
+import shutil
+import os
+
+
+class CompressionUtils(object):
+    """压缩工具类"""
+    
+    @staticmethod
+    def create_zip_from_directory(source_dir, zip_path):
+        """将目录压缩为ZIP文件"""
+        def add_files_to_zip(zipf, folder, base_path=""):
+            for item in os.listdir(folder):
+                item_path = os.path.join(folder, item)
+                arcname = os.path.join(base_path, item)
+                
+                if os.path.isfile(item_path):
+                    zipf.write(item_path, arcname)
+                elif os.path.isdir(item_path):
+                    add_files_to_zip(zipf, item_path, arcname)
+        
+        with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            add_files_to_zip(zipf, source_dir)
+    
+    @staticmethod
+    def combine_binary_and_zip(binary_path, zip_path, output_path):
+        """将二进制文件和ZIP文件合并为单个二进制文件"""
+        # 简单的文件合并：二进制文件 + ZIP文件
+        with open(output_path, 'wb') as outfile:
+            # 先写入二进制启动器
+            with open(binary_path, 'rb') as binfile:
+                outfile.write(binfile.read())
+            
+            # 写入分隔符（可选，用于识别ZIP部分）
+            outfile.write(b'\x50\x4B\x03\x04')  # ZIP文件头作为分隔符
+            
+            # 再写入ZIP文件内容
+            with open(zip_path, 'rb') as zipfile:
+                outfile.write(zipfile.read())
+    
+    @staticmethod
+    def extract_zip_from_combined(combined_path, output_dir):
+        """从合并文件中提取ZIP部分"""
+        # 这里需要根据实际的合并格式来实现
+        # 简化实现：假设ZIP部分在特定偏移量开始
+        with open(combined_path, 'rb') as infile:
+            content = infile.read()
+        
+        # 查找ZIP文件头位置
+        zip_header = b'\x50\x4B\x03\x04'
+        zip_start = content.find(zip_header)
+        
+        if zip_start == -1:
+            raise ValueError("Could not find ZIP section in combined file")
+        
+        # 提取ZIP部分
+        zip_content = content[zip_start:]
+        zip_output_path = os.path.join(output_dir, "extracted.zip")
+        
+        with open(zip_output_path, 'wb') as zipfile:
+            zipfile.write(zip_content)
+        
+        return zip_output_path

PyInstallerEx/utils/file_utils.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+"""
+文件操作工具函数
+"""
+
+from __future__ import print_function
+import hashlib
+import shutil
+import os
+
+
+class FileUtils(object):
+    """文件操作工具类"""
+    
+    @staticmethod
+    def calculate_md5(file_path):
+        """计算文件的MD5哈希值"""
+        hash_md5 = hashlib.md5()
+        with open(file_path, "rb") as f:
+            for chunk in iter(lambda: f.read(4096), b""):
+                hash_md5.update(chunk)
+        return hash_md5.hexdigest()
+    
+    @staticmethod
+    def copy_config_to_dir(config_content, target_dir, config_name="config.json"):
+        """将配置内容复制到目标目录"""
+        config_path = os.path.join(target_dir, config_name)
+        with open(config_path, 'w') as f:
+            f.write(config_content)
+        return config_path
+    
+    @staticmethod
+    def ensure_directory_exists(directory):
+        """确保目录存在，不存在则创建"""
+        if not os.path.exists(directory):
+            os.makedirs(directory)
+    
+    @staticmethod
+    def clean_directory(directory):
+        """清空目录内容"""
+        if os.path.exists(directory):
+            shutil.rmtree(directory)
+        os.makedirs(directory)
+    
+    @staticmethod
+    def get_script_name(script_path):
+        """从脚本路径获取脚本名称（不含扩展名）"""
+        base_name = os.path.basename(script_path)
+        script_name = os.path.splitext(base_name)[0]
+        return script_name

PyInstallerEx/utils/icon_helper.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+"""
+图标设置辅助工具
+用于在 Windows 上为 exe 文件设置图标
+"""
+
+import os
+import sys
+import subprocess
+
+
+def set_icon_with_resource_hacker(exe_path, icon_path, resource_hacker_path=None):
+    """
+    使用 Resource Hacker 为 exe 文件设置图标
+    
+    Args:
+        exe_path: exe 文件路径
+        icon_path: ico 图标文件路径
+        resource_hacker_path: Resource Hacker 可执行文件路径（可选）
+    
+    Returns:
+        bool: 是否成功
+    """
+    # 查找 Resource Hacker
+    if resource_hacker_path is None:
+        # 常见安装路径
+        common_paths = [
+            r"C:\Program Files (x86)\Resource Hacker\ResourceHacker.exe",
+            r"C:\Program Files\Resource Hacker\ResourceHacker.exe",
+            os.path.join(os.environ.get('LOCALAPPDATA', ''), 'Resource Hacker', 'ResourceHacker.exe'),
+        ]
+        
+        for path in common_paths:
+            if os.path.exists(path):
+                resource_hacker_path = path
+                break
+        
+        if not resource_hacker_path or not os.path.exists(resource_hacker_path):
+            print("Error: Resource Hacker not found. Please install it from:")
+            print("http://www.angusj.com/resourcehacker/")
+            return False
+    
+    # 构建命令
+    # ResourceHacker.exe -open input.exe -save output.exe -action addoverwrite -res icon.ico -mask ICONGROUP,MAINICON,
+    cmd = [
+        resource_hacker_path,
+        "-open", exe_path,
+        "-save", exe_path,
+        "-action", "addoverwrite",
+        "-res", icon_path,
+        "-mask", "ICONGROUP,MAINICON,"
+    ]
+    
+    try:
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        if result.returncode == 0:
+            print("Icon set successfully using Resource Hacker")
+            return True
+        else:
+            print("Error: Resource Hacker failed")
+            print(result.stderr)
+            return False
+    except Exception as e:
+        print("Error running Resource Hacker: {0}".format(e))
+        return False
+
+
+if __name__ == "__main__":
+    if len(sys.argv) < 3:
+        print("Usage: python icon_helper.py <exe_path> <icon_path> [resource_hacker_path]")
+        sys.exit(1)
+    
+    exe_path = sys.argv[1]
+    icon_path = sys.argv[2]
+    rh_path = sys.argv[3] if len(sys.argv) > 3 else None
+    
+    success = set_icon_with_resource_hacker(exe_path, icon_path, rh_path)
+    sys.exit(0 if success else 1)

PyInstallerEx/utils/logging.py

@@ -0,0 +1,154 @@
+# -*- coding: utf-8 -*-
+"""
+日志记录和错误处理模块
+"""
+
+from __future__ import print_function
+import sys
+
+
+def setup_logging(verbose=False):
+    """设置统一的日志格式"""
+    # Python 2.7 兼容性处理 - 动态导入 logging 模块并添加缺失的功能
+    import logging
+    
+    # 确保在Python 2.7中也能正确导入logging模块常量
+    if not hasattr(logging, 'INFO'):
+        logging.INFO = 20
+    if not hasattr(logging, 'DEBUG'):
+        logging.DEBUG = 10
+    if not hasattr(logging, 'WARNING'):
+        logging.WARNING = 30
+    if not hasattr(logging, 'ERROR'):
+        logging.ERROR = 40
+    if not hasattr(logging, 'CRITICAL'):
+        logging.CRITICAL = 50
+    
+    level = logging.DEBUG if verbose else logging.INFO
+    
+    # 确保至少有一个处理器
+    try:
+        if hasattr(logging, 'getLogger'):
+            root_logger = logging.getLogger()
+        else:
+            # 如果没有 getLogger，尝试使用 root 属性
+            root_logger = getattr(logging, 'root', None)
+            
+        if root_logger is not None:
+            if not hasattr(root_logger, 'handlers') or not root_logger.handlers:
+                # 添加一个默认的处理器以避免 "No handlers could be found" 警告
+                try:
+                    if hasattr(logging, 'StreamHandler'):
+                        handler = logging.StreamHandler(sys.stdout)
+                        if hasattr(logging, 'Formatter'):
+                            formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+                            handler.setFormatter(formatter)
+                        root_logger.addHandler(handler)
+                except:
+                    # 如果无法创建处理器，至少确保日志级别被设置
+                    pass
+            root_logger.setLevel(level)
+    except:
+        # 忽略日志初始化错误
+        pass
+    
+    # Python 2.7 兼容性处理
+    if hasattr(logging, 'basicConfig'):
+        # Python 2.7.9+ 和 Python 3 中有 basicConfig 函数
+        try:
+            # 尝试使用 handlers 参数
+            logging.basicConfig(
+                level=level,
+                format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+                handlers=[logging.StreamHandler(sys.stdout)]
+            )
+        except (TypeError, AttributeError):
+            # 更早版本的 Python 2.7 不支持 handlers 参数
+            try:
+                logging.basicConfig(
+                    level=level,
+                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+                )
+            except:
+                pass
+    else:
+        # 如果没有 basicConfig 函数，则手动配置根日志记录器
+        try:
+            if hasattr(logging, 'getLogger'):
+                root_logger = logging.getLogger()
+            else:
+                root_logger = getattr(logging, 'root', None)
+                
+            if root_logger is not None:
+                root_logger.setLevel(level)
+                
+                # 清除现有的处理器
+                if hasattr(root_logger, 'handlers'):
+                    for handler in root_logger.handlers[:]:
+                        root_logger.removeHandler(handler)
+                    
+                    # 添加新的处理器
+                    if hasattr(logging, 'StreamHandler'):
+                        try:
+                            handler = logging.StreamHandler(sys.stdout)
+                        except:
+                            # 如果无法创建 StreamHandler，使用备用方案
+                            handler = None
+                    else:
+                        handler = None
+                        
+                    if handler is not None:
+                        if hasattr(logging, 'Formatter'):
+                            try:
+                                formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+                                handler.setFormatter(formatter)
+                            except:
+                                pass  # 忽略格式化器错误
+                        
+                        try:
+                            root_logger.addHandler(handler)
+                        except:
+                            pass  # 忽略添加处理器的错误
+        except:
+            pass
+
+
+class PackageError(Exception):
+    """自定义打包异常"""
+    pass
+
+
+def handle_error(error, message=""):
+    """统一的错误处理"""
+    # Python 2.7 兼容性处理 - 动态导入 logging 模块
+    import logging
+    
+    # 尝试使用 logging 记录错误，如果失败则直接打印
+    try:
+        # 检查是否有 getLogger 函数
+        if hasattr(logging, 'getLogger'):
+            try:
+                logger = logging.getLogger(__name__)
+            except:
+                # 如果无法获取特定名称的 logger，则使用 None
+                logger = None
+        else:
+            logger = None
+            
+        if logger is not None:
+            # 尝试记录错误日志
+            try:
+                logger.error("{0}: {1}".format(message, error))
+            except:
+                # 如果记录日志失败，则使用 print
+                print("ERROR: {0}: {1}".format(message, error))
+        else:
+            # 直接使用 print 输出错误
+            print("ERROR: {0}: {1}".format(message, error))
+    except:
+        # 最后的备用方案
+        print("ERROR: {0}: {1}".format(message, error))
+        
+    # Python 2兼容的异常抛出方式
+    new_error = PackageError("{0}: {1}".format(message, error))
+    raise new_error

PyInstallerEx/utils/system_utils.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+"""
+系统检测和平台相关工具
+"""
+
+from __future__ import print_function
+import platform
+import sys
+import tempfile
+import os
+
+
+class SystemInfo(object):
+    """系统信息检测类"""
+    
+    @staticmethod
+    def get_platform():
+        """获取当前平台标识"""
+        system = platform.system().lower()
+        arch = platform.machine().lower()
+        
+        if system == "windows":
+            return "win"
+        elif system == "linux":
+            if "arm" in arch or "aarch" in arch:
+                # 根据架构位数判断是arm还是arm64
+                if "64" in arch or "aarch64" in arch:
+                    return "linux_arm64"
+                else:
+                    return "linux_arm"
+            else:
+                return "linux_x86"
+        else:
+            raise ValueError("Unsupported platform: {0}_{1}".format(system, arch))
+    
+    @staticmethod
+    def get_temp_dir():
+        """获取系统临时目录"""
+        return tempfile.gettempdir()
+    
+    @staticmethod
+    def get_binary_name(platform_type):
+        """根据平台类型获取对应的二进制启动器名称"""
+        binary_map = {
+            "win": "launcher_windows.exe",
+            "linux_x86": "launcher_linux_x86",
+            "linux_arm": "launcher_linux_arm",
+            "linux_arm64": "launcher_linux_arm64"
+        }
+        
+        if platform_type not in binary_map:
+            raise ValueError("Unsupported platform type: {0}".format(platform_type))
+        
+        return binary_map[platform_type]
+    
+    @staticmethod
+    def is_windows():
+        """判断是否为Windows系统"""
+        return platform.system().lower() in ["windows", "win32", "win64"]