.PhdZddlZddlZddlmZddlmZGddeZGd d eZ Gd d e Z Gd de Z dZ gdZ dS)zEModule implementing the Pool for :mod:``requests_toolbelt.threaded``.N)thread)queuec~eZdZdZdddejfdZdZedZ ed dZ dZ dZ d Z d Zd ZdS) 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 Ncl|tjpd}|dkrtd|_t j_t j_|_|pt_ |pt_ |_ fdtjD_dS)Nrz)Number of processes should at least be 1.cg|]:}tjjjj;S)r SessionThread _new_session _job_queue_response_queue _exc_queue).0_selfs _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/requests_toolbelt/threaded/pool.py z!Pool.__init__..*sV     !2!2!4!4do!%!5t H H   )multiprocessing cpu_count ValueErrorrrQueuerr _processes _identity _initializer_auth_sessionrange_pool)r job_queue initializerauth_generator num_processessessions` r__init__z Pool.__init__s  +577<1M 1  HII I#${}}+--''49#0y      4?++    rcv|||SN)rrr)rs rr zPool._new_session0s*zz$++DMMOO<<===rc ztj}|D]}||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"r )rrputrequest_kwargs)cls exceptionskwargsr"excs rfrom_exceptionszPool.from_exceptions3sQ KMM  . .C MM#, - - - -s11Y1&111rc ddi}||pitj}|D]B}|}|d|i||C|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` methodGETurlr"r )updaterrcopyr+)r-urlsr,r/ request_dictr"r5jobs r from_urlszPool.from_urlsIs!%( N0b111KMM   C##%%C JJs| $ $ $ MM#    s11Y1&111rc#BK |}|dS|V)zoIterate over all the exceptions in the pool. :returns: Generator of :class:`~ThreadException` TN) get_exception)rr0s rr.zPool.exceptionsbs4  $$&&C{III  rc |j\}}t||S#tj$rYdSwxYw)zSGet an exception from the pool. :rtype: :class:`~ThreadException` N)r get_nowaitThreadExceptionrEmpty)rrequestr0s rr=zPool.get_exceptionmsV  1!_7799NWc#7C00 0{   44 .AAc |j\}}t||S#tj$rYdSwxYw)zPGet a response from the pool. :rtype: :class:`~ThreadResponse` N)rr?ThreadResponserrA)rrBresponses r get_responsezPool.get_responseysX  5"&"6"A"A"C"C Wh"'844 4{   44 rCc#BK |}|dS|V)zmIterate over all the responses in the pool. :returns: Generator of :class:`~ThreadResponse` TN)rG)rresps r responseszPool.responsess4  $$&&D|JJJ  rcB|jD]}|dS)z*Join all the threads to the master thread.N)r!join)rsession_threads rjoin_allz Pool.join_alls2"j " "N    ! ! ! ! " "rr))__name__ __module__ __qualname____doc__requestsSessionr'r classmethodr1r;r.r=rGrJrNr rrrr s"/34#X-=    *>>>22[2*222[20    1 1 1 5 5 5   """""rrceZdZdZdZdS) ThreadProxyNctj}||jvr!|||j}t ||S|||S)z/Proxy attribute accesses to the proxied object.)object__getattribute__attrs proxied_attrgetattr)rattrgetrFs r __getattr__zThreadProxy.__getattr__sK% tz ! !s4!233H8T** *3tT?? "r)rOrPrQr\r`r rrrWrWs(L#####rrWc6eZdZdZdZeddgZdZdS)rEa>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() rFr,c"||_||_dSr))r,rF)rr,rFs rr'zThreadResponse.__init__s,  rNrOrPrQrRr\ frozensetr[r'r rrrErEsI  L I'4 5 5E!!!!!rrEc6eZdZdZdZeddgZdZdS)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"||_||_dSr))r,rf)rr,rfs rr'zThreadException.__init__s,"rNrcr rrr@r@sI  L I'5 6 6E#####rr@c|Sr)r ) session_objs rrrs r)r@rEr)rRrrSr_compatrrYrrWrEr@r__all__r rrrmsKKJ"J"J"J"J"6J"J"J"Z # # # # #& # # #!!!!![!!!,#####k###, 8 7 7r