Twisted MySQL adbapi返回字典

时间:2011-12-07 07:07:57

标签: python twisted

有没有办法将字典结果从adbapi查询返回给MySQL?

[name: 'Bob', phone_number: '9123 4567']

默认返回元组。

['Bob', '9123 4567']

简单的Python& MySQL我们可以使用 MySQLdb.cursors.DictCursor 。但如何使用扭曲的adbapi


UPD:我解决了它,但我认为应该有更好的方法。我的解决方案:只需覆盖 adbapi.ConnectionPool 类的* _runInteraction *方法。

class MyAdbapiConnectionPool(adbapi.ConnectionPool):
def _runInteraction(self, interaction, *args, **kw):
    conn = self.connectionFactory(self)
    trans = self.transactionFactory(self, conn)
    try:
        result = interaction(trans, *args, **kw)
        if(result and isinstance(result[0], (list, tuple))):
            colnames = [c[0] for c in trans._cursor.description]
            result = [dict(zip(colnames, item)) for item in result]         
        trans.close()
        conn.commit()
        return result
    except:
        excType, excValue, excTraceback = sys.exc_info()
        try:
            conn.rollback()
        except:
            log.err(None, 'Rollback failed')
        raise excType, excValue, excTraceback

1 个答案:

答案 0 :(得分:9)

您可以将MySQLdb作为DictCursor参数的值传递给cursorclass函数,以指示MySQLdb使用connectConnectionPool允许您将任意参数传递给connect方法:

import MySQLdb
pool = ConnectionPool("MySQLdb", ..., cursorclass=MySQLdb.cursors.DictCursor)
...

当您运行查询时,您将获得dict而不是元组,例如runQuery("SELECT 1")会产生({'1': 1L},)的结果