58be33a8d7
* Added pre commit hook * Run pre commit hook on all files --------- Co-authored-by: Horilla <131998600+horilla-opensource@users.noreply.github.com>
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
"""
|
|
forms.py
|
|
|
|
Horilla forms
|
|
"""
|
|
|
|
from typing import Any, Dict
|
|
|
|
from django import forms
|
|
|
|
from horilla_widgets.widgets.horilla_multi_select_field import HorillaMultiSelectField
|
|
|
|
|
|
class HorillaForm(forms.Form):
|
|
def clean(self) -> Dict[str, Any]:
|
|
for field_name, field_instance in self.fields.items():
|
|
if isinstance(field_instance, HorillaMultiSelectField):
|
|
self.errors.pop(field_name, None)
|
|
if len(self.data.getlist(field_name)) < 1:
|
|
raise forms.ValidationError({field_name: "This field is required"})
|
|
cleaned_data = super().clean()
|
|
employee_data = self.fields[field_name].queryset.filter(
|
|
id__in=self.data.getlist(field_name)
|
|
)
|
|
cleaned_data[field_name] = employee_data
|
|
cleaned_data = super().clean()
|
|
return cleaned_data
|
|
|
|
|
|
class HorillaModelForm(forms.ModelForm):
|
|
def clean(self) -> Dict[str, Any]:
|
|
for field_name, field_instance in self.fields.items():
|
|
if isinstance(field_instance, HorillaMultiSelectField):
|
|
self.errors.pop(field_name, None)
|
|
if len(self.data.getlist(field_name)) < 1:
|
|
raise forms.ValidationError({field_name: "Thif field is required"})
|
|
cleaned_data = super().clean()
|
|
employee_data = self.fields[field_name].queryset.filter(
|
|
id__in=self.data.getlist(field_name)
|
|
)
|
|
cleaned_data[field_name] = employee_data
|
|
cleaned_data = super().clean()
|
|
return cleaned_data
|