Day26.管理员功能

1.管理员功能_修改用户额度

 　　admin.py中的代码：

# 修改用户额度
def change_balance():
    while True:
        # 1）输入需要修改的用户名
        change_user = input('请输入需要修改额度的用户：').strip()
        
        # 2）修改的用户额度
        money = input('请需要修改的用户额度：').strip()
        if not money.isdigit():
            continue

        # 3）调用修改额度接口
        flag, msg = admin_interface.change_balance_interface(change_user, money)
        if flag:
            print(msg)
            break
        else:
            print(msg)

　　admin_interface.py中的代码：

from ATM.db import db_handler

# 修改额度接口
def change_balance_interface(username, input_money):
    user_dic = db_handler.select(username)
    # 用户存在
    if user_dic:
        user_dic['balance'] = int(input_money)
        # 保存修改后的数据
        db_handler.save(user_dic)
        return True, '用户:{}，额度修改为:{}'.format(username, input_money)
    else:
        return False, '用户:{}，不存在'.format(username)

2.管理员功能_添加用户和冻结用户

  　　admin.py中的代码：

# 添加账户
def add_user():
    src.register()

# 冻结账户
def lock_user():
    while True:
        # 1）输入需要冻结的用户名
        change_user = input('请输入需要修改额度的用户：').strip()
        flag, msg =  admin_interface.lock_user_interface(change_user)
        if flag:
            print(msg)
            break
        else:
            print(msg)

　　admin_interface.py中的代码：

# 冻结用户接口
def lock_user_interface(username):
    user_dic = db_handler.select(username)
    if user_dic:
        # 将loked的默认值改为True
        user_dic['locked'] = True
        db_handler.save(user_dic)
        return True, '用户：{}，已冻结'.format(username)
    
    else:
        return False, '用户：{}，不存在'.format(username)