forms: check model and count of custom faceshield windows

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))