vg5dZddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZddlmZmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mGdde j"ZGdd e j"Z?Gd!d"e?Z@Gd#d$e?ZAd%ZBGd&d'eZCGd(d)eZDGd*d+eZEGd,d-e)ZFGd.d/eZGGd0d1eZHGd2d3eZIgd4ZJGd5d6e jZLGd7d8e jZMGd9d:eZNGd;de jZPGd?d@e jZQGdAdBe(ZRGdCdDeZSGdEdFeZTGdGdHeZUGdIdJeZVGdKdLeZWe jdMr4dNdOlYmZZZm[Z[GdPdQeZ\GdRdSe jZ]GdTdUeZ^yy)Vz This module provides Horilla ModelForms for creating and managing leave-related data, including leave type, leave request, leave allocation request, holidays and company leaves. N)datedatetime)Any)forms)apps)ValidationError)Q) ModelForm) TextInputrender_to_string) gettext_lazy)filtersubordinatesemployeemodelreload_queryset) CompanyLeavesHolidays)EmployeeFilter)MultipleFileField)Employee)horilla_middlewares) HorillaFormHorillaModelForm)HorillaMultiSelectField)HorillaMultiSelectWidget)calculate_requested_dayscompany_leave_dates_listget_leave_day_attendanceholiday_dates_listleave_requested_dates)AvailableLeaveLeaveAllocationRequestLeaveallocationrequestComment LeaveRequestLeaverequestCommentLeaverequestFile LeaveType RestrictLeaveyesYesnoNogj@c"eZdZdZfdZxZS)r z Customized ModelForm class with additional functionality for field customization based on the type of widget and setting initial values based on the current request. ct||i|ttjdd}t |j |j jD]\}}|j}t|tjrA|jjjdditj|_nt|tj"tj$tj&fr3|jjjd|j(dt|tj*fr)|jjjddit|tj,r6|jjjd|j(dd d nt|tj.tj0fs|jjjdd i |j2j4|j d _ |j2j4j6|j d _y#Y8xYw#YyxYw)a> Initializes the ModelForm instance. This method customizes field attributes such as CSS classes and placeholders based on the type of widget. It also sets initial values for specific fields based on the current request, particularly for 'employee_id' and 'company_id' fields. requestNclassz oh-input oh-calendar-input w-100oh-input w-100r/ placeholderz/oh-select oh-select-2 select2-hidden-accessible(r/r2rowscolsoh-switch__checkbox employee_id company_id)super__init__getattrr_thread_localsrfieldsitemswidget isinstancer DateInputattrsupdatertodayinitial NumberInput EmailInputr labelSelectTextarea CheckboxInputCheckboxSelectMultipleuser employee_get get_companyselfargskwargsr. field_namefieldrA __class__s /var/www/horilla/leave/forms.pyr<zModelForm.__init__:s $)&)-<rr?r@rArBrrKrDrCrErrFrGrHrIr rJrLrMrNrOrPrQrRs rYr<zConditionForm.__init__rs $)&)-<N(O  r`rct|}d|jvr |jd=d|jvr |jd=t|ddzdz |d<|dst|d<d|d<d|d <d |d <d |d <|S) Nr9 exceed_daysr~r3 limit_leaveTresetyearly reset_based1 reset_month reset_dayr;cleanerrorsroundLEAVE_MAX_LIMITrS cleaned_datarXs rYrzLeaveTypeForm.cleansw} DKK ' M* DKK ' M*%*< +E+I%JQ%N \"M*)8L &$(L !*2L '*-L '(+L %r`ct||i|}|jjdx}rN|D]H}tj j |}t|||jjJyy)Nr9id) leave_type_idr9available_days) r;savedatagetlistrobjectsgetr r~)rSrTrU leave_type employeesr9employeerXs rYrzLeaveTypeForm.saves}W\4262  ))-8 89 8(  #++//;/?", (#-#8#8$&   9r`c$t||i|yN)r;r<rSrTrUrXs rYr<zLeaveTypeForm.__init__s $)&)r`)rZr[r\rrrallrrr9rrrr<r^r_s@rYrfrfsY)!!%%''6'(+!8   K    **r`rfcFeZdZfdZGddZfdZfdZxZS)UpdateLeaveTypeFormctt| |i|g}|jjj D]W\}}||dk(s|j dr(tjdd|}|j||j|Yt|jD]t\}}t|jj||vs/d|jj j"d<d|jj j"d<v|jj$x} r| |jd<yy) N_idz_id$zwdisplay:none;width:100%; height:50px;border: 1px solid hsl(213deg,22%,84%);border-radius: 0rem;padding: 0.8rem 1.25rem;rdTz data-hiddenr)r;rr<instance__dict__r@endswithresubappend enumeratevisible_fieldslistr?keysrWrArDr) rSrTrU empty_fieldsrV field_valueforeign_key_field_nameindexvisible expire_daterXs rYr<zUpdateLeaveTypeForm.__init__s< !414B6B '+}}'='='C'C'E 0 #J "kR&7&&u--/VVFB -K* ''(>?##J/  0((;(;(=> ANE7DKK$$&'.,>N $$**73=A $$**=9  A--@@ @; @6ADKK2 3 Ar`c`eZdZeZdZgdZedddejddid Z y ) UpdateLeaveTypeForm.Metart)r}r~rurxryrzr|r{r)rxrN) rZr[r\r&rr?rr rrCrrr`rYrrs8:g%OP(7vv>N(O r`rct|}d|jvr |jd=t|ddzdz |d<|dst|d<d|d<d|d <d |d <d |d <|S) Nrcountr3rr~Trrrrrrrrs rYrzUpdateLeaveTypeForm.cleansw} DKK ' M* %l7&;a&? @1 D WM*)8L &$(L !*2L '*-L '(+L %r`c$t||i|}yr)r;r)rSrTrUrrXs rYrzUpdateLeaveTypeForm.save sW\4262 r`)rZr[r\r<rrrr^r_s@rYrrs"B*   33r`rc t||}tjj}t |}t jj}t || |jdk(rA|jdk(r2tt| z t fd|D}||z }|Sd} |jdk(r|D] } | |vs| dz } || z }|jdk(rt fd|D} || z }|S)Nr(c3&K|]}|v ywrr).0requested_date total_leavess rY z/cal_effective_requested_days..s /=Nl * rc3&K|]}|v ywrr)rrcompany_leave_datess rYrz/cal_effective_requested_days..&s&""55&r) rrrrrrrexclude_company_leaveexclude_holidayrsetsum) start_dateend_daterrequested_daysrequested_datesholidays holiday_datescompany_leavestotal_leave_count holiday_countrcompany_leave_countrrs @@rYcal_effective_requested_daysrs'+JAO##%H&x0M"**..0N2>:N++u4  ) )U 2C 0C CDE  AP  (*;;   ( (E 1"1 '!]2!Q&M ',m;N  . .% 7"%&&5&# ,.AAN r`ceZdZejej ddiZejej ddiZfdZfdZ dZ Gdd Z xZ S) LeaveRequestCreationFormr{rr|rAc0t|}|jd}|jd}|jd}|jd}|jd}|jd}|jd}tjj |||j |jj } |jd k(r"| tjdtd i||kstjtd ||k(r#||k7rtjtd tjj ||jstjtd| j ddgjrtjtdtjj||} | j| j z} t#||||} t%|||| } t'||}|Dcgc]#}|j(d|j*d%}}t-j.}t1t3|}|j4d|j(|vr/|j7|j+dd|j(| j9|}| j:j<dvr/| j:j>| kr| j:j>n| } | j:j<dk(r| j r| | j z } | |z } | | kstjtd|Scc}w)Nrrr9rstart_date_breakdownend_date_breakdown attachmentr9start_date__lte end_date__gterr(0An attachment is required for this leave request,End date should not be less than start date.DThere is a mismatch in the breakdown of the start date and end date.r9rEmployee has no leave type.. cancelledrejected status__in:Employee has already a leave request for this date range..rrrr-%m carryforwardzcarryforward expireno carryforward)Employee doesn't have enough leave days..) r;rrr#rfilterrrrrequire_attachmentrrrr existsrcarryforward_daysrrryearstrftimerrFrrmonthremoveforcasted_leavesrcarryforward_typecarryforward_max)rSrrrr9rrrroverlapping_requestsavailable_leavetotal_leave_daysreffective_requested_days leave_datesr month_yearrF unique_dates forcated_daysrXs rYrzLeaveRequestCreationForm.clean2sw} !%%l3 ##J/"&&}5 $((9 +//0FG)--.BC!%%l3 +33::#XZ ; '}}     + +u 4!++$aN'X%''@A   !#'99++^ %%,,#=- &(''*H(IJ J  ' 'K3L ' M T T V''NO )0044#=5   * *_-N-N N 2 "68J $@!') $  ,JA GRSt 1]T]]4%8$9:S S C O, kk]!EJJ< (L 8   5>>$#7"8%** F G'88D ,,>>67-->>AQQ  ) ) : :"   ) ) ; ;?P P11//2S2SS M)'+;;''*U(VW W1Ts=(Nct||i|d|jdjjd<|jdjjj ddddd d |jd jjj d dd d|jdjjj ddddd d y)N.jpg, .jpeg, .png, .pdfracceptrz#leaveRequestCreateForm#availableLeaveCount outerHTMLchange%/leave/employee-available-leave-count hx-include hx-targetzhx-swap hx-triggerhx-getr9#id_leave_type_id_parent_divz=/leave/get-employee-leave-types?form=LeaveRequestCreationFormrrrrr;r<r?rArDrErs rYr<z!LeaveRequestCreationForm.__init__s $)&);T L!((..x8 O$++118873&&A    M"))//66;&Y   L!((..5573&&A   r`c&d|i}td|}|SS Render the form fields as HTML table rows with Bootstrap styling. formhorilla_form.htmlr rSrTrUcontext table_htmls rYas_pzLeaveRequestCreationForm.as_p 4.%&97C r`ceZdZeZgdZy)LeaveRequestCreationForm.Metar9rrrrrr descriptionNrZr[r\r#rr?rr`rYrr$  r`r rZr[r\r DateFieldrCrrrr<r!rr^r_s@rYrr.s^ vv>N(OPJuoeooVV>$#7"8%** F G'88D ,,>>67-->>AQQ  ) ) : :"   ) ) ; ;?P P11//2S2SS M)'+;;''*U(VW W1Ts;(Mct||i||jdjjj dddddd|jdjjj d dd d d |jd jjd<|jdjjj ddddddy)Nrz#leaveRequestUpdateFormz#assinedLeaveAvailableCountrrrrr9rz=/leave/get-employee-leave-types?form=LeaveRequestUpdationFormrr rr rrrs rYr<z!LeaveRequestUpdationForm.__init__s $)&) O$++11887:&&A    M"))//66;&Y  N(OPJuoeooVV   r`r,ceZdZdZej ejjejdZ ej e jjejdZ GddZ y)AvailableLeaveForma Form for managing available leave data. This form allows users to manage available leave data by specifying details such as the leave type and employee. Attributes: - leave_type_id: A ModelChoiceField representing the leave type associated with the available leave. - employee_id: A ModelChoiceField representing the employee associated with the available leave. N)rqrA empty_labelceZdZeZgdZy)AvailableLeaveForm.Meta)rr9ruN)rZr[r\r rr?rr`rYrr6Ts >r`r)rZr[r\r]rModelChoiceFieldr&rrSelectMultiplerrr9rrr`rYr3r3=su +E**""&&(##M )%((!!%%'##K ??r`r3c eZdZdZeej jede ddddZ Gd d Z fd Z xZ S) LeaveOneAssignForma= Form for assigning available leave to employees. This form allows administrators to assign available leave to a single employee by specifying the employee and setting the is_active flag. Attributes: - employee_id: A HorillaMultiSelectField representing the employee to assign leave to. rgrhriTrjrrpceZdZdZeZddgZy)LeaveOneAssignForm.Meta3 Meta class for additional options r9ruNrZr[r\r]r rr?rr`rYrr<ps -r`rcNt||i|t|jyr)r;r<rr?rs rYr<zLeaveOneAssignForm.__init__xs! $)&) $r`)rZr[r\r]rrrrrrr9rr<r^r_s@rYr:r:YsV*!!%%''6'(+!8   K..%%r`r:c$eZdZdZGddZy)AvailableLeaveUpdateForma Form for updating available leave data. This form allows users to update available leave data by modifying fields such as available_days, carryforward_days, and is_active. Attributes: - Meta: Inner class defining metadata options. - model: The model associated with the form (AvailableLeave). - fields: A list of fields to include in the form. ceZdZdZeZgdZy)AvailableLeaveUpdateForm.Metar=)rrruNr>rr`rYrrCs Er`rNrZr[r\r]rrr`rYrArA}s FFr`rAc$eZdZdZGddZy)CompanyLeaveForma Form for managing company leave data. This form allows users to manage company leave data by including all fields from the CompanyLeaves model except for is_active. Attributes: - Meta: Inner class defining metadata options. - model: The model associated with the form (CompanyLeaves). - fields: A special value indicating all fields should be included in the form. - exclude: A list of fields to exclude from the form (is_active). ceZdZdZeZdZdgZy)CompanyLeaveForm.Metar=rtruN)rZr[r\r]rrr?rrr`rYrrHs -r`rNrDrr`rYrFrFs   r`rFc"eZdZejej ddiZejej ddiZeje dejZ fdZ fdZ d ZGd d ZxZS) UserLeaveRequestFormr{rr|r DescriptionrJrAct|}|jd}|jd}|jd}|jd}|jd}|jd}tjj |||j |jj}tjj | } | stjtd |jd k(r3|jd } | tjd td i||k(r#||k7rtjtd||kstjtd|j ddgjrtjtdt||||} tjj||} | j | j"z} t%|||| }|| kstjtd|S)Nrrr9rrrrr)r9z6You dont have enough leave days to update the request.r(rrrrrrrz=Employee has already a leave request for this date range.....rrr)r;rrr#rrrrrr rrrrrrrrr)rSrrrr9rrrrassigned_leave_typesrrrrrrXs rYrzUserLeaveRequestForm.cleans]w} !%%l3 ##J/"&&}5 +//0FG)--.BC$((9 +33::#XZ ; 'T]]%%' &  .55<<<U#''JK   + +u 4%)),7J!++$aN'  !#'99++^ X%''@A  ' 'K3L ' M T T V''QR 2 "68J )0044#=5   * *_-N-N N $@!') $  (+;;''*U(VW Wr`cp|jdd}|jdd}tt||i|d|jdj j d<|r_|jj}tjj|jdd }||jd_ |rutjj|dj |jd_ |dj|jd_d|jd_yy) NrGrr rr rTflatid__inr)popr;rJr<r?rArDrrr&rr values_listrqrrGr4)rSrTrUrravailable_leavesrNrXs rYr<zUserLeaveRequestForm.__init__sZZ 40 ::j$/ "D2DCFC;T L!((..x8 '77;;= #,#4#4#;#;'33O$3O$<$ 5IDKK ( 1 4=4E4E4L4Lo.115M5DKK ( 14>o3N3Q3QDKK ( 07;DKK ( 4 r`c&d|i}td|}|Srr rs rYr!zUserLeaveRequestForm.as_pr"r`cBeZdZdZeZgdZdejiZ y)UserLeaveRequestForm.Metar=r%r9N rZr[r\r]r#rr?rrrrr`rYrrYs/    ,5,,. r`r)rZr[r\rr*rCrr CharFieldrrLr&rr<r!rr^r_s@rYrJrJsx vv>N(OPJuoeooVVN(OPJuoeooVV}t }||_|j |j|j@t||}|r|jj|||fSNrr rrr attachemntr%filerrpkr;addrSrmultiple_files_idsrr file_instancerrXs rYrzLeaveCommentForm.save ::  g &JJ&&w/E',QxDMM $!# # < 0 2 %/ """$"))-*:*:;  < 7<'  HNN   2 3r`. rZr[r\r]rarr<r!boolrrr^r_s@rYrrks;!L.4#r`rc$eZdZdZGddZy)!LeaveallocationrequestcommentFormz. Leave Allocation Requestcomment form ceZdZdZeZdZy)&LeaveallocationrequestcommentForm.Metar=rNrZr[r\r]r"rr?rr`rYrrs .r`rNrDrr`rYrrrr`rcVeZdZdZdZGddZfdZdZd dede ffd Z xZ S) LeaveAllocationCommentFormrrceZdZdZeZdZy)LeaveAllocationCommentForm.Metar=rtNrrr`rYrrs .r`rc~t||i|td|jd<d|jd_y)NrrF)r;r<rr?rors rYr<z#LeaveAllocationCommentForm.__init__s9 $)&)0w? G(- G%r`c&d|i}td|}|Srr rs rYr!zLeaveAllocationCommentForm.as_prr`rrcg}d}|jjdrt|jjd}|d|j_g}|D]>}t }||_|j |j|j@t||}|r|jj|||fSrrrs rYrzLeaveAllocationCommentForm.saverr`rrr_s@rYrrs;!L. 4#r`rceZdZejej ddiZejej ddiZdZGddZ fdZ xZ S) RestrictLeaveFormr{rr|rc|jjd}|jjd}|r|r||krttd|S)Nrrz3End date should not be earlier than the start date.)rrrr)rSrrs rYclean_end_datez RestrictLeaveForm.clean_end_datesW&&**<8 $$((4 (x*'<!GH r`c.eZdZeZdZdgZdediZy)RestrictLeaveForm.MetartrutitleTitleN) rZr[r\r'rr?rrrrr`rYrrs#- QwZ r`rctt| |i|d|jdjj d<|jdjj j dddddy) Nr autocomplete departmentz#leaveRestrictFormz#restrictLeaveJobPositionrz!/leave/get-restrict-job-positions)rrrr)r;rr<r?rArDrErs rYr<zRestrictLeaveForm.__init__sk /@@rrrgrOrPrrrUr?rcrrrrdistinctrqrArDrE) rSrTrUr. instance_idrholiday_attendanceattendance_datesrqrXs rYr<zCompensatoryLeaveForm.__init__sJ ' 7 H H1@@)TRGK}}"mm.. GV,GLL.9"<<44%=k&"$(&2249JK$ 8H O,4/X--44t4D""))7<<)HIhj  3;DKK & / KK & - - 3 3 : :!?"*A r`c&d|i}td|}|S)z[ Render the form fields as HTML table rows with Bootstrap styling. rrr rs rYr!zCompensatoryLeaveForm.as_pAs tnG)*=wGJ r`ct|}|jd}|t|dkr t j dt di|jd}d}d}|jr|jj}|D][}tjj||jt|td zjsYd }n|rt j dd i|S) NrrzThis field is required.r9F)r9rrr)rTz;This attendance is already converted to complimentory leave)r;rrlenrrrrrrrrrr r)rSrrrattendance_repeatrrrXs rYrzCompensatoryLeaveForm.cleanIs 7=?L(,,_=M$M(:Q(>++$a(A&BC$'' 6H % K}}"mm.. +  ,44;;$,J<WQ+.*1EEFVX  )-% !++')f  r`) rZr[r\r]rr<r!rr^r_s@rYrrs'   # J   r`rceZdZdZej edejdddZGdd Z y ) "CompensatoryLeaveRequestRejectForma< Form for rejecting a compensatory leave request. This form allows administrators to provide a rejection reason when rejecting a compensatory leave request. Attributes: - reason: A CharField representing the reason for rejecting the compensatory leave request. rprqrrrsr|rLceZdZeZdgZy)'CompensatoryLeaveRequestRejectForm.MetarvN)rZr[r\rrr?rr`rYrrws,E%&Fr`rNrwrr`rYrrgs@ !&'!5>>=Q(RS   ' 'r`rc$eZdZdZGddZy)#CompensatoryLeaveRequestcommentFormz* LeaverequestComment form ceZdZdZeZdZy)(CompensatoryLeaveRequestcommentForm.MetarrN)rZr[r\r]rrr?rr`rYrrs 4E!Fr`rNrDrr`rYrr{s  " "r`r)_r]mathrrrrtypingrdjangor django.appsrdjango.core.exceptionsrdjango.db.modelsr django.formsr django.forms.widgetsr django.template.loaderr django.utils.translationrr base.methodsrr base.modelsrremployee.filtersremployee.formsremployee.modelsrhorillarhorilla_widgets.formsrr2horilla_widgets.widgets.horilla_multi_select_fieldr&horilla_widgets.widgets.select_widgetsr leave.methodsrrrrr leave.modelsr r!r"r#r$r%r&r'CHOICESrrbrfrrrr,r3r:rArFrJrhFormr`rorzrrrrrrrrr is_installedmodelsrrrrrrr`rYrs"  #2"*36I/+,$'?VK    1U8 tQtWo .::z.EOO.b8*M8*v/3-/3d>K yK \~ y~ B??8!%)!%HFyF, y 0m 9m ` UZZ F##2m 9m `  D#uzz#*%UZZ%P+:k+:\ i -y-`  ++\% % P4\"Qc c J'UZZ'( "i "y#r`