打一个 ATM系统的代码
时间: 2023-03-22 18:02:14 浏览: 69
以下是一个简单的ATM系统的代码示例,用Python编写:
```python
class ATM:
def __init__(self, balance):
self.balance = balance
def withdraw(self, amount):
if amount <= 0:
print("请输入一个正整数")
elif amount > self.balance:
print("余额不足")
else:
self.balance -= amount
print("取款成功,剩余余额为:", self.balance)
def deposit(self, amount):
if amount <= 0:
print("请输入一个正整数")
else:
self.balance += amount
print("存款成功,当前余额为:", self.balance)
if __name__ == "__main__":
atm = ATM(1000) # 初始余额为1000元
while True:
print("请选择操作:1.查询余额 2.取款 3.存款 4.退出")
choice = input()
if choice == "1":
print("当前余额为:", atm.balance)
elif choice == "2":
amount = int(input("请输入取款金额:"))
atm.withdraw(amount)
elif choice == "3":
amount = int(input("请输入存款金额:"))
atm.deposit(amount)
elif choice == "4":
print("谢谢使用,再见!")
break
else:
print("无效操作,请重新选择")
```
该代码定义了一个名为ATM的类,具有余额查询、取款、存款和退出操作。在main函数中,实例化了一个ATM对象,并循环等待用户输入操作。当用户选择取款或存款操作时,会要求用户输入金额,并进行相应的处理。最后,用户可以选择退出操作结束程序。