U ZhC@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z ddlmZmZmZmZddlmZmZmZd d lmZd d lmZmZGd d d ZGdddeZGdddZdS)) annotationsN) AutoBackend) SOCKET_OPTIONAsyncNetworkBackend)ConnectionNotAvailableUnsupportedProtocol)OriginProxyRequestResponse) AsyncEventAsyncShieldCancellationAsyncThreadLock)AsyncHTTPConnection)AsyncConnectionInterfaceAsyncRequestInterfacec@sZeZdZdddddZddddd Zdd d d ZddddddZdd ddZd S)AsyncPoolRequestr NonerequestreturncCs||_d|_t|_dSN)r connectionr _connection_acquired)selfrrC/tmp/pip-unpacked-wheel-s37ty00c/httpcore/_async/connection_pool.py__init__szAsyncPoolRequest.__init__zAsyncConnectionInterface | None)rrcCs||_|jdSr)rrset)rrrrrassign_to_connectionsz%AsyncPoolRequest.assign_to_connectionrcCsd|_t|_dSr)rr rrrrrclear_connectionsz!AsyncPoolRequest.clear_connectionN float | Noner)timeoutrcs2|jdkr|jj|dIdH|jdk s,t|jS)Nr&)rrwaitAssertionError)rr&rrrwait_for_connections z$AsyncPoolRequest.wait_for_connectionboolcCs |jdkSr)rr#rrr is_queued'szAsyncPoolRequest.is_queued)N)__name__ __module__ __qualname__rr!r$r*r,rrrrrs rc@seZdZdZd4ddd d d d d d d d dddd ddZdddddZeddddZdddd d!Zddd"d#Z ddd$d%d&Z ddd'd(Z ddd)d*Z d5d+d,d-dd.d/d0Z d1dd2d3ZdS)6AsyncConnectionPoolz5 A connection pool for making HTTP requests. N TFrzssl.SSLContext | Nonez Proxy | Nonez int | Noner%r+intz str | NonezAsyncNetworkBackend | Nonez%typing.Iterable[SOCKET_OPTION] | Noner) ssl_contextproxymax_connectionsmax_keepalive_connectionskeepalive_expiryhttp1http2retries local_addressudsnetwork_backendsocket_optionsrc Cs||_||_|dkrtjn||_|dkr.tjn||_t|j|j|_||_||_||_ ||_ | |_ | |_ | dkrvt n| |_| |_g|_g|_t|_dS)a A connection pool for making HTTP requests. Parameters: ssl_context: An SSL context to use for verifying connections. If not specified, the default `httpcore.default_ssl_context()` will be used. max_connections: The maximum number of concurrent HTTP connections that the pool should allow. Any attempt to send a request on a pool that would exceed this amount will block until a connection is available. max_keepalive_connections: The maximum number of idle HTTP connections that will be maintained in the pool. keepalive_expiry: The duration in seconds that an idle HTTP connection may be maintained for before being expired from the pool. http1: A boolean indicating if HTTP/1.1 requests should be supported by the connection pool. Defaults to True. http2: A boolean indicating if HTTP/2 requests should be supported by the connection pool. Defaults to False. retries: The maximum number of retries when trying to establish a connection. local_address: Local address to connect from. Can also be used to connect using a particular address family. Using `local_address="0.0.0.0"` will connect using an `AF_INET` address (IPv4), while using `local_address="::"` will connect using an `AF_INET6` address (IPv6). uds: Path to a Unix Domain Socket to use instead of TCP sockets. network_backend: A backend instance to use for handling network I/O. socket_options: Socket options that have to be included in the TCP socket when the connection was established. N) _ssl_context_proxysysmaxsize_max_connections_max_keepalive_connectionsmin_keepalive_expiry_http1_http2_retries_local_address_udsr_network_backend_socket_options _connections _requestsr_optional_thread_lock) rr3r4r5r6r7r8r9r:r;r<r=r>rrrr0s0,zAsyncConnectionPool.__init__r r)originrc Cs|jdk r|jjjdkrPddlm}||jjj|jj||j|j|j |j |j dS|jdkrddl m }||jjj|jj|jj||j|j dSddl m}||jjj|jj|jj||j|j|j |j |j d St||j|j|j |j |j|j|j|j |jd S) N)ssocks5ssocks5hr)AsyncSocks5Connection) proxy_origin proxy_auth remote_originr3r7r8r9r=shttp)AsyncForwardHTTPConnection)rS proxy_headersproxy_ssl_contextrUr7r=)AsyncTunnelHTTPConnection) rSrWrXrUr3r7r8r9r=) rQr3r7r8r9r:r;r<r=r>)r@urlschemeZ socks_proxyrRrQauthr?rFrGrHrLZ http_proxyrVheadersr3rYrrIrJrKrM)rrQrRrVrYrrrcreate_connections^      z%AsyncConnectionPool.create_connectionzlist[AsyncConnectionInterface]r"cCs t|jS)a Return a list of the connections currently in the pool. For example: ```python >>> pool.connections [ , , , ] ``` )listrNr#rrr connectionsszAsyncConnectionPool.connectionsr r rc s|jj}|dkrtd|dkr4td|d|jdi}|dd}|jt|}|j |W5QRXzx|j| }W5QRX| |IdH|j |d IdH}z| |jIdH}Wqtk r|YqvXqqvWn`tk rL} z@|j|j|| }W5QRX| |IdH| dW5d} ~ XYnXt|jtjsbtt|j|jt|j||d |jd S) z Send an HTTP request, and return an HTTP response. This is the core implementation that is called into by `.request()` or `.stream()`. z;Request URL is missing an 'http://' or 'https://' protocol.)httphttpswswssz)Request URL has an unsupported protocol 'z://'.r&poolNr')stream pool_requestrf)statusr]content extensions)rZr[decoderrkgetrPrrOappend_assign_requests_to_connections_close_connectionsr*handle_async_requestrrr$ BaseExceptionremove isinstancergtyping AsyncIterabler)r rir]PoolByteStream) rrr[Ztimeoutsr&rhclosingrresponseexcrrrrqsV    z(AsyncConnectionPool.handle_async_requestcs\g}t|jD]r}|r(|j|q|rH|j|||q|rtdd|jD|jkr|j|||qdd|j D}|D]}|j j j fdd|jD}dd|jD}|r|d}| |qt|j|jkr|}|j|| |q|r|d}|j||||}|j|| |q|S)aE Manage the state of the connection pool, assigning incoming requests to connections as available. Called whenever a new request is added or removed from the pool. Any closing connections are returned, allowing the I/O for closing those connections to be handled seperately. cSsg|] }|qSris_idle.0rrrr &szGAsyncConnectionPool._assign_requests_to_connections..cSsg|]}|r|qSrr,r~rrrrr.scs"g|]}|r|r|qSr)Zcan_handle_requestZ is_availabler}rQrrr1s cSsg|]}|r|qSrr{r}rrrr6sr)r_rNZ is_closedrs has_expiredrnr|lenrDrOrrZrQr!rCr^)rclosing_connectionsrZqueued_requestsrhZavailable_connectionsZidle_connectionsrrrrosN                 z3AsyncConnectionPool._assign_requests_to_connections)rxrc s.t|D]}|IdHq W5QRXdSr)raclose)rrxrrrrrpUsz&AsyncConnectionPool._close_connectionsc s6|jt|j}g|_W5QRX||IdHdSr)rPr_rNrp)rrrrrr[s zAsyncConnectionPool.aclosecs|Srrr#rrr __aenter__cszAsyncConnectionPool.__aenter__ztype[BaseException] | NonezBaseException | Noneztypes.TracebackType | None)exc_type exc_value tracebackrcs|IdHdSr)r)rrrrrrr __aexit__fszAsyncConnectionPool.__aexit__strc Cs|jj}|jNdd|jD}dd|jD}|d}|d}|d}|d}W5QRXd|d|d}d |d|d } d |d |d | dS)NcSsg|] }|qSrrrrrrrqsz0AsyncConnectionPool.__repr__..cSsg|] }|qSrr{r}rrrrrsFTz Requests: z active, z queuedz Connections: z idle) __class__r-rPrOrNcount) r class_nameZrequest_is_queuedZconnection_is_idleZnum_active_requestsZnum_queued_requestsZnum_active_connectionsZnum_idle_connectionsZ requests_infoZconnection_inforrr__repr__ns   zAsyncConnectionPool.__repr__) NNr1NNTFrNNNN)NNN)r-r.r/__doc__rr^propertyr`rqrorprrrrrrrrr0+s6(P5GGr0c@s<eZdZdddddddZdd d d Zdd d d ZdS)rwztyping.AsyncIterable[bytes]rr0r)rgrhrfrcCs||_||_||_d|_dS)NF)_stream _pool_request_pool_closed)rrgrhrfrrrrszPoolByteStream.__init__ztyping.AsyncIterator[bytes]r"c Cs\z |j2z3dHW}|Vq6Wn6tk rV}z|IdH|dW5d}~XYnXdSr)rrrr)rpartrzrrr __aiter__s zPoolByteStream.__aiter__c s~|jszd|_t"t|jdr0|jIdHW5QRX|jj |jj|j |j }W5QRX|j |IdHdS)NTr) rrhasattrrrrrPrOrsrrorp)rrxrrrrs  zPoolByteStream.acloseN)r-r.r/rrrrrrrrws rw) __future__rsslrAtypesruZ_backends.autorZ_backends.baserr _exceptionsrrZ_modelsr r r r Z_synchronizationr rrrr interfacesrrrr0rwrrrrs   \