B
    Yc                 @   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
 G dd deZG dd	 d	eZeeG d
d dejZdS )    )gettext)	ModelForm)admin)LeaveCategory)settingsc               @   s   e Zd ZG dd dZdS )LeaveCategoryCreationFormc               @   s(   e Zd ZeZdgZejr$eddg7 ZdS )zLeaveCategoryCreationForm.MetacompanyprefixsuffixN)__name__
__module____qualname__r   modelexcluder   CLOUD_VERSION r   r   =G:\easytimepro\master/mysite/att\admin\leavecategory_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 )LeaveCategoryChangeFormc               @   s(   e Zd ZeZdgZejr$eddg7 ZdS )zLeaveCategoryChangeForm.Metar   r	   r
   N)r   r   r   r   r   r   r   r   r   r   r   r   r      s   r   c                s>   t t| j|| t| dd }|r:|jr:d| jd jjd< d S )NinstanceFcategory_namereadonly)superr   __init__getattrZpkZfieldsZwidgetZattrs)selfargskwargsr   )	__class__r   r   r      s    
z LeaveCategoryChangeForm.__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 )LeaveCategoryAdmin)idr   report_symbolr	   r
   )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LeaveCategoryAdmin.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.utils.translationr   _Zdjango.formsr   mysiter   Zmysite.att.modelsr   Zdjango.confr   r   r   registerZZKModelAdminr    r   r   r   r   <module>   s   	