数据说明:
一月份各省销售数据:csv格式
二月份各省销售数据:json格式
实现要求:将两个文件中的数据存储到数据库中,并反向从数据库中读取数据存储为json格式文件
本文提供数据
完成案例所需基础
案例实现
先在数据库中创建表,依次输入执行以下sql语句
sql
create database py_sql charset utf8;
use py_sql;
create table orders(
order_date date,
order_id varchar(255),
money int,
province varchar(10)
);
Data_define.py
python
# 数据定义的类
class Record:
def __init__(self, date, order_id, money, province):
self.date = date # 订单日期
self.order_id = order_id # 订单id
self.money = money # 订单金额
self.province = province # 销售省份
def __str__(self):
return f"{self.date},{self.order_id},{self.money},{self.province}"
File_define.py
python
# 和文件相关的类定义
import json
from Data_define import Record
# 先定义一个抽象类用来做顶层设计,确定有哪些功能需要实现
class FileReader:
def read_data(self) -> list[Record]:
# 读取文件的数据,读到的每一条数据都转换为Record对象,将它们都封装到list内返回即可
pass
class TextFileReader(FileReader):
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)
f.close()
return record_list
class JsonFileReader(FileReader):
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)
f.close()
return record_list
if __name__ == '__main__':
text_file_reader = TextFileReader("2011年1月销售数据.txt")
json_file_reader = JsonFileReader("2011年2月销售数据JSON.txt")
list1 = text_file_reader.read_data()
list2 = json_file_reader.read_data()
for l in list1:
print(l)
for l in list2:
print(l)
python
# 读取文件,写入MySQL数据库中
from File_define import TextFileReader,JsonFileReader
from Data_define import Record
from pymysql import Connection
text_file_reader = TextFileReader("2011年1月销售数据.txt")
json_file_reader = JsonFileReader("2011年2月销售数据JSON.txt")
jan_data: list[Record] = text_file_reader.read_data() # 1月份数据
feb_data: list[Record] = json_file_reader.read_data() # 2月份数据
# 将2个月份的数据合并为1个list来存储
all_data: list[Record] = jan_data + feb_data
conn = Connection(
host = 'localhost',
port = 3306,
user = 'root',
password = '123456',
autocommit = True
)
cursor = conn.cursor()
conn.select_db('py_sql')
# 组织SQL语句
for record in all_data:
sql = f"insert into orders(order_date, order_id, money, province)" \
f"values('{record.date}','{record.order_id}',{record.money},'{record.province}')"
cursor.execute(sql)
conn.close()
python
# 将写入到数据库的数据读取出来存储为json格式
from pymysql import Connection
import json
conn = Connection(
host = 'localhost',
port = 3306,
user = 'root',
password = '123456',
autocommit = True
)
cursor = conn.cursor()
conn.select_db('py_sql')
cursor.execute("select * from orders")
results: tuple = cursor.fetchall()
f = open('2011年1,2月销售数据.txt','w',encoding='UTF-8')
for r in results:
dict = {"date":f"{r[0]}","order_id":r[1],"money":r[2],"province":r[3]}
#print(dict)
data = json.dumps(dict, ensure_ascii=False)
#print(data)
f.write(data)
f.write("\n")
f.close()
conn.close()