[ python ] 面向对象 - 选课系统
根据源程序进行改写：
原程序地址：http://www.cnblogs.com/lianzhilei/p/5985333.html 如有侵权立即删除。
感谢原作者将完整的代码提供参考。
原作者的代码是通过 shelve 模块实现的，非常完美。具体参考上面的链接地址；以下我改写代码是通过 pickle 模块来实现。主要做为练习：
README
class_system-pickle 选课系统根目录
├── bin
│ ├── __init__.py
│ └── start.py 程序入口
├── conf
│ ├── __init__.py
│ └── settings.py 配置文件 - 主要存储数据库文件目录
├── core
│ ├── __init__.py
│ ├── manage_center.py 主菜单入口
│ ├── school_center.py 学校中心功能点实现
│ ├── student_center.py 学生中心功能点实现
│ └── teacher_center.py 讲师中心功能点实现
├── db
│ ├── __init__.py
│ └── school_dict 数据库文件（运行程序自动初始化生成）
└── modules
├── course.py 课程类
├── grade.py 班级类
├── __init__.py
├── school.py 学校类
├── student.py 学生类
├── teacher.py 讲师类
└── tools.py 工具包 - 对数据库文件读写操作
5 directories, 18 files
代码实现逻辑请阅读原作者链接。
基于 pickle 序列化的代码如下：
bin
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey from conf.settings import school_db_file from core.manage_center import Manage_center if __name__ == '__main__': obj = Manage_center() obj.run()
conf
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import os BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) school_db_file = os.path.join(BASE_DIR, 'db', 'school_dict') teacher_db_file = os.path.join(BASE_DIR, 'db', 'school_dict')
db
（不需要手动创建文件，启动程序自动生成数据文件）
core
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey from core.school_center import School_center from core.student_center import Student_center from core.teacher_center import Teacher_center class Manage_center(object): def __init__(self): pass def run(self): while True: print('1. 学校中心\n' '2. 讲师中心\n' '3. 学生中心\n' '4. 退出') choice = input('>>>').strip() if choice == '1': School_center() elif choice == '2': Teacher_center() elif choice == '3': Student_center() elif choice == '4': break else: print('\033[31;1m输入错误，请重新输入.\033[0m')
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import os, sys, pickle from conf.settings import school_db_file from modules.tools import file_oper from modules.school import School class School_center(object): def __init__(self): if not os.path.exists(school_db_file): self.init_school() self.school_db = file_oper(school_db_file, 'rb') self.manage_run() else: self.school_db = file_oper(school_db_file, 'rb') self.manage_run() def init_school(self): sh = School('上海', '上海市') bj = School('北京', '北京市') data = {'上海': sh, '北京': bj} file_oper(school_db_file, 'wb', data) def manage_run(self): for key in self.school_db: print('学校名：', key) choice_school = input('>>>').strip() if choice_school in self.school_db: self.choice_school = choice_school self.school_obj = self.school_db[choice_school] while True: print('\n欢迎进入【%s】校区\n' '创建课程 add_course\n' '查看课程 check_course\n' '创建班级 add_grade\n' '查看班级 check_grade\n' '创建讲师 add_teacher\n' '查看讲师 check_teacher\n' '退出系统 exit' % choice_school) user_func = input('>>>').strip() if hasattr(self, user_func): getattr(self, user_func)() def add_course(self): course_name = input('\033[34;1m输入课程名:\033[0m') course_price = input('\033[34;1m输入课程价格:\033[0m') course_time = input('\033[34;1m输入课程周期:\033[0m') if course_name not in self.school_obj.school_course: self.school_obj.create_course(course_name, course_price, course_time) self.school_db[self.choice_school] = self.school_obj file_oper(school_db_file, 'wb', self.school_db) print('\033[32;1m课程新增成功。\033[0m') else: print('\033[31;1m课程信息已存在.\033[0m') self.school_obj.create_course(course_name, course_price, course_time) print('\033[32;1m课程更新成功。\033[0m') def check_course(self): self.school_obj.show_course() def add_grade(self): grade_name = input('\033[34;1m输入班级名:\033[0m') course_name = input('\033[34;1m输入班级要上的课程名:\033[0m') if course_name in self.school_obj.school_course: course_obj = self.school_obj.school_course[course_name] if grade_name not in self.school_obj.school_grade: self.school_obj.create_grade(grade_name, course_obj) self.school_db[self.choice_school] = self.school_obj file_oper(school_db_file, 'wb', self.school_db) else: print('\033[31;1m班级信息已存在.\033[0m') else: print('\033[31;1m课程信息不存在.\033[0m') def check_grade(self): self.school_obj.show_grade() def add_teacher(self): teacher_name = input('\033[34;1m输入讲师名:\033[0m') teacher_salary = input('\033[34;1m输入讲师薪资:\033[0m') grade_name = input('\033[34;1m输入讲师关联的班级:\033[0m') if grade_name in self.school_obj.school_grade: grade_obj = self.school_obj.school_grade[grade_name] if teacher_name not in self.school_obj.school_teacher: self.school_obj.create_teacher(teacher_name, teacher_salary, grade_name, grade_obj) print('\033[32;1m讲师新增成功.\033[0m') else: print('\033[31;1m讲师信息已存在.\033[0m') self.school_obj.create_teacher(teacher_name, teacher_salary, grade_name, grade_obj) print('\033[32;1m讲师更新成功.\033[0m') self.school_db[self.choice_school] = self.school_obj file_oper(school_db_file, 'wb', self.school_db) else: print('\033[31;1m班级信息不存在，请先创建班级.\033[0m') def check_teacher(self): self.school_obj.show_teacher() def exit(self, *args): sys.exit('欢迎下次使用选课系统.')
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import os, sys, pickle from conf.settings import school_db_file from modules.tools import file_oper from modules.school import School class Student_center(object): def __init__(self): if os.path.exists(school_db_file): self.school_db = file_oper(school_db_file, 'rb') self.manage_run() else: print('\033[31;1m数据库文件不存在，请先初始化.\033[0m') def manage_run(self): for key in self.school_db: print('学校名：', key) choice_school = input('>>>').strip() if choice_school in self.school_db: self.choice_school = choice_school self.school_obj = self.school_db[choice_school] student_name = input('\033[34;1m输入学生名:\033[0m').strip() student_age = input('\033[34;1m输入学生年龄:\033[0m').strip() self.school_obj.show_grade_course() grade_name = input('\033[34;1m输入要选择的班级名:\033[0m') if grade_name in self.school_obj.school_grade: self.school_obj.create_student(student_name, student_age, grade_name) self.school_db[self.choice_school] = self.school_obj file_oper(school_db_file, 'wb', self.school_db) print('\033[32;1m选课成功.\033[0m') else: print('\033[31;1m班级信息不存在.\033[0m')
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import os, sys from modules.tools import file_oper from conf.settings import school_db_file, teacher_db_file class Teacher_center(object): def __init__(self): if os.path.exists(school_db_file): self.school_db = file_oper(school_db_file, 'rb') self.manage_run() else: print('\033[31;1m数据库不存在，请初始化数据库.') exit(1) # if not os.path.exists(teacher_db_file): # self.init_teacher() # self.manage_run() # else: # self.teacher_db = file_oper(teacher_db_file, 'rb') # self.manage_run() def init_teacher(self): data = {} file_oper(teacher_db_file, 'wb', data) def manage_run(self): for key in self.school_db: print('学校名：', key) choice_school = input('\033[34;1m输入学校名:\033[0m') if choice_school in self.school_db: self.choice_school = choice_school self.school_obj = self.school_db[choice_school] teacher_name = input('\033[34;1m输入讲师名:\033[0m') if teacher_name in self.school_obj.school_teacher: while True: print('\n欢迎进入讲师中心\n' '查看班级信息 check_grade\n' '退出 exit') user_func = input('>>>').strip() if hasattr(self, user_func): getattr(self, user_func)(teacher_name) def check_grade(self, teacher_name): self.school_obj.show_teacher_grade(teacher_name) def exit(self, *args): sys.exit('欢迎下次使用选课系统.')
modules
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey class Course(object): def __init__(self, course_name, course_price, course_time): self.course_name = course_name self.course_price = course_price self.course_time = course_time
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey class Grade(object): def __init__(self, grade_name, course_obj): self.grade_name = grade_name self.course_obj = course_obj self.grade_student = {}
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import pickle from modules.course import Course from modules.grade import Grade from modules.teacher import Teacher from modules.student import Student class School(object): def __init__(self, school_name, school_addr): self.school_name = school_name self.school_addr = school_addr self.school_course = {} self.school_grade = {} self.school_teacher = {} def create_course(self, course_name, course_price, course_time): course_obj = Course(course_name, course_price, course_time) self.school_course[course_name] = course_obj def show_course(self): for key in self.school_course: course_obj = self.school_course[key] print('\033[32;1m课程【%s】\t价格【%s元】\t周期【%s个月】\033[0m' % ( course_obj.course_name, course_obj.course_price, course_obj.course_time)) def create_grade(self, grade_name, course_obj): grade_obj = Grade(grade_name, course_obj) self.school_grade[grade_name] = grade_obj def show_grade(self): for key in self.school_grade: grade_obj = self.school_grade[key] print('\033[32;1m班级名【%s】\t课程【%s】\033[0m' % (grade_obj.grade_name, grade_obj.course_obj.course_name)) def create_teacher(self, teacher_name, teacher_salary, grade_name, grade_obj): teacher_obj = Teacher(teacher_name, teacher_salary) teacher_obj.teacher_add_grade(grade_name, grade_obj) self.school_teacher[teacher_name] = teacher_obj def show_teacher(self): for key in self.school_teacher: teacher_obj = self.school_teacher[key] grade_list = [] for key in teacher_obj.teacher_grade: grade_list.append(key) print('\033[32;1m讲师【%s】\t薪资【%s】\t班级列表【%s】\033[0m' % ( teacher_obj.teacher_name, teacher_obj.teacher_salary, grade_list)) def show_grade_course(self): for key in self.school_grade: grade_obj = self.school_grade[key] course_obj = grade_obj.course_obj print('\033[32;1m班级【%s】\t课程【%s】\t价格【%s元】\t周期【%s个月】\033[0m' % ( grade_obj.grade_name, course_obj.course_name, course_obj.course_price, course_obj.course_time)) def create_student(self, student_name, student_age, grade_name): student_obj = Student(student_name, student_age) grade_obj = self.school_grade[grade_name] grade_obj.grade_student[student_name] = student_obj def show_teacher_grade(self, teacher_name): teacher_obj = self.school_teacher[teacher_name] for key in teacher_obj.teacher_grade: grade_obj = teacher_obj.teacher_grade[key] student_list = [] for st in grade_obj.grade_student: student_list.append(st) print('\033[32;1m班级【%s】课程【%s】学生列表【%s】\033[0m' % ( grade_obj.grade_name, grade_obj.course_obj.course_name, student_list))
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey class Student(object): def __init__(self, student_name, student_age): self.student_name = student_name self.student_age = student_age
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey class Teacher(object): def __init__(self, teacher_name, teacher_salary): self.teacher_name = teacher_name self.teacher_salary = teacher_salary self.teacher_grade = {} def teacher_add_grade(self, grade_name, grade_obj): self.teacher_grade[grade_name] = grade_obj
#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: hkey import pickle def file_oper(file, mode, *args): if mode == 'wb': data = args[0] with open(file, mode) as f: pickle.dump(data, f) elif mode == 'rb': with open(file, mode) as f: data = pickle.load(f) return data
