入门项目 A5-3 interface-user 第三方接口3

时间:2022-04-15 15:31:54
'''
用户接口层
'''

# 导入数据库包下面的处理数据模,为了使用其内部名称空间
from db import db_handler

# 注册接口函数,接收名字与密码两个参数
def register_interface(name, pwd, balance=15000):
    user_dic = db_handler.select(name)  # 用户信息字典 == 数据库包 里面的 查找用户名是否存在得返回值
    # 返回值(不返回则返回了(没有设置返回值),存在则返回用户信息字典)

    if user_dic:  # 如果返回了字典说明用户名在之前已经被注册
        return False, '用户已存在'  # 返回给外界两个值(错误,用户已存在)

    # 以上条件没有成立,说明用户信息不存在,用户的注册名没有被别人注册,可以执行下面代码,将用户信息写入
    user_dic = {  # 定义用户信息字典
        'name': name, 'pwd': pwd, 'balance': balance, 'flow': [], 'shopping_cart': {}
        # 用户名[k]:姓名,
        # 用户密码[k]:密码,
        # 用户流水[K]:空列表后续添加信息,
        # 购物车[k]:空字典后续添加用户选择商品信息
    }
    db_handler.save(user_dic)
    # 调用数据包里面的数据库模下面的保存字典函数,并传入已经处理好的用户信息字典
    return True, '注册成功!'  # (True,'注册成功!')此处return向外界返回了两个值,布尔值True 和 字符串‘注册成功’

# 登陆接口功能函数,接收名字与密码两个参数
def login_interface(name, pwd):
    user_dic = db_handler.select(name)  # 调用数据库的select函数功能查找用户是否存在
    # user_dic 接收返回 (用户信息字典/None)
    if not user_dic:  # if not None = if True ,执行内部代码
        return False, '用户不存在!'  # 返回 错误,用户不存在,此处的False是为了给前端flag两种回复方案
    if pwd == user_dic['pwd']:  # 如果输入的密码等于用户信息字典中密码K的值
        return True, '登陆成功!'  # 返回True,登陆成功
    else:
        return False, '密码错误!'  # 返回False,密码错误

# 查看用户余额接口
def check_balance_interface(name):
    # 用户信息/None
    user_dic = db_handler.select(name)
    if not user_dic['balance']:
        return '请充值..'
    return user_dic['balance']

# 注销功能
def logout_interface():
    from core import src
    src.user_info['name'] = None
    return '注销成功'