Python连接Postgres

Python连接Postgres

PostgreSQL是一个功能强大、开源的对象关系型数据库系统。在Python中，我们可以使用psycopg2库来连接PostgreSQL数据库，并执行SQL语句。

安装psycopg2库

首先，我们需要安装psycopg2库。可以通过pip来安装：

pip install psycopg2

连接到数据库

在Python中，我们可以使用psycopg2库中的connect()方法来连接到PostgreSQL数据库。我们需要提供主机名、数据库名、用户名和密码等连接信息。

import psycopg2

# 连接到数据库
conn = psycopg2.connect(
    dbname="mydatabase",
    user="myuser",
    password="mypassword",
    host="localhost"
)

# 创建一个cursor对象
cur = conn.cursor()

# 执行SQL查询
cur.execute("SELECT * FROM mytable")

# 获取查询结果
rows = cur.fetchall()

# 打印结果
for row in rows:
    print(row)

# 关闭连接
cur.close()
conn.close()

执行SQL语句

在连接到数据库后，我们可以使用cursor对象来执行SQL语句。

# 执行CREATE TABLE语句
cur.execute("CREATE TABLE students (id SERIAL PRIMARY KEY, name VARCHAR);")

# 插入数据
cur.execute("INSERT INTO students (name) VALUES ('Alice');")

# 提交事务
conn.commit()

查询数据

我们也可以执行查询语句来获取数据库中的数据。

# 查询数据
cur.execute("SELECT * FROM students;")

# 获取查询结果
rows = cur.fetchall()

# 打印结果
for row in rows:
    print(row)

参数化查询

在执行SQL语句时，为了防止SQL注入攻击，我们可以使用参数化查询来替换变量。

# 参数化查询
name = "Bob"
cur.execute("INSERT INTO students (name) VALUES (%s);", (name,))

# 提交事务
conn.commit()

错误处理

在连接和执行SQL语句过程中，可能会发生一些错误。我们可以使用try…except语句来捕获异常并处理错误。

try:
    # 执行SQL查询
    cur.execute("SELECT * FROM non_existing_table;")
except psycopg2.Error as e:
    print("Error executing SQL statement:", e)

关闭连接

最后，在完成所有操作后，我们需要关闭连接。

# 关闭连接
cur.close()
conn.close()

通过上述方法，我们可以在Python中连接到PostgreSQL数据库，并执行SQL语句。这为我们提供了灵活、方便的数据交互方式。