From 76272aa7c28848b6d5f879e135f1bc6dbf63cfab Mon Sep 17 00:00:00 2001
From: "Pavel \"LEdoian\" Turinsky" <ledoian@matfyz.cz>
Date: Wed, 28 Apr 2021 19:41:03 +0200
Subject: [PATCH] =?UTF-8?q?Zkr=C3=A1=C5=A1len=C3=AD=20seminar.utils.deadli?=
 =?UTF-8?q?ne?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 seminar/utils.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

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.