diff --git a/cashonly/web/views.py b/cashonly/web/views.py index 4e77d76..04620d0 100644 --- a/cashonly/web/views.py +++ b/cashonly/web/views.py @@ -3,6 +3,7 @@ from django import forms from django.shortcuts import render, get_object_or_404, redirect from django.contrib.auth.decorators import login_required from django.core import paginator +from django.contrib.auth.models import User from cashonly.core.models import * from cashonly.core.services import AccountManager from django.utils.translation import ugettext as _ @@ -17,7 +18,12 @@ import datetime @login_required def overview(request): - a = request.user.account + try: + a = request.user.account + except User.account.RelatedObjectDoesNotExist: + return render(request, 'cashonly/web/index.html', + {'latest_transactions': [], 'latest_purchases': []}) + time = datetime.datetime.now() - datetime.timedelta(hours=12) transactions = Transaction.objects.filter(account=a) \ .filter(timestamp__gte=time).order_by('-timestamp')