问题:
python使用 sqlite3 数据库进行增删改查时发现,sqlite3查询数据返回的元组,会导致我使用csv库导出数据时会报错。
如下图,是sqlite3最初查询返回的数据格式:
源码:
import sqlite3
conn = sqlite3.connect('show.db.php')
cursor = conn.cursor()
def catalog_page_num():
sql_catalog = 'SELECT a.cat_name,COUNT(b.page_id) FROM catalog a LEFT JOIN page b ON a.cat_id=b.cat_id GROUP BY cat_name'
cursor.execute(sql_catalog)
data_catalog = cursor.fetchall()
print(' 类名 文章数')
for d in data_catalog:
print(d)
catalog_page_num()
执行结果:
原因:sqlite3本身并没有原生提供字典形式的游标,(adodb、pymysql等其他库有)
解决:官方文档里已经有预留了相应的实现方案,如下图,重写重写 row_factory 方法后,查询数据返回的就是字典!
import sqlite3
conn = sqlite3.connect('show.db.php')
# 重写row_factory方法,然后调用
def dict_factory(cursor, row):
d = {}
for idx, col in enumerate(cursor.description):
d[col[0]] = row[idx]
return d
def catalog_page_num():
sql_catalog = 'SELECT a.cat_name,COUNT(b.page_id) FROM catalog a LEFT JOIN page b ON a.cat_id=b.cat_id GROUP BY cat_name'
conn.row_factory = dict_factory # 重写 row_factory 方法后,查询数据返回的就是字典!
cursor = conn.cursor()
cursor.execute(sql_catalog)
data_catalog = cursor.fetchall()
print(' 类名 文章数')
for d in data_catalog:
print(d)
catalog_page_num()
执行结果如下:
在此加上csv导出代码,就能正常导出了。