File: //lib/python3.6/site-packages/botocore/__pycache__/awsrequest.cpython-36.pyc
3
�T_�Z � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl Z
d dlmZ d dlmZmZmZmZmZmZ d dlmZ eje�ZG d d
� d
e�ZG d d � d e �Z G d
d� de e�Z G dd� de e�Z G dd� de
�Z G dd� de �Z!d'dd�Z"dd� Z#dd� Z$G dd � d e �Z%G d d � d e �Z&G d d � d e �Z'G d!d"� d"e �Z(G d#d$� d$e �Z)G d%d&� d&e�Z*dS )(� N)�VerifiedHTTPSConnection)�HTTPConnection)�HTTPConnectionPool)�HTTPSConnectionPool)�six)� HTTPHeaders� HTTPResponse�
urlunsplit�urlsplit� urlencode�MutableMapping)�UnseekableStreamErrorc @ s e Zd Zdd� Zdd� ZdS )�AWSHTTPResponsec O s" |j d�| _tj| f|�|� d S )N� status_tuple)�pop�
_status_tupler �__init__)�self�args�kwargs� r � /usr/lib/python3.6/awsrequest.pyr & s zAWSHTTPResponse.__init__c C s( | j d k r| j }d | _ |S tj| �S d S )N)r r � _read_status)r r r r r r * s
z AWSHTTPResponse._read_statusN)�__name__�
__module__� __qualname__r r r r r r r # s r c sv e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z � fdd�Z dd� Z
� ZS )�
AWSConnectiona� Mixin for HTTPConnection that supports Expect 100-continue.
This when mixed with a subclass of httplib.HTTPConnection (though
technically we subclass from urllib3, which subclasses
httplib.HTTPConnection) and we only override this class to support Expect
100-continue, which we need for S3. As far as I can tell, this is
general purpose enough to not be specific to S3, but I'm being
tentative and keeping it in botocore because I've only tested
this against AWS services.
c s* t t| �j||� | j| _d| _d| _d S )NF)�superr r �response_class�_original_response_cls�_response_received�_expect_header_set)r r r )� __class__r r r ? s zAWSConnection.__init__c s& t t| �j� d| _d| _| j| _d S )NF)r r �closer r! r r )r )r" r r r# M s zAWSConnection.closec sT d| _ |jdd�dkr d| _nd| _| j| _tt| �j||||f|�|�}d| _|S )NFZExpect� s 100-continueT)r �getr! r r r r �
_send_request)r �method�url�body�headersr r Zrval)r" r r r& T s
zAWSConnection._send_requestc C sH g }x4|D ],}t |tj�r,|j|jd�� q
|j|� q
W dj|�}|S )Nzutf-8s
)�
isinstancer � text_type�append�encode�join)r Z mixed_bufferZ bytes_buffer�chunk�msgr r r �_convert_to_bytes` s
z AWSConnection._convert_to_bytesNc O s� | j jd� | j| j �}| j d d �= t|t�r:||7 }d }| j|� | jr|tjd� t j
j | j d�rr| j
|� d S tjd� |d k r�| j|� d S )Nr$ z"Waiting for 100 Continue response.� zCNo response seen from server, continuing to send the response body.)r$ r$ )Z_buffer�extendr2 r+ �bytes�sendr! �logger�debug�urllib3�utilZ
wait_for_read�sock�_handle_expect_response)r � message_bodyr r r1 r r r � _send_outputm s
zAWSConnection._send_outputc C s d }x|dkr|j � }qW d S )Ns
)�readline)r �fpZcurrentr r r �_consume_headers� s
z AWSConnection._consume_headersc
C s� | j jdd�}z�|j� }|jd d�}| j|�rN| j|� tjd� | j|� ndt |�dkr�|d j
d�r�tjd� |d j d�t |d �|d j d�f}t
jt|d
�}|| _d | _W d |j� X d S ) N�rbr � z5100 Continue response seen, now sending request body.� s HTTP/zOReceived a non 100 Continue response from the server, NOT sending request body.�asciir3 )r T)r; �makefiler? �split�_is_100_continue_statusrA r7 r8 �_send_message_body�len�
startswith�decode�int� functools�partialr r r r# )r r= r@ �maybe_status_line�partsr r r r r r< � s"
z%AWSConnection._handle_expect_responsec C s |d k r| j |� d S )N)r6 )r r= r r r rI � s z AWSConnection._send_message_bodyc s$ | j rtjd� d S tt| �j|�S )Nz@send() called, but reseponse already received. Not sending data.)r r7 r8 r r r6 )r �str)r" r r r6 � s
zAWSConnection.sendc C s2 |j d d�}t|�dko0|d jd�o0|d dkS )NrC rD r s HTTP/r3 s 100)rG rJ rK )r rP rQ r r r rH � s z%AWSConnection._is_100_continue_status)N)r r r �__doc__r r# r& r2 r> rA r< rI r6 rH �
__classcell__r r )r" r r 3 s
$ % r c @ s e Zd ZdZdS )�AWSHTTPConnectionz8 An HTTPConnection that supports 100 Continue behavior. N)r r r rS r r r r rU � s rU c @ s e Zd ZdZdS )�AWSHTTPSConnectionz9 An HTTPSConnection that supports 100 Continue behavior. N)r r r rS r r r r rV � s rV c @ s e Zd ZeZdS )�AWSHTTPConnectionPoolN)r r r rU �
ConnectionClsr r r r rW � s rW c @ s e Zd ZeZdS )�AWSHTTPSConnectionPoolN)r r r rV rX r r r r rY � s rY c
C s� | }|dk r |d }||d< |j d�}t||d |�}|d rttjj}||d �} d|krh|d| 7 }n |d | 7 }||d
< ||d < |dkr�i |d < dS ) a;
This method prepares a request dict to be created into an
AWSRequestObject. This prepares the request dict by adding the
url and the user agent to the request dict.
:type request_dict: dict
:param request_dict: The request dict (created from the
``serialize`` module).
:type user_agent: string
:param user_agent: The user agent to use for this request.
:type endpoint_url: string
:param endpoint_url: The full endpoint url, which contains at least
the scheme, the hostname, and optionally any path components.
Nr* z
User-Agent� host_prefix�url_pathZ query_string�?z?%sz&%sr( �context)r% �_urljoin�botocore�utils�percent_encode_sequence)
� request_dict� endpoint_urlr] Z
user_agent�rr* rZ r( ra Zencoded_query_stringr r r �prepare_request_dict� s
re c C s2 | }t |d |d |d |d d�}|d |_|S )aK
This method takes a request dict and creates an AWSRequest object
from it.
:type request_dict: dict
:param request_dict: The request dict (created from the
``prepare_request_dict`` method).
:rtype: ``botocore.awsrequest.AWSRequest``
:return: An AWSRequest object based on the request_dict.
r' r( r) r* )r' r( �datar* r] )�
AWSRequestr] )rb rd Zrequest_objectr r r �create_request_object
s
rh c C s� t | �}| s|dkr.|d s$d}qh|d }n:|d jd�r\|jd�r\|d d d� | }n |d | }|d }|d k r�|| }t|d |||d |d f�}|S )N�/rC r3 r rD � ���)r
�endswithrK r )rc r[ rZ �p�new_pathZ
new_netlocZ
reconstructedr r r r^ ! s
r^ c @ sB e Zd ZdZdd� Zdd� Zddd�Zd d
� Zd d � Zd
d� Z dS )�AWSRequestPreparera
This class performs preparation on AWSRequest objects similar to that of
the PreparedRequest class does in the requests library. However, the logic
has been boiled down to meet the specific use cases in botocore. Of note
there are the following differences:
This class does not heavily prepare the URL. Requests performed many
validations and corrections to ensure the URL is properly formatted.
Botocore either performs these validations elsewhere or otherwise
consistently provides well formatted URLs.
This class does not heavily prepare the body. Body preperation is
simple and supports only the cases that we document: bytes and
file-like objects to determine the content-length. This will also
additionally prepare a body that is a dict to be url encoded params
string as some signers rely on this. Finally, this class does not
support multipart file uploads.
This class does not prepare the method, auth or cookies.
c C s< |j }| j|�}| j|�}| j||�}|j}t|||||�S )N)r' � _prepare_url�
_prepare_body�_prepare_headers�
stream_output�AWSPreparedRequest)r �originalr' r( r) r* rs r r r �prepareQ s
zAWSRequestPreparer.preparec C s2 |j }|jr.tt|jj� �dd�}d||f }|S )NT)�doseqz%s?%s)r( �paramsr �list�items)r ru r( rx r r r rp Z s
z AWSRequestPreparer._prepare_urlNc C sl t |jj� �}d|ks d|kr"|S |jdkrh| j|�}|d k rLt|�|d< n t|�}tjd|� d|d< |S ) NzTransfer-EncodingzContent-Length�GET�HEAD�OPTIONSz Failed to determine length of %sZchunked)r{ r| r} ) � HeadersDictr* rz r' �_determine_content_lengthrR �typer7 r8 )r ru Z
prepared_bodyr* ZlengthZ body_typer r r rr a s
z#AWSRequestPreparer._prepare_headersc C s< |\}}t |tj�r |jd�}t |tj�r4|jd�}||fS )Nzutf-8)r+ r r, r. )r �item�key�valuer r r �_to_utf8v s
zAWSRequestPreparer._to_utf8c sB |j }|dkrd}t|t�r>� fdd�|j� D �}t|dd�}|S )z"Prepares the given HTTP body data.r$ Nc s g | ]}� j |��qS r )r� )�.0r� )r r r �
<listcomp>� s z4AWSRequestPreparer._prepare_body.<locals>.<listcomp>T)rw )rf r+ �dictrz r )r ru r) rx r )r r rq ~ s
z AWSRequestPreparer._prepare_bodyc C s~ |sdS yt |�S ttfk
r6 } zW Y d d }~X nX t|d�rzt|d�rz|j� }|jdd� |j� }|j|� || S d S )Nr �seek�tellrC )rJ �AttributeError� TypeError�hasattrr� r� )r r) �eZorig_posZ end_file_posr r r r � s
z,AWSRequestPreparer._determine_content_length)N)
r r r rS rv rp rr r� rq r r r r r ro = s
ro c @ s2 e Zd ZdZeZd
dd�Zdd� Zedd � �Z dS ) rg a
Represents the elements of an HTTP request.
This class was originally inspired by requests.models.Request, but has been
boiled down to meet the specific use cases in botocore. That being said this
class (even in requests) is effectively a named-tuple.
NFc
C sx | j � | _|d kri n|}|| _|| _t� | _|| _|| _|| _|| _ |d k rnx |j
� D ]\}} | | j|< qXW i | _ d S )N) �_REQUEST_PREPARER_CLS�_request_preparerr' r( r r* rf rx � auth_pathrs rz r] )
r r' r( r* rf rx r� rs r� r� r r r r � s
zAWSRequest.__init__c C s | j j| �S )z>Constructs a :class:`AWSPreparedRequest <AWSPreparedRequest>`.)r� rv )r r r r rv � s zAWSRequest.preparec C s$ | j � j}t|tj�r |jd�}|S )Nzutf-8)rv r) r+ r r, r. )r r) r r r r) � s
zAWSRequest.body)NNNNNNF)
r r r rS ro r� r rv �propertyr) r r r r rg � s
rg c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) rt a� A data class representing a finalized request to be sent over the wire.
Requests at this stage should be treated as final, and the properties of
the request should not be modified.
:ivar method: The HTTP Method
:ivar url: The full url
:ivar headers: The HTTP headers to send.
:ivar body: The HTTP body.
:ivar stream_output: If the response for this request should be streamed.
c C s" || _ || _|| _|| _|| _d S )N)r' r( r* r) rs )r r' r( r* r) rs r r r r � s
zAWSPreparedRequest.__init__c C s d}|| j | j| j| jf S )NzD<AWSPreparedRequest stream_output=%s, method=%s, url=%s, headers=%s>)rs r' r( r* )r Zfmtr r r �__repr__� s zAWSPreparedRequest.__repr__c C s� t jt jtf}| jdks$t| j|�r(dS y tjd| j� | jjd� W n: t k
r� } z tjd|� t
| jd��W Y dd}~X nX dS )a+ Resets the streaming body to it's initial position.
If the request contains a streaming body (a streamable file-like object)
seek to the object's initial position to ensure the entire contents of
the object is sent. This is a no-op for static bytes-like body types.
NzRewinding stream: %sr zUnable to rewind stream: %s)Z
stream_object) r Z binary_typer, � bytearrayr) r+ r7 r8 r� � Exceptionr
)r Znon_seekable_typesr� r r r � reset_stream� s
z AWSPreparedRequest.reset_streamN)r r r rS r r� r� r r r r rt � s rt c @ s0 e Zd ZdZdd� Zedd� �Zedd� �ZdS ) � AWSResponsea� A data class representing an HTTP response.
This class was originally inspired by requests.models.Response, but has
been boiled down to meet the specific use cases in botocore. This has
effectively been reduced to a named tuple.
:ivar url: The full url.
:ivar status_code: The status code of the HTTP response.
:ivar headers: The HTTP headers received.
:ivar body: The HTTP response body.
c C s& || _ || _t|�| _|| _d | _d S )N)r( � status_coder~ r* �raw�_content)r r( r� r* r� r r r r s
zAWSResponse.__init__c C s* | j dkr$t� j| jj� �p t� | _ | j S )z!Content of the response as bytes.N)r� r5 r/ r� �stream)r r r r �content! s
zAWSResponse.contentc C s. t jj| j�}|r | jj|�S | jjd�S dS )a
Content of the response as a proper text type.
Uses the encoding type provided in the reponse headers to decode the
response content into a proper text type. If the encoding is not
present in the headers, UTF-8 is used as a default.
zutf-8N)r_ r` Zget_encoding_from_headersr* r� rL )r �encodingr r r �text. s zAWSResponse.textN)r r r rS r r� r� r� r r r r r� s
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
_HeaderKeyc C s || _ |j� | _d S )N)�_key�lower�_lower)r r� r r r r > s z_HeaderKey.__init__c C s
t | j�S )N)�hashr� )r r r r �__hash__B s z_HeaderKey.__hash__c C s t |t�o| j|jkS )N)r+ r� r� )r �otherr r r �__eq__E s z_HeaderKey.__eq__c C s | j S )N)r� )r r r r �__str__H s z_HeaderKey.__str__c C s
t | j�S )N)�reprr� )r r r r r� K s z_HeaderKey.__repr__N)r r r r r� r� r� r� r r r r r� = s
r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )r~ z:A case-insenseitive dictionary to represent HTTP headers. c O s i | _ | j||� d S )N)�_dict�update)r r r r r r r Q s zHeadersDict.__init__c C s || j t|�< d S )N)r� r� )r r� r� r r r � __setitem__U s zHeadersDict.__setitem__c C s | j t|� S )N)r� r� )r r� r r r � __getitem__X s zHeadersDict.__getitem__c C s | j t|�= d S )N)r� r� )r r� r r r � __delitem__[ s zHeadersDict.__delitem__c C s dd� | j D �S )Nc s s | ]}t |�V qd S )N)rR )r� r� r r r � <genexpr>_ s z'HeadersDict.__iter__.<locals>.<genexpr>)r� )r r r r �__iter__^ s zHeadersDict.__iter__c C s
t | j�S )N)rJ r� )r r r r �__len__a s zHeadersDict.__len__c C s
t | j�S )N)r� r� )r r r r r� d s zHeadersDict.__repr__c C s t | j� �S )N)r~ rz )r r r r �copyg s zHeadersDict.copyN) r r r rS r r� r� r� r� r� r� r� r r r r r~ O s r~ )NN)+�sysZloggingrN ZsocketZ urllib3.utilr9 Zurllib3.connectionr r Zurllib3.connectionpoolr r Zbotocore.utilsr_ Zbotocore.compatr r r r r
r r Zbotocore.exceptionsr
Z getLoggerr r7 r �objectr rU rV rW rY re rh r^ ro rg rt r� r� r~ r r r r �<module> s<
#
' d921