def deposit_amount(amount): if amount > MAX_DEPOSIT_PER_TRANS: print( "Sorry! You've attempted to make a deposit exceeding KES 40,000. Kindly retry." ) elif Transaction.number_of_deposits() > MAX_DEPOSIT_FREQ: print( "Sorry! You've exceeded number of deposits per day. kindly try again tommorow" ) else: bal = Transaction.query_balance() + amount # calculate new balance new_trans = Transaction(bal, amount, 'deposit') new_trans.save_transaction() # Save changes if Transaction.amount_deposited_today() > MAX_DEPOSIT: print( "Dear customer, limit deposit per day is KES 150,000. We apologise for the inconvinience" ) new_trans.delete_transaction() else: print("-" * 70) print("Deposit successful! New balance is: KES {}".format( Transaction.query_balance()))
def test_can_calculate_cumulative_deposit(self): self.one = Transaction(100, 10, 'deposit') self.two = Transaction(10, 5, 'deposit') self.one.save_transaction() self.two.save_transaction() self.assertEqual(Transaction.amount_deposited_today(), 15)