"""Multiple classes example for testing AST parsing."""

class BankAccount:
    """Simple bank account class."""
    
    def __init__(self, account_number, initial_balance=0):
        self.account_number = account_number
        self.balance = initial_balance
        self.transaction_history = []
    
    def deposit(self, amount):
        self.balance += amount
        self.transaction_history.append(f"Deposit: +{amount}")
    
    def withdraw(self, amount):
        if amount <= self.balance:
            self.balance -= amount
            self.transaction_history.append(f"Withdrawal: -{amount}")
            return True
        return False
    
    def get_balance(self):
        return self.balance


class Customer:
    """Customer class to hold account information."""
    
    def __init__(self, customer_id, name, email):
        self.customer_id = customer_id
        self.name = name
        self.email = email
        self.accounts = []
    
    def add_account(self, account):
        self.accounts.append(account)
    
    def get_total_balance(self):
        return sum(account.get_balance() for account in self.accounts)


class Bank:
    """Bank class to manage customers and accounts."""
    
    def __init__(self, name):
        self.name = name
        self.customers = {}
        self.next_account_number = 1000
    
    def create_customer(self, name, email):
        customer_id = len(self.customers) + 1
        customer = Customer(customer_id, name, email)
        self.customers[customer_id] = customer
        return customer
    
    def create_account(self, customer_id, initial_balance=0):
        if customer_id in self.customers:
            account = BankAccount(self.next_account_number, initial_balance)
            self.customers[customer_id].add_account(account)
            self.next_account_number += 1
            return account
        return None