py设计ATM取款机代码

在Python中设计一个简单的ATM模拟器，我们可以创建一个`BankAccount`类作为基础账户，然后创建一个`ATM`类来操作这些账户。这里是一个简化的例子：

class BankAccount:
    def __init__(self, account_number, balance=0):
        self.account_number = account_number
        self.balance = balance

    def deposit(self, amount):
        if amount > 0:
            self.balance += amount
            print(f"存入 {amount} 元，当前余额为 {self.balance} 元")

    def withdraw(self, amount):
        if amount <= self.balance:
            self.balance -= amount
            print(f"取出 {amount} 元，当前余额为 {self.balance} 元")
        else:
            print("余额不足，请存入更多钱后再取款.")

class ATM:
    def __init__(self, bank_account):
        self.account = bank_account

    def check_balance(self):
        print(f"您的账户余额为 {self.account.balance} 元")

    def withdraw_money(self, amount):
        self.account.withdraw(amount)

# 使用示例
account = BankAccount(123456, 1000)
atm = ATM(account)
atm.check_balance()  # 检查初始余额
atm.deposit_money(500)  # 存款
atm.withdraw_money(300)  # 取款

在这个例子中，`ATM`对象有一个指向`BankAccount`实例的引用，可以进行存款、取款和查看余额的操作。