迁移服务平台大促

import psycopg2
from psycopg2 import sql

def migrate_data(source_conn_str, dest_conn_str):
    try:
        # 连接到源数据库
        source_conn = psycopg2.connect(source_conn_str)
        source_cursor = source_conn.cursor()
        
        # 连接到目标数据库
        dest_conn = psycopg2.connect(dest_conn_str)
        dest_cursor = dest_conn.cursor()
        
        # 执行数据迁移
        source_cursor.execute("SELECT * FROM old_table")
        rows = source_cursor.fetchall()
        
        for row in rows:
            dest_cursor.execute(
                sql.SQL("INSERT INTO new_table (col1, col2, col3) VALUES (%s, %s, %s)").format(
                    sql.Identifier('col1'), sql.Identifier('col2'), sql.Identifier('col3')
                ), row
            )
        
        # 提交事务并关闭连接
        dest_conn.commit()
        source_cursor.close()
        dest_cursor.close()
        source_conn.close()
        dest_conn.close()
        
        print("数据迁移成功！")
    
    except Exception as e:
        print(f"数据迁移失败：{e}")

# 示例调用
source_connection_string = "dbname=source_db user=user password=password host=localhost port=5432"
destination_connection_string = "dbname=dest_db user=user password=password host=localhost port=5432"
migrate_data(source_connection_string, destination_connection_string)