Python 连接 PostgreSQL 基础教程

PostgreSQL 是一个功能强大的开源关系型数据库管理系统（RDBMS），而 Python 是一种流行的编程语言，用于开发各种应用程序。本教程将详细介绍如何使用 Python 连接到 PostgreSQL 数据库，并进行各种数据库操作。

我们将从设置先决条件开始，然后逐步展示如何创建连接、执行查询、插入数据以及关闭连接。最后，我们将提供一个简单的示例，演示如何连接到一个名为 “user” 的表格，并查询其中的数据。

先决条件

为了连接 PostgreSQL 数据库，您需要具备以下先决条件：

Python 安装：确保您的计算机上安装了 Python。您可以从 Python 官方网站下载并安装 Python。
PostgreSQL 数据库安装：您需要在计算机上安装 PostgreSQL 数据库。您可以从 PostgreSQL 官方网站下载并安装适用于您的操作系统的 PostgreSQL 版本。
Python psycopg2 模块：确保您的 Python 安装了 psycopg2 模块，这是与 PostgreSQL 数据库交互的 Python 驱动。

您可以使用以下命令安装 psycopg2 模块：

pip install psycopg2

连接 PostgreSQL 的步骤

现在，让我们深入了解如何使用 Python 连接到 PostgreSQL 数据库。

步骤 1: 创建 PostgreSQL 连接

首先，您需要创建一个连接到 PostgreSQL 数据库的连接对象。使用 psycopg2 模块来完成这一步骤，并提供连接详细信息。

import psycopg2

# 创建 PostgreSQL 连接
connection = psycopg2.connect(
    user="your_username",
    password="your_password",
    host="your_host",
    port="your_port",
    database="your_database"
)

在上面的代码中，your_username 和 your_password 是登录 PostgreSQL 数据库的凭据，your_host 和 your_port 是数据库服务器的主机名和端口号，your_database 是要连接的数据库名称。根据您的 PostgreSQL 数据库设置，可能需要更改这些值。

步骤 2: 创建游标对象

要执行 SQL 查询，您需要创建一个游标对象。游标充当与数据库进行交互的工具。

# 创建游标
cursor = connection.cursor()

步骤 3: 执行查询

现在，您可以使用游标对象执行 SQL 查询。以下是一个示例，演示如何查询 “user” 表格中的数据：

# 执行查询
query = "SELECT username, email FROM user"
cursor.execute(query)

# 获取查询结果
print("查询结果：")
for (username, email) in cursor:
    print("Username:", username)
    print("Email:", email)

步骤 4: 插入数据

要向表格中插入新数据，您可以使用 INSERT INTO SQL 语句。以下是一个示例，演示如何插入一条用户记录到 “user” 表格中：

# 插入一条数据
insert_query = "INSERT INTO user (username, email) VALUES (%s, %s)"
data = ("new_user", "[email protected]")
cursor.execute(insert_query, data)

# 提交更改
connection.commit()

步骤 5: 关闭连接

当您完成所有数据库操作后，务必关闭 PostgreSQL 连接，以释放资源并确保数据的完整性。

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

示例

以下是一个完整的示例，演示如何连接到 PostgreSQL 数据库、执行查询、插入一条用户记录以及关闭连接：

import psycopg2

# 创建 PostgreSQL 连接
connection = psycopg2.connect(
    user="your_username",
    password="your_password",
    host="your_host",
    port="your_port",
    database="your_database"
)

# 创建游标
cursor = connection.cursor()

# 执行查询
query = "SELECT username, email FROM user"
cursor.execute(query)

# 获取查询结果
print("查询结果：")
for (username, email) in cursor:
    print("Username:", username)
    print("Email:", email)

# 插入一条数据
insert_query = "INSERT INTO user (username, email) VALUES (%s, %s)"
data = ("new_user", "[email protected]")
cursor.execute(insert_query, data)

# 提交更改
connection.commit()

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

总结

本教程介绍了如何使用 Python 连接到 PostgreSQL 数据库。通过遵循上述步骤，您可以开始在 Python 中执行各种数据库操作，并根据您的需求访问和管理数据。希望本文对您在 PostgreSQL 数据库编程方面有所帮助。