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

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()

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值