Your IP : 216.73.216.213
B
�ˀX> � @ sj d Z ddlZddlmZ ddlmZ ddlmZmZ ddlm Z m
Z
e � Z G dd � d e �Z
d
Zd d � ZdS )
z7
betamax.adapter.
==============
adapter for betamax
� N� )�cassette)� BetamaxError)�datetime� timedelta)� BaseAdapter� HTTPAdapterc s\ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zddd�Z ddd�Z
dd� Z � Z S )�BetamaxAdapterz�This object is an implementation detail of the library.
It is not meant to be a public API and is not exported as such.
c sD t t| ��� d | _d | _|�di �| _tf |�| _d | _ i | _
d S )N� old_adapters) �superr �__init__r �
cassette_name�popr
r � http_adapter� serialize�options)�self�kwargs)� __class__� �@/opt/alt/python37/lib/python3.7/site-packages/betamax/adapter.pyr s zBetamaxAdapter.__init__c C s | j rtj�| j �rdS dS )zlCheck if cassette exists on file system.
:returns: bool -- True if exists, False otherwise
TF)r
�os�path�exists)r r r r �cassette_exists" s z BetamaxAdapter.cassette_existsc C s | j �� dS )z&Propagate close to underlying adapter.N)r �close)r r r r r + s zBetamaxAdapter.closec C s | j r| j �� d| _ dS )z Eject currently loaded cassette.N)r Zeject)r r r r �eject_cassette/ s
z BetamaxAdapter.eject_cassettec C s8 || _ || _| j�|�� � | j�di �}i }tjj}| j�d|d �}| j�d�|d< | j�d�|d< | j�d�|d< x*t |�� �D ]\}} | dkr�|�
|� q�W tj||f|| j�d�d �|��| _d| jkr�| jd | j_ | jj �|� t
j}
| j�d
��rt
| jd
�}
t�� } |
| | jj k �r4| j�� dS ) a% Load cassette.
Loads a previously serialized http response as a cassette
:param str cassette_name: (required), name of cassette
:param str serialize: (required), type of serialization i.e 'json'
:options dict options: (required), options for cassette
� placeholders�match_requests_onZpreserve_exact_body_bytesZallow_playback_repeats�record� record_modeN�cassette_library_dir)r r! �re_record_interval)r
r r �update�items�getr ZCassetteZdefault_cassette_options�listr r �
match_optionsr �maxr �utcnowZearliest_recorded_date�clear) r r
r r r Zcassette_optionsZdefault_optionsr �option�valuer" �nowr r r �
load_cassette5 s:
z BetamaxAdapter.load_cassetteFNTc
C sj d}| j }|std��|jr&|�|�}|sF|�� rF| �||||||�}|sXtt||���|�� } | | _| S )zcSend request.
:param request request: request
:returns: A Response object
Nz#No cassette was specified or found.) r r Z interactionsZ
find_matchZ is_recording�send_and_record�unhandled_request_messageZ as_responseZ
connection)
r �request�stream�timeout�verify�cert�proxiesZ interactionZcurrent_cassetteZrespr r r �sendm s
zBetamaxAdapter.sendc C s0 | � |j�}|j|d||||d�}| j�||�S )aj Send request and record response.
The response will be serialized and saved to a
cassette which can be replayed in the future.
:param request request: request
:param bool stream: (optional) defer download until content is accessed
:param float timeout: (optional) time to wait for a response
:param bool verify: (optional) verify SSL certificate
:param str cert: (optional) path to SSL client
:param proxies dict: (optional) mapping protocol to URL of the proxy
:return: Interaction
:rtype: class:`betamax.cassette.Interaction`
T)r2 r3 r4 r5 r6 )� find_adapter�urlr7 r Zsave_interaction) r r1 r2 r3 r4 r5 r6 �adapter�responser r r r/ � s
z BetamaxAdapter.send_and_recordc C s. x(| j �� D ]\}}|�� �|�r |S q W dS )z�Find adapter.
Searches for an existing adapter where the url and prefix match.
:param url str: (required) url of the adapter
:returns: betamax adapter
N)r
r$ �lower�
startswith)r r9 �prefixr: r r r r8 � s zBetamaxAdapter.find_adapter)FNTNN)FNTNN)
�__name__�
__module__� __qualname__�__doc__r r r r r. r7 r/ r8 �
__classcell__r r )r r r s 8
r z�A request was made that could not be handled.
A request was made to {url} that could not be found in {cassette_file_path}.
The settings on the cassette are:
- record_mode: {cassette_record_mode}
- match_options {cassette_match_options}.
c C s t j| j|j|j|jd�S )z*Generate exception for unhandled requests.)r9 Zcassette_file_pathZcassette_record_modeZcassette_match_options)�UNHANDLED_REQUEST_EXCEPTION�formatr9 r
r r' )r1 r r r r r0 � s r0 )rB r � r �
exceptionsr r r Zrequests.adaptersr r �objectZ _SENTINELr rD r0 r r r r �<module> s (