PostgreSQL是一个轻量级的数据库,用python操作时,用到了psycopg2模块,其程序代码少,速度快,而且稳定。
参考:http://www.yiibai.com/html/postgresql/2013/080998.html
环境
ubuntu 16.04
python2
安装
1. 安装postgresql
sudo apt-get install postgresql
(下面2步操作应该可以在python代码中实现)
=》 在数据库中创建user
sudo -u postgres createuser -P uuser, with password ppas
=》 创建数据库DB
sudo -u postgres createdb -O uuser testdb
2. 安装python psycopg2模块
sudo apt-get install python-psycopg2
后面如果需要删除数据库,可以执行:
sudo -u postgres dropdb testdb
python pscopg2 模块中的APIs
SN | API &描述 |
1 | psycopg2.connect(database="testdb", user="uuser", password="ppas", host="127.0.0.1", port="5432") 这个API打开一个连接到PostgreSQL数据库。如果成功打开数据库时,它返回一个连接对象。 (默认情况这里可以不用port这个参数) |
2 | connection.cursor() 该程序创建一个光标将用于整个数据库使用Python编程。 |
3 | cursor.execute(sql [, optional parameters]) 此例程执行SQL语句。可被参数化的SQL语句(即占位符,而不是SQL文字)。 psycopg2的模块支持占位符用%s标志 yiibai.com 例如: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对数据库的操作需要先连接到数据库,如果该数据库不存在,那么它就会被创建,最终返回一个数据库对象。
#!/bin/python import psycopg2 conn = psycopg2.connect(database="testdb", user="uuser",password="ppas", host="127.0.0.1") print "connect success"
创建表
#!/bin/python import psycopg2 conn = psycopg2.connect(database = "testdb", user = "uuer", password = "ppas", host = "127.0.0.1") print "Open database success" cur = conn.cursor() cur.execute('''CREATE TABLE COMPANY (ID INT PRIMARY KEY NOT NULL, NAME TEXT NOT NULL, AGE INT NOT NULL, ADDRESS CHAR(50)''' ) print "Table created successfully" conn.commit() conn.close()
这里如果想让ID为主键,并且自增,可以如此定义: ID SERIAL PRIMARY KEY
insert操作
在上面创建的表中插入记录:
#!/bin/python import psycopg2 conn = psycopg2.connect(database = "testdb", user = "uuer", password = "ppas", host = "127.0.0.1") print "Open database success" cur = conn.cursor() cur.execute("INSERT INTO COMPANY (ID, NAME, AGE, ADDRESS) \ VALUES (1, 'Paul', 32, 'California')"); cur.execute("INSERT INTO COMPANY (ID, NAME, AGE, ADDRESS) \ VALUES (4, 'Mark', 25, 'Rich-Mond')") conn.commit() print "insert succcess" conn.close()
selete操作
#!/bin/python import psycopg2 conn = psycopg2.connect(database = "testdb", user = "uuer", password = "ppas", host = "127.0.0.1") print "Open database success" cur = conn.cursor() cur.execute("SELECT id, name, address from COMPANY") rows = cur.fetchall() for row in rows: print "ID = ", row[0] print "NAME = ", row[1] print "ADDRESS = ", row[2] print "Operation done successfully"; conn.close()
update操作
#!/bin/python import psycopg2 conn = psycopg2.connect(database = "testdb", user = "uuer", password = "ppas", host = "127.0.0.1") print "Open database success" cur = conn.cursor() cur.execute("UPDATE COMPANY set NAME = "newCompy" where ID=1") conn.commit print "Total number of rows updated :", cur.rowcount cur.execute("SELECT id, name, address from COMPANY") rows = cur.fetchall() for row in rows: print "ID = ", row[0] print "NAME = ", row[1] print "ADDRESS = ", row[2] print "Operation done successfully"; conn.close()
delete操作
删除一行
#!/bin/python import psycopg2 conn = psycopg2.connect(database = "testdb", user = "uuer", password = "ppas", host = "127.0.0.1") print "Open database success" cur = conn.cursor() cur.execute("DELETE from COMPANY where ID=2;") conn.commit print "Total number of rows deleted :", cur.rowcount cur.execute("SELECT id, name, address, salary from COMPANY") rows = cur.fetchall() for row in rows: print "ID = ", row[0] print "NAME = ", row[1] print "ADDRESS = ", row[2] print "Operation done successfully"; conn.close()