B
    GDb                 @   sd   d dl mZ d dl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j	Z
d
S )    )	ModelForm)admin)AllowanceTypec               @   s   e Zd ZG dd dZdS )AllowanceTypeCreationFormc               @   s   e Zd ZeZdZdS )zAllowanceTypeCreationForm.Meta)companyN)__name__
__module____qualname__r   modelexclude r   r   BG:\easytimepro\master/mysite/payroll\admin\allowance_type_admin.pyMeta	   s   r   N)r   r   r	   r   r   r   r   r   r      s   r   c                   s*   e Zd ZG dd dZ fddZ  ZS )AllowanceTypeChangeFormc               @   s   e Zd ZeZdZdS )zAllowanceTypeChangeForm.Meta)r   N)r   r   r	   r   r
   r   r   r   r   r   r      s   r   c                s   t t| j|| d S )N)superr   __init__)selfargskwargs)	__class__r   r   r      s    z AllowanceTypeChangeForm.__init__)r   r   r	   r   r   __classcell__r   r   )r   r   r      s   r   c                   s,   e Zd ZdZdZeZeZ fddZ	  Z
S )AllowanceTypeAdmin)idallowance_codeallowance_name)r   r   c                s    t t| |j|jjd}|S )N)Z
company_id)r   r   get_querysetfilteruserZcurrent_company)r   ZrequestZqs)r   r   r   r      s    zAllowanceTypeAdmin.get_queryset)r   r   r	   Zlist_displayZlist_filterr   Zadd_formr   Zformr   r   r   r   )r   r   r      s
   r   N)Zdjango.formsr   mysiter   Zmysite.payroll.modelsr   r   r   registerZZKModelAdminr   r   r   r   r   <module>   s   
