psycopg2是Python编程语言的PostgreSQL数据库的适配器。postgreSQL属于关系型数据库,支持Python通过DB-API,本文假设已经安装和配置好了PostgreSQL,要使用psycopg2的模块,首先必须创建一个Connection对象,它表示数据库然后再可以选择创建游标对象,这将帮助执行的所有SQL语句。
本例中用的操作系统为:
Debian GNU/Linux 8.2 (jessie)12
Python版本为:
Python 2.7.912
psycopg2的安装
Debian下执行以下命令安装 psycopg2:
apt-get install python-psycopg21
安装好后,就可以写Python脚本来访问PostgreSQL了。
导入psycopg2
想要使用psycopg2,必须用import语句导入该包:
import psycopg21
connection类
connection类表示数据库连接对象。由psycopg2.connect()方法创建。
创建connection对象
psycopg2.connect()函数创建一个新的数据库会话(Session)并且返回一个连接对象。该函数的参数为:
dbname – 数据库名字
user – 数据库角色名称
password – 数据库角色密码
host – 数据库地址
port – 端口
connection的主要方法
connection提供了常用的数据库操作:
commit():提交任何未提交的事务(transaction)到数据库。
rollback():回滚。
close():关闭数据库。如果关闭数据库时仍有未提交的事务,则执行回滚操作。
cursor类
创建cursor对象
psycopg2提供了一个cursor类,用来在数据库Session里执行PostgreSQL命令。cursor对象由connection.cursor()方法创建:
cur = conn.cursor()1
cursor的主要方法
execute(query, vars=None):执行SQL语句。
fetchall():获取所有查询结果,返回值为tuple列表。
有了cursor对象,就可以操作数据库了。
建立数据库连接
conn = psycopg2.connect(dbname="mydb", user="postgres",
password="1234", host="127.0.0.1", port="5432")12
创建表
cur.execute( 'CREATE TABLE Employee ('
'name varchar(80),'
'address varchar(80),'
'age int,'
'date date'
')'
)12345678
插入数据
cur.execute("INSERT INTO Employee "
"VALUES('Gopher', 'China Beijing', 100, '2017-05-27')")12
查询数据
cur.execute("SELECT * FROM Employee")
rows = cur.fetchall()for row in rows:
print('name=' + str(row[0]) + ' address=' + str(row[1]) +
' age=' + str(row[2]) + ' date=' + str(row[3]))12345
更新数据
cur.execute("UPDATE Employee SET age=12 WHERE name='Gopher'")1
删除数据
cur.execute("DELETE FROM Employee WHERE name='Gopher'")1
完整示例
#coding=utf-8import psycopg2# 连接数据库conn = psycopg2.connect(dbname="mydb", user="postgres",
password="1234", host="127.0.0.1", port="5432")# 创建cursor以访问数据库cur = conn.cursor()# 创建表cur.execute( 'CREATE TABLE Employee ('
'name varchar(80),'
'address varchar(80),'
'age int,'
'date date'
')'
)# 插入数据cur.execute("INSERT INTO Employee "
"VALUES('Gopher', 'China Beijing', 100, '2017-05-27')")# 查询数据cur.execute("SELECT * FROM Employee")
rows = cur.fetchall()for row in rows:
print('name=' + str(row[0]) + ' address=' + str(row[1]) +
' age=' + str(row[2]) + ' date=' + str(row[3]))# 更新数据cur.execute("UPDATE Employee SET age=12 WHERE name='Gopher'")# 删除数据cur.execute("DELETE FROM Employee WHERE name='Gopher'")# 提交事务conn.commit()# 关闭连接conn.close()
Python psycopg2 模块APIs
以下是psycopg2的重要的的模块例程可以满足Python程序与PostgreSQL数据库的工作。如果寻找一个更复杂的应用程序,那么可以看看Python psycopg2的模块的官方文档。
S.N. | API & 描述 |
---|---|
1 | psycopg2.connect(database="testdb", user="postgres", password="cohondob", host="127.0.0.1", port="5432") 这个API打开一个连接到PostgreSQL数据库。如果成功打开数据库时,它返回一个连接对象。 |
2 | connection.cursor() 该程序创建一个光标将用于整个数据库使用Python编程。 |
3 | cursor.execute(sql [, optional parameters]) 此例程执行SQL语句。可被参数化的SQL语句(即占位符,而不是SQL文字)。 psycopg2的模块支持占位符用%s标志 例如:cursor.execute("insert into people values (%s, %s)", (who, age)) |
4 | curosr.executemany(sql, seq_of_parameters) 该程序执行SQL命令对所有参数序列或序列中的sql映射。 |
5 | curosr.callproc(procname[, parameters]) 这个程序执行的存储数据库程序给定的名称。该程序预计为每一个参数,参数的顺序必须包含一个条目。 |
6 | cursor.rowcount 这个只读属性,它返回数据库中的行的总数已修改,插入或删除最后 execute*(). |
7 | connection.commit() 此方法提交当前事务。如果不调用这个方法,无论做了什么修改,自从上次调用commit()是不可见的,从其他的数据库连接。 |
8 | connection.rollback() 此方法会回滚任何更改数据库自上次调用commit()方法。 |
9 | connection.close() 此方法关闭数据库连接。请注意,这并不自动调用commit()。如果你只是关闭数据库连接而不调用commit()方法首先,那么所有更改将会丢失! |
10 | cursor.fetchone() 这种方法提取的查询结果集的下一行,返回一个序列,或者无当没有更多的数据是可用的。 |
11 | cursor.fetchmany([size=cursor.arraysize]) 这个例程中取出下一个组的查询结果的行数,返回一个列表。当没有找到记录,返回空列表。该方法试图获取尽可能多的行所显示的大小参数。 |
12 | cursor.fetchall() 这个例程获取所有查询结果(剩余)行,返回一个列表。空行时则返回空列表。 |
连接到数据库
Python代码显示了如何连接到一个现有的数据库。如果数据库不存在,那么它就会被创建,最终将返回一个数据库对象。
#!/usr/bin/pythonimport psycopg2 conn = psycopg2.connect(database="testdb", user="postgres", password="pass123", host="127.0.0.1", port="5432")print "Opened database successfully"
在这里,也可以提供数据库testdb的名称,如果数据库成功打开,那么它会给下面的消息:
Open database successfully