PostgreSQL 入门

PostgreSQL 是一个积极开发和成熟的开源数据库。使用 psycopg2 模块，我们可以对数据库执行查询。

使用 pip 安装

pip install psycopg2

基本用法

让我们假设我们在数据库 my_database 中有一个表 my_table，定义如下。

我们可以使用 psycopg2 模块以下列方式在数据库上运行查询。

import psycopg2

# Establish a connection to the existing database 'my_database' using
# the user 'my_user' with password 'my_password'
con = psycopg2.connect("host=localhost dbname=my_database user=my_user password=my_password")

# Create a cursor
cur = con.cursor()

# Insert a record into 'my_table'
cur.execute("INSERT INTO my_table(id, first_name, last_name) VALUES (2, 'Jane', 'Doe');")

# Commit the current transaction
con.commit()

# Retrieve all records from 'my_table'
cur.execute("SELECT * FROM my_table;")
results = cur.fetchall()

# Close the database connection
con.close()

# Print the results
print(results)

# OUTPUT: [(1, 'John', 'Doe'), (2, 'Jane', 'Doe')]