o .j@sddlmZddlZddlmZddlmZddlmZ ej r7ddl m Z ddl mZdd lmZdd lmZdd lmZmZmZmZmZGd d d edZdS)) annotationsN)ABCMeta)asynccontextmanager)timeout)AsyncHTTPResponse) HttpVersion)AsyncDirectStreamAccess)AsyncTrafficPolice) BaseSSLError ProtocolErrorReadTimeoutErrorSSLErrorMustRedialErrorc@seZdZdZd4ddZed5ddZed5d d Zed6d d Z d7ddZ ed8ddZ e d9ddZ e d:ddZe d;ddZe dd)d*Zd?d-d.Zd@d1d2Zd3S)AAsyncExtensionFromHTTPzRepresent an extension that can be negotiated just after a "101 Switching Protocol" HTTP response. This will considerably ease downstream integration.returnNonecCsd|_d|_d|_dSN)_dsa _response_police_officerselfrb/home/thesage/.local/lib/python3.10/site-packages/urllib3/contrib/webextensions/_async/protocol.py__init__s zAsyncExtensionFromHTTP.__init__!typing.AsyncGenerator[None, None]c Cs,d}zzdVWnhty-}zd}|jr t|jdr |jjnd}t|dd|d}~wty[}z#dt|vr?t||d}|jrNt|jdrN|jjnd}t|dd|d}~wtt fyq}z t d|||d}~wwd}W|s|jr| IdHdSdSdS|s|jr| IdHwww) Catch low-level python exceptions, instead re-raising urllib3 variants, so that low-level exceptions are not leaked in the high-level api. On unrecoverable issues, release the connection back to the pool. FNT_poolRead timed out.zread operation timed outConnection broken: ) SocketTimeoutrhasattrrr r strrOSErrorrr closer clean_exitepoolrrr_read_error_catcher sT        z*AsyncExtensionFromHTTP._read_error_catcherc Csd}zazdVWnFty+}z|jrt|jdr|jjnd}t|dd|d}~wty;}zt||d}~wtyO}z td|||d}~wwd}W|sa|jrc| IdHdSdSdS|sr|jrs| IdHwww)rFNrrr T) r!rr"rr r rr$r r%r&rrr_write_error_catcherVs@     z+AsyncExtensionFromHTTP._write_error_catcherdict[str, typing.Any]cCsiSrrrrrrurlopen_kwargsz%AsyncExtensionFromHTTP.urlopen_kwargsresponsercs<|jdus t|jdstd|jj|_|j|_||_dS)zEThe HTTP server gave us the go-to start negotiating another protocol.Nrz-The HTTP extension is closed or uninitialized)_fpr"r$rrr)rr/rrrstarts   zAsyncExtensionFromHTTP.startboolcCs |jduSr)rrrrrcloseds zAsyncExtensionFromHTTP.closedset[HttpVersion]cCt)z3Hint about supported parent SVN for this extension.NotImplementedErrorrrrr supported_svnz$AsyncExtensionFromHTTP.supported_svnr#cCr5rr6rrrrimplementationr.z%AsyncExtensionFromHTTP.implementationset[str]cCr5)z%Recognized schemes for the extension.r6rrrrsupported_schemesr9z(AsyncExtensionFromHTTP.supported_schemesschemecCr5)zJConvert the extension scheme to a known http scheme (either http or https)r6)r=rrrscheme_to_http_schemer9z,AsyncExtensionFromHTTP.scheme_to_http_scheme http_versionrdict[str, str]cCr5)zHSpecific HTTP headers required (request) before the 101 status response.r6)rr?rrrheadersr.zAsyncExtensionFromHTTP.headersct)z"End/Notify close for sub protocol.r6rrrrr%zAsyncExtensionFromHTTP.closestr | bytes | NonecrB)zUnpack the next received message/payload from remote. This call does read from the socket. If the method return None, it means that the remote closed the (extension) pipeline. r6rrrr next_payloadsz#AsyncExtensionFromHTTP.next_payloadbuf str | bytescrB)zDispatch a buffer to remote.r6)rrFrrr send_payloadrCz#AsyncExtensionFromHTTP.send_payloadcallback=typing.Callable[[str | bytes | None], typing.Awaitable[None]]crB)zSet up a callback that will be invoked automatically once a payload is received. Meaning that you stop calling manually next_payload().r6)rrIrrr on_payloadsz!AsyncExtensionFromHTTP.on_payloadN)rr)rr)rr,)r/rrr)rr2)rr4)rr#)rr;)r=r#rr#)r?rrr@)rrD)rFrGrr)rIrJrr)__name__ __module__ __qualname____doc__rrr*r+propertyr-r1r3 staticmethodr8r:r<r>rAr%rErHrKrrrrrs2  5 *           r) metaclass) __future__rtypingabcr contextlibrsocketrr! TYPE_CHECKING_async.responserbackendrbackend._async._baser util._async.traffic_policer exceptionsr r r rrrrrrrs