import os

def merge_sql_files(output_filename="merged_all.sql"):
    # 获取当前目录下所有的文件
    files = [f for f in os.listdir('.') if f.endswith('.sql')]
    
    # 排除掉输出文件本身，防止无限递归写入
    if output_filename in files:
        files.remove(output_filename)
        
    # 按文件名排序（确保合并顺序可控）
    files.sort()

    if not files:
        print("当前目录下没有发现 .sql 文件。")
        return

    with open(output_filename, 'w', encoding='utf-8') as outfile:
        for filename in files:
            with open(filename, 'r', encoding='utf-8') as infile:
                # 写入文件名注释，方便追踪来源
                outfile.write(f"\n-- --- START OF FILE: {filename} ---\n")
                outfile.write(infile.read())
                outfile.write("\n-- --- END OF FILE ---\n")
            print(f"已合并: {filename}")

    print(f"\n成功！所有 SQL 文件已整合至: {output_filename}")

if __name__ == "__main__":
    merge_sql_files()