def trans_add(request):
if request.method == 'POST':
tran = Trans()
tran.type = request.POST['type']
tran.amount = request.POST['amount']
tran.date = datetime.datetime.now()
tran.bank = Bank.objects.get(pk=request.user.bank_id)
tran.officer = User.objects.get(pk=request.user.id)
bank = Bank.objects.get(pk=request.user.bank_id)
if tran.type == 'Debit':
bank.balance = int(bank.balance) + int(tran.amount)
elif tran.type == 'Credit':
bank.balance = int(bank.balance) - int(tran.amount)
bank.save()
tran.save()
return redirect('/transactions/')
elif request.method == 'GET':
return render(request, 'trans-add.html')
