sql数据库迁移到云服务器

import psycopg2
from psycopg2 import sql

# 连接到本地数据库
local_conn = psycopg2.connect(
    dbname="local_db",
    user="local_user",
    password="local_password",
    host="localhost",
    port="5432"
)

# 连接到腾讯云数据库
cloud_conn = psycopg2.connect(
    dbname="cloud_db",
    user="cloud_user",
    password="cloud_password",
    host="cloud_host",
    port="5432"
)

# 创建游标
local_cursor = local_conn.cursor()
cloud_cursor = cloud_conn.cursor()

# 导出本地数据库表结构
local_cursor.execute("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = 'your_table'")
columns = local_cursor.fetchall()

# 在云数据库中创建表
create_table_sql = sql.SQL("CREATE TABLE your_table ({})").format(
    sql.SQL(', ').join(
        sql.SQL("{} {}").format(sql.Identifier(col[0]), sql.SQL(col[1])) for col in columns
    )
)
cloud_cursor.execute(create_table_sql)

# 导出本地数据并插入云数据库
local_cursor.execute("SELECT * FROM your_table")
rows = local_cursor.fetchall()
insert_sql = sql.SQL("INSERT INTO your_table ({}) VALUES ({})").format(
    sql.SQL(', ').join(map(sql.Identifier, [col[0] for col in columns])),
    sql.SQL(', ').join(sql.Placeholder() * len(columns))
)
cloud_cursor.executemany(insert_sql, rows)

# 提交事务并关闭连接
cloud_conn.commit()
local_cursor.close()
cloud_cursor.close()
local_conn.close()
cloud_conn.close()