1、main.py
# 1. 导入管理系统模块 from managerSystem import * # 表格创建 # 2. 启动管理系统 # 保证是当前文件运行才启动管理系统:if --创建对象并调用run方法 if __name__ == '__main__': student_manager = StudentManager() student_manager.run()
2、student.py
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
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 connection def __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 显示功能菜单 -- 打印序号的功能对应关系 -- 静态 @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): 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) 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): 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打开,如果有异常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()