B
?bB @ s d dl Z d dlZd dlmZ d dlZd dlmZ d dlmZ d dl m
Z m Z d dl
mZ d dlmZmZ d dlmZ d d lmZmZ d d
lmZ d d lmZmZ d d l m Z m Z d d
l m Z m!Z! d dl"m#Z# d dlmZ d dl$m%Z% G dd dej&Z'G dd deZ(G dd dej)eZ*dS ) N)Decimal)Q) TruncDate) gettext_lazyactivate) FilterSet)mixins serializers)Response) STATUS_VALIDSTATUS_RESIGN_PENDING)NoneSerializer)ReportUtilGenericViewSetReportGenericFilter)short_date_formatshort_time_format) TransactionTerminalWorkCode)get_system_setting)
C_ATT_RULEc @ s e Zd ZejeddddZejeddddZejeddddZejed d
ddZ ejed d ddZ
ejed
dddZ ejeddddZ ejeddddZ
ejeddddZejeddZejeddZejeddZejeddZejeddd Zejed d d Zejed dZejed dZejed!dZejed"dZe Ze Zd#d$ Z d%d& Z d'd( Z d)d* Z d+d, Z d-d. Z!d/d0 Z"d1d2 Z#d3d4 Z$d5d6 Z%d7d8 Z&G d9d: d:Z'd;S )<TransactionReportSerializerZreport_column_empCodez emp.emp_codeT)labelsourceZ
allow_nullZreport_column_firstNamezemp.first_nameZreport_column_lastNamez
emp.last_nameZreport_column_nickNamez emp.nicknameZreport_column_genderzemp.get_gender_displayZ report_column_departmentCodezemp.department.dept_codeZ report_column_departmentNamezemp.department.dept_nameZreport_column_positionCodezemp.position.position_codeZreport_column_positionNamezemp.position.position_nameZreport_column_punchDate)r Zreport_column_punchTimeZreport_column_punchStateZreport_column_workCodeAliasZreport_column_verifyTypeZget_verify_type_display)r r Zreport_column_dataSourcesZget_source_displayZ report_column_displayed_tempZ report_column_displayed_maskZ report_column_high_temp_colorZ report_column_warn_temp_colorc C s" ddl m} |j}|||j}|S )zTD/MD
Display in report the status for mask
mask_flag: user have mask mask or not when punch
mask_enable: enable mask display with color or not in setting
return: 1-green yes; 2-red no; 3-black yes; 4-black yes
r )get_mask_value)mysite.att.utilsr mask_flag
company_id)selfobjr r result r EG:\easytimepro\master/mysite/att\api\views\view_report_transaction.pyget_displayed_maskE s z.TransactionReportSerializer.get_displayed_maskc C s" ddl m} |j}|||j}|S )ziTD/MD
Check the real temp and max/min in setting to judge display high temp color or not
r )check_use_high_temp)r r# temperaturer )r r r# celsiusr r r r! get_check_display_high_tempQ s z7TransactionReportSerializer.get_check_display_high_tempc C s" ddl m} |j}|||j}|S )ziTD/MD
Check the real temp and max/min in setting to judge display warn temp color or not
r )check_use_warn_temp)r r' r$ r )r r r' r% r r r r! get_check_display_warn_tempZ s z7TransactionReportSerializer.get_check_display_warn_tempc C s" ddl m} |j}|||j}|S )z TD/MD
The result depend on temperature unit use fahrenheit or celsius
:param obj: Transaction obj
:return: value of temp for display in report
r )temperature_update)r r) r$ r )r r r) r% r r r r! get_displayed_tempc s z.TransactionReportSerializer.get_displayed_tempc C s& d}t ||jd}|r |d nd}|S )zTD/MD
Get the color for high temp to display in report
:param obj: Transaction obj
:return: color, like '#ff0000'
temp_mask_setting)r high_temp_colorN)r r )r r setting_namer+ tem_unitr r r! get_high_temp_colorn s z/TransactionReportSerializer.get_high_temp_colorc C s& d}t ||jd}|r |d nd}|S )zTD/MD
Get the color for warning temp to display in report
:param obj: Transaction obj
:return: color, like '#ff0000'
r+ )r warn_temp_colorN)r r )r r r- r+ r. r r r! get_warn_temp_colory s z/TransactionReportSerializer.get_warn_temp_colorc C s ddl m} y || jd jjd}W n* || jd jjjjd}Y nX x$| D ]}|d |kr`|d S q`W d S )Nr )r request func_keysvaluename)
mysite.att.global_cacher getcontextusercurrent_companyZ
departmentZcompanyidvalues)r r4 r r3 vr r r! get_func_key s $ z(TransactionReportSerializer.get_func_keyc C s | |jS )N)r> punch_state)r r r r r! get_punch_state s z+TransactionReportSerializer.get_punch_statec C s
t |jS )N)r date)r r r r r! get_att_date s z(TransactionReportSerializer.get_att_datec C s
t |jS )N)r
punch_time)r r r r r! get_punch_time s z*TransactionReportSerializer.get_punch_timec C s0 d}|j r,tjj|j |jd}|r,|d j}|S )N )coder r ) work_coder objectsfilterr alias)r r rJ Z
work_codesr r r! get_work_code_alias s
z/TransactionReportSerializer.get_work_code_aliasc @ s e Zd ZeZdZdS )z TransactionReportSerializer.Meta)r; emp_code
first_name last_name nick_namegender dept_code dept_name
position_code
position_namerG att_datework_code_aliasrC r? verify_typer displayed_tempdisplayed_maskr, r0 check_display_high_tempcheck_display_warn_tempN)__name__
__module__ __qualname__r modelfieldsr r r r! Meta s ra N)(r\ r] r^ r Z CharField_rL rM rN rO rP rQ rR rS rT ZSerializerMethodFieldrU rC r? rV rW r rX rY r, r0 rZ r[ r" r& r( r* r/ r1 r> r@ rB rD rK ra r r r r! r sh
r c @ sZ e Zd Zejr,ejdddZejdddZn ejdddZejdddZG dd dZ dS ) TransactionReportFilterrA Zgte)Z
field_name lookup_exprZlte)r5 rd c @ s e Zd ZeZdddddgZdS )z TransactionReportFilter.MetaZ employeesZ departmentsZareas
start_dateend_dateN)r\ r] r^ r r_ r` r r r r! ra s ra N)
r\ r] r^ sixZPY3django_filtersZ
DateFilterre rf ra r r r r! rc s rc c s e Zd ZeZej dZ e
Z e e dZ
fddZdd Zdd Zd d
Zd d Zd
d Zdd ZdddZdddZ ZS )TransactionReportViewSetz -punch_time)listZexportc
sj d}d}| j s| jrd| j|kr4tt| d}qr| j|krTtt| d}qrtt| }ntt| }| jjjsV| jjj }d|i}|j
f |}| jjj s| jjj }| jjj
}| jjj}| r|j
| d}| r|j
| d }| r|j
| d}tt|d } | d
krV|
ttd td d B ttd B }|jtd d
}|S )N13z -punch_timerL Zemp__company_id)Zemp__department__in)Z
emp__area__in)Zemp__position__in
resign_emp0)Z emp__statusrC )rA )Z page_wiseZgroup_bysuperri get_querysetorder_byr2 r9 Z is_employeer: rI Z is_superuserZ
get_auth_deptZ
get_auth_areaZ get_auth_posexistsallZdistinctstrr r7 r r r Zannotater )
r ZEMP_WISEZ DATE_WISEquerysetr Z base_filterZ auth_deptZ auth_areaZauth_posrm ) __class__r r! rp s:
z%TransactionReportViewSet.get_querysetc C s | j | jtS )N)serializer_dictr7 actionr
)r r r r! get_serializer_class s z-TransactionReportViewSet.get_serializer_classc C s t d tdS )NZenZmenu_att_transactionReport)r rb )r r r r! get_file_title s z'TransactionReportViewSet.get_file_titlec C s | dd |S )Nr zNo.)insert)r headersr r r! update_export_headers s z.TransactionReportViewSet.update_export_headersc C s: x4t |D ](\}}|d|d i |jddd q
W |S )NzNo. F)Zlast) enumerateupdate move_to_end)r datasidatar r r! update_datas_before_export s z3TransactionReportViewSet.update_datas_before_exportc C s0 |r,| d | d | d | d |S )Nr, r0 rZ r[ )append)r export_headersgroup_by_fieldr r r! update_header_before_export s
z4TransactionReportViewSet.update_header_before_exportc C s0 |dkr| dd |d= |d= |d= |d= |S )NrQ r zNo.)r{ )r r| r r r r! update_group_export_headers s z4TransactionReportViewSet.update_group_export_headersNc C s xt |D ]\}}y| jd}W n d}Y nX |rPd|| d ||< y| jd}W n d}Y nX |r|| r|| d }|| d } || d }
|| d } |
r|rd
| d t|| d ||< n,| r| rd
| d t|| d ||< y| jd } W n d} Y nX | rh|| rh|| }
|
d
krVd|| < n|
dkrhd|| < t|tr|d|d q t|}|dd |||< q W |S )z"only use for att report pdf reportrU NzzrX r, r0 rZ r[ zrY Yesz(YesNoz'Nor r~ rE )r r indexrt
isinstancerj r{ )r r r
extra_datar r att_date_indexdisplayed_temp_indexr, r0 rZ r[ displayed_mask_indexstatusr r r! update_data_list sH
"
z)TransactionReportViewSet.update_data_listc C s x|D ]}xt |D ]\}}y| jd}W n d}Y nX |rht|trhd|| d ||< y| jd}W n d}Y nX |r<|| r<| jd} | jd}
| jd} | jd } || r|| rd
|| d t|| d ||< n8|| r<||
rzrX r, r0 rZ r[ zrY r z(Yesr z'Nor rQ r r~ rE )r r r r rj rt r{ )r r r r r r r= r r Zhigh_temp_color_indexZwarn_temp_color_indexZ check_display_high_temp_indexZ check_display_warn_temp_indexr r r r r! update_group_data_list4 sT
&$
z/TransactionReportViewSet.update_group_data_list)N)N)r\ r] r^ r r_ rH rp Zselect_relatedrq ru rc Z filter_classr rw ry rz r} r r r r r
__classcell__r r )rv r! ri s "
.ri )+ collectionsrg Zdecimalr rh Zdjango.db.modelsr Zdjango.db.models.functionsr Zdjango.utils.translationr rb r Z django_filters.rest_frameworkr Zrest_frameworkr r Zrest_framework.responser
Zmysite.admin.modelsr r Zmysite.att.api.serializersr
Zmysite.att.api.utils_classr r r r r Zmysite.iclock.modelsr r Z mysite.utilsr r6 r ZModelSerializerr rc ZListModelMixinri r r r r! s*