ó o€ \c@s²dZddlZddlZddlmZddlmZdefd„ƒYZd efd „ƒYZ d e fd „ƒYZ d e fd„ƒYZ d„Z d d dgZ dS(sEModule implementing the Pool for :mod:``requests_toolbelt.threaded``.iÿÿÿÿNi(tthreadi(tqueuetPoolcBs}eZdZd d d ejd„Zd„Zed„ƒZ ed d„ƒZ d„Z d„Z d„Z d„Zd „ZRS( s>Pool that manages the threads containing sessions. :param queue: The queue you're expected to use to which you should add items. :type queue: queue.Queue :param initializer: Function used to initialize an instance of ``session``. :type initializer: collections.Callable :param auth_generator: Function used to generate new auth credentials for the session. :type auth_generator: collections.Callable :param int num_threads: Number of threads to create. :param session: :type session: requests.Session c CsÝ|dkr!tjƒpd}n|dkr<tdƒ‚n||_tjƒ|_tjƒ|_||_ |put |_ |p„t |_ ||_ gt|j ƒD]-}tj|jƒ|j|j|jƒ^q£|_dS(Nis)Number of processes should at least be 1.(tNonetmultiprocessingt cpu_countt ValueErrort _job_queueRtQueuet_response_queuet _exc_queuet _processest _identityt _initializert_autht_sessiontrangeRt SessionThreadt _new_sessiont_pool(tselft job_queuet initializertauth_generatort num_processestsessiont_((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyt__init__s     cCs|j|j|jƒƒƒS(N(RR R(R((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR0scKs=tjƒ}x|D]}|j|jƒqW|d||S(s2Create a :class:`~Pool` from an :class:`~ThreadException`\ s. Provided an iterable that provides :class:`~ThreadException` objects, this classmethod will generate a new pool to retry the requests that caused the exceptions. :param exceptions: Iterable that returns :class:`~ThreadException` :type exceptions: iterable :param kwargs: Keyword arguments passed to the :class:`~Pool` initializer. :returns: An initialized :class:`~Pool` object. :rtype: :class:`~Pool` R(RRtputtrequest_kwargs(tclst exceptionstkwargsRtexc((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pytfrom_exceptions3s  cKszidd6}|j|piƒtjƒ}x;|D]3}|jƒ}|ji|d6ƒ|j|ƒq3W|d||S(sëCreate a :class:`~Pool` from an iterable of URLs. :param urls: Iterable that returns URLs with which we create a pool. :type urls: iterable :param dict request_kwargs: Dictionary of other keyword arguments to provide to the request method. :param kwargs: Keyword arguments passed to the :class:`~Pool` initializer. :returns: An initialized :class:`~Pool` object. :rtype: :class:`~Pool` tGETtmethodturlR(tupdateRRtcopyR(RturlsRR t request_dictRR%tjob((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyt from_urlsIs    ccs2x+tr-|jƒ}|dkr%Pn|VqWdS(soIterate over all the exceptions in the pool. :returns: Generator of :class:`~ThreadException` N(tTruet get_exceptionR(RR!((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyRbs    cCsBy|jjƒ\}}Wntjk r0dSXt||ƒSdS(sSGet an exception from the pool. :rtype: :class:`~ThreadException` N(R t get_nowaitRtEmptyRtThreadException(RtrequestR!((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR-ms cCsBy|jjƒ\}}Wntjk r0dSXt||ƒSdS(sPGet a response from the pool. :rtype: :class:`~ThreadResponse` N(R R.RR/RtThreadResponse(RR1tresponse((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyt get_responseys ccs2x+tr-|jƒ}|dkr%Pn|VqWdS(smIterate over all the responses in the pool. :returns: Generator of :class:`~ThreadResponse` N(R,R4R(Rtresp((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyt responses…s    cCs"x|jD]}|jƒq WdS(s*Join all the threads to the master thread.N(Rtjoin(Rtsession_thread((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pytjoin_allsN(t__name__t __module__t__doc__RtrequeststSessionRRt classmethodR"R+RR-R4R6R9(((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR s  t ThreadProxycBseZdZd„ZRS(cCsHtj}||jkr7|||jƒ}t||ƒS|||ƒSdS(s/Proxy attribute accesses to the proxied object.N(tobjectt__getattribute__tattrst proxied_attrtgetattr(RtattrtgetR3((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyt __getattr__™s   N(R:R;RRDRH(((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR@–sR2cBs/eZdZdZeddgƒZd„ZRS(s>A wrapper around a requests Response object. This will proxy most attribute access actions to the Response object. For example, if you wanted the parsed JSON from the response, you might do: .. code-block:: python thread_response = pool.get_response() json = thread_response.json() R3RcCs||_||_dS(N(RR3(RRR3((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR²s (R:R;R<RDt frozensetRCR(((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR2£s R0cBs/eZdZdZeddgƒZd„ZRS(s=A wrapper around an exception raised during a request. This will proxy most attribute access actions to the exception object. For example, if you wanted the message from the exception, you might do: .. code-block:: python thread_exc = pool.get_exception() msg = thread_exc.message t exceptionRcCs||_||_dS(N(RRJ(RRRJ((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyRÈs (R:R;R<RDRIRCR(((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR0¹s cCs|S(N((t session_obj((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyR Ïs(R<RR=tRt_compatRRARR@R2R0R t__all__(((sT/data/av2000/b2b/venv/lib/python2.7/site-packages/requests_toolbelt/threaded/pool.pyts