import hashlib
import pickle
import os
class Iden:
# 用户的父类
def __init__(self, user, pwd, ):
self.user = user
self.pwd = pwd
# 学生类
class Normal(Iden):
def __init__(self, user, pwd, course='无'):
super().__init__(user, pwd)
self.groupwork = 'student'
# 普通用户的注册
def normal_regedist(user, pwd, ):
pwd = hashli(user, pwd)
person = Normal(user, pwd)
picke_write(person, f'./user/{user}')
return True
# 学生选择课程
def use_course(self, course):
self.course = course
picke_write(self, f'./user_course/{self.user}_course')
return '添加成功'
# 读取自己的课程
def read_course(self):
student_user_course = os.listdir('./user_course/')
if f'{self.user}_course' in student_user_course:
return picke_read(f'./user_course/{self.user}_course')
return False
# 管理员类
class Admin(Iden):
def __init__(self, user, pwd, ):
super().__init__(user, pwd)
self.groupwork = 'administrator'
# 管理员的注册，
def __admin_regedist(user, pwd):
pwd = hashli(user, pwd)
person = Admin(user, pwd, )
picke_write(person, f'./user/{user}')
return True
# 查看所有的学生
def see_all_student(self):
student_user = os.listdir('./user/')
for us in student_user:
if us != self.user:
print(f"账号：{picke_read(f'./user/{us}').user}")
# 查看所有的学生的课程
def see_all_student_course(self):
student_user_course = os.listdir('./user_course/')
for us in student_user_course:
course_read(picke_read(f'./user_course/{us}'))
# 使用picke读取文件
def picke_read(path):
with open(path, 'rb')as f:
a = pickle.load(f)
return a
# 使用picke写入文件
def picke_write(self, path):
with open(path, 'wb')as f:
pickle.dump(self, f)
# 打印课程信息,格式化
def course_read(a): print(f'账号：{a.user},课程名称:{a.course.name},课程价格:{a.course.price}，'
f'课程周期:{a.course.month}，课程讲师:{a.course.teacher}')
# 课程类
class Course:
def __init__(self, name, price, month, teacher):
self.name = name
self.price = price
self.month = month
self.teacher = teacher
@staticmethod
# 创建课程
def create_course(name, price, month, teacher):
course = Course(name, price, month, teacher)
print(course.name)
with open('./course/course', 'ab') as f:
pickle.dump(course, f)
return True
@staticmethod
# 读取所有的课程，并将结果以列表的形式返回
def read_all_course():
li = []
with open('./course/course', 'rb') as f:
while True:
try:
li.append(pickle.load(f).__dict__)
except:
break
return li
# 查看所有的课程
def see_course():
course = Course.read_all_course()
for i in range(len(course)):
print(f'序号{i + 1} 课程名称{course[i]["name"]},课程价格{course[i]["price"]},\
课程周期{course[i]["month"]},课程讲师{course[i]["teacher"]}')
return course
# 数据加密
def hashli(user, pwd):
md5 = hashlib.sha256(user.encode('utf-8'))
md5.update(pwd.encode('utf-8'))
return md5.hexdigest()
# 登录验证，正确就返回账户的信息（包含类中的方法）
def login(user, pwd):
lisdis = os.listdir('./user/')
if user in lisdis:
with open(f'./user/{user}', 'rb') as f:
user_pwd_dic = pickle.load(f)
if user == user_pwd_dic.user and hashli(user, pwd) == user_pwd_dic.pwd:
return user_pwd_dic
else:
return
else:
return
# 用户登录输入
def login_input():
while 1:
user = input('请输入用户名>>>').replace(' ', '')
pwd = input('请输入密码>>>').replace(' ', '')
if user and pwd:
break
else:
print('输入有误，请重新输入')
return user, pwd
# 创建课程输入
def courese_input():
name = input('请输入课程名称>>>').replace(' ', '')
price = int(input('请输入课程价格>>>').replace(' ', ''))
month = input('请输入课程周期>>>').replace(' ', '')
teacher = input('请输入授课老师>>>').replace(' ', '')
Course.create_course(name, price, month, teacher)
return
flag = True
while flag:
user, pwd = login_input() # 输入用户密码
a = login(user, pwd) # 登录验证
if a:
if a.groupwork == 'administrator': # 如果身份是管理员身份
print(f'欢迎管理员{a.user}登录')
while flag:
print('1,创建用户。 2,创造课程 3,查看课程 4,查看学生 5,查看学生选课 6,退出程序 ')
choose = input('请输入你要选择的操作').replace(' ', '')
if choose and choose.isnumeric() and int(choose) <= 6 and int(choose) > 0:
if choose == '1':
user, pwd = login_input()
if Normal.normal_regedist(user, pwd): print(f'创建学生用户{user}成功')
elif choose == '2':
b = courese_input()
elif choose == '3':
look_course = see_course()
elif choose == '4':
a.see_all_student()
elif choose == '5':
a.see_all_student_course()
elif choose == '6':
print('欢迎下次使用')
flag = False
break
else:
print('输入不正确，请重新输入')
elif a.groupwork == 'student':
print(f'欢迎学生{a.user}')
while True:
print('1,查看课程 2,选择课程 3,查看所选课程 4,退出程序 ')
choose = input('请输入你要选择的操作').replace(' ', '')
if choose and choose.isnumeric() and int(choose) <= 4 and int(choose) > 0:
if choose == '1':
look_course = see_course()
elif choose == '2':
while flag:
course = see_course()
n = input('请输入课程编号(输入q退出)>>>').replace(' ', '')
if n and n.isnumeric() and int(n) <= len(course) and int(n) > 0:
n = int(n) - 1
choose_course = Course(course[n]['name'], course[n]['price'], course[n]['month'],
course[n]['teacher'])
a_course = a.use_course(choose_course)
print(a_course)
elif n.lower() == 'q':
break
else:
print('输入有误，请重新输入')
elif choose == '3':
is_course = (a.read_course())
if is_course == False:
print('你现在还没有选择课程')
else:
course_read(is_course)
elif choose == '4':
print('欢迎下次使用')
flag = False
break
else:
print('输入有误，请重新输入')
else:
print('用户不存在，或者密码错误')
