diff --git a/mo/imports.py b/mo/imports.py index 51aa689b4e9a77adbbfc85124a688f0e0e552507..9415b42775784582b018c3447e5387280dad45a0 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 503276570d83c2962f82a7b67b8b28be1192d716..aa4c2f7ab5bed415be5cdccb9e4f8410a8a5c27c 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):