Python操作MySQL基础使用

创建到mysql的数据库链接

from pymysql import Connect
# 获取到MySQL数据库的链接对象
conn = Connect(
    host='localhost',   # 主机名（或IP地址）
    port=3306,          # 端口，默认3306
    user='root',        # 账户名
    password='123456'   # 密码
)
# 打印MySQL数据库软件信息
print(conn.get_server_info())
# 关闭到数据库的链接
conn.close()

执行SQL语句

执行非查询性质语句：

from pymysql import Connect

# 获取到MySQL数据库的链接对象
conn = Connect(
    host='localhost',  # 主机名（或IP地址）
    port=3306,  # 端口，默认3306
    user='root',  # 账户名
    password='123456',  # 密码
)

# 执行非查询性质SQL
# 获取游标对象
cursor = conn.cursor()
conn.select_db("test")  # 先选择数据库
# 使用游标对象，执行SQL语句
cursor.execute("create table test_pymysql2(id int,info varchar(225))")
# 关闭到数据库的链接
conn.close()

执行查询性质的SQL语句：

from pymysql import Connect

# 获取到MySQL数据库的链接对象
conn = Connect(
    host='localhost',  # 主机名（或IP地址）
    port=3306,  # 端口，默认3306
    user='root',  # 账户名
    password='123456',  # 密码
)
# 获取游标对象
cursor = conn.cursor()
conn.select_db("test")  # 先选择数据库(是创建好的)
# 执行查询性质SQL
# 使用游标对象，执行SQL语句
cursor.execute("select * from student")  # 要有student表
# 获取查询结果
results: tuple = cursor.fetchall()
for r in results:
    print(r)
# 关闭到数据库的链接
conn.close()

插入数据到MySQL

commit提交

from pymysql import Connect

# 获取到MySQL数据库的链接对象
conn = Connect(
    host='localhost',  # 主机名（或IP地址）
    port=3306,  # 端口，默认3306
    user='root',  # 账户名
    password='123456',  # 密码
)
# 获取游标对象
cursor = conn.cursor()
conn.select_db("test")  # 先选择数据库
# 使用游标对象，执行SQL语句
cursor.execute("insert into `student` (`name`, `age`, `class_id`, `score`, `exam_num`) values ('孙杰', 80, 4, 107.5, 3);")
# 通过commit确认
conn.commit()
# 关闭到数据库的链接
conn.close()

自动commit提交

# 构建到MySQL数据库的链接
conn = Connection(
	host='localhost',  # 主机名（或IP地址）
    port=3306,  # 端口，默认3306
    user='root',  # 账户名
    password='123456',  # 密码
    autocommit=True	   # 设置自动提交
)

综合案例

实现步骤：读取数据 -> 封装数据对象 -> 构建数据库对象 -> 写入数据库

# data_defin.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 typing import List
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"], int(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文件
"""
SQL综合案例，读取文件，写入MySQL数据库中
"""
from typing import List
from file_define import FileReader, TextFileReader, JsonFileReader
from data_define import Record
from pymysql import Connect

text_file_reader = TextFileReader("2011年1月销售数据.txt")
json_file_reader = JsonFileReader("2011年2月销售数据JSON.txt")

jan_data: List[Record] = text_file_reader.read_data()
feb_data: List[Record] = json_file_reader.read_data()
# 将两个月份的数据合并为一个list来存储
all_data: List[Record] = jan_data + feb_data

# 构建MySQL链接对象
conn = Connect(
    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}')"
    # 执行SQL语句
    cursor.execute(sql)
# 关闭MySQL链接对象
conn.close()