Your IP : 216.73.216.213
��DX�# � @ s� d d l Z d d l m Z d d l Z d d l m Z d d l m Z d d l m
Z
d d l m Z e
� Z Gd d � d e
� Z Gd d
� d
e � Z Gd d � d e � Z Gd
d � d e e � Z d g Z d S)� N)� BaseAdapter)�parse)�
exceptions)�_RequestObjectProxy)�_MatcherResponsec @ sp e Z d Z d d � Z d d � Z e d d � � Z e d d � � Z e d d
� � Z e d d � � Z d
S)�_RequestHistoryTrackerc C s
g | _ d S)N)�request_history)�self� r
�F/opt/alt/python35/lib/python3.5/site-packages/requests_mock/adapter.py�__init__ s z _RequestHistoryTracker.__init__c C s | j j | � d S)N)r �append)r �requestr
r
r �_add_to_history s z&_RequestHistoryTracker._add_to_historyc C s, y | j d SWn t k
r' d SYn Xd S)z Retrieve the latest request sent� N���)r �
IndexError)r r
r
r � last_request" s
z#_RequestHistoryTracker.last_requestc C s
| j d k S)Nr )�
call_count)r r
r
r �called* s z _RequestHistoryTracker.calledc C s
| j d k S)Nr )r )r r
r
r � called_once. s z"_RequestHistoryTracker.called_oncec C s
t | j � S)N)�lenr )r r
r
r r 2 s z!_RequestHistoryTracker.call_countN)
�__name__�
__module__� __qualname__r r �propertyr r r r r
r
r
r r s r c @ s e Z d Z d Z d S)� _RunRealHTTPa A fake exception to jump out of mocking and allow a real request.
This exception is caught at the mocker level and allows it to execute this
request through the real requests mechanism rather than the mocker.
It should never be exposed to a user.
N)r r r �__doc__r
r
r
r r 7 s r c sd e Z d Z d Z � f d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z � S)�_Matcherz;Contains all the information about a provided URL to match.c s t t | � j � | | _ | | _ | | _ | | _ | | _ | | _ t | t
j � r� t j | � } | j
j � | _ | j j � | _ | j p� d | _ | j | _ | s� | j j � | _ | j j � | _ n$ d | _ d | _ d | _ d | _ d S)a2
:param bool complete_qs: Match the entire query string. By default URLs
match if all the provided matcher query arguments are matched and
extra query arguments are ignored. Set complete_qs to true to
require that the entire query string needs to match.
�/N)�superr r �_method�_url�
_responses� _complete_qs�_request_headers�
_real_http�
isinstance�six� string_types�urlparse�scheme�lower�_scheme�netloc�_netloc�path�_path�query�_query) r �method�url� responses� complete_qs�request_headers� real_http�case_sensitiveZ url_parts)� __class__r
r r D s( z_Matcher.__init__c C s9 | j t k r d S| j j � | j j � k r5 d Sd S)NTF)r! �ANYr4 r, )r r r
r
r �
_match_methodg s
z_Matcher._match_methodc
C s` | j t k r d St | j d � r> | j j | j � d k S| j rc | j j � | j k rc d S| j r� | j j � | j k r� d S| j
p� d | j k r� d St j
| j � } t j
| j � } xd t j | � D]S \ } } xD | D]< } y | j | g � j | � Wq� t k
r&d SYq� Xq� Wq� W| j r\x! t j | � D] } | rHd SqHWd S)NT�searchFr )r" r< �hasattrr> r5 r- r+ r, r/ r. r0 r1 r* �parse_qsr2 r3 r( � iteritems�get�remove�
ValueErrorr$ �
itervalues)r r Z
request_qsZ
matcher_qs�k�vals�vr
r
r �
_match_urlp s. !!
z_Matcher._match_urlc C s� x� t j | j � D]� \ } } y | j | } Wn[ t k
r� t | t j � sV d Sy | j | j d � } Wn t k
r� d SYn XYn X| | k r d Sq Wd S)NFzutf-8T)r( rA r% �headers�KeyErrorr' � text_type�encode)r r rF rG �headerr
r
r �_match_headers� s
z_Matcher._match_headersc C s+ | j | � o* | j | � o* | j | � S)N)r= rI rO )r r r
r
r �_match� s z_Matcher._matchc C sv | j | � s d S| j r% t � � t | j � d k rO | j j d � } n
| j d } | j | � | j | � S)Nr r )rP r& r r r# �popr Z get_response)r r Zresponse_matcherr
r
r �__call__� s
z_Matcher.__call__)
r r r r r r= rI rO rP rR r
r
)r; r r A s # % r c s^ e Z d Z d Z d � f d d � Z d d � Z d d � Z d d
d � Z d d
� Z � S)�Adapterz:A fake adapter than can return predefined responses.
Fc s) t t | � j � | | _ g | _ d S)N)r rS r �_case_sensitive� _matchers)r r: )r; r
r r � s zAdapter.__init__c K s� t | d | j | �} | j | � x| t | j � D]k } y | | � } Wn' t k
rt t j | � | _ � Yn X| d k r5 t j | � | _ | | _ | Sq5 Wt
j | � � d S)Nr: ) r rT r �reversedrU � Exception�weakref�refZ_matcher�
connectionr Z
NoMockAddress)r r �kwargs�matcher�respr
r
r �send� s
z Adapter.sendc C s d S)Nr
)r r
r
r �close� s z
Adapter.closeNc
K s� | j d d � } | j d i � } | j d d � } | rQ | rQ t d � � n< | rr | sc | rr t d � � n | s� | r� g n | g } d d � | D� } t | | | d | j d | d | d
| �} | j | � | S) z�Register a new URI match and fake response.
:param str method: The HTTP method to match.
:param str url: The URL to match.
r7 Fr8 r& zKYou should specify either a list of responses OR response kwargs. Not both.z?You should specify either response data OR real_http. Not both.c S s g | ] } t | � � q Sr
)r )�.0rF r
r
r �
<listcomp>� s z(Adapter.register_uri.<locals>.<listcomp>r: r9 )rQ � RuntimeErrorr rT � add_matcher)
r r4 r5 Z
response_listr[ r7 r8 r9 r6 r\ r
r
r � register_uri� s&
zAdapter.register_uric C s | j j | � d S)z�Register a custom matcher.
A matcher is a callable that takes a `requests.Request` and returns a
`requests.Response` if it matches or None if not.
:param callable matcher: The matcher to execute.
N)rU r
)r r\ r
r
r rc s zAdapter.add_matcher) r r r r r r^ r_ rd rc r
r
)r; r rS � s $rS )rX Zrequests.adaptersr r( Zsix.moves.urllibr r* Z
requests_mockr Zrequests_mock.requestr Zrequests_mock.responser �objectr< r rW r r rS �__all__r
r
r
r �<module>
s
�O