Zkrášlení seminar.utils.deadline

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.