Your IP : 216.73.216.196
B
�ˀX% � @ s d dl mZ ddlmZ ddlmZmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ydd lmZ W n ek
r� d d lmZ Y nX dd
lZdd
lZdd
lZd!d d �Zd"d
d�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd � Z d d � Z e � Z d d � Z!d
S )#� )�MockHTTPResponse� )�datetime)�PreparedRequest�Response)� HTTPResponse)�CaseInsensitiveDict)�_codes)�RequestsCookieJar)�HTTPHeaderDictNc C s t | d�r| �|pdd�} | S )N�decodezutf-8�replace)�hasattrr )�content�encoding� r �=/opt/alt/python37/lib/python3.7/site-packages/betamax/util.py�coerce_content s
r c C s" t | d�r| �|pd�} t�| �S )N�encodezutf-8)r r �io�BytesIO)�stringr r r r �body_io s
r c C s t | t�r| d S | S )Nr )�
isinstance�list)�valuer r r � from_list s
r c C s� t | dt | dd��}t|d�r&|�� }|s.d}|sDd| j�dd�kr~tjdkrjt|d �rj|�|d
pfd �}t� |��
� |d < nt ||d
�|d
< dS )a: Simple function which takes a response or request and coerces the body.
This function adds either ``'string'`` or ``'base64_string'`` to
``body_dict``. If ``preserve_exact_body_bytes`` is ``True`` then it
encodes the body as a base64 string and saves it like that. Otherwise,
it saves the plain string.
:param r: This is either a PreparedRequest instance or a Response
instance.
:param preserve_exact_body_bytes bool: Either True or False.
:param body_dict dict: A dictionary already containing the encoding to be
used.
�raw�bodyN�read� �gzipzContent-Encoding)� r r r zutf-8�
base64_stringr ) �getattrr r �headers�get�sys� version_infor �base64� b64encoder r )�r�preserve_exact_body_bytesZ body_dictr r r r �add_body% s
r- c C s> | j }ddi}t| ||� |tdd� |�� D ��| j| jd�S )Nr zutf-8c s s" | ]\}}t |d �|gfV qdS )zutf-8N)r )�.0�k�vr r r � <genexpr>K s z-serialize_prepared_request.<locals>.<genexpr>)r r% �method�uri)r% r- �dict�itemsr2 �url)�requestr, r% r r r r �serialize_prepared_requestD s r8 c C s� t � }t� |_| d }t|t�rH|�d�}|pBt�|�dd��� �|_ n||_ dd� | d �
� D �}t |�|_ | d |_
| d |_|S )
Nr r r# r c S s g | ]\}}|t |�f�qS r )r )r. r/ r0 r r r �
<listcomp>\ s z0deserialize_prepared_request.<locals>.<listcomp>r% r2 r3 )r r
Z_cookiesr r4 r& r) � b64decoder r r5 r r% r2 r6 )�
serialized�pr Z
original_body�hr r r � deserialize_prepared_requestR s
r> c C s` d| j i}t| ||� t| jj�}i }x |�� D ]}|�|�||< q0W ||| j| jd�| j d�S )Nr )�code�message)r r% �statusr6 )
r r- r r r% �keysZgetlist� status_code�reasonr6 )�responser, r Z
header_mapr% � header_namer r r �serialize_responsec s
rG c C s� t � }| d d |_t� }xJ| d �� D ]:\}}t|t�rVx&|D ]}|�||� q@W q(|�||� q(W t|�|_| � dd�|_
d| kr�| d d |_ | d d |_ n | d |_ t
|j d
�� |_ t| ||� |S ) Nr r r% r6 r rA r? r@ rC r )r r r r5 r r �addr r% r& r6 rC rD r �upper�add_urllib3_response)r; r+ Z header_dictrF Z header_listZ header_valuer r r �deserialize_responses s"
rK c C sd d| d kr*t �t�| d d �� ��}ntf | d �}t||j|j|dt |�d�}d|_
||_ d S )Nr# r F)rA rD r% Zpreload_contentZoriginal_response) r r r) r: r r r rC rD r Zchunkedr )r; rE r% r r= r r r rJ � s rJ c C s@ t �� �� } y| �d�}W n tk
r. | S X | d |� S d S )N�.)r �utcnow� isoformat�rindex�
ValueError)Zstamp�ir r r � timestamp� s rR c C s" |� | t�}|tkr |� | �}|S )N)r& � _SENTINEL)�option�kwargs�defaultsr r r r � _option_from� s
rW )N)N)"Z
mock_responser r Zrequests.modelsr r Zrequests.packages.urllib3r Zrequests.structuresr Zrequests.status_codesr Zrequests.cookiesr
Z&requests.packages.urllib3._collectionsr � ImportErrorr% r) r r' r r r r- r8 r> rG rK rJ rR �objectrS rW r r r r �<module> s2