Day32.学生注册

1.学生注册_详细代码及执行流程

2.学生注册_学生视图层(student.py)_学生注册register功能函数

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

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

# todo 学生注册
def register():
    while True:
        student_username = input('请输入用户名：').strip()
        student_password = input('请输入密码：').strip()
        student_re_password = input('请确认密码：').strip()

        # todo 小的逻辑判断
        if student_password == student_re_password:
            # 调用接口层，学生注册接口
            flag, msg = student_interface.student_register_interface(
                student_username, student_password
            )
            
            if flag:
                print(msg)
                break
            
            else:
                print(msg)

        else:
            print('两次密码不一致，请重新输入')

3.学生注册_学生接口层(student_interface.py)_学生注册接口student_register_interface功能

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

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

# todo 学生注册接口
def student_register_interface(student_username, student_password):
    # todo 1.判断用户是否存在
    # todo 调用Student类中的，select方法，有该方法去调用db_handler中
    # todo 的select_data功能获取对象
    student_obj = models.Student.select(student_username)
    
    # todo 1.1）若存在不允许注册，返回用户已存在给视图层
    if student_obj:
        return False, '学生用户已存在！'
    
    else:
        # todo 1.2）若不存在则允许注册，调用类实例化得到对象并保存
        student_obj = models.Student(student_username, student_password)

        # todo 对象调用save() 会将student_obj传给save方法
        student_obj.save()
        
        return True, '学生注册成功！'

4.学生注册_存放类(models.py)_学生类Student功能

'''用于存放类的
学校类、学员类、课程类、讲师类、管理员类
'''
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
        self.course_list = []
        self.score = {}     # todo {"course_name": 0}

5.学生注册_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

6.学生注册_学生注册运行结果