Day31.管理员登录功能

1.管理员登录功能_代码导入及执行流程

2.管理员登录功能_视图层admin.py功能

# todo 管理员登录
def login():
    while True:
        admin_username = input('请输入用户名：').strip()
        admin_password = input('请输入密码：').strip()
        # todo 1. 调用管理员登录接口
        flag, msg = admin_interface.admin_login_interface(
            admin_username, admin_password
        )
        if flag:
            # todo 在全局定义字典user_info, 记录当前用户登录状态, 修改其中字典状态
            user_info['user'] = admin_username
            print(admin_username, msg)
            break
        
        else:
            print(msg)

3.管理员登录功能_管理员登录接口admin_interface.py

# todo 管理员登录接口
def admin_login_interface(admin_username, admin_password):
    # todo 1. 判断用户是否存在
    admin_obj = models.Admin.select(admin_username)

    # todo 2. 若不存在，则证明用户不存在并返回给视图层
    if not admin_obj:
        return False, '用户不存在'

    # todo 3. 若存在，则校验密码
    else:
        if admin_password == admin_obj.pwd:
            return True, '用户登录成功'
        else:
            return False, '密码错误'

4.管理员登录功能_models.py存放各种类，其中Admin类调用db_handler.py中的方法

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


class Admin:
    # todo 调用类的时候触发
    # todo admin_username, admin_password
    def __init__(self, user, pwd):
        self.user = user
        self.pwd = pwd
    
    # todo 定义一个数据保存方法
    def save(self):
        # todo 调用db_handler中的save方法，并且需要对其中的方法做传值,即添加self
        # todo 让db_handler中的save_data进行保存对象数据
        db_handler.save_data(self)

    # todo 定义一个数据查找方法
    @classmethod
    def select(cls, username):      # 接收到Admin和username
        # todo obj： 对象 OR None
        obj = db_handler.select_data(
            cls, username)
        return obj

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.管理员登录功能_登录功能运行结果