diff --git a/formity/forms.py b/formity/forms.py index dceff5c..6d63629 100644 --- a/formity/forms.py +++ b/formity/forms.py @@ -1,5 +1,5 @@ from flask_wtf import FlaskForm -from wtforms import StringField, BooleanField, TextAreaField, IntegerField, validators +from wtforms import StringField, BooleanField, TextAreaField, IntegerField, validators, ValidationError from wtforms.widgets import html5 @@ -16,3 +16,15 @@ class FaceshieldRequestForm(FlaskForm): faceshield_full_required = IntegerField('Pilnie potrzebna minimalna ilość pełnych przyłbic', default=0, validators=[validators.InputRequired()], widget=html5.NumberInput()) rules_approval = BooleanField('Jestem z placówki medycznej i zobowiązuję się do nieodsprzedawania otrzymanego sprzętu ochronnego. Jestem świadom, że nie jest to certyfikowany produkt medyczny i nie jest dezynfekowany/sterylny.', validators=[validators.DataRequired()]) + + def validate_faceshield_model(self, field): + model = field.data.strip() + count = int(self.faceshield_front_required.data) + if not model and count > 0: + raise ValidationError('Proszę podać model i markę przyłbicy do której mają zostać wyprodukowane szybki') + + def validate_faceshield_front_required(self, field): + model = self.faceshield_model.data.strip() + count = int(field.data) + if model and count == 0: + raise ValidationError('Proszę podać ilość potrzebnych szybek do przyłbicy {}'.format(model))