vÅÏ^Gã@sdZddlmZddlmZddlmZmZddlm Z dddd d d d d gZ Gdd„de ƒZ Gdd„de ƒZ Gdd„de ƒZGdd „d e ƒZGdd „d ƒZGdd „d eƒZGdd „d eƒZdd „ZdS)zH Base file upload handler classes, and the built-in concrete subclasses é)ÚBytesIO)Úsettings)ÚInMemoryUploadedFileÚTemporaryUploadedFile)Ú import_stringÚUploadFileExceptionÚ StopUploadÚSkipFileÚFileUploadHandlerÚTemporaryFileUploadHandlerÚMemoryFileUploadHandlerÚ load_handlerÚStopFutureHandlersc@seZdZdZdS)rz6 Any error having to do with uploading files. N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©rrúA/tmp/pip-build-8lau8j11/django/django/core/files/uploadhandler.pyrs c@s1eZdZdZddd„Zdd„ZdS)rz= 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. N)Úconnection_reset)ÚselfrrrrÚ__init__szStopUpload.__init__cCs|jr dSdSdS)Nz StopUpload: Halt current upload.z,StopUpload: Consume request data, then halt.)r)rrrrÚ__str__'s zStopUpload.__str__N)rrrrrrrrrrrs c@seZdZdZdS)r zX This exception is raised by an upload handler that wants to skip a given file. N)rrrrrrrrr .s c@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. N)rrrrrrrrr5s c@speZdZdZdZddd„Zddd „Zddd d „Zd d „Zdd„Z dd„Z dS)r z3 Base class for streaming upload handlers. é@éé NcCs:d|_d|_d|_d|_d|_||_dS)N)Ú file_nameÚ content_typeÚcontent_lengthÚcharsetÚcontent_type_extraÚrequest)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 '--'. Nr)rÚ input_dataÚMETArÚboundaryÚencodingrrrÚhandle_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_namerrrrr )rr'rrrrr rrrÚnew_file]s      zFileUploadHandler.new_filecCstdƒ‚dS)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() methodN)ÚNotImplementedError)rÚraw_dataÚstartrrrÚreceive_data_chunkksz$FileUploadHandler.receive_data_chunkcCstdƒ‚dS)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() methodN)r))rÚ file_sizerrrÚ file_completerszFileUploadHandler.file_completecCsdS)zƒ Signal that the upload is complete. Subclasses should perform cleanup that is necessary for this handler. Nr)rrrrÚupload_complete{sz!FileUploadHandler.upload_completeii) rrrrÚ chunk_sizerr&r(r,r.r/rrrrr =s   cs@eZdZdZ‡fdd†Zdd„Zdd„Z‡S)r zA Upload handler that streams data into a temporary file. cs>tƒj||Žt|j|jd|j|jƒ|_dS)zK Create the file object to append to as data is coming in. rN)Úsuperr(rrrrr Úfile)rÚargsÚkwargs)Ú __class__rrr(‡sz#TemporaryFileUploadHandler.new_filecCs|jj|ƒdS)N)r2Úwrite)rr*r+rrrr,Žsz-TemporaryFileUploadHandler.receive_data_chunkcCs#|jjdƒ||j_|jS)Nr)r2ÚseekÚsize)rr-rrrr.‘s z(TemporaryFileUploadHandler.file_complete)rrrrr(r,r.rr)r5rr ƒs  csOeZdZdZddd„Z‡fdd†Zdd„Zd d „Z‡S) 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Ú activated)rr"r#rr$r%rrrr&œsz(MemoryFileUploadHandler.handle_raw_inputcs5tƒj||Ž|jr1tƒ|_tƒ‚dS)N)r1r(r9rr2r)rr3r4)r5rrr(¥s  z MemoryFileUploadHandler.new_filecCs$|jr|jj|ƒn|SdS)z!Add the data to the BytesIO file.N)r9r2r6)rr*r+rrrr,«s z*MemoryFileUploadHandler.receive_data_chunkcCs`|js dS|jjdƒtd|jd|jd|jd|jd|d|jd |jƒS) z2Return a file object if this handler is activated.Nrr2r'Únamerr8rr ) r9r2r7rr'rrrr )rr-rrrr.²s      z%MemoryFileUploadHandler.file_complete)rrrrr&r(r,r.rr)r5rr —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)Úpathr3r4rrrr Ãs N)rÚiorZ django.confrZdjango.core.files.uploadedfilerrZdjango.utils.module_loadingrÚ__all__Ú Exceptionrrr rr r r r rrrrÚs   F,