diff --git a/alkatorapi/views.py b/alkatorapi/views.py index 3f41226..1ccaf82 100644 --- a/alkatorapi/views.py +++ b/alkatorapi/views.py @@ -45,11 +45,14 @@ def register(request): invoice_date = datetime.today() invoice_id = invoice_date.year * 1000000 + invoice_date.month * 10000 + invoice_date.day * 100 - latest_user = User.objects.latest("invoice_id") - if latest_user.invoice_id < invoice_id: + try: + latest_user = User.objects.latest("invoice_id") + if latest_user.invoice_id < invoice_id: + invoice_id = invoice_id + 1 + else: + invoice_id = latest_user.invoice_id + 1 + except User.DoesNotExist: invoice_id = invoice_id + 1 - else: - invoice_id = latest_user.invoice_id + 1 user = User( first_name=request.POST['first_name'],