o
    Eg                     @   sP   d dl mZ d dlmZmZ d dlmZ G dd dejZG dd dejZ	dS )	    )serializers)	CustomersCustomerDocument)datec                   @   s   e Zd ZG dd dZdS )CustomerDocumentSerializerc                   @   s   e Zd ZeZdZdS )zCustomerDocumentSerializer.Meta__all__N)__name__
__module____qualname__r   modelfields r   r   F/var/www/rescue_company/customer_api/serializers/CustomerSerializer.pyMeta   s    r   N)r   r	   r
   r   r   r   r   r   r      s    r   c                   @   s:   e Zd Ze Ze ZG dd dZdd Zdd Z	dS )CustomerSerializerc                   @   s   e Zd ZeZg dZdS )zCustomerSerializer.Meta)	is_active
is_deletedbatchupdatedcreatedgroupsuser_permissionspasswordN)r   r	   r
   r   r   excluder   r   r   r   r      s    r   c                 C   s<   |j rt }|j|j j |j|jf|j j|j jfk  S d S )N)dobr   todayyearmonthday)selfobjr   r   r   r   get_age   s   *zCustomerSerializer.get_agec              
   C   sZ   zt jj|d}t|dd}|j}W |S  ty, } zt| d }W Y d }~|S d }~ww )N)customerT)many)r   objectsfilterr   data	Exceptionprint)r   r    documents_datadocument_serializer	documentser   r   r   get_documents!   s   z CustomerSerializer.get_documentsN)
r   r	   r
   r   SerializerMethodFieldager+   r   r!   r-   r   r   r   r   r      s    
r   N)
rest_frameworkr   accounts.modelsr   r   datetimer   ModelSerializerr   r   r   r   r   r   <module>   s
    