o 2wi#@s"ddlZddlZddlZddlmZdZzddlmZWn ey'dZYn weZ e j p0e j Zer9ej erAddlZeZdZdZdZeZGdddZz ddlmZddlZzdd lmZWn eyodZYnwWn ey{YdSw dd ejjfd d ZGdddeZdS)N)lock)get_default_verify_pathszECDH+AESGCM:DH+AESGCM:ECDH+AES256:DH+AES256:ECDH+AES128:DH+AES:ECDH+HIGH:DH+HIGH:ECDH+3DES:DH+3DES:RSA+AESGCM:RSA+AES:RSA+HIGH:RSA+3DES:ECDH+RC4:DH+RC4:RSA+RC4:!aNULL:!eNULL:!MD5@c@sxeZdZdZdZddeedfddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZdS)ConnectionPoolrFc CsZd|_||_||_||_||_t||_tj ||_ | |_ ||_ ||_||_||_dS)NF)_closed_connection_host_connection_port _request_host _request_portrBoundedSemaphore _semaphoregeventqueue LifoQueue _socket_queue _use_proxyconnection_timeoutnetwork_timeoutsize disable_ipv6) selfconnection_hostconnection_port request_host request_portrrrr use_proxyr\/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/geventhttpclient/connectionpool.py__init__'s   zConnectionPool.__init__cCs2d}|jr tjj}tj|j|j|dtjj}|S)z6resolve (dns) socket information needed to connect it.r)rrsocketAF_INET getaddrinforr SOL_TCP)rfamilyinforrr_resolveAszConnectionPool._resolvecCsbd|_|js/z|jjdd}z|WnYWn tjjy'Ynw|jrdSdS)NTFblock)rremptygetcloserrEmptyrsockrrrr+Ps  zConnectionPool.closecCstj|||}|S)ztcp socket factory.)rr )rr$socktypeprotocolr.rrr_create_tcp_socket\sz!ConnectionPool._create_tcp_socketc Cs|}d}|D]c}z |j|dd}Wnty+}z |s!|}WYd}~qd}~wwz||j|||d}||||j|WStyd}z| |sZ|}WYd}~qd}~w| |rp|t d|j d|j )zgmight be overridden and super for wrapping into a ssl socket or set tcp/socket options NzCannot resolve :) r&r1 Exception settimeoutr_connect_socket after_connectrOSErrorr+ RuntimeError_host_port)r sock_infos first_error sock_infor.errr_create_socketas8     zConnectionPool._create_socketcCsdSNrr-rrrr8szConnectionPool.after_connectcCs|||||SrB)connect _setup_proxy)rr.addressrrrr7s  zConnectionPool._connect_socketcCs`|jr,|td|jd|jdd|d}|}|r%|ddkr.td|dSdS) NzCONNECT r4z HTTP/1.1 utf8is200z#Error response from Proxy server : )rsendbytesr r recvsplitr:)rr.resppartsrrrrDs  zConnectionPool._setup_proxyc Cs^|j|jr tdz|jjddWStjjy.z| WYS|j w)z?get a socket from the pool. This blocks until one is available.zconnection pool closedFr') r acquirerr:rr*rrr,rArelease)rrrr get_sockets  zConnectionPool.get_socketcCs<|jrz|WdSYdS|j||jdS)zreturn a socket to the pool.N)rr+rputr rOr-rrr return_sockets  zConnectionPool.return_socketcCs.z|WnY|js|jdSdS)z'call when the socket is no more usable.N)r+rr rOr-rrrrelease_sockets zConnectionPool.release_socketN)__name__ __module__ __qualname__DEFAULT_CONNECTION_TIMEOUTDEFAULT_NETWORK_TIMEOUTrr&r+r1rAr8r7rDrPrRrSrrrrr#s&  ! r)PROTOCOL_TLS_CLIENT)create_default_contextTreturncCs|pi}z||d}Wnty|}|j|dYnw||_|r(tjj|_d|vr;d|vr;|j|d|ddd|vrF||ddD]}||vrVt ||||qH|S)a Initializes an SSL context with additional SSL options. :param ssl_context_factory: Callable to create an SSL context :param ca_certs: Path to CA certificates file :param check_hostname: Whether to enable hostname checking :param ssl_options: Optional dictionary of additional SSL options :return: Configured SSLContext instance )cafilecertfilekeyfile)r]r^ciphers)options verify_flags) TypeErrorload_verify_locationscheck_hostnamerssl CERT_REQUIRED verify_modeload_cert_chain set_cipherssetattr)ssl_context_factoryca_certsrd ssl_options ssl_contextoptionrrrinit_ssl_contexts*    rpcsFeZdZdZeeejje dZ   d fdd Z fddZ Z S) SSLConnectionPoolafSSLConnectionPool creates connections wrapped with SSL/TLS. :param host: hostname :param port: port :param ssl_options: accepts any options supported by `ssl.wrap_socket` :param ssl_context_factory: use `ssl.create_default_context` by default if provided. It must be a callable that returns a SSLContext. )r_rl cert_reqs ssl_versionFNc  st||_|j|_|j|pi|pt}|dur(t||jd|j |d|_nd|_tj ||||fi|dS)Nrl)rdrm) insecuredefault_optionscopyrmupdaterZrprnsuperr) rrrrrrtrkrmkw __class__rrrs  zSSLConnectionPool.__init__csNt||}|jdurtjj|fi|jS|jd|j}|jj||dS)Nserver_hostname)r|) rxr7rnrre wrap_socketrmr*r )rr.rEr|rzrrr7s  z!SSLConnectionPool._connect_socket)FNN)rTrUrV__doc___DEFAULT_CIPHERS _CA_CERTSrrerfrYrurr7 __classcell__rrrzrrqs  rq)TN)os gevent.queuer gevent.socketrrrer ImportError_certsr\capathpathisdircertifiwhererrWrXobjectIGNOREDrrY gevent.sslrZ SSLContextrprqrrrrsN       '