Python实现一键生成自解压源码文件并打包项目

python self_extract.py

# build_self_extract.py
from pathlib import Path
import json

SOURCE_DIR = Path("./")
OUTPUT_DIR = Path("build")
OUTPUT_PY = OUTPUT_DIR / "self_extract.py"

INCLUDE_EXT = {".py", ".json"}  # 需要打包的源码类型
EXCLUDE_DIRS = {
    ".git", ".venv", "venv", "__pycache__", ".history", "build"
}
EXCLUDE_FILES = {
    "self_extract.py", "build_self_extract.py", "111.py"
}

OUTPUT_DIR.mkdir(exist_ok=True)

files_data = {}

for file in SOURCE_DIR.rglob("*"):
    if file.is_dir():
        continue

    if file.suffix not in INCLUDE_EXT:
        continue

    if any(part in EXCLUDE_DIRS for part in file.parts):
        continue

    if file.name in EXCLUDE_FILES:
        continue

    rel_path = file.relative_to(SOURCE_DIR)
    files_data[str(rel_path)] = file.read_text(
        encoding="utf-8", errors="ignore"
    )

# 生成自解压 py
with OUTPUT_PY.open("w", encoding="utf-8") as f:
    f.write(
        f'''"""
🚀 自解压源码文件
运行后将还原所有源代码并生成 source_code.zip
"""

from pathlib import Path
import zipfile
import json

FILES = json.loads({json.dumps(json.dumps(files_data, ensure_ascii=False))})

BASE_DIR = Path("extracted_source")
ZIP_NAME = "source_code.zip"

def main():
    BASE_DIR.mkdir(exist_ok=True)

    # 写回所有文件
    for path, content in FILES.items():
        file_path = BASE_DIR / path
        file_path.parent.mkdir(parents=True, exist_ok=True)
        file_path.write_text(content, encoding="utf-8", errors="replace")

    # 打包为 zip
    with zipfile.ZipFile(ZIP_NAME, "w", zipfile.ZIP_DEFLATED) as zf:
        for file in BASE_DIR.rglob("*"):
            if file.is_file():
                zf.write(file, arcname=file.relative_to(BASE_DIR))

    print("✅ 源代码已还原并压缩为", ZIP_NAME)

if __name__ == "__main__":
    main()
'''
    )

print(f"✅ 已生成自解压文件: {OUTPUT_PY}")

python build_self_extract.py

python self_extract.py