3.21周末作业

====================周末作业====================

编写ATM程序实现下述功能，数据来源于文件db.txt

# 0、注册功能：用户输入账号名、密码、金额，按照固定的格式存入文件db.txt
user_info = {'user': None}
def register():
        name=input('your name:').strip()
        pwd=input('you password:').strip()
        pwd1=input('once again:').strip()
        if pwd1 == pwd:
            print(f'{name}注册成功')
        else:
            print('注册失败')
# register()
# 1、登录功能：用户名不存在，要求必须先注册，用户名存在&输错三次锁定，登录成功后记录下登录状态（提示：可以使用全局变量来记录）
def login():
    name=input('your name:').strip()
    pwd=input('your password:').strip()
    with open('users.txt','r',encoding='utf-8') as f:
        for line in f:
            user,password=line.strip().split(':')
            if name == user and pwd == password:
                user_info['user']=name
                print('登录成功')
                break
        else:
            print('登录失败')
# login()
# 下述操作，要求登录后才能操作
# 1、充值功能：用户输入充值钱数，db.txt中该账号钱数完成修改
def pay_money():
    dic={}
    username=input('请输入充值用户：').strip()
    with open('db.txt','r',encoding='utf-8') as f:
        for line in f:
            user,money=line.strip().split(':')
            dic[user]=int(money)
    if not user_info.get('user'):
        login()
    if username not in dic:
        print('用户不存在，结束程序')
        return
    while True:
        money=input('充值多少：').strip()
        if not money.isdigit():
            print('请输入纯数字')
            continue
        money=int(money)
        dic[username]+= money
        with open('db.txt','w',encoding='utf-8') as f:
            for user,money in dic.items():
                f.write(f'{user}:{money}\n')
                print('充值成功')
            break

# pay_money()
# 2、转账功能：用户A向用户B转账1000元，db.txt中完成用户A账号减钱，用户B账号加钱
def transfer():
    a_user=input('请输入转账用户：').strip()
    b_user=input('请输入收款用户：').strip()
    transfer_moner=int(input('转多少：').strip())
    dic={}
    with open('db2.txt','r',encoding='utf-8') as f:
        for line in f:
            user,money=line.strip().split(':')
            dic[user]=int(money)
        if not user_info.get('user'):
            login()
        if a_user not in dic:
            print('转账用户不存在')
            return
        if b_user not in dic:
            print('收款用户不存在')
            return
        print('转账前:',dic)
        if dic.get(a_user) >= transfer_moner:
            dic[a_user] -= transfer_moner
            dic[b_user] += transfer_moner
            print('转账后:',dic)
        with open('db2.txt','w',encoding='utf-8') as f:
            for user,money in dic.items():
                f.write(f'{user}:{money}\n')
# 3、提现功能：用户输入提现金额，db.txt中该账号钱数减少
def withdraw():
    username=input('提现用户:').strip()
    get_money=int(input('提现多少：').strip())
    dic={}
    with open('db3.txt','r',encoding='utf-8') as f:
        for line in f:
            user,money=line.strip().split(':')
            dic[user]=int(money)
        if not user_info.get('user'):
            login()
        if username not in dic:
            print('用户不存在')
            return
        if dic.get(username) >= get_money:
            dic[username] -= get_money
        print(dic)
        with open('db3.txt','w',encoding='utf-8') as f:
            for user,money in dic.items():
                f.write(f'{user}:{money}\n')

# 4、查询余额功能：输入账号查询余额
def check_money():
    dic={}
    username = input('查询账户:').strip()
    with open('db.txt','r',encoding='utf-8') as f:
        for line in f:
            user,money=line.strip().split(':')
            dic[user]=int(money)
    if not user_info.get('user'):
        login()
    if username not in dic:
        print('用户不存在')
        return
    print(dic.get(username))


dic={
    '0':('退出',None),
    '1':('登录',login),
    '2':('注册',register),
    '3':('充值',pay_money),
    '4':('转账',transfer),
    '5':('提现',withdraw),
    '6':('查余额',check_money),
}
while True:
    for line in dic:
        print(line,dic[line][0])
    cmd=input('请输入指令:').strip()
    if not cmd.isdigit():
        print('请输入纯数字指令，傻子')
        continue
    if cmd == '0':
        break
    if cmd in dic:
        dic[cmd][1]()
    else:
        print('傻子，没有这个指令')