�
@u`Wc@sH
dd
lm
Z

ddlmZ
ddlmZmZ
ddlmZ
ddlm	Z	
ddl
mZ
ddlm
Z

ydd	lmZ
Wn!ek
r�


dd	lmZ
n
Xdd
lZdd
lZdd
lZd
d�
Zd
d�
Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Ze �Z!d�Z"d
S(i
(
tMockHTTPResponsei����(
tdatetime(tPreparedRequesttResponse(
tHTTPResponse(
tCaseInsensitiveDict(
t_codes(
tRequestsCookieJar(
tHTTPHeaderDictNcCs.t|d
�r*|j
|
pdd�}n|S(Ntdecodesutf-8treplace(thasattrR	(tcontenttencoding((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytcoerce_contents


cCs4t|d
�r'|j
|
pd�
}ntj|�
S(Ntencodesutf-8(RRtiotBytesIO(tstringR
((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytbody_ios


c

Cst|t
�r|d
S|S(Ni(t
isinstancetlist(
tvalue((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pyt	from_lists


cCs�t|d
t|dd��}t|d�r<|j�}n|sKd}n|
sld|jjdd�kr�tjdkr�t|d	�r�|j|d
p�d�
}nt	j
|�
j�|d<nt||d
�|d
<dS(s: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.
    trawtbodytreadttgzipsContent-EncodingiiRR
sutf-8t
base64_stringRN(ii(
tgetattrtNoneRRtheaderstgettsystversion_infoRtbase64t	b64encodeR	R(t
rtpreserve_exact_body_bytest	body_dictR((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytadd_body%s


	


cCsb|j}i
d
d6}t
||
|�
i|d6td�|j�D�
�
d6|jd6|jd6S(Nsutf-8R
Rc
ss-|]#\}
}t|
d�|g
fV
qd
S(sutf-8N(
R(t.0t
kt
v((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pys	<genexpr>KsR tmethodturi(R R)tdicttitemsR-turl(trequestR'R R((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytserialize_prepared_requestDs
	








c
Cs�t�}
t
�|
_|d
}t|t�rj|jd�
}|patj|jdd�j��
|
_	n	||
_	g|dj
�D]\}}|t|�
f^q�}t|�
|
_
|d|
_|d|
_|
S(NRRRRR R-R.(RRt_cookiesRR/R!R$t	b64decodeRRR0RRR R-R1(t
serializedt
pRt
original_bodyR+R,t
h((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytdeserialize_prepared_requestRs
	






!	
5





cCs�i
|jd
6}t
||
|�
t|jj�
}i}x'|j�D]}|j|�
||<qEWi|d6|d6i|jd6|jd6d6|j	d6S(NR
RR tcodetmessagetstatusR1(
R
R)RRR tkeystgetlisttstatus_codetreasonR1(tresponseR'Rt
header_mapR theader_name((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytserialize_responsecs









c
Cs
t�}
|d
d|
_
t�}xa|dj�D]O\}}t|t�rsx1|D]}|j||�
qVWq4|j||�
q4Wt|�
|
_|j	dd�|
_
d|kr�|dd|
_|dd|
_n'|d	|
_t
|
jd
j�|
_t||
|�
|
S(NRR
R R1RR=R;R<R@i(RR
RR0RRtaddRR R!R1R@RARtuppertadd_urllib3_response(R6R&theader_dictRDtheader_listtheader_value((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pytdeserialize_responsess"
	

	











c
Cs�d
|dkr9tj
tj|dd
j��
�
}nt|d�}t|d|
jd|
jd|dt	dt
|�
�
}t	|_||
_dS(NRRR=RAR tpreload_contenttoriginal_response(
RRR$R5RRRR@RAtFalseRtchunkedR(R6RBR RR9((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pyRH�s


#

	
	


	
cCsFtj
�j�}y|jd
�
}
Wntk
r9


|SX||
 SdS(Nt
.(R
tutcnowt	isoformattrindext
ValueError(tstampt
i((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pyt	timestamp�s





cCs4|
j|t
�}|t
kr0|j|�
}n|S(
N(R!t	_SENTINEL(toptiontkwargstdefaultsR((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pyt_option_from�s



(#t
mock_responseRR
trequests.modelsRRtrequests.packages.urllib3Rtrequests.structuresRtrequests.status_codesRtrequests.cookiesRt&requests.packages.urllib3._collectionsRtImportErrorR R$RR"RRRRR)R3R:RERLRHRXtobjectRYR](((s=/opt/alt/python27/lib/python2.7/site-packages/betamax/util.pyt<module>
s2