"""File containing webapp forms.""" from django import forms from django.utils.translation import ugettext_lazy as _ from django.core.exceptions import ValidationError from webapp.models import OhlhafvChallenge, KaehmyForm class KaehmyForm_Form(forms.ModelForm): """Class representing Kaehmy form.""" class Meta: """Meta for class KaehmyForm.""" model = KaehmyForm fields = ['name', 'email', 'phone_number', 'year', 'preset_roles', 'custom_roles', 'custom_role_name', 'custom_role_is_board', 'text'] def clean_phone_number(self): """Clean phone number field.""" number = self.cleaned_data.get('phone_number') if number.isdigit(): return number else: raise ValidationError(_('Invalid value')) class OhlhafvForm(forms.ModelForm): """Class representing Ohlhafv form.""" class Meta: """Meta class for Ohlhafv form.""" model = OhlhafvChallenge fields = ['challenger', 'challenger_email', 'victim', 'victim_email', 'series', 'message']