B
bb @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dlm Z d dl m
Z
d d l mZ d d
lmZmZmZ d d lmZmZ d d lmZ d d
lmZ d dlmZ d dl m Z m Z G dd deZ G dd deZ G dd deZ!e"e
e! dS ) N) ZKModelAdmin)
ZKModelAction)zk_site) gettext_lazy)settings)forms)OutdoorEmployeeSchedule)OutdoorSchedulePlanner) ModelFormValidationErrorModelMultipleChoiceField)
DepartmentEmployee)AdminRuntimeWarning)Q)
C_ATT_RULE) STATUS_VALIDSTATUS_RESIGN_PENDINGc s e Zd ZejdddZeej ddZ
ee j ddZ ej
eddZej
eddZejedej dZG d d
d
Zd d Z fd
dZdddZ ZS ) AddOutdoorEmployeeScheduleFormF)initialrequired)querysetr ZscheduleEmp_field_startTime)labelZscheduleEmp_field_endTimeZ!scheduleEmp_field_schedulePlanner)r r c @ s e Zd ZeZdZdS )z#AddOutdoorEmployeeScheduleForm.Meta)dept_emp
departmentschedule_employee
start_dateend_dateschedule_plannerN)__name__
__module__ __qualname__r ZmodelZfields r" r" BG:\easytimepro\master/mysite/att\admin\outdoorempschedule_admin.pyMeta s r$ c C sF | j dd }| j dd }|rB| j dd rBtjj| j d d}|S )NZ
schd_employeer r )Zdepartment__in) cleaned_datagetr objectsfilter)selfvalueZ
is_departmentr" r" r# clean_schd_employee! s
z2AddOutdoorEmployeeScheduleForm.clean_schd_employeec s t t| }|dd }|dd }|dd }|dd }|sRttddd|sfttddd|rn|s~ttd dd||krttd
ddx>|D ]6}tjj|j ||d }|
rttd |j ddqW |S )
Nr r r r Z!scheduledAssign_error_empNotFoundz-1)codeZ&scheduledAssign_error_scheduleNotFoundZ scheduledAssign_error_datePeriodZ%scheduledAssign_error_datePeriodError)employeeZstart_date__lteZ
end_date__gteZatt_outdoor_schedule) superr cleanr& r _r r' r( idexists
first_name)r) r% r r Z employeesr empZoverlaps) __class__r" r# r/ ( s&
z$AddOutdoorEmployeeScheduleForm.cleanTc
K s | j d }| j d }| j d }| j d }g }xj|D ]b}t|||||d d} || |jr|||jk sp||jk r|ttdt|dkr2tj| q2W |rtj| d S ) Nr r r r
created_by)r- r r r Z assigned_byZ&att_outdoor_assign_schedule_validation ) r% r appendZ hire_dater r0 lenr' Z bulk_create)
r) Zcommitkwargsr r r r Zobjsr4 objr" r" r# save> s
z#AddOutdoorEmployeeScheduleForm.save)T)r r r! r Z BooleanFieldr r r
r' get_querysetr r r Z DateFieldr0 r r ZModelChoiceFieldr allr r$ r+ r/ r<
__classcell__r" r" )r5 r# r s
r c @ s0 e Zd ZedZedZedZeZdd Z dS )AddOutdoorEmployeeScheduleZ%clientDetails_action_addClientDetailsZ,clientDetails_action_addClientDetailsHelpTxtZ0clientDetails_action_addClientDetailsDescriptionc O s* t | jj}| r&|j| jjjd d S )N)r6 )r requestZPOSTZis_validr< userZusername)r) argsr: Zformr" r" r# actionX s z!AddOutdoorEmployeeSchedule.actionN)
r r r! r0 Z verbose_nameZhelp_txtshort_descriptionr Z action_formrD r" r" r" r# r@ R s
r@ c s e Zd ZdZdZefZdddZdd Ze de_
d d
Z e d e _
d d
Z e de _
dd Z
e de
_
dd Ze de_
fddZdd Z ZS ) OutdoorEmployeeScheduleAdmin) r1 emp_coder3 last_namer r r mapviewtextview)Zemployee__emp_codeZemployee__first_nameZemployee__last_namer r r Nc C s dS )NFr" )r) rA r; r" r" r# has_change_permissione s z2OutdoorEmployeeScheduleAdmin.has_change_permissionc C s |j jS )N)r- rG )r) r; r" r" r# rG h s z%OutdoorEmployeeScheduleAdmin.emp_codeZemp_field_employeeCodec C s |j jS )N)r- r3 )r) r; r" r" r# r3 m s z'OutdoorEmployeeScheduleAdmin.first_nameZemp_field_firstNamec C s |j jS )N)r- rH )r) r; r" r" r# rH r s z&OutdoorEmployeeScheduleAdmin.last_nameZemp_field_lastNamec C s* d}d dt|j d t| d gS )N ,zk)joinstrr1 )r) r; typer" r" r# rI w s z$OutdoorEmployeeScheduleAdmin.mapviewZoutdoor_map_viewc C s* d}d dt|j d t| d gS )NrL rM zj)rN rO r1 )Zselfselfr; rP r" r" r# rJ s z%OutdoorEmployeeScheduleAdmin.textviewZoutdoor_text_viewc s |j j}tt| |j|d}|j js|j j}|j j}|j j }|
rX|j| d}|
rt|j| d }|
r|j| d}t
t|d}|dkr|ttdtd dB ttdB }|S )N)
company_id)Zemployee__department__in)Zemployee__area__in)Zemployee__position__in
resign_emp0)Zemployee__status)rB Zcurrent_companyr. rF r= r( Z is_superuserZ
get_auth_deptZ
get_auth_areaZ get_auth_posr2 r> ZdistinctrO r r& r r r )r) rA rQ ZqsZ auth_deptZ auth_areaZauth_posrR )r5 r" r# r= s$ z)OutdoorEmployeeScheduleAdmin.get_querysetc
C s ddl m} | j}g }x>| jD ]4}|| jkrH| j| rR|| j| q || q W || _| |}|| _|j r| j}d| _t ||}|| _x|D ]} | d= | d= qW |S g S )Nr )yield_export_results rI rJ )
Zmysite.admin.admin_listrT list_displayZexport_fields_dictr8 Z_setup_changelist result_listcountZempty_value_displaylist)
r) rA rT Ztemp_list_displayZ export_fieldZeachZclZtemp_empty_value_displayrW xr" r" r# get_export_data s*
z,OutdoorEmployeeScheduleAdmin.get_export_data)N)r r r! rV Z list_filterr@ ZactionsrK rG r0 rE r3 rH rI rJ r= r[ r? r" r" )r5 r# rF ^ s
rF )#ZsixZ mysite.adminr r Zmysite.admin.sitesr Zdjango.utils.translationr r0 Z django.confr r Zmysite.att.modelsr r Z django.formsr
r r Zmysite.personnel.modelsr
r Zmysite.admin.exceptionsr Zdjango.db.modelsr Zmysite.att.global_cacher Zmysite.admin.modelsr r r r@ rF registerr" r" r" r# s$ > [