#!/usr/bin/env python
# -*- coding:utf-8 -*-
#-Author-Lian 

import pickle,os
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
#數(shù)據(jù)地址
__db_main = BASE_DIR + r"\database\main_dict"
__db_teacher = BASE_DIR + r"\database\teacher_dict"
 
class School(object):
 #創(chuàng)建學校

 def __init__(self,name,addr):

 self.name = name

 self.addr = addr

 def cat_school(self):
 print("學校名：【%s】\t地址：【%s】"%(self.name,self.addr))
 

 def hire_teacher(self,dict,course,teacher,file):
 #數(shù)據(jù)庫添加講師信息
 dict[self][course] = {"teacher":teacher}
 file_oper(file,"wb", dict)
 
 def create_course(self,dict,course,file):
 # 數(shù)據(jù)庫添加課程資料
 dict[self][course]={}
 file_oper(file,"wb", dict)
 def create_grade(self,dict,teacher_dict,course,grade,teacher,file1,file2):
 #數(shù)據(jù)庫添加班級信息
 dict[self][course]["grade"] = grade
 file_oper(file1, "wb", dict)
 teacher_dict[teacher] = {"grade":grade}
 file_oper(file2, "wb", teacher_dict)

class Course():
 #創(chuàng)建課程
 def __init__(self,name,price,time):
 self.name = name
 self.price = price
 self.time = time
 def cat_course(self):
 #查看課程信息
 print("課程：【%s】\t價格：【￥%s】\t周期：【%s個月】"
  %(self.name,self.price,self.time))

class Grade():
 # 創(chuàng)建班級
 def __init__(self,name,course,teacher):
 student = set([])
 self.name = name
 self.course = course
 self.teacher = teacher
 self.student = student

 

 def cat_grade(self):
 #查看班級信息
 print("班級：【%s】\t課程：【%s】\t講師：【%s】"
  %(self.name, self.course, self.teacher))

 

 def add_student(self,student_name,dict,teacher,file):
 self.student.add(student_name)
 dict[teacher]={"grade":self}
 file_oper(file, "wb", dict)

class People():
 def __init__(self,name,age):
 self.name = name
 self.age = age
class Teacher(People):
 # 創(chuàng)建講師
 def __init__(self,name,age,school,course,role="講師"):
 super(Teacher,self).__init__(name,age)
 self.role = role
 self.school = school
 self.course = course
 def cat_teacher(self):
 #查看老師資料和課程
 print('課程【%s】\t講師【%s】'%(self.course,self.name))


def file_oper(file,mode,*args):

 #數(shù)據(jù)庫寫入、讀取操作

 if mode == "wb":
 with open(file, mode) as f:
  dict = args[0]
  f.write(pickle.dumps(dict))

 if mode == "rb":
 with open(file, mode) as f:
  dict = pickle.loads(f.read())
  return dict

def information(dict,mode,*args):

 '''通過匹配mode模式，打印相應的輸出信息'''
 if args:
 dict_info, set_info = {}, args[0]
 else:
 dict_info,set_info = {},set([])
 if dict:
 for key in dict:
  if mode == "course":
  key.cat_course()
  if mode == "main":
  key.cat_school()
  if mode == "teacher" and key == "teacher":
  dict[key].cat_teacher()
  # dict_info[key] = dict[key]
  set_info.add(dict[key].name)
  if mode == "grade" and key == "grade":
  dict[key].cat_grade()
  set_info.add(dict[key].name)
  if mode == "teacher_center":
  pass
  if type(key) != str: #key值不是字符串
  dict_info[key.name] = key
 return dict_info,set_info

 

