【python】读取 操作 使用 db文件，并转化为其他文件类型

问题描述

在我同级目录下有一个"MorkDork.db"文件，我现在想知道里面有几张表格table、表格的名字、表头结构。并且转化为其他格式的文件（eg：csv格式）

注：可使用filecmp.cmp验证转化的正确与否

1、使用SQL语句"“select name from sqlite_master where type=‘table’ order by name”"，查找表格的名字。

实例代码如下：

# coding:utf-8
import sqlite3
conn = sqlite3.connect("MorkDork.db")
cursor = conn.cursor()
sql = """select name from sqlite_master where type='table' order by name"""
cursor.execute(sql)
result = cursor.fetchall()
print(result)
print(type(result))
conn.close()

>>>[(u'prisoners',)]
>>><type 'list'>

可以看出，“MorkDork.db"文件中有1张表格，表格名字为"prisoners”。

2、使用SQL语句"“pragma table_info(prisoners))”"，查找"prisoners"表格中的表头结构。

实例代码如下：

import sqlite3
conn = sqlite3.connect("MorkDork.db")
cursor = conn.cursor()
sql = """pragma table_info(prisoners)"""
cursor.execute(sql)
result = cursor.fetchall()
print(result)
print(type(result))
conn.close()
>>>[(0, u'name', u'text'), (1, u'id', u'int')]
>>><type 'list'>

可以看出“prisoners”表中有name、id 两列。
注：在执行查询语句后，Python将返回一个循环器，包含有查询获得的多个记录。可以循环读取，也可以使用sqlite3提供的fetchone()和fetchall()方法读取记录：

完整代码：

遍历fetchall()方法读取的数据存储csv：

import sqlite3
import csv
import filecmp 

def sql2csv (fileIn, fileOut) :
	conn = sqlite3.connect(fileIn)
	cursor = conn.cursor()
	# sql = """select name from sqlite_master where type='table' order by name"""
	sql = """select * from prisoners"""
	cursor.execute(sql)
	results = cursor.fetchall()
	with open(fileOut,'w',newline='') as fp:
	    writer=csv.writer(fp)
	    writer.writerows(results)
	conn.close()

sql2csv("MorkDork.db", "morkdork.csv")
print(filecmp.cmp("morkdork.csv", "morkdork_good.csv"))

>>>True