PyMySQL和OrderedDict

Question

我一直在使用PyMySQL一段时间,并创建了我自己的包装器,我习惯于简写编写查询.尽管如此,我一直在使用OrderedDict创建CSV文件,因为我需要保持顺序相同但我意识到如果我使用PyMySQL来查询数据库,我将无法得到数据库回馈的顺序.如果我只想转储东西而不是手写订单,这对于检查CSV文件有点烦人.

我的问题是,如何将PyMySQL与OrderedDict一起使用？目前我的代码如下:

import pymysql
conn = pymysql.connect(host='localhost', user='root', passwd='', db='test')
cursor = conn.cursor(pymysql.cursors.DictCursor)

所以每当我查询时,我都会收到一本字典:

cursor.execute("""SELECT * FROM test""")
for row in cursor:
    pp(row)  # gives me dictionary

我想要的是,当我浏览游标时,我实际上是按照它们从数据库中进入的顺序检索列的OrderedDict.

就像是:

cursor = conn.cursor(pymysql.cursors.OrderedDict)

Answer 1

您可以使用cursors.DictCursorMixin并将其更改dict_type为collections.OrderedDict(默认值为dict):

from collections import OrderedDict
from pymysql.cursors import DictCursorMixin, Cursor

class OrderedDictCursor(DictCursorMixin, Cursor):
    dict_type = OrderedDict

然后您可以使用新的游标类,如下所示

cursor = conn.cursor(OrderedDictCursor)