python 获取sqlite3数据库的表名和表字段名的实例

来源：jb51　　时间：2019/7/17 13:11:15　　对本文有异议

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)
 
#之所以保存为元组，一是可避免误操作修改字段名，二是元组巧用转化字符串，可
#直接用于SQL的insert语句中。例如下面代码可得到第一个表的带括号字段名集合：
'''
  sql_col_name=str(col_names[0]).replace('\'','')
'''

接下来，用自动获取的字段名做点事吧。

比如，我一个sqlite数据库的表数据是从Excel表中导入的，Excel表中一些误录入的回车符也被一同导入了，在sqlite表中会显示为‘\n'的字符，我想把所有字段中的这个字符都去掉，我是通过以下代码实现的。

#删除第一个表中所有文本型字段中的回车符（注：存入数据库中的回车符取出
#后会展现为原始字符'\n'）
cu.execute('select * from '+ tab_name[0])
cdset=cu.fetchall()
cdset=[list(line) for line in cdset] #将结果集的元组转为列表，才能修改。
for line in cdset:
  for x in range(len(line)):
  #注：作为For循环计数变量不能在循环体中修改，此处不能用x in line代替
    if type(line[x])==str and (r'\n' in line[x]):  #只能修改字符型数据
      line[x]=line[x].replace(r'\n','')
      sql="update {bm} set {cm}='{vm}' where {cm2}={vm2}".format(
        bm=tab_name[0],cm=col_names[0][x],vm=line[x],cm2=col_names
        [0][0],vm2=line[0])
        #此处cm2必须是表中具有排它性的主键字段,非字符型字段
      cu.execute(sql)
conn.commit()