Python+psycopg2：实现跨数据库表两字段数据迁移及注意事项

以下是使用 Python 结合`psycopg2`（以 PostgreSQL 数据库为例）实现将一个数据库中的表的两个字段保存到另一个数据库中的代码示例。 ```python import psycopg2 # 源数据库连接信息 source_db_config = { 'host': 'source_host', 'port': 'source_port', 'database': 'source_database', 'user': 'source_user', 'password': 'source_password' } # 目标数据库连接信息 target_db_config = { 'host': 'target_host', 'port': 'target_port', 'database': 'target_database', 'user': 'target_user', 'password': 'target_password' } try: # 连接到源数据库 source_conn = psycopg2.connect(source_db_config) source_cursor = source_conn.cursor() # 连接到目标数据库 target_conn = psycopg2.connect(target_db_config) target_cursor = target_conn.cursor() # 从源数据库中选择两个字段的数据 select_query = "SELECT column1, column2 FROM source_table" source_cursor.execute(select_query) rows = source_cursor.fetchall() # 将数据插入到目标数据库中 insert_query = "INSERT INTO target_table (column1, column2) VALUES (%s, %s)" for row in rows: target_cursor.execute(insert_query, row) # 提交事务 target_conn.commit() print("数据插入成功") except (Exception, psycopg2.Error) as error: print("发生错误:", error) finally: # 关闭游标和连接 if source_cursor: source_cursor.close() if source_conn: source_conn.close() if target_cursor: target_cursor.close() if target_conn: target_conn.close() ``` 代码 1. 首先定义了源数据库和目标数据库的连接信息。 2. 使用`psycopg2`库分别连接到源数据库和目标数据库。 3. 从源数据库的指定表中选择两个字段的数据。 4. 将选择的数据逐行插入到目标数据库的指定表中。 5. 提交事务以确保数据插入生效。 6. 最后关闭游标和数据库连接。 我在以往的开发过程中，碰到过以下几点需要格外小心处理的地方： 1. 确保源数据库和目标数据库的表结构匹配，尤其是字段名和数据类型。 2. 处理数据插入过程中可能出现的异常，如重复键错误等。 3. 注意数据库连接的关闭，避免资源泄漏。 ######[AI写代码神器 | 625点数解答 | 2025-08-14 11:56:59]