按数据库拆分包含多个数据库的mysqldump文件

mysqldump -u 用户名 -p 数据库名 > 导出文件名.sql

import os

filename = 'mysqldump.sql'
output_dir = 'output'

if not os.path.exists(output_dir):
    os.makedirs(output_dir)

with open(filename, 'r') as f:
    lines = f.readlines()

current_db = None
current_file = None

for line in lines:
    if line.startswith('CREATE DATABASE'):
        db_name = line.split(' ')[2].strip(';')
        current_db = db_name
        current_file = open(os.path.join(output_dir, f'{db_name}.sql'), 'w')
    elif line.startswith('USE'):
        db_name = line.split(' ')[1].strip(';')
        current_db = db_name
        current_file = open(os.path.join(output_dir, f'{db_name}.sql'), 'w')
    elif line.startswith('INSERT INTO'):
        table_name = line.split(' ')[2].strip('`').strip(';')
        current_file.write(f'USE {current_db};\n')
        current_file.write(f'INSERT INTO `{table_name}` VALUES\n')
    else:
        current_file.write(line)

if current_file:
    current_file.close()