Python深度培训笔记七

学员管理系统（带数据库，每秒自动保存到数据库）

student模块

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}'

StudentManager模块

import pymysql
import time

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

#启动管理系统

if __name__ == '__main__':
    student_manager = StudentManager()
    student_manager.run()

# 定义学生信息表操作类
class StudentManager:
    def __init__(self):
        # 获取数据库连接
        self.db = pymysql.connect(host="localhost", user="root", password="password", db="jdbc", charset="utf8")

        # 创建游标对象
        self.cursor = self.db.cursor()

    while True:
        print('请选择如下功能：')
        print('1:添加学员')
        print('2:删除学员')
        print('3:修改学员信息')
        print('4:查询学员信息')
        print('5:显示所有学员信息')
        print('7:自动保存学员信息')
        print('8:退出系统')

    key = input("请输入对应选择：")
    if key == "5":
        print("=" * 12, "学员信息浏览", "=" * 14)
        StudentManager.show_student()
        input("按回车继续")
    elif key == "1":
        print("=" * 12, "学员信息添加", "=" * 14)
        stu = {}
        stu['name'] = input("请输入要添加的学员姓名：")
        stu['gender'] = input("请输入要添加的学员性别：")
        stu['tel'] = input("请输入要添加的学员电话：")
        StudentManager.add_student(stu['name'], stu['gender'], stu['tel'])
        StudentManager.show_student()
        input("按回车继续")
    elif key == "7":
        print("=" * 12, "自动保存学员信息", "=" * 14)
        print("自动每秒保存数据到mysql数据库中")
        StudentManager.save_student()
        input("按回车继续")

        # 定义信息查询函数
        def show_student(self):
            # 定义sql语句进行查询
            sql = "select * from student"
            try:
                # 执行sql操作
                self.cursor.execute(sql)
                # 解析结果
                while True:
                    data = self.cursor.fetchone()
                    if data == None:
                        break;
                    print(data)
            except:
                print("操作失败！")

        # 定义添加学员信息函数
        def add_student(self, name, gender, tel):
            self.name = name
            self.gender = gender
            self.tel = tel
            # 定义sql语句进行插入
            sql = "insert into student (name,gender,tel) values('%s','%s','%s')" % (self.name, self.gender, self.tel)
            try:
                # 执行sql操作
                self.cursor.execute(sql)
                # 提交数据信息
                self.db.commit()
            except:
                self.db.rollback()
                print("操作失败！")

        def save_student(self):
            sql1 = "insert into student (name,gender,tel) values('周自珩','男','123')"
            sql2 = "insert into student (name,gender,tel) values('夏习清','男','345')"
            try:
                # 执行数据插入操作
                execute = self.cursor.execute(sql1)
                # 每隔一秒自动保存一次数据
                time.sleep(1)
                self.cursor.execute(sql2)
                # 提交数据信息
                self.db.commit()
            except:
                self.db.rollback()
                print("操作失败！")

managerSystem模块

from student import *

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

    # 一. 程序入口函数
    def run(self):
        # 1. 加载文件里面的学员数据
        self.load_student()
        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:
                # 退出系统 -- 退出循环
                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)
                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):
        # 1. 打开文件
        f = open('student.data', 'w')

        # 2. 文件写入数据
        # 2.1 [学员对象] 转换成 [字典]
        new_list = [i.__dict__ for i in self.student_list]

        # 2.2 文件写入 字符串数据
        f.write(str(new_list))

        # 3. 关闭文件
        f.close()

    # 2.8 加载学员信息
    def load_student(self):
        # 1. 打开文件：尝试r打开，如果有异常w
        try:
            f = open('student.data', 'r')
        except:
            f = open('student.data', 'w')
        else:
            # 2. 读取数据：文件读取出的数据是字符串还原列表类型；[{}] 转换 [学员对象]
            data = f.read()  # 字符串
            new_list = eval(data)
            for i in new_list:
                self.student_list = [Student(i['name'], i['gender'], i['tel'])]
            # self.student_list = [Student(i['name'], i['gender'], i['tel']) for i in new_list]
        finally:
            # 3. 关闭文件
            f.close()