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)
