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	 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)
ParallelDb)forms)ZKPasswordInputc               @   s&   e Zd ZejedZG dd dZdS )ParallelDbCreationForm)widgetc               @   s   e Zd ZeZdZdS )zParallelDbCreationForm.Meta)companyN)__name__
__module____qualname__r   modelexclude r   r   <G:\easytimepro\master/mysite/base\admin\parallel_db_admin.pyMeta   s   r   N)r
   r   r   r   	CharFieldr   passwordr   r   r   r   r   r   	   s   r   c                   s6   e Zd ZejedZG dd dZ fddZ  Z	S )ParallelDbChangeForm)r   c               @   s   e Zd ZeZdZdS )zParallelDbChangeForm.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ParallelDbChangeForm.__init__)
r
   r   r   r   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 )ParallelDbAdmin)ZdatabaseZ
ip_addressZportZdatabase_nameuserr   Zservice_typeZduplicate_punch)r   c                s    t t| |j|jjd}|S )N)Z
company_id)r   r   get_querysetfilterr   Zcurrent_company)r   ZrequestZqs)r   r   r   r   %   s    zParallelDbAdmin.get_queryset)r
   r   r   Zlist_displayZhidden_fieldsr   Zadd_formr   Zformr   r   r   r   )r   r   r      s
   r   N)Zdjango.formsr   mysiter   Zmysite.base.modelsr   Zmysite.adminr   Zmysite.admin.forms.widgetsr   r   r   registerZZKModelAdminr   r   r   r   r   <module>   s   