datai/datai-scenes/datai-scene-salesforce/docs/scripts/整合所有sql.py

30 lines
1.0 KiB
Python

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()