Python查询PostgreSQL数据库

news2024/12/24 21:45:50

哈喽，大家好，我是木头左！

Python与PostgreSQL的连接

需要了解如何在Python中连接到PostgreSQL数据库。这通常涉及到使用一个库，如psycopg2，它是Python中用于PostgreSQL的最流行的适配器。安装psycopg2非常简单，可以通过pip进行安装：

pip install psycopg2

安装完成后，可以使用以下代码来建立与PostgreSQL数据库的连接：

import psycopg2

try:
    connection = psycopg2.connect(user="your_username",
                                  password="your_password",
                                  host="127.0.0.1",
                                  port="5432",
                                  database="your_database")
    cursor = connection.cursor()
    print("Connected to PostgreSQL database!")
except (Exception, psycopg2.Error) as error:
    print("Error while connecting to PostgreSQL", error)
finally:
    if connection:
        cursor.close()
        connection.close()
        print("PostgreSQL connection is closed!")

这段代码将尝试连接到本地运行的PostgreSQL数据库，并创建一个游标对象，该对象允许执行SQL命令。

执行SQL脚本

一旦建立了连接，就可以使用Python来执行SQL脚本了。这些脚本可以是创建表、插入数据、查询数据等任何有效的SQL命令。以下是一个简单的例子，展示如何创建一个名为users的表，并向其中插入一些数据：

import psycopg2

# 连接到数据库
connection = psycopg2.connect(user="your_username",
                              password="your_password",
                              host="127.0.0.1",
                              port="5432",
                              database="your_database")
cursor = connection.cursor()

# 创建users表
create_table_query = """
CREATE TABLE users (
    id SERIAL PRIMARY KEY,
    name VARCHAR(100),
    age INTEGER
);
"""
cursor.execute(create_table_query)

# 插入数据
insert_data_query = "INSERT INTO users (name, age) VALUES (%s, %s);"
data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
cursor.executemany(insert_data_query, data)
connection.commit()

# 查询数据
select_data_query = "SELECT * FROM users;"
cursor.execute(select_data_query)
rows = cursor.fetchall()
for row in rows:
    print(row)

# 关闭连接
cursor.close()
connection.close()

在这个例子中，首先创建了一个users表，然后插入了一些用户数据，最后执行了一个查询来获取所有用户的信息。通过这种方式，可以非常灵活地执行各种SQL操作。

错误处理和事务管理

在执行数据库操作时，错误处理和事务管理是非常重要的。psycopg2提供了异常类来帮助捕获和处理可能发生的错误。此外，还可以使用commit()和rollback()方法来管理事务。以下是一个包含错误处理和事务管理的示例：

import psycopg2
from psycopg2 import Error

try:
    connection = psycopg2.connect(user="your_username",
                                  password="your_password",
                                  host="127.0.0.1",
                                  port="5432",
                                  database="your_database")
    cursor = connection.cursor()
    
    # 创建表
    create_table_query = """
    CREATE TABLE users (
        id SERIAL PRIMARY KEY,
        name VARCHAR(100),
        age INTEGER
    );
    """
    cursor.execute(create_table_query)
    
    # 插入数据
    insert_data_query = "INSERT INTO users (name, age) VALUES (%s, %s);"
    data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
    cursor.executemany(insert_data_query, data)
    
    # 提交事务
    connection.commit()
    
except Exception as error:
    # 发生错误时回滚事务
    print("An error occurred:", error)
    connection.rollback()
    
finally:
    # 关闭连接
    cursor.close()
    connection.close()