3

�P\Z�@s�dZd
dl
Z
d
dlmZ
dZdddd	�Ze
jd
ddg�ZGd
d
�d
e�Zdd�Z	dd�Z
dd�Zddd�
Zdd�Z
dd�Zd dd�
Zd!dd�
ZdS)"zGThis module provides functions for dumping information about responses.�N)
�compat�
dump_response�dump_alls0.9s1.0s1.1)�	�
��PrefixSettings�request�responsecseZ
dZ�f
d
d�Z�ZS)rcs$t|
�
}
t|�
}t
t|�j||
|�S)
N)�_coerce_to_bytes�superr�__new__)�clsr	r
)
�	__class__��/usr/lib/python3.6/dump.pyr
s


zPrefixSettings.__new__)�__name__�
__module__�__qualname__r
�
__classcell__rr)
rrrs
c
Cs<t|j
d
d�r8i}
|jj}|jd�
r,d|
d<||
d<|
SdS)NZ
proxy_managerFzhttps://ZCONNECT�method�request_path)�getattrZ
connectionr	�url�
startswith)r
�
proxy_infoZrequest_urlrrr�_get_proxy_informations







rcCst|�
d
t|
�
dS)Ns: s
)
r)�name�valuerrr�_format_header&s

rcCsVtj
|�
}|
jd
�
}|dk	r,t|�
}||fSt|j�
}|jrN|dt|j�
7}||fS)Nr�
?)rZurlparse�getr�pathZquery)rr�uriZ	proxy_urlrrrr�_build_request_path+s









r$cCs�|dkri}|
j}t
|jd
|j��
}t|j|�\}}|j||d|d�

|jj�}t
|jd|j	��
}	|j|d|	d�

x(|j
�D]\}
}|j|t|
|��

q�W|j|d�

|jr�t
|jtj�r�|j|t
|j�
�

n
|jd�

|jd�

dS)Nr�
 s HTTP/1.1
ZHostsHost: s
s,<< Request body is not a string-like type >>)r	r�poprr$r�extend�headers�copyZnetloc�itemsrZbody�
isinstancerZ
basestring)r	�prefixes�bytearrr�prefixrrr#r(Zhost_headerrrrrr�_dump_request_data9s"












r/c	Cs�|
j}|j
}tj|jd
�}|j|d|dt|j�
jd�
dt	|j
�
d�

|j}x8|j�D],}x&|j
|�
D]}|j|t||��

qrWqbW|j|d�

|j|j�

dS)Nr sHTTP/r%�asciis
)r
�raw�
HTTP_VERSIONSr!�versionr'�strZstatus�encoder�reasonr(�keysZgetlistrZcontent)	r
r,r-r.r1Zversion_strr(rrrrr�_dump_response_dataWs
4


r8c

Cs0t|t
�r t|d
�r |jd�
}|dk	r,|SdS)Nr5zutf-8�)r+�bytes�hasattrr5)
�datarrrrns


r�< �> cCsX|d
k	r|nt�}t
|
|�}t|d�s.td�
�
t|�
}t|j|||d�
t|||�
|S)aDump a single request-response cycle's information.

    This will take a response object and dump only the data that requests can
    see for that single request-response cycle.

    Example::

        import requests
        from requests_toolbelt.utils import dump

        resp = requests.get('https://api.github.com/users/sigmavirus24')
        data = dump.dump_response(resp)
        print(data.decode('utf-8'))

    :param response:
        The response to format
    :type response: :class:`requests.Response`
    :param request_prefix: (*optional*)
        Bytes to prefix each line of the request data
    :type request_prefix: :class:`bytes`
    :param response_prefix: (*optional*)
        Bytes to prefix each line of the response data
    :type response_prefix: :class:`bytes`
    :param data_array: (*optional*)
        Bytearray to which we append the request-response cycle data
    :type data_array: :class:`bytearray`
    :returns: Formatted bytes of request and response information.
    :rtype: :class:`bytearray`
    Nr	z"Response has no associated request)
r)�	bytearrayrr;�
ValueErrorrr/r	r8)r
�request_prefix�response_prefixZ
data_arrayr<r,rrrrrus








cCsBt�}t
|jd
d
��
}|j|�

x|D]}t||
||�
q(W|S)a�Dump all requests and responses including redirects.

    This takes the response returned by requests and will dump all
    request-response pairs in the redirect history in order followed by the
    final request-response.

    Example::

        import requests
        from requests_toolbelt.utils import dump

        resp = requests.get('https://httpbin.org/redirect/5')
        data = dump.dump_all(resp)
        print(data.decode('utf-8'))

    :param response:
        The response to format
    :type response: :class:`requests.Response`
    :param request_prefix: (*optional*)
        Bytes to prefix each line of the request data
    :type request_prefix: :class:`bytes`
    :param response_prefix: (*optional*)
        Bytes to prefix each line of the response data
    :type response_prefix: :class:`bytes`
    :returns: Formatted bytes of request and response information.
    :rtype: :class:`bytearray`
    N)r?�list�history�appendr)r
rArBr<rDrrrr�s



)rr)
N)r=r>N)r=r>)�__doc__�collectionsZrequestsr�__all__r2�
namedtupleZ_PrefixSettingsrrrr$r/r8rrrrrrr�<module>
s$







+