611 lines
18 KiB
Python
611 lines
18 KiB
Python
"""
|
|
filters.py
|
|
|
|
This page is used to register filter for recruitment models
|
|
|
|
"""
|
|
|
|
import uuid
|
|
import django_filters
|
|
from django import forms
|
|
from recruitment.models import (
|
|
Candidate,
|
|
InterviewSchedule,
|
|
Recruitment,
|
|
SkillZone,
|
|
SkillZoneCandidate,
|
|
Stage,
|
|
RecruitmentSurvey,
|
|
SurveyTemplate,
|
|
)
|
|
from base.filters import FilterSet
|
|
|
|
# from django.forms.widgets import Boo
|
|
|
|
|
|
class CandidateFilter(FilterSet):
|
|
"""
|
|
Filter set class for Candidate model
|
|
|
|
Args:
|
|
FilterSet (class): custom filter set class to apply styling
|
|
"""
|
|
|
|
name = django_filters.CharFilter(field_name="name", lookup_expr="icontains")
|
|
# for pipeline use
|
|
candidate_name = django_filters.CharFilter(
|
|
method="pipeline_search", lookup_expr="icontains"
|
|
)
|
|
start_date = django_filters.DateFilter(
|
|
field_name="recruitment_id__start_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
end_date = django_filters.DateFilter(
|
|
field_name="recruitment_id__end_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
scheduled_from = django_filters.DateFilter(
|
|
field_name="joining_date",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end = django_filters.DateFilter(
|
|
field_name="probation_end",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end_till = django_filters.DateFilter(
|
|
field_name="probation_end",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end_from = django_filters.DateFilter(
|
|
field_name="probation_end",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
schedule_date = django_filters.DateFilter(
|
|
field_name="schedule_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
scheduled_till = django_filters.DateFilter(
|
|
field_name="joining_date",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
recruitment = django_filters.CharFilter(
|
|
field_name="recruitment_id__title", lookup_expr="icontains"
|
|
)
|
|
|
|
portal_sent = django_filters.BooleanFilter(
|
|
field_name="onboarding_portal",
|
|
method="filter_mail_sent",
|
|
widget=django_filters.widgets.BooleanWidget(),
|
|
)
|
|
joining_set = django_filters.BooleanFilter(
|
|
field_name="joining_date",
|
|
method="filter_joining_set",
|
|
widget=django_filters.widgets.BooleanWidget(),
|
|
)
|
|
|
|
def pipeline_search(self, queryset, _, value):
|
|
"""
|
|
This method is used to include the candidates when they in the recruitment/stages
|
|
"""
|
|
queryset = (
|
|
queryset.filter(name__icontains=value)
|
|
| queryset.filter(stage_id__stage__icontains=value)
|
|
| queryset.filter(stage_id__recruitment_id__title__icontains=value)
|
|
).distinct()
|
|
return queryset
|
|
|
|
class Meta:
|
|
"""
|
|
Meta class to add the additional info
|
|
"""
|
|
|
|
model = Candidate
|
|
fields = [
|
|
"recruitment",
|
|
"recruitment_id",
|
|
"stage_id",
|
|
"schedule_date",
|
|
"email",
|
|
"mobile",
|
|
"country",
|
|
"state",
|
|
"city",
|
|
"zip",
|
|
"gender",
|
|
"start_onboard",
|
|
"hired",
|
|
"canceled",
|
|
"is_active",
|
|
"recruitment_id__company_id",
|
|
"job_position_id",
|
|
"recruitment_id__closed",
|
|
"recruitment_id__is_active",
|
|
"job_position_id__department_id",
|
|
"recruitment_id__recruitment_managers",
|
|
"stage_id__stage_managers",
|
|
"stage_id__stage_type",
|
|
"joining_date",
|
|
"skillzonecandidate_set__skill_zone_id",
|
|
"skillzonecandidate_set__candidate_id",
|
|
"portal_sent",
|
|
"joining_set",
|
|
"rejected_candidate__reject_reason_id",
|
|
"offer_letter_status",
|
|
"candidate_rating__rating",
|
|
"candidate_interview__employee_id",
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.form.fields["is_active"].initial = True
|
|
for field in self.form.fields.keys():
|
|
self.form.fields[field].widget.attrs["id"] = f"{uuid.uuid4()}"
|
|
|
|
def filter_mail_sent(self, queryset, name, value):
|
|
return queryset.filter(onboarding_portal__isnull=(not value))
|
|
|
|
def filter_joining_set(self, queryset, name, value):
|
|
return queryset.filter(joining_date__isnull=(not value))
|
|
|
|
|
|
BOOLEAN_CHOICES = (
|
|
("", ""),
|
|
("false", "False"),
|
|
("true", "True"),
|
|
)
|
|
|
|
|
|
class RecruitmentFilter(FilterSet):
|
|
"""
|
|
Filter set class for Recruitment model
|
|
|
|
Args:
|
|
FilterSet (class): custom filter set class to apply styling
|
|
"""
|
|
|
|
candidate_name = django_filters.CharFilter(
|
|
field_name="title", method="pipeline_search"
|
|
)
|
|
search_onboarding = django_filters.CharFilter(
|
|
field_name="title", method="onboarding_search"
|
|
)
|
|
description = django_filters.CharFilter(lookup_expr="icontains")
|
|
start_date = django_filters.DateFilter(
|
|
field_name="start_date", widget=forms.DateInput(attrs={"type": "date"})
|
|
)
|
|
end_date = django_filters.DateFilter(
|
|
field_name="end_date", widget=forms.DateInput(attrs={"type": "date"})
|
|
)
|
|
start_from = django_filters.DateFilter(
|
|
field_name="start_date",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
end_till = django_filters.DateFilter(
|
|
field_name="end_date",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
search = django_filters.CharFilter(method="filter_by_name")
|
|
is_active = django_filters.ChoiceFilter(
|
|
choices=[
|
|
(True, "Yes"),
|
|
(False, "No"),
|
|
],
|
|
initial="Yes",
|
|
empty_label="Yes",
|
|
)
|
|
|
|
class Meta:
|
|
"""
|
|
Meta class to add the additional info
|
|
"""
|
|
|
|
model = Recruitment
|
|
fields = [
|
|
"recruitment_managers",
|
|
"company_id",
|
|
"title",
|
|
"is_event_based",
|
|
"start_date",
|
|
"end_date",
|
|
"closed",
|
|
"is_active",
|
|
"is_published",
|
|
"job_position_id",
|
|
]
|
|
|
|
def filter_by_name(self, queryset, _, 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 ""
|
|
|
|
job_queryset = queryset.filter(
|
|
open_positions__job_position__icontains=value
|
|
) | queryset.filter(title__icontains=value)
|
|
if first_name and last_name:
|
|
queryset = queryset.filter(
|
|
recruitment_managers__employee_first_name__icontains=first_name,
|
|
recruitment_managers__employee_last_name__icontains=last_name,
|
|
)
|
|
elif first_name:
|
|
queryset = queryset.filter(
|
|
recruitment_managers__employee_first_name__icontains=first_name
|
|
)
|
|
elif last_name:
|
|
queryset = queryset.filter(
|
|
recruitment_managers__employee_last_name__icontains=last_name
|
|
)
|
|
|
|
queryset = queryset | job_queryset
|
|
return queryset.distinct()
|
|
|
|
def pipeline_search(self, queryset, _, value):
|
|
"""
|
|
This method is used to search recruitment
|
|
"""
|
|
queryset = (
|
|
queryset.filter(title__icontains=value)
|
|
| queryset.filter(stage_set__stage__icontains=value)
|
|
| queryset.filter(candidate__name__icontains=value)
|
|
)
|
|
return queryset.distinct()
|
|
|
|
def onboarding_search(self, queryset, _, value):
|
|
"""
|
|
This method is used to search recruitment
|
|
"""
|
|
queryset = (
|
|
queryset.filter(title__icontains=value)
|
|
| queryset.filter(onboarding_stage__stage_title__icontains=value)
|
|
| queryset.filter(
|
|
candidate__onboarding_stage__candidate_id__name__icontains=value
|
|
)
|
|
)
|
|
return queryset.distinct()
|
|
|
|
|
|
class StageFilter(FilterSet):
|
|
"""
|
|
Filter set class for Stage model
|
|
|
|
Args:
|
|
FilterSet (class): custom filter set class to apply styling
|
|
"""
|
|
|
|
search = django_filters.CharFilter(method="filter_by_name")
|
|
candidate_name = django_filters.CharFilter(method="pipeline_search")
|
|
|
|
class Meta:
|
|
"""
|
|
Meta class to add the additional info
|
|
"""
|
|
|
|
model = Stage
|
|
fields = [
|
|
"recruitment_id",
|
|
"recruitment_id__job_position_id",
|
|
"recruitment_id__job_position_id__department_id",
|
|
"recruitment_id__company_id",
|
|
"recruitment_id__recruitment_managers",
|
|
"stage_managers",
|
|
"stage_type",
|
|
]
|
|
|
|
def filter_by_name(self, queryset, _, 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 ""
|
|
recruitment_query = queryset.filter(recruitment_id__title__icontains=value)
|
|
# Filter the queryset by first name and last name
|
|
stage_queryset = queryset.filter(stage__icontains=value)
|
|
if first_name and last_name:
|
|
queryset = queryset.filter(
|
|
stage_managers__employee_first_name__icontains=first_name,
|
|
stage_managers__employee_last_name__icontains=last_name,
|
|
)
|
|
elif first_name:
|
|
queryset = queryset.filter(
|
|
stage_managers__employee_first_name__icontains=first_name
|
|
)
|
|
elif last_name:
|
|
queryset = queryset.filter(
|
|
stage_managers__employee_last_name__icontains=last_name
|
|
)
|
|
|
|
queryset = queryset | stage_queryset | recruitment_query
|
|
return queryset
|
|
|
|
def pipeline_search(self, queryset, _, value):
|
|
"""
|
|
This method is used to search recruitment
|
|
"""
|
|
queryset = (
|
|
queryset.filter(stage__icontains=value)
|
|
| queryset.filter(candidate__name__icontains=value)
|
|
| queryset.filter(recruitment_id__title__icontains=value)
|
|
)
|
|
return queryset.distinct()
|
|
|
|
|
|
class SurveyFilter(FilterSet):
|
|
"""
|
|
SurveyFIlter
|
|
"""
|
|
|
|
options = django_filters.CharFilter(
|
|
lookup_expr="icontains",
|
|
label="Options",
|
|
field_name="options",
|
|
)
|
|
|
|
question = django_filters.CharFilter(
|
|
lookup_expr="icontains",
|
|
label="Question",
|
|
field_name="question",
|
|
)
|
|
|
|
class Meta:
|
|
"""
|
|
class Meta for additional options
|
|
"""
|
|
|
|
model = RecruitmentSurvey
|
|
fields = "__all__"
|
|
exclude = [
|
|
"sequence",
|
|
]
|
|
|
|
|
|
class SurveyTemplateFilter(django_filters.FilterSet):
|
|
"""
|
|
SurveyTemplateFilter
|
|
"""
|
|
|
|
question = django_filters.CharFilter(
|
|
lookup_expr="icontains",
|
|
label="Title",
|
|
field_name="title",
|
|
)
|
|
|
|
class Meta:
|
|
model = SurveyTemplate
|
|
fields = "__all__"
|
|
|
|
|
|
class CandidateReGroup:
|
|
"""
|
|
Class to keep the field name for group by option
|
|
"""
|
|
|
|
fields = [
|
|
("", "select"),
|
|
("recruitment_id", "Recruitment"),
|
|
("job_position_id", "Job Position"),
|
|
("hired", "Hired"),
|
|
("country", "Country"),
|
|
("stage_id", "Stage"),
|
|
("joining_date", "Date Joining"),
|
|
("probation_end", "Probation End"),
|
|
("offer_letter_status", "Offer Letter Status"),
|
|
("offer_letter_status", "Offer Letter Status"),
|
|
("rejected_candidate__reject_reason_id", "Reject Reason"),
|
|
("skillzonecandidate_set__skill_zone_id", "Skill Zone"),
|
|
]
|
|
|
|
|
|
class SkillZoneFilter(FilterSet):
|
|
"""
|
|
Skillzone Filter
|
|
"""
|
|
|
|
search = django_filters.CharFilter(field_name="title", lookup_expr="icontains")
|
|
|
|
class Meta:
|
|
"""
|
|
class Meta for additional options
|
|
"""
|
|
|
|
model = SkillZone
|
|
fields = [
|
|
"is_active",
|
|
"skillzonecandidate_set__candidate_id",
|
|
"skillzonecandidate_set__candidate_id__recruitment_id",
|
|
"skillzonecandidate_set__candidate_id__job_position_id",
|
|
"skillzonecandidate_set__candidate_id__stage_id__stage_type",
|
|
]
|
|
|
|
|
|
class SkillZoneCandFilter(FilterSet):
|
|
"""
|
|
Skillzone Candidate FIlter
|
|
"""
|
|
|
|
search = django_filters.CharFilter(
|
|
field_name="candidate_id__name", method="cand_search"
|
|
)
|
|
start_date = django_filters.DateFilter(
|
|
field_name="candidate__id__recruitment_id__start_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
end_date = django_filters.DateFilter(
|
|
field_name="candidate__id__recruitment_id__end_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
scheduled_from = django_filters.DateFilter(
|
|
field_name="candidate__id__joining_date",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end = django_filters.DateFilter(
|
|
field_name="candidate__id__probation_end",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end_till = django_filters.DateFilter(
|
|
field_name="candidate__id__probation_end",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
probation_end_from = django_filters.DateFilter(
|
|
field_name="candidate__id__probation_end",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
schedule_date = django_filters.DateFilter(
|
|
field_name="candidate__id__schedule_date",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
scheduled_till = django_filters.DateFilter(
|
|
field_name="candidate__id__joining_date",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
recruitment = django_filters.CharFilter(
|
|
field_name="candidate__id__recruitment_id__title", lookup_expr="icontains"
|
|
)
|
|
|
|
portal_sent = django_filters.BooleanFilter(
|
|
field_name="candidate__id__onboarding_portal",
|
|
method="filter_mail_sent",
|
|
widget=django_filters.widgets.BooleanWidget(),
|
|
)
|
|
joining_set = django_filters.BooleanFilter(
|
|
field_name="candidate__id__joining_date",
|
|
method="filter_joining_set",
|
|
widget=django_filters.widgets.BooleanWidget(),
|
|
)
|
|
|
|
class Meta:
|
|
"""
|
|
class Meta for additional options
|
|
"""
|
|
|
|
model = SkillZoneCandidate
|
|
fields = [
|
|
"is_active",
|
|
"candidate_id",
|
|
"candidate_id__recruitment_id",
|
|
"candidate_id__stage_id",
|
|
"candidate_id__schedule_date",
|
|
"candidate_id__email",
|
|
"candidate_id__mobile",
|
|
"candidate_id__country",
|
|
"candidate_id__state",
|
|
"candidate_id__city",
|
|
"candidate_id__zip",
|
|
"candidate_id__gender",
|
|
"candidate_id__start_onboard",
|
|
"candidate_id__hired",
|
|
"candidate_id__canceled",
|
|
"candidate_id__is_active",
|
|
"candidate_id__recruitment_id__company_id",
|
|
"candidate_id__job_position_id",
|
|
"candidate_id__recruitment_id__closed",
|
|
"candidate_id__recruitment_id__is_active",
|
|
"candidate_id__job_position_id__department_id",
|
|
"candidate_id__recruitment_id__recruitment_managers",
|
|
"candidate_id__stage_id__stage_managers",
|
|
"candidate_id__stage_id__stage_type",
|
|
"candidate_id__joining_date",
|
|
"candidate_id__skillzonecandidate_set__skill_zone_id",
|
|
"candidate_id__skillzonecandidate_set__candidate_id",
|
|
"candidate_id__rejected_candidate__reject_reason_id",
|
|
"candidate_id__offer_letter_status",
|
|
"candidate_id__candidate_rating__rating",
|
|
]
|
|
exclude = [
|
|
"reason",
|
|
"objects",
|
|
]
|
|
|
|
def cand_search(self, queryset, _, value):
|
|
"""
|
|
This method to include candidate when search skill zone
|
|
"""
|
|
return (
|
|
queryset.filter(candidate_id__name__icontains=value)
|
|
| queryset.filter(skill_zone_id__title__icontains=value)
|
|
).distinct()
|
|
|
|
|
|
class InterviewFilter(FilterSet):
|
|
"""
|
|
Filter set class for Candidate model
|
|
|
|
Args:
|
|
FilterSet (class): custom filter set class to apply styling
|
|
"""
|
|
|
|
search = django_filters.CharFilter(field_name="candidate_id__name", lookup_expr="icontains")
|
|
|
|
scheduled_from = django_filters.DateFilter(
|
|
field_name="interview_date",
|
|
lookup_expr="gte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
# schedule_date = django_filters.DateFilter(
|
|
# field_name="interview_date",
|
|
# widget=forms.DateInput(attrs={"type": "date"}),
|
|
# )
|
|
scheduled_till = django_filters.DateFilter(
|
|
field_name="interview_date",
|
|
lookup_expr="lte",
|
|
widget=forms.DateInput(attrs={"type": "date"}),
|
|
)
|
|
|
|
class Meta:
|
|
"""
|
|
Meta class to add the additional info
|
|
"""
|
|
|
|
model = InterviewSchedule
|
|
fields = [
|
|
"candidate_id",
|
|
"employee_id",
|
|
"interview_date",
|
|
# "recruitment",
|
|
# "recruitment_id",
|
|
# "stage_id",
|
|
# "schedule_date",
|
|
# "email",
|
|
# "mobile",
|
|
# "country",
|
|
# "state",
|
|
# "city",
|
|
# "zip",
|
|
# "gender",
|
|
# "start_onboard",
|
|
# "hired",
|
|
# "canceled",
|
|
# "is_active",
|
|
# "recruitment_id__company_id",
|
|
# "job_position_id",
|
|
# "recruitment_id__closed",
|
|
# "recruitment_id__is_active",
|
|
# "job_position_id__department_id",
|
|
# "recruitment_id__recruitment_managers",
|
|
# "stage_id__stage_managers",
|
|
# "stage_id__stage_type",
|
|
# "joining_date",
|
|
# "skillzonecandidate_set__skill_zone_id",
|
|
# "skillzonecandidate_set__candidate_id",
|
|
# "portal_sent",
|
|
# "joining_set",
|
|
# "rejected_candidate__reject_reason_id",
|
|
# "offer_letter_status",
|
|
# "candidate_rating__rating",
|
|
# "candidate_interview__employee_id",
|
|
]
|