405 lines
13 KiB
Python
405 lines
13 KiB
Python
|
"""
|
||
|
|
This module contains custom Django filters for filtering querysets related to Shift Requests,
|
||
|
|
Work Type Requests, Rotating Shift and Rotating Work Type Assign.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
import django_filters
|
||
|
|
from django import forms
|
||
|
|
from django.utils.translation import gettext as __
|
||
|
|
from django_filters import CharFilter, DateFilter, filters
|
||
|
|
|
||
|
|
from base.models import (
|
||
|
|
CompanyLeaves,
|
||
|
|
Holidays,
|
||
|
|
PenaltyAccounts,
|
||
|
|
RotatingShiftAssign,
|
||
|
|
RotatingWorkTypeAssign,
|
||
|
|
ShiftRequest,
|
||
|
|
WorkTypeRequest,
|
||
|
|
)
|
||
|
|
from horilla.filters import FilterSet, filter_by_name
|
||
|
|
|
||
|
|
|
||
|
|
class ShiftRequestFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
Custom filter for Shift Requests.
|
||
|
|
"""
|
||
|
|
|
||
|
|
requested_date = django_filters.DateFilter(
|
||
|
|
field_name="requested_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
requested_date__gte = django_filters.DateFilter(
|
||
|
|
field_name="requested_date",
|
||
|
|
lookup_expr="gte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
requested_date__lte = django_filters.DateFilter(
|
||
|
|
field_name="requested_date",
|
||
|
|
lookup_expr="lte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
search = CharFilter(method=filter_by_name)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
"""
|
||
|
|
A nested class that specifies the model and fields for the filter.
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = "__all__"
|
||
|
|
model = ShiftRequest
|
||
|
|
fields = [
|
||
|
|
"id",
|
||
|
|
"employee_id",
|
||
|
|
"requested_date",
|
||
|
|
"previous_shift_id",
|
||
|
|
"shift_id",
|
||
|
|
"requested_till",
|
||
|
|
"approved",
|
||
|
|
"canceled",
|
||
|
|
"employee_id__employee_first_name",
|
||
|
|
"employee_id__employee_last_name",
|
||
|
|
"employee_id__is_active",
|
||
|
|
"employee_id__gender",
|
||
|
|
"employee_id__employee_work_info__job_position_id",
|
||
|
|
"employee_id__employee_work_info__department_id",
|
||
|
|
"employee_id__employee_work_info__work_type_id",
|
||
|
|
"employee_id__employee_work_info__employee_type_id",
|
||
|
|
"employee_id__employee_work_info__job_role_id",
|
||
|
|
"employee_id__employee_work_info__reporting_manager_id",
|
||
|
|
"employee_id__employee_work_info__company_id",
|
||
|
|
"employee_id__employee_work_info__shift_id",
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, data=None, queryset=None, *, request=None, prefix=None):
|
||
|
|
super().__init__(data=data, queryset=queryset, request=request, prefix=prefix)
|
||
|
|
for field in self.form.fields.keys():
|
||
|
|
self.form.fields[field].widget.attrs["id"] = f"{uuid.uuid4()}"
|
||
|
|
|
||
|
|
|
||
|
|
class WorkTypeRequestFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
Custom filter for Work Type Requests.
|
||
|
|
"""
|
||
|
|
|
||
|
|
requested_date = django_filters.DateFilter(
|
||
|
|
field_name="requested_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
requested_date__gte = django_filters.DateFilter(
|
||
|
|
field_name="requested_till",
|
||
|
|
lookup_expr="gte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
requested_date__lte = django_filters.DateFilter(
|
||
|
|
field_name="requested_till",
|
||
|
|
lookup_expr="lte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
search = CharFilter(method=filter_by_name)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
"""
|
||
|
|
A nested class that specifies the model and fields for the filter.
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = "__all__"
|
||
|
|
model = WorkTypeRequest
|
||
|
|
fields = [
|
||
|
|
"id",
|
||
|
|
"employee_id",
|
||
|
|
"requested_date",
|
||
|
|
"previous_work_type_id",
|
||
|
|
"approved",
|
||
|
|
"work_type_id",
|
||
|
|
"canceled",
|
||
|
|
"employee_id__employee_first_name",
|
||
|
|
"employee_id__employee_last_name",
|
||
|
|
"employee_id__is_active",
|
||
|
|
"employee_id__gender",
|
||
|
|
"employee_id__employee_work_info__job_position_id",
|
||
|
|
"employee_id__employee_work_info__department_id",
|
||
|
|
"employee_id__employee_work_info__work_type_id",
|
||
|
|
"employee_id__employee_work_info__employee_type_id",
|
||
|
|
"employee_id__employee_work_info__job_role_id",
|
||
|
|
"employee_id__employee_work_info__reporting_manager_id",
|
||
|
|
"employee_id__employee_work_info__company_id",
|
||
|
|
"employee_id__employee_work_info__shift_id",
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, data=None, queryset=None, *, request=None, prefix=None):
|
||
|
|
super().__init__(data=data, queryset=queryset, request=request, prefix=prefix)
|
||
|
|
for field in self.form.fields.keys():
|
||
|
|
self.form.fields[field].widget.attrs["id"] = f"{uuid.uuid4()}"
|
||
|
|
|
||
|
|
|
||
|
|
class RotatingShiftAssignFilters(FilterSet):
|
||
|
|
"""
|
||
|
|
Custom filter for Rotating Shift Assign.
|
||
|
|
"""
|
||
|
|
|
||
|
|
search = CharFilter(method=filter_by_name)
|
||
|
|
|
||
|
|
next_change_date = django_filters.DateFilter(
|
||
|
|
field_name="next_change_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
start_date = django_filters.DateFilter(
|
||
|
|
field_name="start_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
"""
|
||
|
|
A nested class that specifies the model and fields for the filter.
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = "__all__"
|
||
|
|
model = RotatingShiftAssign
|
||
|
|
fields = [
|
||
|
|
"employee_id",
|
||
|
|
"rotating_shift_id",
|
||
|
|
"next_change_date",
|
||
|
|
"start_date",
|
||
|
|
"based_on",
|
||
|
|
"rotate_after_day",
|
||
|
|
"rotate_every_weekend",
|
||
|
|
"rotate_every",
|
||
|
|
"current_shift",
|
||
|
|
"next_shift",
|
||
|
|
"is_active",
|
||
|
|
"employee_id__employee_work_info__job_position_id",
|
||
|
|
"employee_id__employee_work_info__department_id",
|
||
|
|
"employee_id__employee_work_info__work_type_id",
|
||
|
|
"employee_id__employee_work_info__employee_type_id",
|
||
|
|
"employee_id__employee_work_info__job_role_id",
|
||
|
|
"employee_id__employee_work_info__reporting_manager_id",
|
||
|
|
"employee_id__employee_work_info__company_id",
|
||
|
|
"employee_id__employee_work_info__shift_id",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class RotatingWorkTypeAssignFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
Custom filter for Rotating Work Type Assign.
|
||
|
|
"""
|
||
|
|
|
||
|
|
search = CharFilter(method=filter_by_name)
|
||
|
|
|
||
|
|
next_change_date = django_filters.DateFilter(
|
||
|
|
field_name="next_change_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
start_date = django_filters.DateFilter(
|
||
|
|
field_name="start_date", widget=forms.DateInput(attrs={"type": "date"})
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
"""
|
||
|
|
A nested class that specifies the model and fields for the filter.
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = "__all__"
|
||
|
|
model = RotatingWorkTypeAssign
|
||
|
|
fields = [
|
||
|
|
"employee_id",
|
||
|
|
"rotating_work_type_id",
|
||
|
|
"next_change_date",
|
||
|
|
"start_date",
|
||
|
|
"based_on",
|
||
|
|
"rotate_after_day",
|
||
|
|
"rotate_every_weekend",
|
||
|
|
"rotate_every",
|
||
|
|
"current_work_type",
|
||
|
|
"next_work_type",
|
||
|
|
"is_active",
|
||
|
|
"employee_id__employee_work_info__job_position_id",
|
||
|
|
"employee_id__employee_work_info__department_id",
|
||
|
|
"employee_id__employee_work_info__work_type_id",
|
||
|
|
"employee_id__employee_work_info__employee_type_id",
|
||
|
|
"employee_id__employee_work_info__job_role_id",
|
||
|
|
"employee_id__employee_work_info__reporting_manager_id",
|
||
|
|
"employee_id__employee_work_info__company_id",
|
||
|
|
"employee_id__employee_work_info__shift_id",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class ShiftRequestReGroup:
|
||
|
|
"""
|
||
|
|
Class to keep the field name for group by option
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = [
|
||
|
|
("", "Select"),
|
||
|
|
("employee_id", "Employee"),
|
||
|
|
("shift_id", "Requested Shift"),
|
||
|
|
("previous_shift_id", "Current Shift"),
|
||
|
|
("requested_date", "Requested Date"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class WorkTypeRequestReGroup:
|
||
|
|
"""
|
||
|
|
Class to keep the field name for group by option
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = [
|
||
|
|
("", "Select"),
|
||
|
|
("employee_id", "Employee"),
|
||
|
|
("work_type_id", "Requested Work Type"),
|
||
|
|
("previous_work_type_id", "Current Work Type"),
|
||
|
|
("requested_date", "Requested Date"),
|
||
|
|
("employee_id__employee_work_info__department_id", "Department"),
|
||
|
|
("employee_id__employee_work_info__job_position_id", "Job Position"),
|
||
|
|
("employee_id__employee_work_info__reporting_manager_id", "Reporting Manager"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class RotatingWorkTypeRequestReGroup:
|
||
|
|
"""
|
||
|
|
Class to keep the field name for group by option
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = [
|
||
|
|
("", "Select"),
|
||
|
|
("employee_id", "Employee"),
|
||
|
|
("rotating_work_type_id", "Rotating Work Type"),
|
||
|
|
("current_work_type", "Current Work Type"),
|
||
|
|
("based_on", "Based On"),
|
||
|
|
("employee_id__employee_work_info__department_id", "Department"),
|
||
|
|
("employee_id__employee_work_info__job_role_id", "Job Role"),
|
||
|
|
("employee_id__employee_work_info__reporting_manager_id", "Reporting Manager"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class RotatingShiftRequestReGroup:
|
||
|
|
"""
|
||
|
|
Class to keep the field name for group by option
|
||
|
|
"""
|
||
|
|
|
||
|
|
fields = [
|
||
|
|
("", "Select"),
|
||
|
|
("employee_id", "Employee"),
|
||
|
|
("rotating_shift_id", "Rotating Shift"),
|
||
|
|
("based_on", "Based On"),
|
||
|
|
("employee_id__employee_work_info__department_id", "Department"),
|
||
|
|
("employee_id__employee_work_info__job_role_id", "Job Role"),
|
||
|
|
("employee_id__employee_work_info__reporting_manager_id", "Reporting Manager"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class HolidayFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
Filter class for Holidays model.
|
||
|
|
|
||
|
|
This filter allows searching Holidays objects based on name and date range.
|
||
|
|
"""
|
||
|
|
|
||
|
|
search = filters.CharFilter(field_name="name", lookup_expr="icontains")
|
||
|
|
from_date = DateFilter(
|
||
|
|
field_name="start_date",
|
||
|
|
lookup_expr="gte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
to_date = DateFilter(
|
||
|
|
field_name="end_date",
|
||
|
|
lookup_expr="lte",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
|
||
|
|
start_date = DateFilter(
|
||
|
|
field_name="start_date",
|
||
|
|
lookup_expr="exact",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
|
||
|
|
end_date = DateFilter(
|
||
|
|
field_name="end_date",
|
||
|
|
lookup_expr="exact",
|
||
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
"""
|
||
|
|
Meta class defines the model and fields to filter
|
||
|
|
"""
|
||
|
|
|
||
|
|
model = Holidays
|
||
|
|
fields = {
|
||
|
|
"recurring": ["exact"],
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, data=None, queryset=None, *, request=None, prefix=None):
|
||
|
|
super().__init__(data=data, queryset=queryset, request=request, prefix=prefix)
|
||
|
|
for field in self.form.fields.keys():
|
||
|
|
self.form.fields[field].widget.attrs["id"] = f"{uuid.uuid4()}"
|
||
|
|
self.form.fields["from_date"].label = (
|
||
|
|
f"{self.Meta.model()._meta.get_field('start_date').verbose_name} From"
|
||
|
|
)
|
||
|
|
self.form.fields["to_date"].label = (
|
||
|
|
f"{self.Meta.model()._meta.get_field('end_date').verbose_name} Till"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class CompanyLeaveFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
Filter class for CompanyLeaves model.
|
||
|
|
|
||
|
|
This filter allows searching CompanyLeaves objects based on
|
||
|
|
name, week day and based_on_week choices.
|
||
|
|
"""
|
||
|
|
|
||
|
|
name = filters.CharFilter(field_name="based_on_week_day", lookup_expr="icontains")
|
||
|
|
search = filters.CharFilter(method="filter_week_day")
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
""" "
|
||
|
|
Meta class defines the model and fields to filter
|
||
|
|
"""
|
||
|
|
|
||
|
|
model = CompanyLeaves
|
||
|
|
fields = {
|
||
|
|
"based_on_week": ["exact"],
|
||
|
|
"based_on_week_day": ["exact"],
|
||
|
|
}
|
||
|
|
|
||
|
|
def filter_week_day(self, queryset, _, value):
|
||
|
|
week_qry = CompanyLeaves.objects.none()
|
||
|
|
weekday_values = []
|
||
|
|
week_values = []
|
||
|
|
WEEK_DAYS = [
|
||
|
|
("0", __("Monday")),
|
||
|
|
("1", __("Tuesday")),
|
||
|
|
("2", __("Wednesday")),
|
||
|
|
("3", __("Thursday")),
|
||
|
|
("4", __("Friday")),
|
||
|
|
("5", __("Saturday")),
|
||
|
|
("6", __("Sunday")),
|
||
|
|
]
|
||
|
|
WEEKS = [
|
||
|
|
(None, __("All")),
|
||
|
|
("0", __("First Week")),
|
||
|
|
("1", __("Second Week")),
|
||
|
|
("2", __("Third Week")),
|
||
|
|
("3", __("Fourth Week")),
|
||
|
|
("4", __("Fifth Week")),
|
||
|
|
]
|
||
|
|
|
||
|
|
for day_value, day_name in WEEK_DAYS:
|
||
|
|
if value.lower() in day_name.lower():
|
||
|
|
weekday_values.append(day_value)
|
||
|
|
for day_value, day_name in WEEKS:
|
||
|
|
if value.lower() in day_name.lower() and value.lower() != __("All").lower():
|
||
|
|
week_values.append(day_value)
|
||
|
|
week_qry = queryset.filter(based_on_week__in=week_values)
|
||
|
|
elif value.lower() in __("All").lower():
|
||
|
|
week_qry = queryset.filter(based_on_week__isnull=True)
|
||
|
|
return queryset.filter(based_on_week_day__in=weekday_values) | week_qry
|
||
|
|
|
||
|
|
|
||
|
|
class PenaltyFilter(FilterSet):
|
||
|
|
"""
|
||
|
|
PenaltyFilter
|
||
|
|
"""
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = PenaltyAccounts
|
||
|
|
fields = "__all__"
|