3
P\Z @ s d Z ddlZddlmZ d Zdddd Zejd
d d gZG d
d
d
eZdd Z dd Z
dd Z d ddZ dd Z
dd Zd d dZd!d dZdS )"zGThis module provides functions for dumping information about responses. N)compat
dump_responsedump_alls 0.9s 1.0s 1.1)
PrefixSettingsrequestresponsec s e Zd Z fddZ ZS )r c s$ t |}t |}tt| j| ||S )N)_coerce_to_bytessuperr __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 | jddr8i }| jj}|jdr,d|d< ||d< |S d S )NZ
proxy_managerFzhttps://ZCONNECTmethod 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 )namevaluer 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 Zurlparsegetr 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 |
| qW |j|d | j rt
| j tjr|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 extendheaderscopyZnetlocitemsr Zbody
isinstancer Z
basestring) r prefixesbytearrr 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|jjd 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' strZstatusencoder 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| dr | jd} | d k r,| S dS )Nr5 zutf-8 )r+ byteshasattrr5 )datar r r r n s
r < > c C sX |dk r |nt }t||}t| ds.tdt| }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_prefixresponse_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? listhistoryappendr )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 s$
+