MySQL迁移年末促销

import mysql.connector

# 导出数据库
def export_database(host, user, password, database, filename):
    conn = mysql.connector.connect(host=host, user=user, password=password, database=database)
    cursor = conn.cursor()
    with open(filename, 'w') as f:
        for line in cursor.execute("SHOW TABLES"):
            table = line[0]
            cursor.execute(f"SELECT * FROM {table}")
            rows = cursor.fetchall()
            for row in rows:
                f.write(f"{table},{','.join(map(str, row))}\n")
    cursor.close()
    conn.close()

# 导入数据库
def import_database(host, user, password, database, filename):
    conn = mysql.connector.connect(host=host, user=user, password=password, database=database)
    cursor = conn.cursor()
    with open(filename, 'r') as f:
        for line in f:
            table, data = line.strip().split(',', 1)
            values = tuple(map(str.strip, data.split(',')))
            placeholders = ', '.join(['%s'] * len(values))
            query = f"INSERT INTO {table} VALUES ({placeholders})"
            cursor.execute(query, values)
    conn.commit()
    cursor.close()
    conn.close()

# 使用示例
export_database('old_host', 'user', 'password', 'old_db', 'exported_data.txt')
import_database('new_host', 'user', 'password', 'new_db', 'exported_data.txt')