def school_center():

 #學校管理中心

 Flag = True

 while Flag:
 dict_main = file_oper(__db_main,"rb") #主字典
 res_dict = information(dict_main,"main")[0] #打印學校信息
 school_name = input("\33[34;0m輸入要選擇的學校名\33[0m：").strip()
 if school_name in res_dict:
  school = res_dict[school_name]  #匹配選擇的學校
  while Flag:
  print("\33[32;1m歡迎進入【%s】學校\33[0m".center(50, "*")%school.name)
  choice = options(list_school) #打印當前選項
  if choice == "1":
   while True:
   print("\33[32;0m學?！?s】目前已經(jīng)有的班級信息\33[0m".center(40, "-")%school.name)
   teacher_dict = file_oper(__db_teacher,"rb")
   res_course = information(dict_main[school], "None")[0]
   set_info = set([])
   if res_course: # 打印課程與講師對應關系
    for i in res_course:
    k = res_course[i]
    res_grade = information(dict_main[school][k], "grade",set_info)[1]
   if_cont = input("\n\33[34;0m是否要創(chuàng)建班級 【y】創(chuàng)建 【b】退出\33[0m:")
   if if_cont == "y":
    grade_name = input("\33[34;0m輸入要創(chuàng)建班級的名稱\33[0m:").strip()
    course_name = input("\33[34;0m輸入要班級要上的課程\33[0m:").strip()
    if course_name in res_course:
    course = res_course[course_name]
    if dict_main[school][course]:
     teacher = dict_main[school][course]["teacher"]
     if grade_name not in res_grade:
     grade = Grade(grade_name, course_name, teacher.name)
     school.create_grade(dict_main, teacher_dict, course, grade, teacher, __db_main,
       __db_teacher)
     else:

     print("\33[31;0m錯誤：當前班級已經(jīng)存在\33[0m")

    else:

     print("\33[31;0m錯誤：當前課程還沒有講師\33[0m")

    else:

    print("\33[31;0m錯誤:課程【%s】不存在，請先創(chuàng)建課程\33[0m" % course_name)
   if if_cont == "b":
    break
 
  if choice == "2":
   #招聘講師
   while True:
   print("\33[32;0m學?！?s】目前已經(jīng)有的課程與講師\33[0m".center(40, "-")%school.name)
   res_course = information(dict_main[school],"None")[0]
   set_info = set([])
   if res_course:   #打印課程與講師對應關系
    for i in res_course:
    k = res_course[i]
    res_teacher = information(dict_main[school][k], "teacher",set_info)[1]
    if not res_teacher:
     print("課程【%s】\t講師【None】" %(i))

   if_cont = input("\n\33[34;0m是否要招聘講師 【y】招聘 【b】退出\33[0m:")
   if if_cont == "y":
    teacher_name = input("\33[34;0m輸入要招聘講師的名字\33[0m:").strip()
    teacher_age = input("\33[34;0m輸入要招聘講師的年齡\33[0m:").strip()
    course_name = input("\33[34;0m輸入講師【%s】要授課的課程\33[0m:"%teacher_name).strip()
    if course_name in res_course:
     course = res_course[course_name] #創(chuàng)建講師并寫入數(shù)據(jù)庫
     if teacher_name not in res_teacher:
     teacher = Teacher(teacher_name,teacher_age,school.name,course_name)
     school.hire_teacher(dict_main, course, teacher, __db_main)
     else:
     print("\33[31;0m錯誤:教師【%s】已經(jīng)被聘用\33[0m" %teacher_name)
    else:
    print("\33[31;0m錯誤:課程【%s】不存在，請先創(chuàng)建課程\33[0m" %course_name)
   if if_cont == "b":
    break
  if choice == "3":
   #創(chuàng)建課程
   while True:
   print("\33[32;0m學?！?s】目前已經(jīng)有的課程\33[0m".center(40,"-")%school.name)
   res_dict = information(dict_main[school],"course")[0] #打印課程信息賦值給字典course_dict
   if_cont = input("\n\33[34;0m是否要創(chuàng)建課程 【y】創(chuàng)建 【b】退出\33[0m:")
   if if_cont == "y":
    course_name = input("\33[34;0m輸入要創(chuàng)建的課程\33[0m:").strip()
    if course_name not in res_dict: #課程不存在，創(chuàng)建
    price = input("\33[34;0m輸入課程 【%s】 的價格\33[0m:" % (course_name))
    time = input("\33[34;0m輸入課程 【%s】 的周期（月）\33[0m:" % (course_name))
    course = Course(course_name, price, time) #創(chuàng)建課程course
    school.create_course(dict_main,course, __db_main) #關聯(lián)學校和課程
    else:     #課程存在
    print("\33[31;0m錯誤：當前課程 【%s】 已經(jīng)存在\33[0m" % (course_name))
   if if_cont == "b":
    break
 

  if choice == "4":
   Flag = False
 if Flag:
  print("\33[31;0m錯誤：輸入的學校 【%s】 不存在\33[0m"%(school_name))
