import psycopg2
import pymysql
# 连接PostgreSQL
def connect_postgresql():
conn = psycopg2.connect(
dbname="your_dbname",
user="your_username",
password="your_password",
host="your_host",
port="your_port"
)
return conn
# 连接MySQL
def connect_mysql():
conn = pymysql.connect(
host="your_host",
user="your_username",
password="your_password",
db="your_dbname",
charset='utf8mb4'
)
return conn
# 向PostgreSQL批量插入数据
def insert_data_postgresql(data):
conn = connect_postgresql()
cur = conn.cursor()
for row in data:
cur.execute("INSERT INTO your_table (column1, column2) VALUES (%s, %s)", row)
conn.commit()
conn.close()
# 向MySQL批量插入数据
def insert_data_mysql(data):
conn = connect_mysql()
cur = conn.cursor()
for row in data:
cur.execute("INSERT INTO your_table (column1, column2) VALUES (%s, %s)", row)
conn.commit()
conn.close()
# 示例数据
data_postgresql = [(1, 'Alice'), (2, 'Bob')]
data_mysql = [(3, 'Charlie'), (4, 'David')]
# 执行批量插入
insert_data_postgresql(data_postgresql)
insert_data_mysql(data_mysql)
在这个例子中，我们定义了两个函数connect_postgresql和connect_mysql来分别连接PostgreSQL和MySQL数据库。然后定义了两个函数insert_data_postgresql和insert_data_mysql来批量向对应的数据库表中插入数据。最后，我们使用示例数据执行了这两个函数。在实际应用中，你需要替换your_dbname, your_username, your_password, your_host, your_port, your_table, column1, column2以及data变量的内容。