sql文件拆分

for i in range(1, 2):
# 读取原始 SQL 文件
input_file = ‘output_’+str(i)+‘.sql’
output_file = ‘output2_’+str(i)+‘.sql’

batch_size = 1000

with open(input_file, 'r', encoding='utf-8') as file:
    lines = file.readlines()

with open(output_file, 'w', encoding='utf-8') as file:
    file.write(lines[0])  # 写入第一行

    for i in range(1, len(lines), batch_size):
        batch = lines[i:i+batch_size]
        file.write('INSERT INTO `table_name` VALUES ')
        values = []
        for line in batch:
            values.append(line.split('VALUES', 1)[1][1:-2])
        file.write(','.join(values).rstrip(';'))
        file.write(';\n')  # 写入合并后的SQL语句
        file.write('COMMIT;\n')  # 执行COMMIT语句

    print("转换完成并已写入到"+output_file+"文件中")