B
    ba                 @   sP   d dl mZ d dlmZ d dlmZ d dlmZ e	eG dd dej
ZdS )    )admin)gettext_lazy)forms)AttShiftc                   sD   e Zd ZdZdZejZdd Ze	de_
dd Z fdd	Z  ZS )
AttShiftAdmin)idaliasshift_timetableZ
cycle_unitZshift_cycleZ
auto_shift)r   c             C   sB   |j jddd}g }xt|D ]}|r|| qW dt|S )NZtime_interval__aliasT)Zflat,)Zshiftdetail_setZvalues_listlistappendjoinset)selfobjZtbsZtemp_tbstb r   8G:\easytimepro\master/mysite/att\admin\attshift_admin.pyr	      s    zAttShiftAdmin.shift_timetableZshift_field_timeIntervalc             C   s   ddl m} ||S )Nr   )shift_detail)Zmysite.att.viewsr   )r   requestr   r   r   r   r      s    zAttShiftAdmin.shift_detailc                s    t t| |j|jjd}|S )N)Z
company_id)superr   get_querysetfilteruserZcurrent_company)r   r   Z	query_set)	__class__r   r   r   (   s    zAttShiftAdmin.get_queryset)__name__
__module____qualname__Zlist_displayZlist_filterr   ZAttShiftCreationFormZformr	   _Zshort_descriptionr   r   __classcell__r   r   )r   r   r      s   

r   N)mysiter   Zdjango.utils.translationr   r   Z
mysite.attr   Zmysite.att.modelsr   registerZZKModelAdminr   r   r   r   r   <module>   s
   