o Bh=@sVddlmZddlmZddlmZddlmZmZddlm Z Gdddej Z d S) )forms)ValidationError)timezone)datetime timedelta)OvertimeRequestcseZdZfddZejddZejejddddd d Z ejejddddd d Z ej ej dd d ddddZ GdddZfddZdfdd ZZS)OvertimeRequestFormcs|dd|_tj|i||jr|j|j_t}|jdd}|j dkr5|j|j dddd}n |j|j ddd}t j dd| | d d |jd_dS) Nemployeerday yearmonthr rr date form-control)typeclassminmaxattrs)popr super__init__instancernowrreplacerrr DateInput isoformatfieldswidget)selfargskwargstodaycurrent_month_firstprev_month_first __class__%/var/www/html/wtlms/overtime/forms.pyrs     zOvertimeRequestForm.__init__z0Select a date from the current or previous month) help_texttimer)rrrz/Enter the time you started work (after 5:00 PM))r#r.zEEnter the time you finished work (can be next day for overnight work))rrows)Provide a reason for the overtime requestF)r#r.requiredc@s(eZdZeZgdZddddddZdS)zOvertimeRequestForm.Meta)r start_timeend_timereasonsupporting_documentzSelect the date of overtimezEnter the time you started workz Enter the time you finished workr2z1Upload any supporting documentation (PDF, images)N)__name__ __module__ __qualname__rmodelr" help_textsr,r,r,r-Meta4s r=c st}|d}|d}|d}|jstdt|||gs$|St}||kr4tddi|j dd}|j |j krG|j |j krGn0||krq|j dkr\|j |j dddd }n |j |j ddd }||krptdd intddi||krtdd id dl m }|||jj} | r|S|dkr|jdkrtdditjj|j|gddj|jr|jjndd} | rtd|S)Nrr4r5zEmployee is requiredz(Cannot request overtime for future datesrr r rrzFOvertime can only be requested for the current month or previous monthz)End time cannot be the same as start timer)Holidayz5Start time must be after regular work hours (5:00 PM))PENDING HR_APPROVED REG_APPROVED)r r status__in)pkz9You already have an active overtime request for this date)rcleangetr rallrrrrrrovertime.modelsr> is_holidaycompanyweekdayhourrobjectsfilterexcluderrEexists) r$ cleaned_datarr4r5 current_dater(r)r>rJ overlappingr*r,r-rF?sR               zOvertimeRequestForm.cleanTcs,tjdd}|jr|j|_|r||S)NF)commit)rsaver )r$rUrr*r,r-rVs zOvertimeRequestForm.save)T)r8r9r:rr DateFieldr TimeField TimeInputr4r5 CharFieldTextarear6r=rFrV __classcell__r,r,r*r-r s*  Mr N) djangordjango.core.exceptionsr django.utilsrrrmodelsr ModelFormr r,r,r,r-s