B
    b                 @   s   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z
 d dlmZmZ d dlmZ G dd	 d	eZG d
d deZeeG dd deZdS )    )admin)LeaveFormula)ZKModelAdmin)LeaveCategory)gettext_lazy)ModelChoiceField	ModelForm)get_current_company_idc                   s:   e Zd Zeej dZ fddZG dd dZ	  Z
S )LeaveFormulaChangeForm)querysetc                s2   t t| j|| t }tjj|d| jd _d S )N)
company_idcategory)	superr
   __init__r	   r   objectsfilterfieldsr   )selfargskwargsr   )	__class__ AG:\easytimepro\master/mysite/payroll\admin\leave_formula_admin.pyr      s    zLeaveFormulaChangeForm.__init__c               @   s   e Zd ZeZdZdS )zLeaveFormulaChangeForm.Meta__all__N)__name__
__module____qualname__r   modelr   r   r   r   r   Meta   s   r   )r   r   r   r   r   r   get_querysetr   r   r   __classcell__r   r   )r   r   r
      s   r
   c                   s:   e Zd Zeej dZ fddZG dd dZ	  Z
S )AddLeaveFormulaForm)r   c                s2   t t| j|| tjj| jjjd| j	d _
d S )N)r   r   )r   r!   r   r   r   r   requestusercurrent_companyr   r   )r   r   r   )r   r   r   r      s    zAddLeaveFormulaForm.__init__c               @   s   e Zd ZeZdZdS )zAddLeaveFormulaForm.Metar   N)r   r   r   r   r   r   r   r   r   r   r   !   s   r   )r   r   r   r   r   r   r   r   r   r   r    r   r   )r   r   r!      s   r!   c                   s8   e Zd Zg ZdZdZdZdZeZ	e
Z fddZ  ZS )LeaveFormulaAdmin)idnamer   formularemark)r'   r   r(   r)   r   c                s    t t| |j|jjd}|S )N)Zcategory__company_id)r   r%   r   r   r#   r$   )r   r"   Zqs)r   r   r   r   0   s    zLeaveFormulaAdmin.get_queryset)r   r   r   ZactionsZlist_displayZlist_filterZsort_fieldsZ
cache_keysr!   Zadd_formr
   Zformr   r    r   r   )r   r   r%   &   s   r%   N)mysiter   Zmysite.payroll.modelsr   Zmysite.admin.kernelr   Z%mysite.att.models.model_leavecategoryr   Zdjango.utils.translationr   _Zdjango.formsr   r   Zmysite.att.utilsr	   r
   r!   registerr%   r   r   r   r   <module>   s   