设计一个抽象类,定义文件读取的相关功能,并使用子类实现具体功能,因为文件的数据格式不同,我们读取文件的具体实现逻辑也不一样,所以定义一个抽象类,通过子类去实现具体数据的具体读取方法。
新建一个python文件,名为:file_define.py,先定义一个抽象类FileRecord用来做顶层设计,确定有哪些功能(read_data方法)需要实现,并分别定义TextFileRecord类和JsonFileRecord类,继承FileRecord类,复写父类的read_data方法,
具体代码如下:
from data_define import Record
import json
先定义一个抽象类用来做顶层设计,确定有哪些功能需要实现
class FileRecord:
def read_data(self) -> list[Record]:
读取文件的数据,读到的每一条数据都转换为Record对象,将他们都封装到List内并返回即可
pass
class TextFileRecord(FileRecord):
def init(self, path):
self.path = path # 定义成员变量记录文件路径
复写(实现抽象方法)父类的方法
def read_data(self) -> list[Record]:
f = open(self.path, “r”, encoding=“UTF-8”)
record_list: list[Record] = []
for line in f.readlines():
line = line.strip() # 消除读取到的每一行数据中的\n
data_list = line.split(“,”)
record = Record(data_list[0], data_list[1], int(data_list[2]), data_list[3])
record_list.append(record)
print(record_list)
f.close()
return record_list
class JsonFileRecord(FileRecord):
def init(self, path):
self.path = path # 定义成员变量记录文件路径
复写(实现抽象方法)父类的方法
def read_data(self) -> list[Record]:
f = open(self.path, “r”, encoding=“UTF-8”)
record_list: list[Record] = []
for line in f.readlines():
data_dict = json.loads(line)
record = Record(data_dict[“date”], data_dict[“order_id”], data_dict[“money”], data_dict[“province”])
record_list.append(record)
print(record_list)
f.close()
return record_list
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)
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。
深知大多数软件测试工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!
因此收集整理了一份《2024年软件测试全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上软件测试开发知识点,真正体系化!
由于文件比较大,这里只是将部分目录大纲截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且后续会持续更新
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注软件测试)
一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
实战项目、讲解视频,并且后续会持续更新**
如果你觉得这些内容对你有帮助,可以添加V获取:vip1024b (备注软件测试)
[外链图片转存中…(img-wqQ6mPHE-1712984321684)]
一个人可以走的很快,但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎扫码加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!