python 操作 PostgreSQL 模块：psycopg2

2015 年 5 月 14 日11:26:57 评论 28,243 views

官方资料：http://www.psycopg.org/psycopg/docs/

安装：

  1. yum -y install python-psycopg2 （安装的版本可能是2.0）
  2. pip install psycopg2 （安装版本可能是2.6，目前新版，支持更多功能，建议）

 

如果以上2个方法安装失败，可下载源码安装包：

  1. wget  http://initd.org/psycopg/tarballs/PSYCOPG-2-6/psycopg2-2.6.tar.gz
  2. tar xf psycopg2-2.6.tar.gz
  3. cd psycopg2-2.6
  4. yum install -y postgresql-devel*
  5. python setup.py build
  6. sudo python setup.py install
  1. 使用：
  2. import psycopg2
  4. 连接数据库
  5. conn = psycopg2.connect(database=db_name,user=db_user,password=db_pass,host=db_ip,port=db_port)
  7. 建立游标
  8. cursor = conn.cursor()
  10. 执行SQL语句
  11. cursor.execute(sql)
  13. 提交
  14. conn.commit()
  16. 关闭
  17. conn.close()
 封装读、写数据库的函数，方便后面代码的调用
  1. import psycopg2
  2. import logging
  4. db_name = "****"
  5. db_user = "****"
  6. db_pass = "****"
  7. db_ip = "****"
  8. error_log =  "****"
  10. #定义日志输出格式
  11. logging.basicConfig(level=logging.ERROR,
  12.         format = '%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
  13.         datefmt = '%Y-%m-%d %H:%M:%S',
  14.         filename = error_log,
  15.         filemode = 'a')
  17. def writeDb(sql,data):
  18.     """
  19.     连接mysql数据库（写），并进行写的操作，如果连接失败，会把错误写入日志中，并返回false，如果sql执行失败，也会把错误写入日志中，并返回false，如果所有执行正常，则返回true
  20.     """
  21.     try:
  22.         conn = psycopg2.connect(database=db_name,user=db_user,password=db_pass,host=db_ip,port=5432)
  23.         cursor = conn.cursor()
  24.     except Exception,e:
  25.         print e
  26.         logging.error('数据库连接失败:%s' % e)
  27.         return False
  29.     try:
  30.         cursor.execute(sql,data)
  31.         conn.commit()   #提交事务
  32.     except Exception,e:
  33.         conn.rollback()   #如果出错，则事务回滚
  34.         logging.error('数据写入失败:%s' % e)
  35.         return False
  36.     finally:
  37.         cursor.close()
  38.         conn.close()
  39.     return True
  42. def readDb(sql):
  43.     """
  44.     连接mysql数据库（从），并进行数据查询，如果连接失败，会把错误写入日志中，并返回false，如果sql执行失败，也会把错误写入日志中，并返回false，如果所有执行正常，则返回查询到的数据，这个数据是经过转换的，转成字典格式，方便模板调用，其中字典的key是数据表里的字段名
  45.     """
  46.     try:
  47.         conn = psycopg2.connect(database=db_name,user=db_user,password=db_pass,host=db_ip,port=5432)
  48.         cursor = conn.cursor()
  49.     except Exception,e:
  50.         print e
  51.         logging.error('数据库连接失败:%s' % e)
  52.         return False
  54.     try:
  55.         cursor.execute(sql)
  56.         data = [dict((cursor.description[i][0], value) for i, value in enumerate(row)) for row in cursor.fetchall()]     #转换数据，字典格式
  57.     except Exception,e:
  58.         logging.error('数据执行失败:%s' % e)
  59.         return False
  60.     finally:
  61.         cursor.close()
  62.         conn.close()
  63.     return data
 代码调用上面封装好的数据库writeDb函数，并传递sql和value
  1. @route('/addkeyword',method='POST')
  2. def addkeyword():
  3.     s = request.environ.get('beaker.session')
  4.     if s.get('user',None) == None:
  5.         return redirect('/login')
  6.     keyword = request.forms.get('keyword').strip()
  7.     keycode = request.forms.get('keycode').strip()
  8.     notes = request.forms.get('notes').strip()
  9.     parentid = request.forms.get('site').strip()
  10.     sortid = 1
  11.     admin_id = s.get('userid','末知')
  12.     admin_name = s.get('user','末知')
  13.     if not keyword or not keycode or not notes or not parentid or not sortid:
  14.         return '-2'
  15.     sql = "insert into searchcode(code,code_name,notes,parentid,sortid,admin_id,admin_name) values(%s,%s,%s,%s,%s,%s,%s)"  
  16.     data = (keycode,keyword,notes,int(parentid),int(sortid),int(admin_id),admin_name,)  
  17.     status = writeDb(sql,data)  
  18.     if status:
  19.         return '0'
  20.     else:
  21.         return '-1'

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: