Python与数据库之学员管理系统

一、系统需求

使⽤⾯向对象编程思想完成学员管理系统的开发，具体如下：
系统要求：

• 学员数据存储在MySQL数据库中
• 系统功能：添加学员、删除学员、修改学员信息、查询学员信息、显示所有学员信息、保存学员信息及退出系统等功能。保存学院信息功能可以实现将学员信息每秒自动保存至数据库中。

二、准备程序⽂件

1.分析

为了⽅便维护代码，⼀般⼀个⻆⾊⼀个程序⽂件；项⽬要有主程序⼊⼝，习惯为 main.py

2.创建程序⽂件

程序⽂件如下：

• 程序⼊⼝⽂件：main.py
• 学员⽂件：student.py
• 管理系统⽂件：managerSystem.py

三、书写程序

1.main.py

# 1. 导入管理系统模块
from managerSystem import *


# 2. 启动管理系统
# 保证是当前文件运行才启动管理系统：if --创建对象并调用run方法
if __name__ == '__main__':
    student_manager = StudentManager()  # 实例化对象
    student_manager.run()  # 调用run函数

2.student.py

• 学员信息包含：姓名、性别、⼿机号；
• 添加 str 魔法⽅法，⽅便查看学员对象信息
• 代码：

class Student(object):
    def __init__(self, name, gender, tel):
        # 姓名、性别、手机号
        self.name = name
        self.gender = gender
        self.tel = tel

    def __str__(self):
        return f'{self.name}, {self.gender}, {self.tel}'

3.managerSystem.py

• 存储数据的位置：本地数据库
  • 加载学员数据
  • 修改数据后保存到数据库
• 存储数据的形式：列表存储学员对象
  • 系统功能
  • 添加学员
  • 删除学员
  • 修改学员
  • 查询学员信息
  • 显示所有学员信息
  • 保存学员信息
  • 退出系统
    源代码：

from student import *
import pymysql
import threading

# 数据库信息：
DBHOST = 'localhost'  # 主机名
DBUSER = 'root'  # 用户名
DBPASS = '1234'  # 密码
DBNAME = 'kaohe_student'  # 数据库名


