class RemoveStationForm(forms.Form):
validators = [MinLengthValidator(5)]
center_number = forms.IntegerField(
error_messages={
'invalid': _(u"Expecting only numbers for center number")
},
validators=validators,
widget=forms.NumberInput(attrs=disable_copy_input),
label=_(u"Center Number"))
station_number = forms.IntegerField(
min_value=min_station_value,
max_value=max_station_value,
widget=forms.TextInput(attrs=disable_copy_input),
label=_(u"Station Number"))
tally_id = forms.IntegerField(widget=forms.HiddenInput())
def __init__(self, *args, **kwargs):
super(RemoveStationForm, self).__init__(*args, **kwargs)
self.fields['center_number'].widget.attrs['autofocus'] = 'on'
def clean(self):
if self.is_valid():
cleaned_data = super(RemoveStationForm, self).clean()
center_number = cleaned_data.get('center_number')
station_number = cleaned_data.get('station_number')
tally_id = cleaned_data.get('tally_id')
try:
center = Center.objects.get(code=center_number,
tally__id=tally_id)
stations = center.stations.all()
valid_station_numbers = [s.station_number for s in stations]
if not int(station_number) in valid_station_numbers:
raise forms.ValidationError(
_('Invalid Station Number for this Center'))
except Center.DoesNotExist:
raise forms.ValidationError(_('Center Number does not exist'))
else:
check_results_for_forms(
center.resultform_set.filter(
station_number=station_number))
return cleaned_data
def save(self):
if self.is_valid():
center_number = self.cleaned_data.get('center_number')
station_number = self.cleaned_data.get('station_number')
tally_id = self.cleaned_data.get('tally_id')
station = Station.objects.filter(center__code=center_number,
station_number=station_number,
center__tally__id=tally_id)
return station.first()
class RemoveCenterForm(forms.Form):
validators = [MinLengthValidator(5)]
center_number = forms.IntegerField(
error_messages={
'invalid': _(u"Expecting only numbers for center number")
},
validators=validators,
widget=forms.NumberInput(attrs=disable_copy_input),
label=_(u"Center Number"))
tally_id = forms.IntegerField(widget=forms.HiddenInput())
def __init__(self, *args, **kwargs):
super(RemoveCenterForm, self).__init__(*args, **kwargs)
self.fields['center_number'].widget.attrs['autofocus'] = 'on'
def clean(self):
if self.is_valid():
cleaned_data = super(RemoveCenterForm, self).clean()
center_number = cleaned_data.get('center_number')
tally_id = cleaned_data.get('tally_id')
try:
center = Center.objects.get(code=center_number,
tally__id=tally_id)
except Center.DoesNotExist:
raise forms.ValidationError(_('Center Number does not exist'))
else:
check_results_for_forms(center.resultform_set.all())
return cleaned_data
def save(self):
if self.is_valid():
center_number = self.cleaned_data.get('center_number')
tally_id = self.cleaned_data.get('tally_id')
try:
center = Center.objects.get(code=center_number,
tally__id=tally_id)
except Center.DoesNotExist:
raise forms.ValidationError(_('Center Number does not exist'))
else:
return center
class CenterDetailsForm(forms.Form):
validators = [MinLengthValidator(5)]
center_error_messages = {
'invalid': _(u"Expecting only numbers for center number")
}
center_number = forms.IntegerField(
error_messages=center_error_messages,
validators=validators,
widget=forms.NumberInput(attrs=disable_copy_input),
label=_(u"Center Number"))
center_number_copy = forms.IntegerField(
error_messages=center_error_messages,
validators=validators,
widget=forms.NumberInput(attrs=disable_copy_input),
label=_(u"Center Number Copy"))
station_number = forms.IntegerField(
min_value=min_station_value,
max_value=max_station_value,
widget=forms.TextInput(attrs=disable_copy_input),
label=_(u"Station Number"))
station_number_copy = forms.IntegerField(
min_value=min_station_value,
max_value=max_station_value,
widget=forms.TextInput(attrs=disable_copy_input),
label=_(u"Station Number Copy"))
tally_id = forms.IntegerField(widget=forms.HiddenInput)
def __init__(self, *args, **kwargs):
super(CenterDetailsForm, self).__init__(*args, **kwargs)
self.fields['center_number'].widget.attrs['autofocus'] = 'on'
def clean(self):
"""Ensure that the center number and center number copy match, as well
as the station number and station number copy. Also validate that the
center number is a valid center number and that the station number is
for a station in that center.
"""
if self.is_valid():
cleaned_data = super(CenterDetailsForm, self).clean()
center_number = cleaned_data.get('center_number')
center_number_copy = cleaned_data.get('center_number_copy')
station_number = cleaned_data.get('station_number')
station_number_copy = cleaned_data.get('station_number_copy')
tally_id = cleaned_data.get('tally_id')
if center_number != center_number_copy:
raise forms.ValidationError(_('Center Numbers do not match'))
if station_number != station_number_copy:
raise forms.ValidationError(_('Station Numbers do not match'))
try:
center = Center.objects.get(code=center_number,
tally__id=tally_id)
valid_station_numbers = [
list(d.values())[0]
for d in center.stations.values('station_number')
]
if not int(station_number) in valid_station_numbers:
raise forms.ValidationError(
_('Invalid Station Number for this Center'))
if not center.active:
raise forms.ValidationError(_('Center is disabled'))
station = Station.objects.get(center=center,
station_number=station_number)
if not station.active:
raise forms.ValidationError(_('Station is disabled'))
elif station.sub_constituency:
ballot = station.sub_constituency.get_ballot()
if ballot and not ballot.active:
raise forms.ValidationError(_('Race is disabled.'))
except Center.DoesNotExist:
raise forms.ValidationError(_('Center Number does not exist'))
except Station.DoesNotExist:
raise forms.ValidationError(_('Station Number does not exist'))
return cleaned_data