迁移服务平台双11活动

import psycopg2

def migrate_data(source_conn_str, target_conn_str):
    try:
        # 连接源数据库
        source_conn = psycopg2.connect(source_conn_str)
        source_cursor = source_conn.cursor()
        
        # 连接目标数据库
        target_conn = psycopg2.connect(target_conn_str)
        target_cursor = target_conn.cursor()
        
        # 执行数据迁移
        source_cursor.execute("SELECT * FROM source_table")
        rows = source_cursor.fetchall()
        
        for row in rows:
            target_cursor.execute("INSERT INTO target_table VALUES (%s, %s, %s)", row)
        
        # 提交事务
        target_conn.commit()
        
        print("数据迁移成功！")
    except Exception as e:
        print(f"数据迁移失败: {e}")
    finally:
        source_cursor.close()
        source_conn.close()
        target_cursor.close()
        target_conn.close()

# 示例连接字符串
source_conn_str = "dbname=source_db user=user password=password host=localhost port=5432"
target_conn_str = "dbname=target_db user=user password=password host=localhost port=5432"

migrate_data(source_conn_str, target_conn_str)