<think>我们需要安装适用于Python的PostgreSQL驱动程序，最常用的是psycopg2。 步骤： 1. 安装psycopg2库：可以使用pip直接安装。注意，如果安装纯Python版本（不带C优化），可以安装psycopg2-binary。 2. 连接数据库并执行操作。 以下是详细步骤和示例代码： </think> ### Python 连接 PostgreSQL 数据库指南 Python 连接 PostgreSQL 需要安装适配器库，以下是详细步骤和代码示例： #### **1. 安装 PostgreSQL 适配器** 最常用的库是 `psycopg2`（支持 PostgreSQL 9.5+）或它的简化版本 `psycopg2-binary` ```bash # 使用 pip 安装（推荐） pip install psycopg2 # 或安装预编译版本（无需本地编译依赖） pip install psycopg2-binary ``` > **注意**：如果安装 `psycopg2` 失败，你需要先安装系统依赖： > - **Linux (Debian/Ubuntu)**: `sudo apt-get install libpq-dev python3-dev` > - **macOS (Homebrew)**: `brew install postgresql` > - **Windows**: 直接安装预编译包 (`psycopg2-binary` 是最简单选择) #### **2. 基础连接与查询示例** ```python import psycopg2 try: # 建立数据库连接 conn = psycopg2.connect( dbname='your_database', user='your_username', password='your_password', host='localhost', port=5432 ) # 创建游标执行SQL cur = conn.cursor() cur.execute("SELECT version();") db_version = cur.fetchone() print("PostgreSQL Version:", db_version[0]) # 创建表 cur.execute(""" CREATE TABLE IF NOT EXISTS users ( id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, email VARCHAR(100) UNIQUE ) """) # 插入数据 cur.execute( "INSERT INTO users (name, email) VALUES (%s, %s)", ("Alice", "alice@example.com") ) # 提交事务 conn.commit() # 查询数据 cur.execute("SELECT * FROM users") for row in cur.fetchall(): print(f"ID: {row[0]}, Name: {row[1]}, Email: {row[2]}") except psycopg2.Error as e: print(f"Database error: {e}") finally: # 确保关闭连接 if conn: cur.close() conn.close() print("Connection closed") ``` #### **3. 使用连接池 (推荐生产环境使用)** ```python from psycopg2.pool import SimpleConnectionPool # 创建连接池（最小2个，最大5个连接） POOL = SimpleConnectionPool( 2, 5, dbname='your_db', user='your_user', password='your_pwd', host='localhost' ) try: # 从池中获取连接 conn = POOL.getconn() cur = conn.cursor() cur.execute("SELECT count(*) FROM users") print(f"Total users: {cur.fetchone()[0]}") finally: # 归还连接到池 if conn: cur.close() POOL.putconn(conn) ``` #### **4. 使用 SQLAlchemy ORM (推荐大型项目)** ```python # 安装附加依赖 pip install sqlalchemy psycopg2 from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.orm import declarative_base, sessionmaker Base = declarative_base() # 定义数据模型 class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String(50)) email = Column(String(100), unique=True) # 创建引擎和会话 engine = create_engine('postgresql://user:password@localhost:5432/your_db') Base.metadata.create_all(engine) # 创建表结构 Session = sessionmaker(bind=engine) session = Session() # ORM操作示例 new_user = User(name="Bob", email="bob@example.com") session.add(new_user) session.commit() # 查询所有用户 users = session.query(User).all() for user in users: print(f"User {user.id}: {user.name} <{user.email}>") ``` ### 关键概念说明 1. **连接参数**: - `host`: 数据库服务器地址 - `port`: 默认是 **5432** - `dbname`: 数据库名称 - `user`/`password`: 认证信息 2. **安全实践**: - **永远不要硬编码凭证** - 使用环境变量或配置文件 ```python import os conn = psycopg2.connect(os.getenv("DATABASE_URL")) ``` - 使用 **SSL 连接** 云数据库: ```python conn = psycopg2.connect("host=db.server.com dbname=prod user=admin password=secret sslmode=require") ``` 3. **性能优化**: - 使用连接池避免频繁创建连接 - 对于批量操作使用 `cursor.executemany()` - 设置合适的 `fetchsize` 处理大结果集 ---