Day32.学生选择学校

1.学生选择学校_详细代码及执行流程

2.学生选择学校_学生视图（student.py）中choice_school功能

'''学生视图'''
from lib import common
from interface import student_interface
from lib import common_interface

# todo 全局记录用户登陆状态
student_info = {'user': None}

# todo 学生选择学校
@common.auth('student')
def choice_school():
    print('学生选择学校')
    while True:
        # todo 1. 获取所有学校，让学生选择
        flag, school_list = common_interface.get_all_school_interface()
        if not flag:
            print(school_list)
            break
        else:
            for index, school_name in enumerate(school_list):
                print('编号：{}， 学校名：{}'.format(index, school_name))
        
        # todo 2. 让学生输入学校编号
        choice = input('请输入选择的学校编号：').strip()
        if not choice.isdigit():
            print('输入有误')
            continue
        else:
            choice = int(choice)

        if choice not in range(len(school_list)):
            print('输入编号有误！')
            continue

        school_name = school_list[choice]

        # todo 3. 开始调用学生选择学校接口
        flag, msg = student_interface.add_school_interface(
            school_name, student_info.get('user'))
        
        if flag:
            print(msg)
            break
        else:
            print(msg)
            break

3.学生选择学校_公共接口（common_interface.py）中获取所有学校名称接口

'''公共接口'''
import os
from conf import settings
from db import models

# todo 获取所有学校名称接口
def get_all_school_interface():
    # todo 1. 获取学校文件夹路径
    school_dir = os.path.join(
        settings.DB_PATH, 'School'
    )

    # todo 2. 判断文件夹是否存在
    if not os.path.exists(school_dir):
        return False, '没有学校，请先联系管理员'
    
    else:
        # todo 3. 文件夹若存在，则获取文件夹中所有的文件的名字
        school_list = os.listdir(school_dir)
        return True, school_list

4.学生选择学校_学生接口层（student_interface.py）中学生选择学校接口

'''学生接口层'''
from db import models

# todo 全局记录用户登陆状态
student_info = {'user': None}

# todo 学生选择学校接口
def add_school_interface(school_name, student_name):
    # todo 1. 判断当前学生是否存在学校
    student_obj = models.Student.select(student_name)
    
    if student_obj.school:
        return False, '当前学生已经选过学校了'
    
    # todo 2. 若不存在学校，则给调用学生对象中选择学校的方法，实现学生添加学校
    student_obj.add_school(school_name)

    return True, '选择学校!'

5.学生选择学校_类层（models.py）

'''用于存放类的
学校类、学员类、课程类、讲师类、管理员类
'''
from db import db_handler


# todo 提供select和save方法写入父类, 让所有子类都继承
class Base:
    # todo 定义一个数据查找方法 ----> 登录、查看数据
    @classmethod
    def select(cls, username):      # 接收到类Admin和username
        # todo obj： 对象 OR None
        obj = db_handler.select_data(
            cls, username)
        return obj

    # todo 定义一个数据保存方法 ----> 注册、保存、更新数据
    def save(self):
        # todo 调用db_handler中的save方法，并且需要对其中的方法做传值,即添加self
        # todo 让db_handler中的save_data进行保存对象数据
        db_handler.save_data(self)

# todo 学生类
class Student(Base):
    def __init__(self, user, pwd):
        self.user = user
        self.pwd = pwd
        # todo 每个学生只能有一个校区
        self.school = None
        # todo 一个学生可以选择多门课程
        self.course_list = []
        # todo 学生课程分数
        self.score = {}     # todo {"course_name": 0}
    
    # todo 学生选择学校
    def add_school(self, school_name):
        self.school = school_name
        self.save()

6.学生选择学校_db_handler.py用于保存对象与获取对象

'''
用于保存对象与获取对象
'''
import os
from conf import settings
import pickle

# todo 保存数据
def save_data(obj):
    # todo 1. 获取对象的保存文件夹路径
    # todo 以类名 当作 文件夹的名字
    # todo obj.__class__：获取当前对象的类，即models.py中调用该方法的类
    # todo obj.__class__.__name__：获取当前类的名字，即models.py中调用该方法的类名
    class_name = obj.__class__.__name__
    user_dir_path = os.path.join(
        settings.DB_PATH, class_name
    )

    # todo 2.判断文件夹是否存在，不存在则创建文件夹
    if not os.path.exists(user_dir_path):
        os.mkdir(user_dir_path)

    # todo 3.拼接当前用户的pickle文件路径，以用户作为文件名
    user_path = os.path.join(
        user_dir_path, obj.user     # 当前用户名字
    )

    # todo 4.打开文件，保存对象，通过pickle
    with open(user_path, 'wb') as f:
        pickle.dump(obj, f)
    
# todo 查看数据
def select_data(cls, username):  # 拿到models.py中刚传过来的类名（即文件夹名称）和username，
    # todo 有cls获取类名
    class_name = cls.__name__
    user_dir_path = os.path.join(
        settings.DB_PATH, class_name
    )
    # todo 2.判断文件夹是否存在，不存在则创建文件夹
    if not os.path.exists(user_dir_path):
        os.mkdir(user_dir_path)

    # todo 3.拼接当前用户的pickle文件路径，以用户作为文件名
    user_path = os.path.join(
        user_dir_path, username     # 当前用户名字
    )

    # todo 4.判断文件如果存在，再打开，并返回，若不存在，则代表用户不存在
    if os.path.exists(user_path):
        # todo 5.打开文件，获取对象
        with open(user_path, 'rb') as f:
            obj = pickle.load(f)
            return obj
    else:
        return None
        

7.学生选择学校_运行结果