
    (iv                     ~    d dl mZmZ d dlmZ d dlmZmZ d dlm	Z	 d dl
mZ d dlmZ  G d dej                        Zy	)
    )viewsetsstatus)Response)DateRangeSerializerDailyCallSummarySerializer)DailyCallsService)Call)Companyc                   L    e Zd Zej                  j                         Zd Zd Zy)DailyCallsViewSetc                    | j                   j                  }|j                  r| j                  S |j                  st
        j                  j                         S 	 t        j                  j                  |j                  j                        }| j                  j                  |      S # t        j                  $ r! t
        j                  j                         cY S w xY w)N)id)company)requestuseris_superuserquerysetactive_companyr	   objectsnoner
   getr   DoesNotExistfilter)selfr   r   s      E/var/www/html/dp2/backend_v2.1/server/apps/calls/views/daily_calls.pyget_querysetzDailyCallsViewSet.get_queryset   s    ||  == ""<<$$&&	'oo))T-@-@-C-C)DG }}##G#44 ## 	'<<$$&&	's   4B* *1CCc                 x   t        |j                        }|j                  d       |j                  d   }|j                  d   }| j	                  | j                               }t        j                  ||||j                        }t        |d      }t        d|j                  it        j                        S )	N)dataT)raise_exception
start_dateend_date)manydaily_calls)r   )r   query_paramsis_validvalidated_datafilter_querysetr   r   get_summaryr   r   r   r   r   HTTP_200_OK)r   r   date_serializerr    r!   qssummary
serializers           r   listzDailyCallsViewSet.list   s    -73G3GH   6$33LA
"11*=!!$"3"3"56#//J',,W/dC

8ASASTT    N)	__name__
__module____qualname__r	   r   allr   r   r.    r/   r   r   r      s    ||!H5 Ur/   r   N)rest_frameworkr   r   rest_framework.responser   apps.calls.serializersr   r   apps.calls.services.daily_callsr   apps.calls.modelsr	   apps.companies.modelsr
   GenericViewSetr   r4   r/   r   <module>r<      s+    + , R = " ) U//  Ur/   