python 操做数据库: postgresql

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()