U
    Vc                     @   s   G d d dZ dS )c                   @   s$   e Zd Zdd Zdd Zdd ZdS )SerializerValidationMiddlewarec                 C   s   || _ || _d S N)get_responseserializer_class)selfr   r    r   //var/www/wikiazma_server/wikiazma/middleware.py__init__   s    z'SerializerValidationMiddleware.__init__c                 C   s   |  |}|S r   )r   )r   requestresponser   r   r   __call__	   s    
z'SerializerValidationMiddleware.__call__c                 C   s@   |j dkr| j|jd}n| j|jd}|jddr<|j|_d S )NGET)dataT)raise_exception)methodr   r   r   is_validvalidated_datamiddleware_serializer_data)r   r	   	view_func	view_argsview_kwargs
serializerr   r   r   process_view   s
    
z+SerializerValidationMiddleware.process_viewN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s   r   N)r   r   r   r   r   <module>       