首先,需要安装相应的Python模块,可以使用以下命令安装:
pip install psycopg2
接下来,可以使用以下代码连接PostgreSQL数据库:
import psycopg2
# 建立连接
conn = psycopg2.connect(host="localhost", database="testdb", user="postgres", password="password")
# 创建游标
cur = conn.cursor()
# 执行SQL语句
cur.execute("INSERT INTO users (name, email) VALUES (%s, %s)", ("pidancode.com", "pidancode@gmail.com"))
# 提交事务
conn.commit()
# 关闭游标和连接
cur.close()
conn.close()
以上代码实现了连接名为testdb
的PostgreSQL数据库,使用用户名为postgres
,密码为password
,并向名称为users
的表中插入一条记录。
需要注意的是,在执行SQL语句时,可以使用参数化的方式,使用%s
作为占位符,将实际的参数传递给execute()
方法的第二个参数中。这样可以避免SQL注入等安全问题,并且可以更加方便地处理各种数据类型。