O něco blbuvzdornější import účastníků

mo/csv.py

@@ -7,6 +7,7 @@
 #   - quoting pomocí uvozovek
 
 import csv
+from difflib import get_close_matches
 from enum import auto
 from dataclasses import dataclass, fields
 from typing import Type, List, IO, Sequence
@@ -94,6 +95,19 @@ class MissingHeaderError(RuntimeError):
     pass
 
 
+class UnknownColumnError(RuntimeError):
+    def __init__(self, unknown_name: str, columns: set):
+        best_matches = get_close_matches(unknown_name, columns, n=1)
+        if not best_matches:
+            self.message = (
+                "Neznámý sloupec '{}'. Podporovaná množina sloupců je tato: {}".format(
+                    unknown_name, columns))
+        else:
+            self.message = (
+                "Neznámý sloupec '{}', měli jste na mysli '{}'?".format(
+                    unknown_name, best_matches[0]))
+
+
 def write(file: IO, fmt: FileFormat, row_class: Type[Row], rows: Sequence[Row]):
     writer = csv.writer(file, dialect=fmt.get_dialect())
 
@@ -120,6 +134,9 @@ def read(file: IO, fmt: FileFormat, row_class: Type[Row]):
             header = r
             if not any(h in columns for h in header):
                 raise MissingHeaderError()
+            for h in header:
+                if not h in columns:
+                    raise UnknownColumnError(h, columns)
         else:
             row = row_class()
             not_empty = False

mo/imports.py

@@ -7,7 +7,7 @@ from sqlalchemy.orm import joinedload, Query
 from typing import List, Optional, Any, Dict, Type, Union
 
 import mo.csv
-from mo.csv import FileFormat, MissingHeaderError
+from mo.csv import FileFormat, MissingHeaderError, UnknownColumnError
 import mo.db as db
 import mo.rights
 import mo.users
@@ -162,13 +162,15 @@ class Import:
 
         # Ve snaze zabránit Excelu v interpretování ročníku jako kalendářního data
         # lidé připisují všechny možné i nemožné znaky, které vypadají jako apostrof :)
-        rocnik = re.sub('[\'"\u00b4\u2019]', "", rocnik)
+        rocnik = re.sub('^[^\d]', "", rocnik)
 
-        if (school.is_ss and re.fullmatch(r'\d/\d', rocnik)
-                or school.is_zs and re.fullmatch(r'\d', rocnik)):
-            return rocnik
+        if (school.is_ss and not re.fullmatch(r'\d/\d', rocnik)):
+            return self.error(f'Ročník pro střední školu ({school.place.name}) zapisujte ve formátu číslice/číslice')
+
+        if (school.is_zs and not re.fullmatch(r'\d', rocnik)):
+            return self.error(f'Ročník pro základní školu ({school.place.name}) zapisujte jako číslici 1–9')
 
-        return self.error('Ročník neodpovídá typu školy: pro základní je to číslice, pro střední číslice/číslice')
+        return rocnik
 
     def parse_born(self, rok: str) -> Optional[int]:
         if not re.fullmatch(r'\d{4}', rok):
@@ -379,7 +381,9 @@ class Import:
                 try:
                     rows: List[mo.csv.Row] = mo.csv.read(file=file, fmt=self.fmt, row_class=self.row_class)
                 except MissingHeaderError:
-                    return self.error('Souboru chybí hlavička s názvy sloupců')
+                    return self.error('Souboru chybí první řádek s názvy sloupců')
+                except UnknownColumnError as e:
+                    return self.error(e.message)
         except UnicodeDecodeError:
             return self.error(f'Soubor není v kódování {self.fmt.get_charset()}')
         except Exception as e:

mo/web/templates/org_generic_import.html

@@ -12,11 +12,11 @@ Import dat do {% if contest %}soutěžní oblasti {{ contest.place.name }}{% els
 {% if errs %}
 <h3>Chyby při importu</h3>
 
-<pre><div class="alert alert-danger" role="alert">{{ "" -}}
+<div class="alert alert-danger" role="alert" style="white-space: pre">{{ "" -}}
 {% for e in errs %}
 {{ e }}
 {% endfor %}
-</div></pre>
+</div>
 {% endif %}
 
 <p>Zde je možné importovat účastníky soutěže, dozor na soutěžních místech a opravovatele.