U
    ڲg%                     @   s   d Z ddlZddl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 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ZG dd dZG dd deZG dd deZG dd deZG dd deZdS )z
Parsers are used to parse the content of incoming HTTP requests.

They give us a generic way of being able to handle various media types
on the request, such as form content or json encoded data.
    N)settings)StopFutureHandlers)	QueryDict)	ChunkIter)MultiPartParser)MultiPartParserError)parse_header_parameters)	renderers)
ParseError)api_settings)jsonc                   @   s   e Zd Zdd ZdS )DataAndFilesc                 C   s   || _ || _d S )N)datafiles)selfr   r    r   :/tmp/pip-unpacked-wheel-11h17kvo/rest_framework/parsers.py__init__   s    zDataAndFiles.__init__N)__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @   s   e Zd ZdZdZdddZdS )
BaseParserz
    All parsers should extend `BaseParser`, specifying a `media_type`
    attribute, and overriding the `.parse()` method.
    Nc                 C   s   t ddS )z
        Given a stream to read from, return the parsed representation.
        Should return parsed data, or a `DataAndFiles` object consisting of the
        parsed data and files.
        z.parse() must be overridden.N)NotImplementedError)r   stream
media_typeparser_contextr   r   r   parse'   s    zBaseParser.parse)NNr   r   r   __doc__r   r   r   r   r   r   r       s   r   c                   @   s*   e Zd ZdZdZejZej	Z
dddZdS )
JSONParserz&
    Parses JSON-serialized data.
    zapplication/jsonNc              
   C   s|   |pi }| dtj}z.t||}| jr2tjnd}tj||dW S  t	k
rv } zt
dt| W 5 d}~X Y nX dS )zX
        Parses the incoming bytestream as JSON and returns the resulting data.
        encodingN)parse_constantzJSON parse error - %s)getr   DEFAULT_CHARSETcodecs	getreaderstrictr   Zstrict_constantload
ValueErrorr
   str)r   r   r   r   r    Zdecoded_streamr!   excr   r   r   r   8   s    zJSONParser.parse)NN)r   r   r   r   r   r	   ZJSONRendererZrenderer_classr   ZSTRICT_JSONr&   r   r   r   r   r   r   0   s
   r   c                   @   s   e Zd ZdZdZdddZdS )
FormParserz
    Parser for form data.
    z!application/x-www-form-urlencodedNc                 C   s&   |pi }| dtj}t| |dS )zt
        Parses the incoming bytestream as a URL encoded form,
        and returns the resulting QueryDict.
        r    )r    )r"   r   r#   r   read)r   r   r   r   r    r   r   r   r   M   s    zFormParser.parse)NNr   r   r   r   r   r+   G   s   r+   c                   @   s   e Zd ZdZdZdddZdS )r   zF
    Parser for multipart form data, which may include file data.
    zmultipart/form-dataNc              
   C   s   |pi }|d }| dtj}|j }||d< |j}z&t||||}| \}	}
t|	|
W S  t	k
r } zt
dt| W 5 d}~X Y nX dS )a	  
        Parses the incoming bytestream as a multipart encoded form,
        and returns a DataAndFiles object.

        `.data` will be a `QueryDict` containing all the form parameters.
        `.files` will be a `QueryDict` containing all the form files.
        requestr    CONTENT_TYPEzMultipart form parse error - %sN)r"   r   r#   METAcopyupload_handlersDjangoMultiPartParserr   r   r   r
   r)   )r   r   r   r   r-   r    metar1   parserr   r   r*   r   r   r   r   ]   s    
zMultiPartParser.parse)NNr   r   r   r   r   r   W   s   r   c                   @   s0   e Zd ZdZdZdddZdddZd	d
 ZdS )FileUploadParserz&
    Parser for file upload data.
    z*/*zFFileUpload parse error - none of upload handlers can handle the streamz`Missing filename. Request should include a Content-Disposition header with a filename parameter.)	unhandledno_filenameNc              	   C   s  |pi }|d }| dtj}|j}|j}| |||}|sJt| jd | d| dd}	zt| d| dd	}
W n t	t
fk
r   d
}
Y nX |D ]4}||||
d
|}|d
k	rti d|d i  S qdd |D }tdg| }t||}d	gt| }t|D ]P\}}z|d
||	|
| W n. tk
rZ   |d
|d  }Y  q`Y nX q|D ]R}t|D ]B\}}t|}|||| }||  |7  < |d
krp qdqpqdt|D ]4\}}||| }|d
k	rti d|i  S qt| jd d
S )a  
        Treats the incoming bytestream as a raw file upload and returns
        a `DataAndFiles` object.

        `.data` will be None (we expect request body to be a file content).
        `.files` will be a `QueryDict` containing one 'file' element.
        r-   r    r7   ZHTTP_CONTENT_TYPEr.    ZHTTP_CONTENT_LENGTHCONTENT_LENGTHr   Nfile   c                 S   s   g | ]}|j r|j qS r   )
chunk_size).0xr   r   r   
<listcomp>   s      z*FileUploadParser.parse.<locals>.<listcomp>ir6   )r"   r   r#   r/   r1   get_filenamer
   errorsintr(   	TypeErrorZhandle_raw_inputr   minr   len	enumerateZnew_filer   Zreceive_data_chunkZfile_complete)r   r   r   r   r-   r    r3   r1   filenamecontent_typecontent_lengthhandlerresultZpossible_sizesr<   chunksZcountersindexchunkZchunk_lengthZfile_objr   r   r   r   ~   sf    





 

zFileUploadParser.parsec              
   C   s   t t |d d W  5 Q R  S Q R X t tttL |d j}t|d \}}d|krp|d W  5 Q R  S |d W  5 Q R  S Q R X dS )z
        Detects the uploaded file name. First searches a 'filename' url kwarg.
        Then tries to parse Content-Disposition header.
        kwargsrG   r-   ZHTTP_CONTENT_DISPOSITIONz	filename*N)
contextlibsuppressKeyErrorAttributeErrorr(   r/   r   )r   r   r   r   r3   dispositionparamsr   r   r   r@      s    
zFileUploadParser.get_filename)NN)r   r   r   r   r   rA   r   r@   r   r   r   r   r5   t   s   
Cr5   )r   r$   rP   Zdjango.confr   Zdjango.core.files.uploadhandlerr   Zdjango.httpr   Zdjango.http.multipartparserr   r   r2   r   Zdjango.utils.httpr   Zrest_frameworkr	   Zrest_framework.exceptionsr
   Zrest_framework.settingsr   Zrest_framework.utilsr   r   r   r   r+   r5   r   r   r   r   <module>   s&   