B
    CDb                 @   sl   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	 d dl
mZ d dlmZ G dd	 d	ejeZd
S )    )mixins)PaymentRecord)serializers)CloudViewSet)PaymentRecordFilter)NoneSerializer))get_company_query_without_superuser_extrac                   s@   e Zd ZeZej ZeZ	de
jiZdd Z fddZ  ZS )PaymentRecordViewSetlistc             C   s   | j | jtS )N)serializer_dictgetactionr   )self r   DG:\easytimepro\master/mysite/cloud\api\views\payment_record_views.pyget_serializer_class   s    z)PaymentRecordViewSet.get_serializer_classc                s$   t | j}tt|  j|d}|S )N)Zorder__company__in)r   Zrequestsuperr	   get_querysetfilter)r   Zall_companiesqueryset)	__class__r   r   r      s    
z!PaymentRecordViewSet.get_queryset)__name__
__module____qualname__r   ZmodelZobjectsallr   r   Zfilter_classr   ZPaymentRecordSerializerr   r   r   __classcell__r   r   )r   r   r	      s   

r	   N)Zrest_frameworkr   Zmysite.cloud.modelsr   Zmysite.cloud.apir   Zmysite.cloud.utilsr   Zmysite.cloud.filtersr   Zmysite.cloud.util_serializersr   r   ZListModelMixinr	   r   r   r   r   <module>   s   