python-ATM

money=5000000
name=input("Enter your name: ")
def main():
    print('''------主菜单------
    查询余额\t1
    存款\t\t2
    取款\t\t3
    退出\t\t4
    ''')
    return input("请输入")
def find(a):
    if a:
        print("-------查询--------")
    print(f"{name}余额剩余{money}")

def cun(num):
    print("-------存款--------")
    global money
    money+=num
    print(f"{name}存款金额{num}")
    find(False)

def qu(num):
    print("-------取款--------")
    global money
    money-=num
    print(f"{name}取款金额{num}")
    find(False)

while True:
    k=main()
    if k=="1":
        find(True)
        continue
    elif k=="2":
        num=int(input("请输入存款金额"))
        cun(num)
        continue
    elif k=="3":
        num=int(input("请输入取款金额"))
        qu(num)
        continue
    else:
        break