U ƽb @sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZmZdd lmZdd lmZdd lmZdd lmZddlmZGdddZGdddZGdddeZGdddeZGdddeZ GdddeZdS)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)parse)settings)StopFutureHandlers) QueryDict) ChunkIter)MultiPartParser)MultiPartParserError parse_header) force_str) renderers) ParseError) api_settings)jsonc@seZdZddZdS) DataAndFilescCs||_||_dS)N)datafiles)selfrrr:/tmp/pip-unpacked-wheel-iohwuy7z/rest_framework/parsers.py__init__szDataAndFiles.__init__N)__name__ __module__ __qualname__rrrrrrsrc@seZdZdZdZdddZdS) BaseParserz All parsers should extend `BaseParser`, specifying a `media_type` attribute, and overriding the `.parse()` method. NcCs tddS)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)rstream media_typeparser_contextrrrr&szBaseParser.parse)NNrrr__doc__rrrrrrrsrc@s*eZdZdZdZejZej Z dddZ dS) JSONParserz& Parses JSON-serialized data. zapplication/jsonNc Cs||pi}|dtj}z.t||}|jr2tjnd}tj||dWSt k rv}zt dt |W5d}~XYnXdS)zX Parses the incoming bytestream as JSON and returns the resulting data. encodingN)parse_constantzJSON parse error - %s) getrDEFAULT_CHARSETcodecs getreaderstrictrZstrict_constantload ValueErrorr str)rrrrr!Zdecoded_streamr"excrrrr7szJSONParser.parse)NN) rrrrrr Z JSONRendererZrenderer_classr Z STRICT_JSONr'rrrrrr /s r c@seZdZdZdZdddZdS) FormParserz Parser for form data. z!application/x-www-form-urlencodedNcCs&|pi}|dtj}t||dS)zt Parses the incoming bytestream as a URL encoded form, and returns the resulting QueryDict. r!)r!)r#rr$rread)rrrrr!rrrrLszFormParser.parse)NNrrrrrr,Fsr,c@seZdZdZdZdddZdS)rzF Parser for multipart form data, which may include file data. zmultipart/form-dataNc Cs|pi}|d}|dtj}|j}||d<|j}z&t||||}|\} } t| | WSt k r} zt dt | W5d} ~ XYnXdS)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_TYPEzMultipart form parse error - %sN) r#rr$METAcopyupload_handlersDjangoMultiPartParserrrrr r*) rrrrr.r!metar2parserrrr+rrrr\s   zMultiPartParser.parse)NNrrrrrrVsrc@s8eZdZdZdZdddZd ddZd d Zd d ZdS)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 Cs|pi}|d}|dtj}|j}|j}||||}|sJt|jd|d|dd} zt|d|dd } Wnt t fk rd } YnX|D]4} | ||| d |} | d k rt id | d iSqd d|D} t dg| }t||}d gt|}t|D]P\}} z| d || | |Wn.tk rZ|d |d }Yq`YnXq|D]R}t|D]B\}} t|}| |||}|||7<|d krpqdqpqdt|D]4\}} | ||}|d k rt id |iSqt|jdd 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!r8ZHTTP_CONTENT_TYPEr/ZHTTP_CONTENT_LENGTHCONTENT_LENGTHrNfilecSsg|]}|jr|jqSr) chunk_size).0xrrr sz*FileUploadParser.parse..ir7)r#rr$r0r2 get_filenamer errorsintr) TypeErrorZhandle_raw_inputrminrlen enumerateZnew_filerZreceive_data_chunkZ file_complete)rrrrr.r!r4r2filename content_typecontent_lengthhandlerresultZpossible_sizesr=chunksZcountersindexchunkZ chunk_lengthZfile_objrrrr}sf       zFileUploadParser.parsec Csz|ddWStk r"YnXzD|dj}t|d}|d}d|kr\||WSt|dWStttfk rYnXdS)z Detects the uploaded file name. First searches a 'filename' url kwarg. Then tries to parse Content-Disposition header. kwargsrHr.ZHTTP_CONTENT_DISPOSITIONr< filename*N)KeyErrorr0r encodeget_encoded_filenamer AttributeErrorr))rrrrr4 disposition filename_parmrrrrAs  zFileUploadParser.get_filenamec CsVt|d}z |dd\}}}t|}Wn$ttfk rPt|d}YnX|S)zw Handle encoded filenames per RFC6266. See also: https://tools.ietf.org/html/rfc2231#section-4 rQ'rH)r splitrunquoter) LookupError)rrWZencoded_filenamecharsetlangrHrrrrTs z%FileUploadParser.get_encoded_filename)NN) rrrrrrBrrArTrrrrr6ss Cr6)rr%urllibrZ django.confrZdjango.core.files.uploadhandlerrZ django.httprZdjango.http.multipartparserrrr3rr Zdjango.utils.encodingr Zrest_frameworkr Zrest_framework.exceptionsr Zrest_framework.settingsr Zrest_framework.utilsrrrr r,r6rrrrs&