import py_compile
import os
import hashlib

# 配置
script_dir = 'script'  # 输入文件夹
output_dir = 'out_'       # 输出根目录
preserve_structure = True  # 是否保留文件夹结构（True/False）

# 确保 script 文件夹存在
if not os.path.exists(script_dir):
    print(f"Error: Directory '{script_dir}' does not exist.")
    exit(1)

# 遍历 script 文件夹及其子文件夹
for root, _, files in os.walk(script_dir):
    for filename in files:
        if filename.endswith('.py'):
            # 构造完整的 .py 文件路径
            py_file = os.path.join(root, filename)
            
            # 计算相对路径
            relative_path = os.path.relpath(py_file, script_dir)
            base_name = os.path.splitext(filename)[0]
            
            if preserve_structure:
                # 保留文件夹结构：输出到 output_dir 下的相对路径
                pyc_relative_dir = os.path.dirname(relative_path)
                pyc_dir = os.path.join(output_dir, pyc_relative_dir)
                os.makedirs(pyc_dir, exist_ok=True)
                pyc_filename = f"{base_name}.pyc"
                pyc_file = os.path.join(pyc_dir, pyc_filename)
            else:
                # 不保留结构：平铺到 output_dir，使用哈希前缀避免冲突
                hash_prefix = hashlib.md5(relative_path.encode()).hexdigest()[:8]
                pyc_filename = f"{hash_prefix}_{base_name}.pyc"
                pyc_file = os.path.join(output_dir, pyc_filename)
            
            try:
                # 编译 .py 文件为 .pyc 文件
                py_compile.compile(py_file, cfile=pyc_file)
                print(f"Compiled {py_file} to {pyc_file}")
            except Exception as e:
                print(f"Failed to compile {py_file}: {e}")