在使用ORM之前,一直在用psycopg2去操作数据库。原因在于比较喜欢写原生的sql语句,虽然开发速度比使用ORM慢,但是自我感觉可靠,而且在一些复杂sql时候更方便(不用处理里面的关系映射,这非常不好理解, 也可能是自己太笨了-_-)。然而也遇到一些问题,使用fetchall()方法或者fetchone()方法获取的数据的到的结果往往是一个元组。只能通过索引获取相应的数据,相比字典可操作行相对较低,因为不知道什么时候获取字段就可能增加,而获取索引就会相应的改变。
我之前的处理方法是这样的
data = [dict((cursor.description[i][0], value) for i, value in enumerate(row)) for row in cursor.fetchall()]
在仔细查看文档发现,其实psycopg2已经有这样的处理了, 只需要使用cursor_factory=psycopg2.extras.RealDictCursor参数就可以了
贴上代码,欢迎批评指正~
# -*-coding: utf-8 -*-
fromflask import current_app
import psycopg2
import psycopg2.extrasclass DbModel(object):
def __init__(self, autocommit=True):
self.conn= psycopg2.connect(host=current_app.config[‘HOST‘],
port=current_app.config[‘PORT‘],
user=current_app.config[‘USER‘],