U 5i.@sdZddlZddlmZddlmZddlmZmZddl m Z ddd d d d d dgZ Gddde Z Gddde ZGdd d e ZGddde ZGdd d ZGdd d eZGdd d eZdd ZdS)zH Base file upload handler classes, and the built-in concrete subclasses N)BytesIO)settings)InMemoryUploadedFileTemporaryUploadedFile import_stringUploadFileException StopUploadSkipFileFileUploadHandlerTemporaryFileUploadHandlerMemoryFileUploadHandler load_handlerStopFutureHandlersc@seZdZdZdS)rz6 Any error having to do with uploading files. N__name__ __module__ __qualname____doc__rrC/tmp/pip-unpacked-wheel-gcyiijx6/django/core/files/uploadhandler.pyrsc@s"eZdZdZdddZddZdS) r z= This exception is raised when an upload must abort. FcCs ||_dS)z If ``connection_reset`` is ``True``, Django knows will halt the upload without consuming the rest of the upload. This will cause the browser to show a "connection reset" error. Nconnection_reset)selfrrrr__init__szStopUpload.__init__cCs|jr dSdSdS)Nz StopUpload: Halt current upload.z,StopUpload: Consume request data, then halt.rrrrr__str__'szStopUpload.__str__N)F)rrrrrrrrrrr s c@seZdZdZdS)r zX This exception is raised by an upload handler that wants to skip a given file. Nrrrrrr .sc@seZdZdZdS)rz Upload handlers that have handled a file and do not want future handlers to run should raise this exception instead of returning None. Nrrrrrr5sc@sReZdZdZdZdddZdddZddd Zd d Zd d Z ddZ ddZ dS)r z3 Base class for streaming upload handlers. iNcCs(d|_d|_d|_d|_d|_||_dSN) file_name content_typecontent_lengthcharsetcontent_type_extrarequest)rr#rrrrCs zFileUploadHandler.__init__cCsdS)a Handle the raw input from the client. Parameters: :input_data: An object that supports reading via .read(). :META: ``request.META``. :content_length: The (integer) value of the Content-Length header from the client. :boundary: The boundary from the Content-Type header. Be sure to prepend two '--'. NrrZ input_dataZMETAr boundaryencodingrrrhandle_raw_inputKsz"FileUploadHandler.handle_raw_inputcCs(||_||_||_||_||_||_dS)z Signal that a new file has been started. Warning: As with any data from the client, you should not trust content_length (and sometimes won't even get it). N) field_namerrr r!r")rr(rrr r!r"rrrnew_file]s zFileUploadHandler.new_filecCs tddS)z{ Receive data from the streamed upload parser. ``start`` is the position in the file of the chunk. zJsubclasses of FileUploadHandler must provide a receive_data_chunk() methodNNotImplementedErrorrraw_datastartrrrreceive_data_chunkksz$FileUploadHandler.receive_data_chunkcCs tddS)z Signal that a file has completed. File size corresponds to the actual size accumulated by all the chunks. Subclasses should return a valid ``UploadedFile`` object. zEsubclasses of FileUploadHandler must provide a file_complete() methodNr*r file_sizerrr file_completerszFileUploadHandler.file_completecCsdS)z Signal that the upload is complete. Subclasses should perform cleanup that is necessary for this handler. Nrrrrrupload_complete{sz!FileUploadHandler.upload_completecCsdS)z Signal that the upload was interrupted. Subclasses should perform cleanup that is necessary for this handler. Nrrrrrupload_interruptedsz$FileUploadHandler.upload_interrupted)N)N)NN) rrrr chunk_sizerr'r)r/r2r3r4rrrrr =s    cs8eZdZdZfddZddZddZdd ZZS) r zA Upload handler that streams data into a temporary file. cs,tj||t|j|jd|j|j|_dS)zK Create the file object to append to as data is coming in. rN)superr)rrrr!r"filerargskwargs __class__rrr)sz#TemporaryFileUploadHandler.new_filecCs|j|dSr)r7writer,rrrr/sz-TemporaryFileUploadHandler.receive_data_chunkcCs|jd||j_|jS)Nr)r7seeksizer0rrrr2s z(TemporaryFileUploadHandler.file_completecCsFt|drB|j}z|jt|Wntk r@YnXdS)Nr7)hasattrr7Ztemporary_file_pathcloseosremoveFileNotFoundError)rZ temp_locationrrrr4s   z-TemporaryFileUploadHandler.upload_interrupted) rrrrr)r/r2r4 __classcell__rrr;rr s  cs:eZdZdZd ddZfddZddZd d ZZS) r zS File upload handler to stream uploads into memory (used for small files). NcCs|tjk|_dS)zf Use the content_length to signal whether or not this handler should be used. N)rZFILE_UPLOAD_MAX_MEMORY_SIZE activatedr$rrrr'sz(MemoryFileUploadHandler.handle_raw_inputcs&tj|||jr"t|_tdSr)r6r)rFrr7rr8r;rrr)sz MemoryFileUploadHandler.new_filecCs|jr|j|n|SdS)z!Add the data to the BytesIO file.N)rFr7r=r,rrrr/sz*MemoryFileUploadHandler.receive_data_chunkc Cs8|js dS|jdt|j|j|j|j||j|jdS)z2Return a file object if this handler is activated.Nr)r7r(namerr?r!r") rFr7r>rr(rrr!r"r0rrrr2s z%MemoryFileUploadHandler.file_complete)N) rrrrr'r)r/r2rErrr;rr s  cOst|||S)a= Given a path to a handler, return an instance of that handler. E.g.:: >>> from django.http import HttpRequest >>> request = HttpRequest() >>> load_handler('django.core.files.uploadhandler.TemporaryFileUploadHandler', request) r)pathr9r:rrrrs )rrBiorZ django.confrZdjango.core.files.uploadedfilerrZdjango.utils.module_loadingr__all__ Exceptionrr r rr r r rrrrrs,   M,