File: //usr/lib/python3/dist-packages/django/core/files/__pycache__/uploadhandler.cpython-310.pyc
o
�a. � @ s� d Z ddlZddlmZ ddlmZ ddlmZmZ ddl m
Z
g d�Z G dd � d e �Z
G d
d � d e
�ZG d d
� d
e
�ZG dd� de
�ZG dd� d�ZG dd� de�ZG dd� de�Zdd� ZdS )zH
Base file upload handler classes, and the built-in concrete subclasses
� N)�BytesIO)�settings)�InMemoryUploadedFile�TemporaryUploadedFile��
import_string)�UploadFileException�
StopUpload�SkipFile�FileUploadHandler�TemporaryFileUploadHandler�MemoryFileUploadHandler� load_handler�StopFutureHandlersc @ � e Zd ZdZdS )r z6
Any error having to do with uploading files.
N��__name__�
__module__� __qualname__�__doc__� r r �A/usr/lib/python3/dist-packages/django/core/files/uploadhandler.pyr � r c @ s" e Zd ZdZddd�Zdd� ZdS ) r z=
This exception is raised when an upload must abort.
Fc C s
|| _ 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)�selfr r r r �__init__ s
zStopUpload.__init__c C s | j rdS dS )Nz StopUpload: Halt current upload.z,StopUpload: Consume request data, then halt.r �r r r r �__str__' s zStopUpload.__str__N)F)r r r r r r r r r r r s
r c @ r )r
zX
This exception is raised by an upload handler that wants to skip a given file.
Nr r r r r r
. r r
c @ r )r z�
Upload handlers that have handled a file and do not want future handlers to
run should raise this exception instead of returning None.
Nr r r r r r 5 s r c @ sR e Zd ZdZdZddd�Zddd�Zddd �Zd
d � Zd d
� Z dd� Z
dd� Z dS )r z3
Base class for streaming upload handlers.
i Nc C s( d | _ d | _d | _d | _d | _|| _d S �N)� file_name� content_type�content_length�charset�content_type_extra�request)r r% r r r r C s
zFileUploadHandler.__init__c C � dS )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�encodingr r r �handle_raw_inputK s z"FileUploadHandler.handle_raw_inputc C s( || _ || _|| _|| _|| _|| _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_namer r! r" r# r$ )r r- r r! r" r# r$ r r r �new_file] s
zFileUploadHandler.new_filec C � t d��)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() method��NotImplementedError�r �raw_data�startr r r �receive_data_chunkk s z$FileUploadHandler.receive_data_chunkc C r/ )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() methodr0 �r � file_sizer r r �
file_completer s z FileUploadHandler.file_completec C r& )z�
Signal that the upload is complete. Subclasses should perform cleanup
that is necessary for this handler.
Nr r r r r �upload_complete{ � z!FileUploadHandler.upload_completec C r& )z�
Signal that the upload was interrupted. Subclasses should perform
cleanup that is necessary for this handler.
Nr r r r r �upload_interrupted� r: z$FileUploadHandler.upload_interruptedr )NN) r r r r �
chunk_sizer r, r. r5 r8 r9 r; r r r r r = s
r c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
r zA
Upload handler that streams data into a temporary file.
c s0 t � j|i |�� t| j| jd| j| j�| _dS )zK
Create the file object to append to as data is coming in.
r N)�superr. r r r! r# r$ �file�r �args�kwargs�� __class__r r r. � s z#TemporaryFileUploadHandler.new_filec C s | j �|� d S r )r> �writer2 r r r r5 � s z-TemporaryFileUploadHandler.receive_data_chunkc C s | j �d� || j _| j S )Nr )r> �seek�sizer6 r r r r8 � s z(TemporaryFileUploadHandler.file_completec C sH t | d�r"| j�� }z
| j�� t�|� W d S ty! Y d S w d S )Nr> )�hasattrr> �temporary_file_path�close�os�remove�FileNotFoundError)r �
temp_locationr r r r; � s
��z-TemporaryFileUploadHandler.upload_interrupted) r r r r r. r5 r8 r; �
__classcell__r r rB r r � s r c s: e Zd ZdZd dd�Z� fdd�Zdd� Zd d
� Z� ZS ) r
zS
File upload handler to stream uploads into memory (used for small files).
Nc C s |t jk| _dS )zf
Use the content_length to signal whether or not this handler should be
used.
N)r �FILE_UPLOAD_MAX_MEMORY_SIZE� activatedr' r r r r, � s z(MemoryFileUploadHandler.handle_raw_inputc s* t � j|i |�� | jrt� | _t� �d S r )r= r. rP r r> r r? rB r r r. � s
�z MemoryFileUploadHandler.new_filec C s | j r | j�|� dS |S )z!Add the data to the BytesIO file.N)rP r> rD r2 r r r r5 � s z*MemoryFileUploadHandler.receive_data_chunkc C s8 | j sdS | j�d� t| j| j| j| j|| j| jd�S )z2Return a file object if this handler is activated.Nr )r> r- �namer! rF r# r$ ) rP r> rE r r- r r! r# r$ r6 r r r r8 � s �z%MemoryFileUploadHandler.file_completer ) r r r r r, r. r5 r8 rN r r rB r r
� s
r
c O s t | �|i |��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)
<TemporaryFileUploadHandler object at 0x...>
r )�pathr@ rA r r r r � s
r )r rJ �ior � django.confr � django.core.files.uploadedfiler r �django.utils.module_loadingr �__all__� Exceptionr r r
r r r r
r r r r r �<module> s M ,