Python:操作PostgreSQL数据库(使用DB API2.0)

postgres0

2012-02-14

来看看实现，首先我们要下载提供DB API2.0接口的模块psycopg2,路径：http://initd.org/psycopg/download/，此模块API手册地址为：http://initd.org/psycopg/docs/index.html

一、实现：

[python]

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#导入日志及psycopg2模块
import logging
import logging.config
import psycopg2
#日志配置文件名
LOG_FILENAME = 'logging.conf'
#日志语句提示信息
LOG_CONTENT_NAME = 'pg_log'
def log_init(log_config_filename, logname):
'''''
Function:日志模块初始化函数
Input：log_config_filename:日志配置文件名
lognmae:每条日志前的提示语句
Output: logger
author: socrates
date:2012-02-13
'''
logging.config.fileConfig(log_config_filename)
logger = logging.getLogger(logname)
return logger
def operate_postgre_tbl_product():
'''''
Function:操作pg数据库函数
Input：NONE
Output: NONE
author: socrates
date:2012-02-13
'''
pgdb_logger.debug("operate_postgre_tbl_product enter...")
#连接数据库
try:
pgdb_conn = psycopg2.connect(database = 'kevin_test', user = 'dyx1024', password = '888888', host = '192.168.230.128')
except Exception, e:
print e.args[0]
pgdb_logger.error("conntect postgre database failed, ret = %s" % e.args[0])
return
pgdb_logger.info("conntect postgre database(kevin_test) succ.")
pg_cursor = pgdb_conn.cursor()
#删除表
sql_desc = "DROP TABLE IF EXISTS tbl_product3;"
try:
pg_cursor.execute(sql_desc)
except Exception, e:
print 'drop table failed'
pgdb_logger.error("drop table failed, ret = %s" % e.args[0])
pg_cursor.close()
pgdb_conn.close()
return
pgdb_conn.commit()
pgdb_logger.info("drop table(tbl_product3) succ.")
#创建表
sql_desc = '''''CREATE TABLE tbl_product3(
i_index INTEGER,
sv_productname VARCHAR(32)
);'''
try:
pg_cursor.execute(sql_desc)
except Exception, e:
print 'create table failed'
pgdb_logger.error("create table failed, ret = %s" % e.args[0])
pg_cursor.close()
pgdb_conn.close()
return
pgdb_conn.commit()
pgdb_logger.info("create table(tbl_product3) succ.")
#插入记录
sql_desc = "INSERT INTO tbl_product3(sv_productname) values('apple')"
try:
pg_cursor.execute(sql_desc)
except Exception, e:
print 'insert record into table failed'
pgdb_logger.error("insert record into table failed, ret = %s" % e.args[0])
pg_cursor.close()
pgdb_conn.close()
return
pgdb_conn.commit()
pgdb_logger.info("insert record into table(tbl_product3) succ.")
#查询表方法一
sql_desc = "select * from tbl_product3"
try:
pg_cursor.execute(sql_desc)
except Exception, e:
print 'select record from table tbl_product3 failed'
pgdb_logger.error("select record from table tbl_product3 failed, ret = %s" % e.args[0])
pg_cursor.close()
pgdb_conn.close()
return
for row in pg_cursor:
print row
pgdb_logger.info("%s", row)
print '*' * 20
#查询表方法二
sql_desc = "select * from tbl_test_port"
try:
pg_cursor.execute(sql_desc)
except Exception, e:
print 'select record from table tbl_test_port failed'
pgdb_logger.error("select record from table tbl_test_port failed, ret = %s" % e.args[0])
pg_cursor.close()
pgdb_conn.close()
return
for row in pg_cursor.fetchall():
print row
pgdb_logger.info("%s", row)
#关闭数据库连接
pg_cursor.close()
pgdb_conn.close()
pgdb_logger.debug("operate_sqlite3_tbl_product leaving...")
if __name__ == '__main__':
#初始化日志系统
pgdb_logger = log_init(LOG_FILENAME, LOG_CONTENT_NAME)
#操作数据库
operate_postgre_tbl_product()