o
    8gp                     @   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	m
Z
mZmZmZ d dlmZ d dlmZmZ ee	G dd	 d	ejZG d
d dejZeddZG dd dejZeeG dd dejZejee eje eje
 eje eje dS )    )admin)path)method_decorator)csrf_exempt)	CustomersCompanyUserCompanyCustomerDocumentTransactions)custom_logout_view)CategoryBatchc                   @      e Zd ZdZdS )CustomersAdmin)qr_codeemail
first_name	last_namephoneN__name__
__module____qualname__list_display r   r   )/var/www/rescue_company/accounts/admin.pyr          r   c                       s$   e Zd Zee fddZ  ZS )CustomAdminSitec                    s"   t   }tdtddg}|| S )Nzlogout/custom_logoutname)superget_urlsr   r   )selfurlscustom_urls	__class__r   r   r"      s   
zCustomAdminSite.get_urls)r   r   r   r   r   r"   __classcell__r   r   r&   r   r      s    r   custom_adminr   c                       s$   e Zd ZdZdZ fddZ  ZS )TransactionsAdmin)transaction_updated_by)companyamountr+   createdupdatedc                    s$   |j s|j|_t |||| d S )N)pkuserr+   r!   
save_model)r#   requestobjformchanger&   r   r   r2   #   s   zTransactionsAdmin.save_model)r   r   r   excluder   r2   r(   r   r   r&   r   r*      s    r*   c                   @   r   )
BatchAdmin)batch_number
batch_namecategorycountcolorbgcolorr.   Nr   r   r   r   r   r8   *   r   r8   N)django.contribr   django.urlsr   django.utils.decoratorsr   django.views.decorators.csrfr   accounts.modelsr   r   r   r	   r
   accounts.views.logoutr   qr_manager.modelsr   r   register
ModelAdminr   	AdminSiter   custom_admin_siter*   r8   siter   r   r   r   <module>   s&    

