快捷導(dǎo)航

Python進行PostgreSQL數(shù)據(jù)庫連接的詳細使用指南

更新時間：2025年05月18日 09:02:18 作者：懶大王愛吃狼

在Python中連接PostgreSQL數(shù)據(jù)庫,最常用的庫是psycopg2,本文為大家詳細介紹了Python使用psycopg2庫操作PostgreSQL的詳細步驟,需要的可以了解下

安裝psycopg2

首先需要安裝psycopg2庫：

pip install psycopg2
# 或者使用二進制版本（安裝更快）
pip install psycopg2-binary

基本連接與操作

1. 建立數(shù)據(jù)庫連接

import psycopg2

# 建立連接
conn = psycopg2.connect(
    dbname="your_database",
    user="your_username",
    password="your_password",
    host="your_host",
    port="your_port"
)

# 創(chuàng)建游標(biāo)對象
cur = conn.cursor()

2. 執(zhí)行SQL查詢

# 執(zhí)行簡單查詢
cur.execute("SELECT * FROM your_table LIMIT 5;")

# 獲取結(jié)果
rows = cur.fetchall()
for row in rows:
    print(row)

3. 執(zhí)行參數(shù)化查詢（防止SQL注入）

# 使用參數(shù)化查詢
user_id = 5
cur.execute("SELECT * FROM users WHERE id = %s;", (user_id,))
user = cur.fetchone()
print(user)

4. 插入數(shù)據(jù)

# 插入單條數(shù)據(jù)
cur.execute(
    "INSERT INTO users (name, email) VALUES (%s, %s) RETURNING id;",
    ('John Doe', 'john@example.com')
)
user_id = cur.fetchone()[0]
conn.commit()  # 必須提交事務(wù)
print(f"插入的用戶ID: {user_id}")

???????# 批量插入
users_data = [
    ('Alice', 'alice@example.com'),
    ('Bob', 'bob@example.com'),
    ('Charlie', 'charlie@example.com')
]
cur.executemany(
    "INSERT INTO users (name, email) VALUES (%s, %s);",
    users_data
)
conn.commit()

5. 更新數(shù)據(jù)

cur.execute(
    "UPDATE users SET email = %s WHERE id = %s;",
    ('new_email@example.com', 1)
)
conn.commit()

6. 刪除數(shù)據(jù)

cur.execute(
    "DELETE FROM users WHERE id = %s;",
    (5,)
)
conn.commit()

使用上下文管理器（推薦）

# 使用with語句自動管理連接
with psycopg2.connect(
    dbname="your_database",
    user="your_username",
    password="your_password",
    host="your_host"
) as conn:
    with conn.cursor() as cur:
        cur.execute("SELECT * FROM users;")
        for row in cur:
            print(row)
    # 不需要顯式調(diào)用commit()或close()，with語句會自動處理

使用連接池（適用于Web應(yīng)用）

對于Web應(yīng)用等需要頻繁連接數(shù)據(jù)庫的場景，可以使用連接池：

from psycopg2 import pool

???????# 創(chuàng)建連接池
connection_pool = pool.SimpleConnectionPool(
    minconn=1,
    maxconn=10,
    dbname="your_database",
    user="your_username",
    password="your_password",
    host="your_host"
)

# 從連接池獲取連接
conn = connection_pool.getconn()
cur = conn.cursor()
cur.execute("SELECT * FROM users;")
# ... 執(zhí)行操作 ...

???????# 將連接返回給連接池
connection_pool.putconn(conn)

使用SQLAlchemy（ORM方式）

如果你更喜歡使用ORM，可以安裝SQLAlchemy：

pip install sqlalchemy psycopg2-binary

然后使用：

from sqlalchemy import create_engine, text

# 創(chuàng)建引擎
engine = create_engine('postgresql://user:password@localhost:5432/dbname')

# 執(zhí)行查詢
with engine.connect() as connection:
    result = connection.execute(text("SELECT * FROM users;"))
    for row in result:
        print(row)