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')