ˀX@sdZddlZddlmZd Zdddd d d iZejd d dgZGdd d eZddZ ddZ ddZ dddZ ddZ ddZdddddZddddZdS)!zGThis module provides functions for dumping information about responses.N)compat dump_responsedump_all s0.9 s1.0 s1.1PrefixSettingsrequestresponsecs"eZdZfddZS)rcs4t|}t|}tt|j|||S)N)_coerce_to_bytessuperr__new__)clsr r ) __class__>/usr/lib/python3/dist-packages/requests_toolbelt/utils/dump.pyr s  zPrefixSettings.__new__)__name__ __module__ __qualname__r rr)rrrs cCsRt|jddrNi}|jj}|jdr@d|d<||d<|SdS)NZ proxy_managerFzhttps://ZCONNECTmethod request_path)getattrZ connectionr url startswith)r proxy_infoZ request_urlrrr_get_proxy_informations   rcCst|dt|dS)Ns: s )r )namevaluerrr_format_header&srcCsmtj|}|jd}|dk r4||fSt|j}|jrc|dt|j7}||fS)Nr?)rZurlparsegetr pathZquery)rruriZ proxy_urlrrrr_build_request_path+s   r#c CsO|dkri}|j}t|jd|j}t|j|\}}|j||d|d|jj}t|jd|j } |j|d| dx4|j D]&\} } |j|t | | qW|j|d|j r>t |j tjr1|j|t|j n |jd|jddS)Nr s HTTP/1.1 ZHostsHost: s z,<< Request body is not a string-like type >>)r r poprr#rextendheaderscopyZnetlocitemsrZbody isinstancerZ basestring) r prefixesbytearrrprefixrrr"r'Z host_headerrrrrr_dump_request_data8s"    r.c Cs|j}|j}tj|jd}|j|d|dt|jjddt |j d|j }xH|j D]:}x1|j |D] }|j|t||qWqW|j|d|j|jdS)NrsHTTP/r$asciis )r raw HTTP_VERSIONSr versionr&strZstatusencoder reasonr'keysZgetlistrZcontent) r r+r,r-r0Z version_strr'rrrrr_dump_response_dataVs  > "r7cCs2t|t r.t|dr.|jd}|S)Nr4zutf-8)r*byteshasattrr4)datarrrr msr s< s> cCs~|dk r|nt}t||}t|dsEtdt|}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 requestr) bytearrayrr9 ValueErrorrr.r r7)r request_prefixresponse_prefixZ data_arrayr:r+rrrrrss  cCsWt}t|jdd}|j|x!|D]}t||||q6W|S)aDump 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 r=r>r:r@rrrrs    )z dump_responsezdump_all)__doc__ collectionsZrequestsr__all__r1 namedtupleZ_PrefixSettingsrrrr#r.r7r rrrrrrs$       +