如何用Python操作PostgreSQL数据库？

PostgreSQL作为一款功能强大的开源关系型数据库，凭借其稳定性、可扩展性和对SQL标准的完整支持，成为许多开发者的。Python通过psycopg2库提供了与PostgreSQL数据库交互的便捷方式，无论是数据分析、Web开发还是自动化脚本，都能高效完成数据库操作。如何使用Python连接PostgreSQL、执行CRUD操作以及处理事务，帮助开发者快速掌握这一实用技能。

一、安装必要的库

在开始之前，确保已安装Python和PostgreSQL环境。通过pip安装psycopg2库：

pip install psycopg2-binary

如果需要更高级的功能（如连接池），可以选择安装psycopg2完整版：

pip install psycopg2

二、连接PostgreSQL数据库

使用psycopg2连接数据库需要提供主机名、数据库名、用户名和密码等信息。以下是一个基本连接示例：

import psycopg2

conn = psycopg2.connect(
    host="localhost",
    database="mydatabase",
    user="postgres",
    password="yourpassword"
)
cursor = conn.cursor()
print("数据库连接成功！")

注意事项：

生产环境中建议将敏感信息存储在环境变量中。
操作完成后需手动关闭连接：cursor.close()和conn.close()。

三、执行SQL查询

通过游标（cursor）可以执行SQL语句并获取结果。以下示例演示查询和插入操作：

1. 查询数据

cursor.execute("SELECT * FROM employees")
rows = cursor.fetchall()
for row in rows:
    print(row)

2. 插入数据

cursor.execute(
    "INSERT INTO employees (name, department) VALUES (%s, %s)",
    ("张三", "技术部")
)
conn.commit()  # 提交事务

四、事务管理

PostgreSQL默认开启事务，需显式提交或回滚：

try:
    cursor.execute("UPDATE accounts SET balance = balance - 100 WHERE user_id = 1")
    cursor.execute("UPDATE accounts SET balance = balance + 100 WHERE user_id = 2")
    conn.commit()  # 提交事务
except Exception as e:
    conn.rollback()  # 发生错误时回滚
    print(f"事务失败: {e}")

五、使用上下文管理器简化操作

通过with语句自动管理连接和事务：

with psycopg2.connect(database="mydb") as conn:
    with conn.cursor() as cursor:
        cursor.execute("SELECT version()")
        print(cursor.fetchone())

这种方式无需手动关闭连接或处理异常。

六、高级功能

1. 批量插入

使用executemany提升插入效率：

data = [("李四", "市场部"), ("王五", "人事部")]
cursor.executemany(
    "INSERT INTO employees (name, department) VALUES (%s, %s)",
    data
)

2. 返回插入的ID

获取自增主键值：

cursor.execute(
    "INSERT INTO products (name) VALUES (%s) RETURNING id",
    ("笔记本电脑",)
)
product_id = cursor.fetchone()[0]

通过以上步骤，你已经掌握了Python操作PostgreSQL的核心方法。实际开发中，还可以结合ORM工具（如SQLAlchemy或Django ORM）进一步简化操作。根据项目需求选择合适的方式，让数据库交互更加高效可靠。

(www.nzw6.com)