在使用pymysql的时候，通过fetchall()或fetchone()可以获得查询结果，但这个返回数据是不包含字段信息的（不如php方便）。查阅pymysql源代码后，其实获取查询结果源代码也是非常简单的，直接调用cursor.description即可。

譬如:

1. db=pymysql.connect(...) 
2. cur=db.cursor() 
3. cur.execute(sql) 
4. print(cur.description) 
5. result=cur.fetchall() 
6. data_dict=[] 
7. forfieldincur.description: 
8.   data_dict.append(field[0]) 
9. print(data_dict) 

在pymysql的 pymysql/cursors.py 中，找到 class Cursor 可以看到如下代码：

1. def__init__(self, connection): 
2.   self.connection=connection 
3.   self.description=None 
4.   self.rownumber=0 
5.  self.rowcount=-1 
6.   self.arraysize=1 
7.   self._executed=None 
8.  self._result=None 
9.  self._rows=None 
10.  self._warnings_handled=False 

因此，调用 cur.rowcount 是可以迅速返回查询结果记录数的，不需要通过 len() 获得。

Tags: Python pymysql

分享到：