#!/usr/bin/env python

# -*- coding: utf-8 -*-

import sys

import os

sys.path.append(os.path.dirname(os.path.dirname(__file__)))

import random

import time

import pickle

import os

from lib import modles

from lib.modles import Admin

from config import setting

def create_course(admin_obj):

    teacher_list = pickle.load(open(setting.TEACHER_db_dir, 'rb'))

    for num, item in enumerate(teacher_list, 1):

        print(num, item.name, item.age, item.create_time, item.create_admin.username)

    course_list = []

    while True:

        name = input('请输入课程(q退出)：')

        if name == 'q':

            break

        cost = input('请输入课时费：')

        num = input('选择老师（序号）：')

        obj = modles.Course(name, cost, teacher_list[int(num)-1], admin_obj)

        course_list.append(obj)

    if os.path.exists(setting.COURSE_db_dir):

        exists_list = pickle.load(open(setting.COURSE_db_dir, 'rb'))

        course_list.extend(exists_list)

    pickle.dump(course_list, open(setting.COURSE_db_dir, 'wb'))

def create_teacher(admin_obj):

    teacher_list = []

    while True:

        teacher_name = input('请输入老师姓名（q表示退出）')

        if teacher_name == 'q':

            break

        teacher_age = input('请输入老师年龄')

        obj = modles.Teacher(teacher_name, teacher_age, admin_obj)

        teacher_list.append(obj)

    if os.path.exists(setting.TEACHER_db_dir):

        exists_list = pickle.load(open(setting.TEACHER_db_dir, 'rb'))

        teacher_list.extend(exists_list)

    pickle.dump(teacher_list, open(setting.TEACHER_db_dir, 'wb'))

def login(user, pwd):

    if os.path.exists(os.path.join(setting.BASE_ADMIN_DIR, user)):

        # 这有bug，已解决，导入Admin和添加文件路径os.path.join(setting.BASE_ADMIN_DIR

        admin_obj = pickle.load(open(os.path.join(setting.BASE_ADMIN_DIR, user), 'rb'))

        if admin_obj.login(user, pwd):

            while True:

                sel = input("1、创建老师: 2、创建课程：q、退出")

                if sel == '':

                    create_teacher(admin_obj)

                elif sel == '':

                    create_course(admin_obj)

                else:

                    break

        else:

            return 1

    else:

        return 0

def register(user, pwd):

    admin_obj = modles.Admin()

    admin_obj.register(user, pwd)

def main():

    inp = input("1、管理员登录：2、管理员注册：\n >>>>>")

    user = input('请输入用户名：')

    pwd = input('请输入密码：')

    if inp == '':

        ret = login(user, pwd)

        if ret == 1:

            print('密码错误')

        elif ret == 0:

            print('用户不存在')

    elif inp == '':

        register(user, pwd)

if __name__ == "__main__":

    main()

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import os

BASE_DIR = os.path.dirname(os.path.dirname(__file__))

BASE_ADMIN_DIR = os.path.join(BASE_DIR, "db", "adminall")

TEACHER_db_dir = os.path.join(BASE_DIR, "db", "teacher_list")

COURSE_db_dir = os.path.join(BASE_DIR, "db", "course_list")

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import random

import time

import pickle

from config import setting

import os

class Teacher:

    """

    封装老师的相关信息

    """

    def __init__(self, name, age, admin):

        self.name = name

        self.age = age

        self.__assets = 0

        self.create_time = time.strftime('%Y-%m-%d %H:%M:%S')

        self.create_admin = admin

    def gain(self, cost):

        """

        :param cost: 资产的增加

        :return:

        """

        self.__assets += cost

    def decrease(self, cost):

        """

        :param cost: 资产的减少

        :return:资产

        """

        self.__assets -= cost

class Course:

    """

    课程相关信息

    """

    def __init__(self,course_name, cost, teacher_obj, admin):

        self.course_name = course_name

        self.cost = cost

        self.teacher_obj = teacher_obj

        self.create_time = time.strftime('%Y-%m-%d %H:%M:%S')

        self.create_admin = admin

    def have_lesson(self):

        """

    课程上课，自动给相关联的任课老师增加课时费

        :return:课程内容返回上课者

        """

        self.teacher.gain(self.cost)

        content = random.randrange(10, 100)

        r = time.strftime('%Y-%m-%d %H:%M:%S')

        temp = "课程：%s: 老师：%s: 内容: %d: 时间：" %(self.course_name, self.teacher, content, r)

        return temp

    def absence(self):

        """

        教学事故

        """

class Admin:

    def __init__(self):

        self.username = None

        self.password = None

    def login(self, user, pwd):

        """

        管理员登录

        :param user:用户名

        :param pwd: 密码

        :return:

        """

        if self.username == user and self.password == pwd:

            return True

        else:

            return False

    def register(self, user, pwd):

        """

        管理员注册

        :param user: 用户名

        :param pwd: 密码

        :return:

        """

        self.username = user

        self.password = pwd

        path = os.path.join(setting.BASE_ADMIN_DIR,self.username)

        pickle.dump(self, open(path, 'xb'))