Python读取文件数据写入到数据库中，并反向从数据库读取保存到本地_python中def read_data(self) -&gt； list[record]

本文链接：https://blog.csdn.net/2401_84253850/article/details/137887286

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新软件测试全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以添加V获取：vip1024b （备注软件测试）

正文

if name == ‘main’:

测试读取文本文件

text_file_record = TextFileRecord(“G:\资料\2011年1月销售数据.txt”)
lin1 = text_file_record.read_data()
for l1 in lin1:
print(l1)

测试读取json文件

json_file_record = JsonFileRecord(“G:\资料\2011年2月销售数据JSON.txt”)
lin2 = json_file_record.read_data()
for l2 in lin2:
print(l2)
复制代码

运行结果：

（3）合并数据

新建一个python文件main.py，读取数据并将1月份数据和2月份数据合并

from file_define import FileRecord, TextFileRecord, JsonFileRecord
from data_define import Record

text_file_record = TextFileRecord(“G:\资料\2011年1月销售数据.txt”)
json_file_record = JsonFileRecord(“G:\资料\2011年2月销售数据JSON.txt”)
jan_data: list[Record] = text_file_record.read_data() # 读取1月份数据并保存到jan_data
feb_data: list[Record] = json_file_record.read_data() # 读取2月份数据并保存到feb_data
all_data: list[Record] = jan_data + feb_data # 合并两个月份的数据

print(all_data)

复制代码

合并数据之后的结果：

构建数据库链接并写入数据库

我们首先需要导包：from pymysql import Connection，我在代码中分别创建了sell数据库和orders表，其实可以直接通过数据库可视化工具创建，创建完成后直接执行insert插入语句即可，参考代码如下：

构建数据库链接

conn = Connection(
host=“localhost”,
port=3306,
user=“root”,
password=“123456”,
autocommit=True # 执行sql语句时自动提交
)

构建游标对象

cur = conn.cursor()
cur.execute(“SHOW DATABASES”)
database_list = []
for db in cur:
database_list.append(db[0])
print(database_list) # 打印出当前所有数据库
if “sell” in database_list: # 如果数据库存在则不创建，如果不存在则创建数据库sell
pass
else:
creat_data_base = “CREATE DATABASE sell”
cur.execute(creat_data_base)

conn.select_db(“sell”)
cur.execute(“SHOW TABLES”)
tab_list = []
for tab in cur:
tab_list.append(tab[0])
print(tab_list) # 打印当前库中的所有表
if “orders” in tab_list: # 如果表存在则不创建，如果不存在则创建表orders
pass
else:
cur.execute(“CREATE TABLE orders(order_date DATE,order_id VARCHAR(255),money INT,province VARCHAR(10))”)

遍历所有数据，并执行插入语句

for record in all_data:
sql = f"INSERT INTO sell.orders(order_date,order_id,money,province) VALUES (‘{record.date}’,‘{record.order_id}’,{record.money},‘{record.province}’)"
print(sql)
cur.execute(sql)

关闭数据库链接

cur.close()
复制代码

运行代码成功之后，可在数据库中看到，数据已经正常写入orders表中：

从数据库读取数据并保存到本地

相对写入数据来说，读取数据保存到本地更加简单，先从数据库读取数据，然后处理数据，之后保存到本地文件即可。

（1）保存为txt文件

代码如下：

from pymysql import Connection
from data_define import Record

构建数据库链接

conn = Connection(
host=“localhost”,
port=3306,
user=“root”,
password=“123456”,
autocommit=True # 执行sql语句时自动提交
)

构建游标对象

cur = conn.cursor()
cur.execute(“select * from sell.orders”) # 读取orders表中的所有数据

record_list = []
for data in cur:
record = Record(data[0], data[1], data[2], data[3])
record_list.append(record) # 将读取的每一数据封装成一个Record对象并添加到record_list列表中

f = open(“E:\sell_orders_data.txt”, ‘w’) # 文件存在，则直接覆盖文件中的内容写入，不存在，先创建文件再写入数据
for sell_data in record_list:

只能写入str类型的，则把读取的Record对象转化为str，每个字段用","分割

str_data = str(sell_data.date) + “,” + sell_data.order_id + “,” + str(sell_data.money) + “,” + sell_data.province
f.write(str_data)
f.write(“\n”)
f.close()
复制代码

查看保存的txt文件：

（2）保存为excel文件

代码如下：

import xlwt

写入到excel文件中

调用xlwt模块中的Workbook方法来创建一个excel表格类型文件，style_compression设置是否压缩，赋值为0表示不压缩。

book = xlwt.Workbook(encoding=‘utf-8’, style_compression=0)

用book对象调用add_sheet方法来建立一张sheet表，这里面的第一个参数很明显就是设置sheet表格的名称，第二个参数cell_overwrite_ok用于确认同一个cell单元是否可以重设值，这里赋值为True就表示可重设值。

sheet = book.add_sheet(‘销售数据表’, cell_overwrite_ok=True)

用一个元组col自定义列的数量以及各列的属性名

col = (‘销售日期’, ‘订单ID’, ‘销售额’, ‘省份’)

写入表头

for i in range(0, 4):
sheet.col(i).width = 5000 # 设置宽度
sheet.write(0, i, col[i])
datastyle = xlwt.XFStyle() # 创建一个样式对象，初始化样式
datastyle.num_format_str = ‘yyyy-mm-dd’ # 设置日期格式

循环record_list数据并写入

for row in range(1, len(record_list) + 1):
sheet.write(row, 0, record_list[row - 1].date, datastyle) # 如果不加datastyle，写出来的日期列格式是不对的
sheet.write(row, 1, record_list[row - 1].order_id)
sheet.write(row, 2, record_list[row - 1].money)
sheet.write(row, 3, record_list[row - 1].province)

book.save(‘E:\订单销售额.xls’)

f.close()
复制代码

查看保存的excel文件：