|hdZddlZddlZddlmZddlmZGddeZGd d eZ Gd d e Z Gd de Z dZ gdZ y)zEModule implementing the Pool for :mod:``requests_toolbelt.threaded``.N)thread)queueczeZdZdZdddej fdZdZedZ ed dZ dZ dZ d Z d Zd Zy) Poola>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_process: Number of threads to create. :param session: :type session: requests.Session Nc |tjxsd}|dkr td||_t j |_t j |_||_|xst|_ |xst|_ ||_ t|jDcgc]F}tj|j!|j|j |jHc}|_ycc}w)Nrz)Number of processes should at least be 1.)multiprocessing cpu_count ValueError _job_queuerQueue_response_queue _exc_queue _processes _identity _initializer_auth_sessionranger SessionThread _new_session_pool)self job_queue initializerauth_generator num_processessession_s ^/var/www/html/test/engine/venv/lib/python3.12/site-packages/requests_toolbelt/threaded/pool.py__init__z Pool.__init__s  +557<1M 1 HI I#${{}++-''49#0y  4??+   !2!2!4doo!%!5!5t H   s'A C;c^|j|j|jSN)rrr)rs r!rzPool._new_session0s"zz$++DMMO<==c tj}|D]}|j|j|dd|i|S)a2Create 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)rrputrequest_kwargs)cls exceptionskwargsrexcs r!from_exceptionszPool.from_exceptions3sE KKM  .C MM#,, - .1Y1&11r%c ddi}|j|xsitj}|D]6}|j}|jd|i|j |8|dd|i|S)aCreate 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` methodGETurlrr')updaterrcopyr()r*urlsr)r, request_dictrr2jobs r! from_urlszPool.from_urlsIsz!%( N0b1KKM  C##%C JJs| $ MM#   1Y1&11r%c#<K |j}|y|w)zoIterate over all the exceptions in the pool. :returns: Generator of :class:`~ThreadException` N) get_exception)rr-s r!r+zPool.exceptionsbs+ $$&C{I c |jj\}}t||S#tj$rYywxYw)zSGet an exception from the pool. :rtype: :class:`~ThreadException` N)r get_nowaitThreadExceptionrEmpty)rrequestr-s r!r:zPool.get_exceptionmsD  1!__779NWc#7C0 0{{  +AAc |jj\}}t||S#tj$rYywxYw)zPGet a response from the pool. :rtype: :class:`~ThreadResponse` N)rr=ThreadResponserr?)rr@responses r! get_responsezPool.get_responseysG  5"&"6"6"A"A"C Wh"'84 4{{  rAc#<K |j}|y|w)zmIterate over all the responses in the pool. :returns: Generator of :class:`~ThreadResponse` N)rE)rresps r! responseszPool.responsess+ $$&D|J r;cF|jD]}|jy)z*Join all the threads to the master thread.N)rjoin)rsession_threads r!join_allz Pool.join_alls!"jj "N    ! "r%r$)__name__ __module__ __qualname____doc__requestsSessionr"r classmethodr.r8r+r:rErHrLr'r%r!rr sd"/34#X-=-= *>22*220  1 5 "r%rceZdZdZdZy) ThreadProxyNctj}||jvr|||j}t ||S|||S)z/Proxy attribute accesses to the proxied object.)object__getattribute__attrs proxied_attrgetattr)rattrgetrDs r! __getattr__zThreadProxy.__getattr__sD%% tzz !4!2!23H8T* *tT? "r%)rMrNrOrZr^r'r%r!rUrUs L#r%rUc.eZdZdZdZeddgZdZy)rCa>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() rDr)c ||_||_yr$)r)rD)rr)rDs r!r"zThreadResponse.__init__s,  r%NrMrNrOrPrZ frozensetrYr"r'r%r!rCrCs# L '4 5E!r%rCc.eZdZdZdZeddgZdZy)r>a=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 exceptionr)c ||_||_yr$)r)rd)rr)rds r!r"zThreadException.__init__s,"r%Nrar'r%r!r>r>s# L '5 6E#r%r>c|Sr$r') session_objs r!rrs r%)r>rCr)rPr rQr_compatrrWrrUrCr>r__all__r'r%r!rksSKJ"6J"Z #& #![!,#k#, 8r%