B
    b	                 @   sd   d dl mZmZ d dlmZ d dlmZmZ d dlm	Z	 d dl
mZ eeG dd dejZdS )	    )adminconfig)TempSchedule)TEMP_SCHEDULE_RULE_FLAGTEMP_SCHEDULE_WORK_TYPE)AddTempSchedule)gettext_lazyc                   sr   e Zd ZdZejd ZdddZdd Zdd	 Z	d
d Z
dd Zede_d fdd	Z fddZegZ  ZS )TempScheduleAdmin)idget_employeeZtime_interval
start_timeend_time	work_type	rule_flag)r   r   Nc             C   s   dS )NF )selfrequestobjr   r   <G:\easytimepro\master/mysite/att\admin\tempschedule_admin.pyhas_change_permission   s    z'TempScheduleAdmin.has_change_permissionc             C   s   dS )NFr   )r   r   r   r   r   has_add_permission   s    z$TempScheduleAdmin.has_add_permissionc             C   s   t t}||j S )N)dictr   r   )r   r   choice_dictr   r   r   r      s    zTempScheduleAdmin.work_typec             C   s   t t}||j S )N)r   r   r   )r   r   r   r   r   r   r      s    zTempScheduleAdmin.rule_flagc             C   sb   |j }|rZg }|jr$|t|j |jr:|t|j |jrP|t|j d|S dS d S )N  )ZemployeeZemp_codeappendstrZ
first_name	last_namejoin)r   r   ZempZ	name_listr   r   r   r   "   s    
zTempScheduleAdmin.get_employeer   c                s6   i }|dkr| j |d< || tt| j||f|S )z7
        Use special form during user creation
        NZform)Zadd_formupdatesuperr	   get_form)r   r   r   kwargsdefaults)	__class__r   r   r!   7   s
    

zTempScheduleAdmin.get_formc                sH   |j j}tt| |j|j jd}|j jsD| rD|j|	 d}|S )N)Zemployee__company_id)Zemployee__department__in)
userZget_auth_deptr    r	   get_querysetfilterZcurrent_companyZis_superuserexistsall)r   r   Z	auth_deptZqs)r$   r   r   r&   A   s    zTempScheduleAdmin.get_queryset)N)N)__name__
__module____qualname__Zlist_displayr   ZEMPLOYEE_LIST_FILTERZlist_filterr   r   r   r   r   _Zshort_descriptionr!   r&   r   Zactions__classcell__r   r   )r$   r   r	      s   



	r	   N)mysiter   r   Zmysite.att.modelsr   Zmysite.att.models_choicesr   r   Zmysite.att.actionsr   Zdjango.utils.translationr   r-   registerZZKModelAdminr	   r   r   r   r   <module>   s   