ATM系统-python练习

ATM系统

def transfer(a):
    id = input("请输入对方账号：")
    if id == cardid[a]:
        print("【不能给自己转账】")
    elif id in cardid:
        b = input("转账金额：")
        if b > money[a]:
            print("【你的余额不足】")
        else:
            c = cardid.index(id)
            money[c] = int(money[c]) + int(b)
            money[a] = int(money[a]) - int(b)
            print("【转账成功】")
    else:
        print("【账号不存在】")
        
        
def inquire(a):
    print("【余额为{}】".format(money[a]))
    
cardid = ["1001","1002","1003"]
password = ["123456","654321","888888"]
money = ["100","200","300"]
import datetime
print(datetime.datetime.now().year)
print("====================================\n欢迎使用ATM系统\n当前时间: {}年{}月{}日\n====================================\n".format(datetime.datetime.now().year,datetime.datetime.now().month,datetime.datetime.now().day))
while True:
    print("请登录")
    a = input("卡号：")
    if a in cardid:
        site = cardid.index(a)
        b = input("密码：")
        if b == password[site]: 
            print("\n【登录成功】")
            while True:
                choice = int(input("请选择【1.查询余额 2.转帐 3.退出】:"))
                if choice == 1:
                    inquire(site)
                elif choice == 2:
                    transfer(site)
                else:
                    break
        else:
            print("【密码错误】")
    else:
        print("【账户不存在】")