在Python中执行PostgreSQL查询和操作

2023-04-04 00:00:00 查询 执行 操作

要在Python中执行PostgreSQL查询和操作，需要使用PostgreSQL官方提供的Python驱动程序——psycopg2。具体步骤如下：
1. 安装psycopg2
可以使用pip命令轻松安装psycopg2：

pip install psycopg2
  1. 连接到数据库
    使用psycopg2.connect()方法创建到数据库的连接。需要提供数据库名称、用户名、密码、主机和端口等参数。
import psycopg2
conn = psycopg2.connect(
    dbname="mydb",
    user="myuser",
    password="mypassword",
    host="localhost",
    port="5432"
)
  1. 执行查询
    在psycopg2中，可以使用cursor()方法创建一个游标对象。然后，使用execute()方法执行查询，并用fetchall()方法获取结果集。
cur = conn.cursor()
cur.execute("SELECT * FROM mytable")
rows = cur.fetchall()
for row in rows:
    print(row)
  1. 插入数据
    要插入数据到PostgreSQL中，需要使用INSERT语句。在psycopg2中，可以使用execute()方法设置参数，然后使用commit()方法提交到数据库。
cur = conn.cursor()
cur.execute("INSERT INTO mytable (name, url) VALUES (%s, %s)", ("pidancode.com", "https://www.pidancode.com"))
conn.commit()
  1. 关闭连接
    使用完毕后，需要调用close()方法关闭数据库连接。
conn.close()

完整代码示例：

import psycopg2
# 连接数据库
conn = psycopg2.connect(
    dbname="mydb",
    user="myuser",
    password="mypassword",
    host="localhost",
    port="5432"
)
# 查询数据
cur = conn.cursor()
cur.execute("SELECT * FROM mytable")
rows = cur.fetchall()
for row in rows:
    print(row)
# 插入数据
cur = conn.cursor()
cur.execute("INSERT INTO mytable (name, url) VALUES (%s, %s)", ("pidancode.com", "https://www.pidancode.com"))
conn.commit()
# 关闭连接
conn.close()

相关文章