class StudentManager(object):
    def __init__(self):
        # 存储学员数据 -- 列表
        self.student_list = []

    # 一. 程序入口函数
    def run(self):
        # 1. 加载文件里面的学员数据
        self.load_student()

        def save():
            # print("启动定时器")
            self.save_student()
            timer = threading.Timer(1, function=save)  # 重新创建定时器，使得定时器能够一直工作
            timer.start()  # 启动定时器

        t1 = threading.Timer(1, function=save)  # 创建定时器
        t1.start()  # 启动定时器

        while True:
            # 2. 显示功能菜单
            self.show_menu()

            # 3. 用户输入目标功能序号
            menu_num = int(input('请输入您需要的功能序号：'))

            # 4. 根据用户输入的序号执行不同的功能 -- 如果用户输入1，执行添加
            if menu_num == 1:
                # 添加学员
                self.add_student()
            elif menu_num == 2:
                # 删除学员
                self.del_student()
            elif menu_num == 3:
                # 修改学员信息
                self.modify_student()
            elif menu_num == 4:
                # 查询学员信息
                self.search_student()
            elif menu_num == 5:
                # 显示所有学员信息
                self.show_student()
            elif menu_num == 6:
                # 保存学员信息
                self.save_student()
            elif menu_num == 7:
                t1.cancel()  # 取消定时器
                # 退出系统 -- 退出循环
                break

    # 二. 系统功能函数
    # 2.1 显示功能菜单 -- 打印序号的功能对应关系 -- 静态
    @staticmethod  # 静态方法无需实例化即可调用，也可以实例化后调用
    def show_menu():
        print('请选择如下功能：')
        print('1:添加学员')
        print('2:删除学员')
        print('3:修改学员信息')
        print('4:查询学员信息')
        print('5:显示所有学员信息')
        print('6:保存学员信息')
        print('7:退出系统')

    # 2.2 添加学员
    def add_student(self):
        # 1. 用户输入姓名、性别、手机号
        name = input('请输入您的姓名：')
        gender = input('请输入您的性别：')
        tel = input('请输入您的手机号：')

        # 2. 创建学员对象 -- 类 ？类在student文件里面  先导入student模块，再创建对象
        student = Student(name, gender, tel)

        # 3. 将该对象添加到学员列表
        self.student_list.append(student)

        print(self.student_list)
        print(student)

    # 2.3 删除学员
    def del_student(self):
        # 1. 用户输入目标学员姓名
        del_name = input('请输入要删除的学员姓名：')

        # 2. 遍历学员列表，如果用户输入的学员存在则删除学员对象，否则提示学员不存在
        for i in self.student_list:
            if del_name == i.name:
                # 删除该学员对象
                self.student_list.remove(i)
                print("删除成功！")
                break
        else:
            # 循环正常价结束执行的代码：循环结束都没有删除任何一个对象，所以说明用户输入的目标学员不存在
            print('查无此人！')

        # print(self.student_list)

    # 2.4 修改学员信息
    def modify_student(self):
        # 1. 用户输入目标学员姓名
        modify_name = input('请输入要修改的学员姓名：')

        # 2. 遍历列表数据，如果学员存在修改姓名性别手机号，否则提示学员不存在
        for i in self.student_list:
            if modify_name == i.name:
                i.name = input('姓名：')
                i.gender = input('性别：')
                i.tel = input('手机号：')
                print(f'修改学员信息成功，姓名{i.name}, 性别{i.gender}, 手机号{i.tel}')
                break
        else:
            print('查无此人！')

    # 2.5 查询学员信息
    def search_student(self):
        # 1. 用户输入目标学员姓名
        search_name = input('请输入您要搜索的学员姓名：')

        # 2. 遍历列表。如果学员存在打印学员信息，否则提示学员不存在
        for i in self.student_list:
            if search_name == i.name:
                print(f'姓名是{i.name}, 性别是{i.gender}, 手机号是{i.tel}')
                break
        else:
            print('查无此人！')

    # 2.6 显示所有学员信息
    def show_student(self):
        # 1. 打印表头
        print('姓名\t性别\t手机号')

        # 2. 打印学员数据
        for i in self.student_list:
            print(f'{i.name}\t{i.gender}\t{i.tel}')

    # 2.7 保存学员信息
    def save_student(self):
        try:
            db = pymysql.connect(host=DBHOST, user=DBUSER, password=DBPASS, database=DBNAME)  # 连接数据库
            # print("数据库连接成功！")
            cur = db.cursor()  # 声明一个游标，用以操作数据库
            cur.execute("DROP TABLE IF EXISTS Student")  # 执行SQL语句，创建表之前先查询是否存在，如果存在则删除，以此来保证数据不会重复
            sqlQuery = "CREATE TABLE Student(Name CHAR(20) NOT NULL ,Gender CHAR(20),Tel CHAR(20))"  # 编辑SQL语句：创建表，表名为 Student，具有Name，Email，Age
            cur.execute(sqlQuery)  # 执行SQL语句
            # print("表格创建成功！")
            new_list = [i.__dict__ for i in self.student_list]  # [学员对象] 转换成 [字典]
            for a in new_list:
                sqlQuery = "INSERT INTO Student(Name, Gender, Tel) VALUE (%s, %s, %s)"  # 编辑SQL语句：向表中插入信息
                value = (a['name'], a['gender'], a['tel'])  # 要插入的具体信息
                cur.execute(sqlQuery, value)  # 执行SQL语句
                db.commit()  # 提交到数据库
            # print("数据保存成功！")
        except pymysql.Error as e:  # 连接不成功打印错误
            print("数据保存失败：" + str(e))
            db.rollback()  # 如果发生错误则回滚

        db.close()  # 关闭数据库连接

    # 2.8 加载学员信息
    def load_student(self):
        try:
            # 连接数据库
            db = pymysql.connect(host=DBHOST, user=DBUSER, password=DBPASS, database=DBNAME)  # 连接数据库
            print("数据库连接成功！")

            # 创建表格
            cur = db.cursor()  # 声明一个游标，用以操作数据库
            sqlQuery = "CREATE TABLE Student(Name CHAR(20) NOT NULL ,Gender CHAR(20),Tel CHAR(20))"  # 编辑SQL语句：创建表，表名为 Student，具有Name，Gender，Tel
            cur.execute(sqlQuery)  # 执行SQL语句
            print("表格创建成功！")
        except (pymysql.err.OperationalError):  # 若表格已创建会报错，此语句用以捕获异常(1050, "Table 'student' already exists")
            print("表格已创建")
        finally:
            # 查询表中数据
            sqlQuery = "SELECT * FROM Student"  # 编辑SQL语句：查询数据库 Student 中的内容
            cur.execute(sqlQuery)  # 执行SQL语句
            new_list = cur.fetchall()  # 接收全部的返回结果
            for row in new_list:  # 遍历以打印出数据库信息
                name = row[0]
                grade = row[1]
                num = row[2]
                self.student_list.append(Student(name, grade, num))  # 添加至学员信息列表中
        db.close()  # 关闭数据库连接

四. 总结

• 函数
  • 定义和调⽤
  • 参数的使⽤
• ⾯向对象
  • 定义类
  • 创建对象
  • 定义和调⽤实例属性
  • 定义和调⽤实例⽅法
• 数据类型
  • 列表
    • 增加删除数据
    • 列表推导式
  • 字典
  • 字符串
• 数据库操作
  • 连接数据库
  • 数据库中创建新表
  • 向表中插入数据
  • 查询表中数据
  • 更新表中数据