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

'''
用户接口层
'''

# 导入数据库包下面的处理数据模，为了使用其内部名称空间
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 '注销成功'