From a24c8602c2507eaee2b060db6d4362bbc48b37c5 Mon Sep 17 00:00:00 2001
From: Jiri Kalvoda <jirikalvoda@kam.mff.cuni.cz>
Date: Fri, 23 Jul 2021 23:26:16 +0200
Subject: [PATCH] =?UTF-8?q?P=C5=99esunut=C3=AD=20obecn=C3=BDch=20validac?=
 =?UTF-8?q?=C3=AD=20z=20mo.imports=20do=20mo.users?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mo/imports.py | 29 +++++++++--------------------
 mo/users.py   | 27 +++++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 20 deletions(-)

diff --git a/mo/imports.py b/mo/imports.py
index 51aa689b..9415b427 100644
--- a/mo/imports.py
+++ b/mo/imports.py
@@ -149,21 +149,16 @@ class Import:
         return place
 
     def parse_school(self, kod: str) -> Optional[db.Place]:
-        if kod == "":
-            return self.error('Škola je povinná')
-
         if kod in self.school_place_cache:
             return self.school_place_cache[kod]
 
-        place = db.get_place_by_code(kod, fetch_school=True)
-        if not place:
-            return self.error(f'Škola s kódem "{kod}" nenalezena'+
-                    ('. Nechybí vám # na začátku?' if re.fullmatch(r'\d+', kod) else ''))
-
-        if place.type != db.PlaceType.school:
-            return self.error(f'Kód školy "{kod}" neodpovídá škole')
+        try:
+            place = mo.users.validate_and_find_school(kod)
+        except mo.CheckError as e:
+            return self.error(str(e))
 
         self.school_place_cache[kod] = place
+
         return place
 
     def parse_grade(self, rocnik: str, school: Optional[db.School]) -> Optional[str]:
@@ -174,16 +169,10 @@ class Import:
         # lidé připisují všechny možné i nemožné znaky, které vypadají jako apostrof :)
         rocnik = re.sub('[\'"\u00b4\u2019]', "", rocnik)
 
-        if (not re.fullmatch(r'\d(/\d)?', rocnik)):
-            return self.error(f'Ročník má neplatný formát, musí to být buď číslice, nebo číslice/číslice')
-
-        if (not school.is_zs and re.fullmatch(r'\d', rocnik)):
-            return self.error(f'Ročník pro střední školu ({school.place.name}) zapisujte ve formátu číslice/číslice')
-
-        if (not school.is_ss and re.fullmatch(r'\d/\d', rocnik)):
-            return self.error(f'Ročník pro základní školu ({school.place.name}) zapisujte jako číslici 1–9')
-
-        return rocnik
+        try:
+            return mo.users.normalize_grade(rocnik, school)
+        except mo.CheckError as e:
+            return self.error(str(e))
 
     def parse_born(self, rok: str) -> Optional[int]:
         if not re.fullmatch(r'\d{4}', rok):
diff --git a/mo/users.py b/mo/users.py
index 50327657..aa4c2f7a 100644
--- a/mo/users.py
+++ b/mo/users.py
@@ -12,6 +12,33 @@ import mo.db as db
 import mo.util
 import mo.tokens
 
+def normalize_grade(rocnik:str, school:db.School) -> str:
+    """ Aktuálně provádí jen kontrolu formátu. """
+
+    if not re.fullmatch(r'\d(/\d)?', rocnik):
+        raise mo.CheckError(f'Ročník má neplatný formát, musí to být buď číslice, nebo číslice/číslice')
+
+    if not school.is_zs and re.fullmatch(r'\d', rocnik):
+        raise mo.CheckError(f'Ročník pro střední školu ({school.place.name}) zapisujte ve formátu číslice/číslice')
+
+    if not school.is_ss and re.fullmatch(r'\d/\d', rocnik):
+        raise mo.CheckError(f'Ročník pro základní školu ({school.place.name}) zapisujte jako číslici 1–9')
+
+    return rocnik
+
+def validate_and_find_school(kod:str) -> db.Place:
+        if kod == "":
+            raise mo.CheckError('Škola je povinná')
+
+        place = db.get_place_by_code(kod, fetch_school=True)
+        if not place:
+            raise mo.CheckError(f'Škola s kódem "{kod}" nenalezena'+
+                    ('. Nechybí vám # na začátku?' if re.fullmatch(r'\d+', kod) else ''))
+
+        if place.type != db.PlaceType.school:
+            raise mo.CheckError(f'Kód školy "{kod}" neodpovídá škole')
+
+        return place
 
 def normalize_email(addr: str) -> str:
     if not re.fullmatch(r'.+@.+', addr):
-- 
GitLab