managerSystem.py

import pymysql
import time
DBHOST = 'localhost'
DBUSER = 'root'
DBPASS = '*****'
DBNAME = 'dbtest'from student import *class StudentManager(object):def get_connectiong(self):# 连接数据库connection = pymysql.connect(host = 'localhost',user= 'root',password = '85553890wangzhiy',database = 'dbtest')return  connectiondef __init__(self):# 存储学员数据 -- 列表self.student_list = []# 一. 程序入口函数def run(self):# 1. 加载文件里面的学员数据self.load_student()while True:# 2. 显示功能菜单self.show_menu()self.get_connectiong()# 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 显示功能菜单 -- 打印序号的功能对应关系 -- 静态@staticmethoddef show_menu():print('请选择如下功能：')print('1:添加学员')print('2:删除学员')print('3:修改学员信息')print('4:查询学员信息')print('5:显示所有学员信息')print('6:保存学员信息')print('7:退出系统')# 2.2 添加学员def add_student(self):connection= self.get_connectiong()#数据库连接cur = connection.cursor(cursor=pymysql.cursors.DictCursor)sql="insert into student(Name,gender,tel) values('%s','%s','%s')"# 1. 用户输入姓名、性别、手机号name = input('请输入您的姓名：')gender = input('请输入您的性别：')tel = input('请输入您的手机号：')# 2. 创建学员对象 -- 类 ？类在student文件里面  先导入student模块，再创建对象student = Student(name, gender, tel)# 3. 将该对象添加到学员列表self.student_list.append(student)try:cur.execute(sql % (name,gender,tel)) #插入数据connection.commit()except pymysql.Error as e:print('表格创建失败' + str(e))finally:connection.close()  # 关闭连接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)breakelse:# 循环正常价结束执行的代码：循环结束都没有删除任何一个对象，所以说明用户输入的目标学员不存在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}')breakelse: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}')breakelse:print('查无此人！')# 2.6 显示所有学员信息def show_student(self):connection = self.get_connectiong()  # 数据库连接cur = connection.cursor(cursor=pymysql.cursors.DictCursor)sql = "insert into student(Name,gender,tel) values('%s','%s','%s')"# 1. 打印表头print('姓名\t性别\t手机号')# 2. 打印学员数据for i in self.student_list:print(f'{i.name}\t{i.gender}\t{i.tel}')def timer(n):while True:try:cur.execute(sql % (i.name, i.gender, i.tel))  # 插入数据connection.commit()except pymysql.Error as e:print('表格创建失败' + str(e))time.sleep(n)  # sleep一段时间再执行timer(1)#每秒向数据库输入数据# 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打开，如果有异常wtry: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()