40 lines
2.0 KiB
Python
Executable File
40 lines
2.0 KiB
Python
Executable File
import django_filters
|
|
from django_filters import CharFilter
|
|
from django import forms
|
|
import uuid
|
|
|
|
def filter_by_name(queryset, name, value):
|
|
"""
|
|
Filter queryset by first name or last name.
|
|
"""
|
|
# Split the search value into first name and last name
|
|
parts = value.split()
|
|
first_name = parts[0]
|
|
last_name = ' '.join(parts[1:]) if len(parts) > 1 else ''
|
|
|
|
# Filter the queryset by first name and last name
|
|
if first_name and last_name:
|
|
queryset = queryset.filter(employee_id__employee_first_name__icontains=first_name, employee_id__employee_last_name__icontains=last_name)
|
|
elif first_name:
|
|
queryset = queryset.filter(employee_id__employee_first_name__icontains=first_name)
|
|
elif last_name:
|
|
queryset = queryset.filter(employee_id__employee_last_name__icontains=last_name)
|
|
|
|
return queryset
|
|
|
|
class FilterSet(django_filters.FilterSet):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
for field_name, field in self.filters.items():
|
|
filter_widget = self.filters[field_name]
|
|
widget = filter_widget.field.widget
|
|
if isinstance(widget, (forms.NumberInput, forms.EmailInput,forms.TextInput)):
|
|
filter_widget.field.widget.attrs.update({'class': 'oh-input w-100'})
|
|
elif isinstance(widget,(forms.Select,)):
|
|
filter_widget.field.widget.attrs.update({'class': 'oh-select oh-select-2 select2-hidden-accessible','id':uuid.uuid4()})
|
|
elif isinstance(widget,(forms.Textarea)):
|
|
filter_widget.field.widget.attrs.update({'class': 'oh-input w-100'})
|
|
elif isinstance(widget, (forms.CheckboxInput,forms.CheckboxSelectMultiple,)):
|
|
filter_widget.field.widget.attrs.update({'class': 'oh-switch__checkbox'})
|
|
elif isinstance(widget,(forms.ModelChoiceField)):
|
|
filter_widget.field.widget.attrs.update({'class': 'oh-select oh-select-2 select2-hidden-accessible',}) |