【Python】ATM机操作界面设计

# 定义全局变量money name
money = 5000000
# 这里的客户姓名需要输入，暂定为空
name = None

# 要求客户输入姓名
name= input("请输入您的姓名:")

# 定义查询函数
def query(show_header):
        if show_header:
            # 如果show_header为真就输出表头(此处是为了后面函数的调用)
                print("--------------查询余额----------------")
        print(f"{name},您好，您的余额剩余{money}元")

# 定义存款函数
def saving(num):
            global money
            money += num
            print("-----------------存款------------------")
            print(f"{name},您好,您存款{num}元成功。")
            # 调用查询余额函数，存款之后，自动显示余额
            query(False)
            # 给调用函数一个False参数，他不输出表头

# 定义取款函数
def get_money(num):
            global money
            money -= num
            #  此处是考虑到如果用户取款金额大于余额就显示余额不足
            if money < 0:
                    print(f"余额不足")
                    money +=num
            else:
                    print("-----------------取款----------------")
                    print(f"{name},您好,您取款{num}元成功。")
                    # 此处的查询余额只在取款成功时输出
                    query(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:
            keyboard_input = main()
            # 用户输入的为字符，需要转换成整数形式
            # 查询余额
            if keyboard_input == "1":
                    query(True)
                    continue
            # 存款
            elif keyboard_input == "2":
                    num = int(input("您想要存多少钱？请输入："))
                    saving(num)
                    continue
              # 取款
            elif keyboard_input == "3":
                    num = int(input("您要取多少钱？请输入?"))
                    get_money(num)
                    continue
               # 主界面
            else:
                    print("程序退出啦")
                    break