diff --git a/mo/web/org_place.py b/mo/web/org_place.py index a37e46bd16d191ea64427ed5d555912c4e7cb4f8..c6566888496e07a8677ca8ed263ef8f40d83dcd3 100644 --- a/mo/web/org_place.py +++ b/mo/web/org_place.py @@ -2,6 +2,7 @@ from flask import render_template, g, redirect, url_for, flash, request from flask_wtf import FlaskForm import locale from markupsafe import Markup +import re from sqlalchemy import func, and_ from sqlalchemy.orm import joinedload from typing import List, Optional @@ -110,8 +111,13 @@ class PlaceEditForm(FlaskForm): submit = wtforms.SubmitField('Uložit') def validate_code(form: FlaskForm, field: mo_fields.String) -> None: - if field.data is not None and field.data.startswith('#'): + code = field.data + if code is None: + pass + elif code.startswith('#'): raise wtforms.ValidationError('Kód nesmí začínat na znak "#".') + elif re.fullmatch(r'\d+', code): + raise wtforms.ValidationError('Kód nesmí být složen pouze z číslic.') class PlaceSchoolEditForm(PlaceEditForm):