vg/LdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZGd d eZGd d eZGd deZGddeZGddeZGddeZGddeZGddZGddZGddeZGddZy)z= Module containing custom filter classes for various models. N)forms)Q) FilterSet)reload_queryset)AssetAssetAssignment AssetCategory AssetRequestc"eZdZdZfdZxZS)CustomFilterSeta Custom FilterSet class that applies specific CSS classes to filter widgets. The class applies CSS classes to different types of filter widgets, such as NumberInput, EmailInput, TextInput, Select, Textarea, CheckboxInput, CheckboxSelectMultiple, and ModelChoiceField. The CSS classes are applied to enhance the styling and behavior of the filter widgets. ct||i|t|jj|jjj D]\}}|j |}|jj}t|tjtjtjfr)|jjjddinft|tj fr)|jjjddin"t|tj"r(|jjjddint|tj$tj&fr2|jjjjddint|tj(r(|jjjddinBt|tj*r(|jjjdddt|t,j.sd |_y) Nclasszoh-input w-100zoh-select oh-select-2zoh-switch__checkboxzoh-select oh-select-2 datezoh-input w-100)typer icontains)super__init__rformfieldsitemsfiltersfieldwidget isinstancer NumberInput EmailInput TextInputattrsupdateSelectTextarea CheckboxInputCheckboxSelectMultipleModelChoiceField DateFielddjango_filters CharFilter lookup_expr)selfargskwargs field_namer filter_widgetr __class__s !/var/www/horilla/asset/filters.pyrzCustomFilterSet.__init__s $)&) (()!%!1!1!7!7!9" 0 J LL4M"((//F**E,<,r@r1r0rArLbrBr1rAct||i||jjD]@}t t j |jjjd<ByNid rrrvisible_fieldsstruuiduuid4rrrr*r+r,visibler/s r0rzAssetFilter.__init__mX $)&)yy//1 AG/24::F(~((4GHH  A r1rFc\eZdZdZej dZGddZfdZxZ S)CustomAssetFilterzF Custom filter set for asset assigned to employees instances. r)r)ceZdZdZeZddgZy)CustomAssetFilter.Metaa@ Specifies the model and fields to be used for filtering AssetAssignment instances. Attributes: model (class): The model class AssetAssignment to be filtered. fields (list): The fields to include in the filter, referring to related AssetAssignment fields. asset_id__asset_nameasset_id__asset_statusNr2r3r4r5r r?rr@r1r0rAres   " $ r1rAct||i||jjD]@}t t j |jjjd<ByrNrPrUs r0rzCustomAssetFilter.__init__rWr1) r2r3r4r5r'r(rfrArr6r7s@r0rcrc|s35>44M  AAr1rccheZdZdZej dZdefdZGddZ fdZ xZ S) AssetRequestFilterz7 Custom filter set for AssetRequest instances. rGrHr_c|jd}|jjj}|D]*}||j ||j |zz},|j S)9 This method is used to search employees  )5requested_employee_id__employee_first_name__icontains)4requested_employee_id__employee_last_name__icontainssplitr?objectsnoner[r\r*r]r^r_valuesemptyrrs r0rGz AssetRequestFilter.search_methodsS!&&++- EJO //IN"E ~~r1ceZdZdZeZdZy)AssetRequestFilter.Metaa Specifies the model and fields to be used for filtering AssetRequest instances. Attributes: model (class): The model class AssetRequest to be filtered. fields (str): A special value "__all__" to include all fields of the model in the filter. r=N)r2r3r4r5r r?rr@r1r0rArys r1rAct||i||jjD]@}t t j |jjjd<ByrNrPrUs r0rzAssetRequestFilter.__init__rWr1 r2r3r4r5r'r(r`rRrGrArr6r7s@r0rkrks>'^ & &o >F  "  AAr1rkcheZdZdZej dZdefdZGddZ fdZ xZ S) AssetAllocationFilterz: Custom filter set for AssetAllocation instances. rGrHr_c|jd}|jjj}|D]*}||j ||j |zz},|j S)rmrn)7assigned_to_employee_id__employee_first_name__icontains)6assigned_to_employee_id__employee_last_name__icontainsrqrus r0rGz#AssetAllocationFilter.search_methodsS!&&++- ELQ //KP"E ~~r1ceZdZdZeZdZy)AssetAllocationFilter.MetaD Specifies the model and fields to be used for filtering AssetAllocation instances. Attributes: model (class): The model class AssetAssignment to be filtered. fields (str): A special value "__all__" to include all fields of the model in the filter. r=Nrhr@r1r0rAr  r1rAct||i||jjD]@}t t j |jjjd<ByrNrPrUs r0rzAssetAllocationFilter.__init__rWr1r{r7s@r0r}r}s>'^ & &o >F  "  AAr1r}cleZdZdZej dZGddZfdZdZ fdZ xZ S) AssetCategoryFilterz8 Custom filter set for AssetCategory instances. rGrHceZdZeZdZy)AssetCategoryFilter.Metar=N)r2r3r4r r?rr@r1r0rArs r1rAct||i||jjD]@}t t j |jjjd<ByrNrPrUs r0rzAssetCategoryFilter.__init__rWr1cv|s|S|jt|t|zjS)zW Search method to filter by asset category name or related asset name. )asset_category_name__icontains)asset__asset_name__icontains)r[rr\r*r]namer_s r0rGz!AssetCategoryFilter.search_methods;O U 3U3 4  (* r1ct||}|jrQd|jvrCt|jj}|j |j dd}|jS)zH Filters queryset and applies AssetFilter if necessary. asset__pk)datapkT)flat) asset__pk__in)rfilter_querysetrrFqsr[ values_listr\)r*r]assetsr/s r0rz#AssetCategoryFilter.filter_queryset ss 7*84 99 1 dii033F$00D0A'H  ""r1) r2r3r4r5r'r(r`rArrGrr6r7s@r0rrs;'^ & &o >FA ##r1rceZdZdZgdZy)AssetRequestReGroup: Class to keep the field name for group by option )r!)requested_employee_idEmployee)rJzAsset Category)asset_request_datez Request Date)asset_request_statusStatusNr2r3r4r5rr@r1r0rrFr1rceZdZdZgdZy)AssetAllocationReGroupr)rassigned_to_employee_idr assigned_datez Assigned Date return_datez Return DateNrr@r1r0rr)sFr1rceZdZdZej ddZej ddZejdd e jd d i  Z ejdde jd d i  Z ejdd e jd d i  Z ejdde jd d i  ZdZGddZy)AssetHistoryFilterz^ Custom filter set for AssetAssignment instances for filtering in asset history view. rfr)r-r) return_status exclude_none)r-rIrgterr)r)r-r)rlterc4|dk(r|jd}|S)z` Exclude objects with a null return_status from the queryset if value is "True" TrueF)return_status__isnull)r[rs r0rzAssetHistoryFilter.exclude_noneVs  F?UCHr1ceZdZdZeZdZy)AssetHistoryFilter.Metarr=Nrhr@r1r0rAr^rr1rAN)r2r3r4r5r'r(r`returned_assets DateFilterr DateInputreturn_date_gtereturn_date_lteassigned_date_gteassigned_date_lterrAr@r1r0rr6s'^ & &){F0n//">O0n// uff%56O 0n// uff%56O 211"uff%56 211"uff%56   r1rceZdZdZgdZy)AssetHistoryReGroupr)r)asset_idrrrrNrr@r1r0rrlrr1r)r5rSr'djangordjango.db.modelsrr base.methodsrmodelsrr r r r r9rFrcrkr}rrrrrr@r1r0rs $(GG10i10hVV*/DAA:'A'AT(AO(AV*#/*#Z    33l  r1