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()