【Python】使用MySQL综合案例

数据说明:

一月份各省销售数据：csv格式

二月份各省销售数据：json格式

实现要求：将两个文件中的数据存储到数据库中，并反向从数据库中读取数据存储为json格式文件

本文提供数据

完成案例所需基础

【Python】基础知识(函数与数据容器)

【Python】JSON与jsonpath

【Python】文件操作

【Python】模块和包

【Python】面向对象

【MySQL】基础操作(DDL,DML,DCL,DQL)

【Python】操作MySQL基础使用

案例实现

先在数据库中创建表，依次输入执行以下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

# 数据定义的类

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

# 和文件相关的类定义
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)

main.py

# 读取文件，写入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()

toJson.py

# 将写入到数据库的数据读取出来存储为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()