From 21de28e71d25ef89647920c4a7412a0dfee79303 Mon Sep 17 00:00:00 2001 From: Martin Quarda Date: Tue, 6 Aug 2024 08:36:43 +0200 Subject: [PATCH] fix #4 --- alkatorapi/views.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) 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'],