def teacher_center():

 #講師中心

 print("\33[32;1m歡迎進入講師中心\33[0m".center(50, "*"))
 teacher_dict = file_oper(__db_teacher, "rb")

 dict_info = information(teacher_dict,"teacher_center")[0] #驗證登錄

 teacher_name = input("\n\33[34;0m輸入要登錄講師的名字\33[0m:").strip()

 if teacher_name in dict_info:
  while True:
  print("\33[32;1m歡迎進入講師【%s】的管理中心\33[0m".center(40, "*")%teacher_name)
  choice = options(list_teacher)
  teacher = dict_info[teacher_name]
  grade = teacher_dict[teacher]["grade"]
  if choice == "1":
   print("\33[32;0m講師【%s】的班級信息\33[0m".center(40,"-")%teacher.name)
   print("學校【%s】\t課程【%s】\t班級【%s】\t"%(teacher.school,teacher.course,grade.name))
   any = input("\n\33[34;0m輸入任意鍵退出當前\33[0m:")
  if choice == "2":
   print("\33[32;0m講師【%s】的班級學員列表\33[0m".center(40, "-") % teacher.name)
   print("班級【%s】\n學員【%s】"%(grade.name,grade.student))
   any = input("\n\33[34;0m輸入任意鍵退出當前\33[0m:")
  if choice == "3":
   break
 else:
  print("\33[31;0m錯誤：講師【%s】 不存在\33[0m"%(teacher_name))
def student_center():
 #學員中心
 print("\33[32;1m歡迎進入學員中心中心\33[0m".center(50, "*"))
 while True:
 choice = options(list_student)  #打印學生中心選項
 if choice == "1":
  student_name = input("\33[34;0m輸入學員的名字\33[0m：")
  dict = file_oper(__db_main, "rb")
  teacher_dict = file_oper(__db_teacher,"rb")
  school_dict = information(dict,"main")[0] #打印當前可選的學校
  school_name = input("\33[34;0m輸入要選擇的學校名\33[0m：").strip()
  if school_name in school_dict:
  school = school_dict[school_name]
  if dict[school]:
   course_dict = information(dict[school],"course")[0] # 打印當前學校下的課程
   course_name = input("\33[34;0m輸入要選擇的課程\33[0m：").strip()
   if course_name in course_dict:
   course = course_dict[course_name]
   if dict[school][course].get("grade"):
    for i in teacher_dict:
    if course.name == i.course:
     teacher = i
     grade = teacher_dict[teacher]["grade"]
    print("課程【%s】的費用為【%s】"%(course.name,course.price))
    if_pay = input("\33[34;0m是否支付當前費用 支付【y】\33[0m:")
    if if_pay == "y":  #上面全部匹配成功，選課成功
    grade.add_student(student_name,teacher_dict,teacher,__db_teacher)
    print("\33[31;0m選課成功\33[0m")
    any = input("\n\33[34;0m輸入任意鍵退出當前\33[0m:")
   else:
    print("\33[31;0m錯誤：課程沒有班級\33[0m")
   else:
   print("\33[31;0m錯誤：課程不存在\33[0m")
  else:
   print("\33[31;0m錯誤：當前學校沒有課程\33[0m")
 if choice == "2":
  break

def options(list):
 #打印可選擇的操作模式，并返回選擇值
 for i, v in enumerate(list):
 print(i+1, v)
 choice = input("\33[34;0m選擇要進入模式\33[0m:")
 return choice 

def start():
 '''程序開始'''
 while True:
 print("\33[35;1m歡迎進入選課系統(tǒng)\33[0m".center(50, "#"))
 choice = options(list_main) #打印選項
 if choice == "1":
  student_center() #學生中心
 if choice == "2":
  teacher_center() #教師中心
 if choice == "3":
  school_center()  #學校中心
 if choice == "4":
  break

def init_database():
 '''數(shù)據(jù)庫初始化，不存在則創(chuàng)建，存在跳過'''
 bj = School("北京","北京市")
 sh = School("上海","上海市")
 if not os.path.exists(__db_teacher):
 dict = {bj:{},sh:{}}
 file_oper(__db_main,"wb",dict)
 if not os.path.exists(__db_teacher):
 dict = {}
 file_oper(__db_teacher,"wb",dict)

if __name__ == '__main__':
 init_database() #初始化數(shù)據(jù)庫
 list_main = ["學生中心", "講師中心", "學校中心","退出"]
 list_school = ["創(chuàng)建班級", "招聘講師", "創(chuàng)建課程","返回"]
 list_teacher = ["查看班級", "查看班級學員列表","返回" ]
 list_student = ["學員注冊","返回"]
 start()