Psycopg3 数据库连接与查询操作实践指南

初识Psycopg3的基本用法

Psycopg3是Python中一个流行的PostgreSQL数据库适配器，它提供了与PostgreSQL数据库交互的接口。对于刚接触Psycopg3的开发者来说，理解其基本用法是至关重要的第一步。

连接数据库与创建表

使用Psycopg3连接数据库非常简单，通过connect()方法即可建立连接。连接成功后，我们可以创建一个游标对象来执行SQL命令：

import psycopg

# 建立数据库连接
with psycopg.connect("dbname=test user=postgres") as conn:
    # 创建游标
    with conn.cursor() as cur:
        # 创建测试表
        cur.execute("""
            CREATE TABLE test (
                id serial PRIMARY KEY,
                num integer,
                data text)
            """)

数据插入与参数化查询

Psycopg3支持参数化查询，这不仅能防止SQL注入，还能自动处理Python对象到PostgreSQL类型的转换：

# 插入单条数据
cur.execute(
    "INSERT INTO test (num, data) VALUES (%s, %s)",
    (100, "abc'def"))

查询数据与游标操作

查询数据时，开发者需要注意游标的操作方式。常见的误区是过早调用fetchone()方法，这会导致后续遍历游标时无法获取数据：

# 查询数据（正确做法）
cur.execute("SELECT * FROM test")
for record in cur:
    print(record)  # 直接遍历游标获取所有记录

# 查询数据（易错做法）
cur.execute("SELECT * FROM test")
first_record = cur.fetchone()  # 获取第一条记录
# 此时游标位置已移动，后续遍历可能无法获取完整数据

批量操作与事务提交

Psycopg3提供了executemany()方法用于批量插入数据，这比循环执行单条插入语句效率更高：

# 批量插入数据
data = [(101, "data1"), (102, "data2"), (103, "data3")]
cur.executemany(
    "INSERT INTO test (num, data) VALUES (%s, %s)",
    data)

所有数据库修改操作（如INSERT、UPDATE、DELETE）都需要显式调用commit()才能使更改永久生效：

# 提交事务
conn.commit()

最佳实践建议

1. 使用上下文管理器：通过with语句管理连接和游标，确保资源正确释放
2. 避免过早获取数据：在需要遍历结果集时，直接迭代游标而不是先调用fetch方法
3. 参数化查询：始终使用参数化查询而非字符串拼接，确保安全性
4. 批量操作：对于大量数据操作，优先考虑executemany()方法
5. 显式提交：记住在修改数据后调用commit()

通过掌握这些基本概念和最佳实践，开发者可以更高效地使用Psycopg3进行PostgreSQL数据库操作，避免常见的陷阱和误区。