3
�ga �
@ s� d dl Z d dlmZmZmZmZmZ d dlmZ d dl Z d dl
m Z d dl m
Z
ddlmZmZ G dd � d �ZG d
d � d �Zdeeeee
ef eegef eeeeee
ef d
�dd�Zdeeeee
ef d�dd�Ze jed�dd�ZdS )� N)�Callable�IO�Iterable�Tuple�Union)� urlencode)� super_len)�MultipartEncoder� )�MultipartRequestDataDict�RequestDataDictc @ s6 e Zd Zeed�dd�Zeeeef d�dd�Z dS )�ChunkedUploadStream)�stream�callbackc C s || _ || _d S )N)r r )�selfr r � r � /usr/lib/python3.6/uploads.py�__init__
s z ChunkedUploadStream.__init__)�returnc c s$ x | j D ]}| j|� |V qW d S )N)r r )r �chunkr r r �__iter__ s
z ChunkedUploadStream.__iter__N)
�__name__�
__module__� __qualname__r r r r �str�bytesr r r r r r
s r
c @ s8 e Zd Zd
Zed�dd�Zeeee f d�dd�Z
d S ) � ChunkedMultipartUploadStream�d i )�encoderc C s
|| _ d S )N)r )r r r r r r s z%ChunkedMultipartUploadStream.__init__)r c c s$ x | j j| j�}|sP |V qW d S )N)r �read�
chunk_size)r r r r r r s
z%ChunkedMultipartUploadStream.__iter__Ni � ) r r r r r r r r r r r r r r r r s r F)�body�body_read_callback�content_length_header_valuer c s� t | d�}t| t�r t| dd�} |r4|r0| j� S | S |sV|r�tdd� | gD �� d�} n`t| �sv|d kr�| r�| j� } n| j�� �fdd�}|| _|r�t| t�r�t| d �} n t| � d�} | S )
Nr T)Zdoseqc s s | ]}|j � V qd S )N)�encode)�.0r r r r � <genexpr>; s z'prepare_request_body.<locals>.<genexpr>)r r c s �| � }� |� |S )Nr )�argsr )r" � orig_readr r �new_readP s z&prepare_request_body.<locals>.new_read)r ) �hasattr�
isinstancer r r r
r r r )r! r" r# ZchunkedZofflineZ is_file_liker) r )r" r( r �prepare_request_body% s4
r, )�data�boundary� content_typer c C sH t | j� |d�}|r6|j� }d|kr<|� d|j� �}n|j}|} | |fS )N)Zfieldsr. z boundary=z ; boundary=)r �items�stripZboundary_valuer/ )r- r. r/ r r r r �#get_multipart_data_and_content_typee s r2 )�request�alwaysc C s� t j� }t| jt�r | jj� }n t| jd�r8| jj� }n| j}|j|�}||j � 7 }t
|�t
|�k }|sl|r�|| _d| j d< tt
|��| j d< d S )Nr ZdeflatezContent-EncodingzContent-Length) �zlibZ compressobjr+ r! r r$ r* r �compress�flush�lenZheaders)r3 r4 ZdeflaterZ
body_bytesZ
deflated_dataZ
is_economicalr r r �compress_requesty s
r9 )NFF)NN)r5 Ztypingr r r r r Z urllib.parser ZrequestsZrequests.utilsr Zrequests_toolbeltr Z cli.dictsr r r
r r r �intr, r2 ZPreparedRequest�boolr9 r r r r �<module> s" <