From 8efceb1d01f56921bf7df46b913d36c8f473978c Mon Sep 17 00:00:00 2001 From: Konstantin Martini Date: Sat, 3 Jan 2015 04:42:14 +0100 Subject: [PATCH] added calculation on db values --- matecnt/views.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/matecnt/views.py b/matecnt/views.py index 993c275..84eed30 100644 --- a/matecnt/views.py +++ b/matecnt/views.py @@ -101,12 +101,16 @@ def checkout(request): count = int(request.REQUEST.get('count', 1)) drink_data = get_drink(drink) + total = drink_data.prize * count + drinker = get_user(user) + drinker.credit -= total + drinker.save() ctx = { 'heading': 'Codes scanned: user=%r drink=%r amount=%s' % (user, drink, count), - 'drinker': get_user(user), + 'drinker': drinker, 'drink': drink_data, - 'total': drink_data.prize * count, + 'total': total, } return render(request, 'checkout.html', ctx) @@ -122,9 +126,13 @@ def charge(request): except KeyError as ex: return HttpResponse(status=400) + drinker = get_user(user) + drinker.credit += charge + drinker.save() + ctx = { 'heading': 'Charging: user=%r amount=%s' % (user, charge), - 'drinker': get_user(user), + 'drinker': drinker, 'amount': charge, } return render(request, 'charge.html', ctx)