diff --git a/seminar/utils.py b/seminar/utils.py index 935bc687..9390f1a3 100644 --- a/seminar/utils.py +++ b/seminar/utils.py @@ -323,15 +323,15 @@ def deadline(datum): datum = datum.date() rok = datum.year # Dva ročníky podezřelé z obsahování dat - pozdejsi_rocnik = m.Rocnik.objects.filter(prvni_rok=rok) - drivejsi_rocnik = m.Rocnik.objects.filter(prvni_rok=rok-1) - if any([ - pozdejsi_rocnik.count() > 1, - drivejsi_rocnik.count() > 1, - ]): - raise ValueError(f"Více ročníků začíná/končí stejným rokem: {rok}") - pozdejsi_rocnik = pozdejsi_rocnik.first() if pozdejsi_rocnik.count() > 0 else None - drivejsi_rocnik = drivejsi_rocnik.first() if drivejsi_rocnik.count() > 0 else None + try: + pozdejsi_rocnik = m.Rocnik.objects.get(prvni_rok=rok) + except m.Rocnik.DoesNotExist: + pozdejsi_rocnik = None + + try: + drivejsi_rocnik = m.Rocnik.objects.get(prvni_rok=rok-1) + except m.Rocnik.DoesNotExist: + drivejsi_rocnik = None if drivejsi_rocnik is not None: # Předpokládáme, že neexistuje číslo, které má deadline ale nemá finální deadline.