Python----存取款练习

my_money = 100
atm_money = 0


# 存款函数
def save_money():
    global my_money, atm_money
    num = int(input("请输入您要存款的金额:"))
    if num <= my_money:
        atm_money += num
        my_money -= num
        print(f"存款成功！\n您的银行卡余额为:{atm_money}元!")
        print(f"您身上的余额为:{my_money}元")
    else:
        print("您放入的金额不足!")


# 取款函数
def get_money():
    global my_money, atm_money
    num = int(input("请输入您要取款的金额:"))
    if num <= atm_money:
        atm_money -= num
        my_money += num
        print(f"取款成功！\n您的银行卡余额为:{atm_money}元")
        print(f"您身上的余额为:{my_money}元")
    else:
        print("您的卡上余额不足！")


def main():
    while True:
        print("请根据按键操作: 1.存钱 2.取钱 3.退出")
        num = int(input("请输入您的选择:"))
        if num == 1:
            save_money()
        elif num == 2:
            get_money()
        elif num == 3:
            print("请收好您的物品，欢迎下次光临")
            break
        else:
            print("按键错误，请重新输入！")


# 默认程序入口
if __name__ == '__main__':
    main()