siRuZdZddlZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z m Z ddl m Z dd lmZdd lmZdd lmZmZmZmZmZmZmZmZmZmZmZGd d ej<ZGddejZ GddeZ!Gdde Z"gdZ#GddeZ$GddejZ%Gdde Z&GddeZ'GddZ(Gdd eZ)Gd!d"eZ*Gd#d$eZ+Gd%d&e Z,Gd'd(eZ-Gd)d*e Z.Gd+d,e Z/y)-z Module: filters.py This module contains custom Django filters and filter sets for the PMS (Performance Management System) app. N) timedelta) relativedelta)forms)timezone) DateFilterDateFromToRangeFilter) FilterSet)reload_queryset)HorillaFilterSet) AnonymousFeedbackBonusPointSettingEmployeeBonusPointEmployeeKeyResultEmployeeObjectiveFeedback KeyResultMeetings ObjectivePeriodQuestionTemplateceZdZdZdZy)DateRangeFilterz A custom Django filter for filtering querysets based on date ranges. This filter allows you to filter a queryset based on date ranges such as 'today', 'yesterday', 'week', or 'month' in the 'created_at' field of model. cZ|r|dk(rOtjjj}|jd}|j |}|dk(rZtjj }|tj dz }|jd}|j |}|dk(rttjj }|tj |jz }|tj dz}|j ||g }|S|d k(rvtjj }tj|j|jd}|tj d z} |j || g }|S Ntodayz%Y-%m-%d)created_at__startswith yesterdaydaysweek)created_at__rangemonth datetimenowdatestrftimefilterrrweekdayyearr$ selfqsvaluerformatted_dater start_of_week end_of_weekstart_of_month end_of_months //var/www/python-projects/worksol/pms/filters.pyr+zDateRangeFilter.filter,a  ))--/446!& !;YYnYE # ++-!H$6$6A$>> !*!3!3J!?YYnYE ++- %(:(: (P P +h.@.@a.HH YY-1MYN '! ++-!)uzz5;;!J-0B0B0KK YY.,1OYP N)__name__ __module__ __qualname____doc__r+r9r7rr$s r9rceZdZfdZxZS)CustomFilterSetcBt||i|t|jj|jjj D]\}}|j |}|jj}t|tjtjtjfr)|jjjddint|tj fr(|jjjddint|tj"r(|jjjddint|tj$tj&fr(|jjjddinAt|tj(r'|jjjddit|t*j,sd|_y)Nclassoh-input w-100z/oh-select oh-select-2 select2-hidden-accessiblezoh-switch__checkbox icontains)super__init__r formfieldsitemsfiltersfieldwidget isinstancer NumberInput EmailInput TextInputattrsupdateSelectTextarea CheckboxInputCheckboxSelectMultipleModelChoiceFielddjango_filters CharFilter lookup_expr)r/argskwargs field_namerK filter_widgetrL __class__s r7rFzCustomFilterSet.__init__Is $)&) (()!%!1!1!7!7!9 0 J LL4M"((//F**E,<,r9r7Metarhts r9ror1c\|jd}|jjj}|D]c}||j ||j |z|j |z|j |z|j |zz}e|j S)G This method is used to search employees and objective  )(managers__employee_first_name__icontains)'managers__employee_last_name__icontains))assignees__employee_first_name__icontains)(assignees__employee_last_name__icontainstitle__icontainssplitrnobjectsnoner+distinctr/queryset_r1valuesemptyrzs r7rdz#ActualObjectiveFilter.search_method~sS!&&++- EO//%/PQ//E/RS//5/QR//5/9 :E ~~r9N r:r;r<r=rXrYsearchrostrrdr>r9r7rcrcms1'^ & &o >F    r9rcceZdZdZej dZej dZej dZej dZ ej dZ e d Z e ejd d d  Ze ejd d d  Ze ejd d d  Ze ejd d d  ZGddZdefdZy)ObjectiveFilterz Custom filter set for EmployeeObjective records. This filter set allows to filter EmployeeObjective records based on various criteria. idr] key_result_idprogress_percentageobjective_id__managersrdre created_atr)oh-input w-100typerBrQrLceZdZdZeZgdZy)ObjectiveFilter.MetaT A nested class that specifies the model and fields for the filter. ) objectivestatus employee_idr start_date updated_atend_daterj objective_idN)r:r;r<r=rrnrHr>r9r7rors "  r9ror1c|jd}|jjj}|D]=}||j ||j |z|j |zz}?|S)zI This method is used to search in managers and objective rr)objective_id__title__icontains)6objective_id__managers__employee_first_name__icontains)5objective_id__managers__employee_last_name__icontains)rzrnr{r|r+r~s r7rdzObjectiveFilter.search_methodsS!&&++- EuE//KP"//JO" E  r9N)r:r;r<r=rXrYemployee_objectiverm'employee_objective__progress_percentagerirrcreated_at_date_rangerr DateInputrrrrrorrdr>r9r7rrs 322dC(A(A(A")%/Hn.G.G(/+)~((4LMH &^ & &o >F+|Duf?P%QRJuf?P%QRJuf?P%QRJuf?P%QRH   $r9r))AlloverdueOverdue due_todayz Due Today due_this_weekz Due This Weekdue_this_monthzDue This MonthcVeZdZdZej dedejZ ejdZ ejdZ ejd Zed ej j#d d d Zeej(d dd Zeej(d dd Zedej j#d d d Zedej j#d d d ZGddZdZdZdefdZy)FeedbackFilterz Custom filter set for Feedback records. This filter set allows to filter Feedback records based on various criteria. z Quick Datefilter_due_date)labelchoicesrfrLfilter_offboarding_employeesrerdrDrZrr)rCrr)r]rLrrrrceZdZdZeZgdZy)FeedbackFilter.Metar) manager_idr colleague_idsubordinate_idquestion_template_idrrjrremployee_key_results_idcyclic_feedbackcyclic_feedback_days_countcyclic_feedback_periodcyclic_next_start_datecyclic_next_end_datestart_date_rangeend_date_rangeN)r:r;r<r=rrnrHr>r9r7rors  r9roc|jd}tjj}|dk(r|j |S|dk(r|j |S|dk(r@|t |j z }|t d z}|j ||f S|d k(rB|jd }|td zt d z }|j ||f S|S)! Filter due date Closed)status__iexactr end_date__ltrrrrr"end_date__rangerrdaymonths) excluderr(r)r+rr,replacer) r/rnamer1rr3r4 first_daylast_days r7rzFeedbackFilter.filter_due_date(s##8#< ##% I ???6 6 k !??E?2 2 o %!I5==?$CCM')*;;K??M;3O?P P & & ! ,I!M$;;ya?PPH??Ix3H?I Ir9cp|j| |j| z}|jS)z. Filter offboarding employees )(employee_id__offboardingemployee__isnull)&employee_id__resignationletter__isnull)r+r}r/rrr1s r7rz+FeedbackFilter.filter_offboarding_employees>sC??9>Y# OOu9O MN  ""r9r1c|j}|d}t|dkDrdj|ddnd}|j}|r8|r6|j |||j |z}|j S|r5|j ||j |z}|j S|r%|j | |j |z}|j S)  Search Method rrrrNr)+employee_id__employee_first_name__icontains*employee_id__employee_last_name__icontains)review_cycle__icontainsrr)rzlenjoinr|r+r})r/rrr1parts first_name last_nameqrsts r7rdzFeedbackFilter.search_methodGs 1X +.u:>CHHU12Y'r }} )???D}}???D}} ??;D#>?D}}r9N)r:r;r<r=rX ChoiceFilterDUE_DATE_CHOICESr HiddenInputdue_date_quick_filter BooleanFilteroffboarding_employeesrYr review_cyclerwidgets RangeWidgetrrrrrrrrrorrrrdr>r9r7rrsd 8N77   u  "  9N88-'^ & &o >F,>,,EL+%%11!,<=2  uf?P%QRJuf?P%QRH -%%11!,<=2  +%%11!,<=2 N  4,#r9rceZdZdZedZejddejddiZ ejdd ejddiZ Gd d Z y ) AnonymousFeedbackFilterz Custom filter set for AnonymousFeedback records. This filter set allows to filter AnonymousFeedback records based on various criteria. rrgterr)rr]rZrLlteceZdZdZeZdZy)AnonymousFeedbackFilter.Metar__all__Nr:r;r<r=r rnrHr>r9r7rorr "r9roN) r:r;r<r=rrrXrrrrrror>r9r7rr_sv ,|D***uff%56J )~((uff%56H r9rcHeZdZdZej ddgdZejddZejdd Z ejdd Z ejd d Z ejdd Z ejd d Zd ZGddZy)KeyResultFilterz KeyResult Filter rDuerrr)due_next_monthzDue Next Monthrrfrrrrr]rZrr)rZr]rcftjj}|dk(r|j|S|dk(r@|t |j z }|t dz}|j||fS|dk(rB|j d }|td zt dz }|j||fS|d k(rO|td zj d }|td zt dz }|j||fS|d k(r|j| S|S)rrrrrr"rrrrrrrrrr(r)r+rr,rrr/rrr1rstartends r7rzKeyResultFilter.filter_due_dates+ ##% K ??E?2 2 o %I5==?;;E)++C??E3r9r7ror !r9roN)r:r;r<r=rXrdue NumberFilterkr_progress_percentage__gtekr_progress_percentage__lterYkr_start_date_fromkr_start_date_tillkr_end_date_fromkr_end_date_tillrror>r9r7rr{s &. % %   C#>."="=(e##>."="=(e#322l322l1~00j1~00j2r9rcVeZdZdZej dZGddZdefdZ y) ActualKeyResultFilterz( Filter through KeyResult model rdreceZdZeZgdZy)ActualKeyResultFilter.Meta) progress_type target_valuerl company_idN)r:r;r<rrnrHr>r9r7rors Lr9ror1c|jd}|jjj}|D]}||j |z}|j S)rqrrrwryr~s r7rdz#ActualKeyResultFilter.search_methods_S!&&++- FEX__e_DEE F~~r9Nrr>r9r7rrs3'^ & &o >FMM    r9rceZdZdZgdZy)ObjectiveReGroupz: Class to keep the field name for group by option ))rselect)rOwner)rStatusN)r:r;r<r=rHr>r9r7rrsFr9rceZdZdZej dZejddejddi Z ejdd ejddi Z ejd dejddi Z ejd d ejddi Z ejd dejddi Zejd d ejddi Zejddejddi Zejdd ejddi Zej$ddgdZej(ddZej(dd Zej(dZej(dZdZdZdZdZdZdZdZGdd Z d!e!fd"Z"y#)$EmployeeObjectiveFilterz0 Filter through EmployeeObjective model rdrerrrr)rrrrkr_start_date_till_method)rfrZrLkr_end_date_from_methodkr_end_date_till_methodrrrrrrkr_progress_gtekr_progress_ltec&|j|S) Kr filter )$employee_key_result__start_date__gter+rs r7kr_start_date_from_methodz1EmployeeObjectiveFilter.kr_start_date_from_method$EJJr9c&|j|S)r')$employee_key_result__start_date__lter)rs r7r!z1EmployeeObjectiveFilter.kr_start_date_till_method*r+r9c&|j|S)r')"employee_key_result__end_date__gter)rs r7r"z/EmployeeObjectiveFilter.kr_end_date_from_method0%HHr9c&|j|S)r')"employee_key_result__end_date__lter)rs r7r#z/EmployeeObjectiveFilter.kr_end_date_till_method6r0r9c&|j|S)# Kr prgress filter )-employee_key_result__progress_percentage__gter)rs r7r$z'EmployeeObjectiveFilter.kr_progress_gte<USSr9c&|j|S)r4)-employee_key_result__progress_percentage__lter)rs r7r%z'EmployeeObjectiveFilter.kr_progress_lteBr6r9cftjj}|dk(r|j|S|dk(r@|t |j z }|t dz}|j||fS|dk(rB|j d }|td zt dz }|j||fS|d k(rO|td zj d }|td zt dz }|j||fS|d k(r|j| S|S)rr)employee_key_result__end_daterrr")$employee_key_result__end_date__rangerrrrrr)!employee_key_result__end_date__ltrrs r7rz'EmployeeObjectiveFilter.filter_due_dateHs. ##% K ???G G o %I5==?;;E)++C??PS ?U U & &MMaM(E=22iQ6GGC??PS ?U U & &]!44==!=DE=22iQ6GGC??PS ?U U i ??U?K Kr9ceZdZeZgdZy)EmployeeObjectiveFilter.Meta)rrjrrrrN)r:r;r<rrnrHr>r9r7ror>as! r9ror1c6|jd}|jjj}|D]P}||j |z|j |z|j |z|j |z}R|j S)rqrrrr)objective__icontains)*employee_key_result__key_result__icontainsryr~s r7rdz%EmployeeObjectiveFilter.search_methodlsS!&&++- E??u?UW??e?TV???>@??e?T V  ~~r9N)#r:r;r<r=rXrYrrrrstart_date_fromstart_date_till end_date_from end_date_tillrrrrrr r progress_percentage__gteprogress_percentage__lter r r*r!r"r#r$r%rrorrdr>r9r7r r sA'^ & &o >F/n//uff%56O 0n//uff%56O .N--uff%56M .N--uff%56M 322*uff%56 322*uff%56 1~00(uff%56 1~00(uff%56 &. % %   C ;~::(e  ;~::(e #>."="=EV"W"=."="=EV"WK K I I T T 2     r9r c8eZdZejddZej ddejddiZ ej ddejddiZ ej dd ejddiZ Gd d Z y ) MeetingsFiltertitlerDrr)rrr date__gte date__lteceZdZeZgdZy)MeetingsFilter.Meta) rmanagerquestion_template is_active.employee_id__employee_work_info__department_id+employee_id__employee_work_info__company_id0employee_id__employee_work_info__job_position_id-employee_id__employee_work_info__work_type_id)employee_id__employee_work_info__shift_id5employee_id__employee_work_info__reporting_manager_idN)r:r;r<rrnrHr>r9r7rorNs  r9roN) r:r;r<rXrYrrrrr) date_greater date_lessror>r9r7rIrI~s &^ & &'{ SF $> $ $uff%56 D ->,,uff%56L *))uff%56I   r9rIc&eZdZdZej ddZej dZedZ e e jdd d   Z e e jdd d   Zd ZGddZdefdZddddfd ZxZS)AnonymousFilterz Custom filter set for Anonymous records. This filter set allows to filter Anonymous records based on various criteria. rdrD)rfrZrrrr)rrrrcZ|r|dk(rOtjjj}|jd}|j |}|dk(rZtjj }|tj dz }|jd}|j |}|dk(rttjj }|tj |jz }|tj dz}|j ||g }|S|d k(rvtjj }tj|j|jd}|tj d z} |j || g }|Srr&r.s r7r+zAnonymousFilter.filterr8r9ceZdZdZeZdZy)AnonymousFilter.MetarrNrr>r9r7ror^rr9ror1c&|j|S)r)feedback_subject__icontainsr))r/rrr1s r7rdzAnonymousFilter.search_methods5AAr9N)requestprefixc4tt| ||||y)N)datarrarb)rEr[rF)r/rdrrarbr_s r7rFzAnonymousFilter.__init__s  ot-'& . r9)NN)r:r;r<r=rXrYrrrrrrrrrr+rorrdrFr`ras@r7r[r[s '^ & &o; WF,>,,EL+|Duf?P%QRJuf?P%QRH 6BB  D  r9r[cHeZdZejddZGddZy)QuestionTemplateFilterrPrDrceZdZeZdgZy)QuestionTemplateFilter.MetarPN)r:r;r<rrnrHr>r9r7rorhs   r9roNr:r;r<rXrYrror>r9r7rfrfs# &^ & &&KF  r9rfcHeZdZejddZGddZy) PeriodFilter period_namerDrceZdZeZdgZy)PeriodFilter.MetarlN)r:r;r<rrnrHr>r9r7rorn!s  r9roNrir>r9r7rkrks# &^ & & kF  r9rkc0eZdZdZGddZdefdZy)BonusPointSettingFilter0 Filter through BonusPointSetting model ceZdZeZdZy)BonusPointSettingFilter.MetarN)r:r;r<r rnrHr>r9r7rorsCr r9ror1c|jd}|jjj}|D]*}||j |z|j |z},|j Srqrrrrryr~s r7rdz%BonusPointSettingFilter.search_methodGxS!&&++- E??u?UW??e?TV  ~~r9Nr:r;r<r=rorrdr>r9r7rprp(s4    r9rpc0eZdZdZGddZdefdZy)EmployeeBonusPointFilterrqceZdZeZdZy)EmployeeBonusPointFilter.MetarN)r:r;r<rrnrHr>r9r7ror{\s "r9ror1c|jd}|jjj}|D]*}||j |z|j |z},|j Sruryr~s r7rdz&EmployeeBonusPointFilter.search_method`rvr9Nrwr>r9r7ryryWs    r9ry)0r=r'rdjangorXdateutil.relativedeltarr django.utilsrrr base.filtersr base.methodsr horilla.filtersr pms.modelsr r rrrrrrrrrFilterrr@rcrrrrrrrr rIr[rfrkrpryr>r9r7rs2 0!<"(,    !n++!H"0n.."0J , DDoDN}%}@n668@o@F , .  Z .Z z  %  \? &? D  Y    #  , i, ^ y r9