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)