3
�P\Z � @ s� d Z ddlZddlmZ d Zdddd �Zejd
d d g�ZG d
d
� d
e�Zdd� Z dd� Z
dd� Z d dd�Z dd� Z
dd� Zd d d�Zd!d d�ZdS )"zGThis module provides functions for dumping information about responses.� N)�compat�
dump_response�dump_alls 0.9s 1.0s 1.1)� �
� �PrefixSettings�request�responsec s e Zd Z� fdd�Z� ZS )r c s$ t |�}t |�}tt| �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__r r )r r r s c C s< t | jdd�r8i }| jj}|jd�r,d|d< ||d< |S d S )NZ
proxy_managerFzhttps://ZCONNECT�method� request_path)�getattrZ
connectionr �url�
startswith)r
�
proxy_infoZ request_urlr r r �_get_proxy_information s
r c C s t | �d t |� d S )Ns : s
)r )�name�valuer r r �_format_header& s r c C sV t j| �}|jd�}|d k r,t|�}||fS t|j�}|jrN|dt|j� 7 }||fS )Nr � ?)r Zurlparse�getr �pathZquery)r r �uriZ proxy_urlr r r r �_build_request_path+ s
r$ c C s� |d kr 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� d S )Nr � s HTTP/1.1
ZHosts Host: s
s, << Request body is not a string-like type >>)r r �popr r$ r �extend�headers�copyZnetloc�itemsr Zbody�
isinstancer Z
basestring) r �prefixes�bytearrr �prefixr r r# r( Z host_headerr r r r r �_dump_request_data9 s"
r/ c C s� |j }| j}tj|jd�}|j|d | d t|j�jd� d t | j
� d � |j }x8|j � D ],}x&|j
|�D ]}|j|t||� � qrW qbW |j|d � |j| j� d S )Nr s HTTP/r% �asciis
)r
�raw�
HTTP_VERSIONSr! �versionr' �strZstatus�encoder �reasonr( �keysZgetlistr Zcontent) r
r, r- r. r1 Z version_strr( r r r r r �_dump_response_dataW s 4 r8 c C s0 t | t� r t| d�r | jd�} | d k r,| S dS )Nr5 zutf-8� )r+ �bytes�hasattrr5 )�datar r r r n s
r � < � > c C sX |dk r |nt � }t||�}t| d�s.td��t| �}t| j|||d� t| ||� |S )a Dump 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 )� bytearrayr r; �
ValueErrorr r/ r r8 )r
�request_prefix�response_prefixZ
data_arrayr< r, r r r r r u s
c C sB t � }t| jdd� �}|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
rA rB r< rD r r r r � s
)r r )N)r= r> N)r= r> )�__doc__� collectionsZrequestsr �__all__r2 �
namedtupleZ_PrefixSettingsr r r r$ r/ r8 r r r r r r r �<module> s$
+