Python中对sqlite3数据库进行操作时,经常需要用到字段名,然而对于sqlite使用select语句并不能象MySql等数据库一样返回带字段名的字典数据集。特别是对于一个不熟悉的sqlite数据库,写代码时如果需要借助工具查看,那其实是有点对不起python的。
下面两段代码就可以轻易获得一个sqlite数据库中所有数据表名和字段名的列表:
# python 获取sqlite3数据库mydb.db中的表名和表字段名
import sqlite3
conn=sqlite3.connect('mydb.db')
cu=conn.cursor()
#获取表名,保存在tab_name列表
cu.execute("select name from sqlite_master where type='table'")
tab_name=cu.fetchall()
tab_name=[line[0] for line in tab_name]
#获取表的列名(字段名),保存在col_names列表,每个表的字段名集为一个元组
col_names=[]
for line in tab_name:
cu.execute('pragma table_info({})'.format(line))
col_name=cu.fetchall()
col_name=[x[1] for x in col_name]
col_names.append(col_name)
col_name=tuple(col_name)
#之所以保存为元组,一是可避免误操作修改字段名,二是元组巧用转化字符串,可