实现atm提款简易代码

项目要求：程序启动后要求输入客户姓名
查询余额、存款、取款后都会返回主菜单
存款、取款后，都应显示一下当前余额
客户选择退出或输入错误，程序会退出，否则一直运行

代码实现

money = 500000
name = input("请输入你的姓名：")
def cha(show_header):if show_header:print("-----查询余额-----")print(f"{name},您好,您的余额剩余:{money}")
def cun(num):global moneymoney = money + numprint("-----存款-----")print(f"{name},您好，您存款{num}元成功")cha(False)
def qu(num):global moneymoney = money - numprint("-----取钱-----")print(f"{name},您好，您取出{num}元成功")cha(False)
def main():print("-----主菜单-----")print(f"{name},您好，欢迎来到黑马ATM，请选择操作：")print("查询余额\t[输入1]")print("存款\t\t[输入2]")print("取钱\t\t[输入3]")print("退出\t\t[输入4]")return input("您的选择是：")
while True:keyword_input = main()if keyword_input == "1":cha(True)continueelif keyword_input == "2":num = int(input("你想要存入多少钱，请输入："))cun(num)continueelif keyword_input == "3":num = int(input("你想要取出多少钱，请输入："))qu(num)continueelse:print("程序退出")break