Your IP : 216.73.216.196
B
�ˀX� � @ sT d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
G dd� de �Z dS ) � )�matchers� serializers)�BetamaxAdapter)�Cassette)�
Configuration)�Optionsc @ sv e Zd ZdZdi fdd�Zdd� Zdd� Zed d
� �Ze d d � �Z
ed
d� �Z edd� �Z dd� Z
dd� Zdd� ZdS )�Betamaxa This object contains the main API of the request-vcr library.
This object is entirely a context manager so all you have to do is:
.. code::
s = requests.Session()
with Betamax(s) as vcr:
vcr.use_cassette('example')
r = s.get('https://httpbin.org/get')
Or more concisely, you can do:
.. code::
s = requests.Session()
with Betamax(s).use_cassette('example') as vcr:
r = s.get('https://httpbin.org/get')
This object allows for the user to specify the cassette library directory
and default cassette options.
.. code::
s = requests.Session()
with Betamax(s, cassette_library_dir='tests/cassettes') as vcr:
vcr.use_cassette('example')
r = s.get('https://httpbin.org/get')
with Betamax(s, default_cassette_options={
're_record_interval': 1000
}) as vcr:
vcr.use_cassette('example')
r = s.get('https://httpbin.org/get')
Nc C sJ || _ |j�� | _t| jd�| _t� | _| jj� |p6i � |rF|| j_
d S )N)Z old_adapters) �sessionZadapters�copy�
http_adaptersr �betamax_adapterr �config�default_cassette_options�update�cassette_library_dir)�selfr r r � r �A/opt/alt/python37/lib/python3.7/site-packages/betamax/recorder.py�__init__0 s
zBetamax.__init__c C s | � � | S )N)�start)r r r r � __enter__C s zBetamax.__enter__c G s | � � t|�rdS d S )NF)�stop�any)r Zex_argsr r r �__exit__G s zBetamax.__exit__c C s t � S )aA Help to configure the library as a whole.
.. code::
with Betamax.configure() as config:
config.cassette_library_dir = 'tests/cassettes/'
config.default_cassette_options['match_options'] = [
'method', 'uri', 'headers'
]
)r r r r r � configureP s zBetamax.configurec C s | j jS )zwReturn the cassette that is currently in use.
:returns: :class:`Cassette <betamax.cassette.Cassette>`
)r �cassette)r r r r �current_cassette^ s zBetamax.current_cassettec C s | � t j| j< dS )z�Register a new request matcher.
:param matcher_class: (required), this must sub-class
:class:`BaseMatcher <betamax.matchers.BaseMatcher>`
N)r Zmatcher_registry�name)Z
matcher_classr r r �register_request_matcherf s z Betamax.register_request_matcherc C s | j }| � tj|< dS )z�Register a new serializer.
:param matcher_class: (required), this must sub-class
:class:`BaseSerializer <betamax.serializers.BaseSerializer>`
N)r r Zserializer_registry)Zserializer_classr r r r �register_serializero s zBetamax.register_serializerc C s$ x | j D ]}| j�|| j� qW dS )z*Start recording or replaying interactions.N)r r �mountr )r �kr r r r z s z
Betamax.startc C s> | j �� | j �� x$| j�� D ]\}}| j�||� q W dS )z)Stop recording or replaying interactions.N)r Zeject_cassette�closer �itemsr r )r r! �vr r r r � s
z Betamax.stopc K s^ t |�}|d p|d }| jj|d< t�| jj|||d �}|rR| j�|||� ntd��| S )a� Tell Betamax which cassette you wish to use for the context.
:param str cassette_name: relative name, without the serialization
format, of the cassette you wish Betamax would use
:param str serialize_with: the format you want Betamax to serialize
the cassette with
:param str serialize: DEPRECATED the format you want Betamax to
serialize the request and response data to and from
� serializeZserialize_withr �recordz4Cassette must have a valid name and may not be None.)r r
r r Z
can_be_loadedr Z
load_cassette�
ValueError)r Z
cassette_name�kwargsr% Zcan_loadr r r � use_cassette� s
zBetamax.use_cassette)�__name__�
__module__� __qualname__�__doc__r r r � staticmethodr �propertyr r r r r r) r r r r r s %
r N)
� r r Zadapterr r r r r �optionsr �objectr r r r r �<module> s