o .j`@s2ddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl mZddlmZddlmZmZmZmZdd lmZdd lmZmZmZmZmZmZmZm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&dd l'm(Z(ddl)m*Z*ej+rddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3Gddde#Z4dS)) annotationsN)asynccontextmanager)timeout)HTTPHeaderDict) _TYPE_BODY) C_INT_MAX CHUNK_AMT_MAXDECODE_GROWTH_FACTORDECODE_MIN_RAW_REFERENCE)AsyncLowLevelResponse) BaseSSLError HTTPErrorIncompleteRead ProtocolErrorReadTimeoutErrorResponseNotReadySSLErrorMustRedialError)ContentDecoder HTTPResponse) is_fp_closedBytesQueueBuffer)Retry)AsyncHTTPConnection)Message)AsyncHTTPConnectionPool)AsyncExtensionFromHTTP)AsyncTrafficPolicec@seZdZ                 dZd[d#d$Zd\d'd(Zed]d*d+Zd^d,d-Zed_d/d0Z ed`d2d3Z dad6d7Z dbd9d:Z edcdtyping.Mapping[str, str] | typing.Mapping[bytes, bytes] | Nonestatusintversionreason str | Nonepreload_contentbooldecode_contentoriginal_responseAsyncLowLevelResponse | NonepoolAsyncHTTPConnectionPool | None connectionAsyncHTTPConnection | NonemsgMessage | Noneretries Retry | Noneenforce_content_lengthrequest_method request_url auto_closepolice_officer.AsyncTrafficPolice[AsyncHTTPConnection] | NonereturnNonecCst|tr ||_nt||_zt||_Wn ty!d|_Ynw||_||_||_d|_ ||_ d|_ d|_ | |_ d|_d|jvr\|jdd}dd|dD}d|vr\d |_d|_| |_||_d|_d|_d|_||_d|_| durtjd td d | |_|rt|ttfr||_| |_ | |_!t"|d r||_d|_#||_$|%|j$|_&t'|_(d|_)||_*||_+|j*dur|j*,||j!|j*j-dur|j*j-,||j dSdSdS)NrFztransfer-encodingr!css|]}|VqdSN)strip).0encrCS/home/thesage/.local/lib/python3.10/site-packages/urllib3_future/_async/response.py Xsz-AsyncHTTPResponse.__init__..,chunkedTzjPassing msg=.. is deprecated and no-op in urllib3.future and is scheduled to be removed in a future major.r) stacklevelread). isinstancerr#r&r% ValueErrorr'r(r,_has_decoded_content _request_url_retries _extensionr5rGgetlowersplit_decoderr7r:_body_uncached_read_occurred_fp_original_response_fp_bytes_readwarningswarnDeprecationWarningr3strbytes_pool _connectionhasattr chunk_left_request_method _init_lengthlength_remainingr_decoded_buffer_last_raw_read_size_police_officer_preloaded_contentmemorizeparent)selfr"r#r%r'r(r*r,r-r/r1r3r5r7r8r9r:r;tr_enc encodingsrCrCrD__init__,sr        zAsyncHTTPResponse.__init__b bytearraycs>|t|IdH}t|dkrdS||dt|<t|S)Nr)rIlen)rkrotemprCrCrDreadintos  zAsyncHTTPResponse.readinto!typing.AsyncGenerator[None, None]c Cs@d}zxzdVWnHty}zt|jdd|d}~wty;}zdt|vr/t||t|jdd|d}~wttfyQ}z td|||d}~wwd}W|sj|j r_|j |j rj|j IdH|j rx|j rz| dSdSdS|s|j r|j |j r|j IdH|j r|j r| www)z Catch low-level python exceptions, instead re-raising urllib3 variants, so that low-level exceptions are not leaked in the high-level api. On exit, release the connection back to the pool. FNzRead timed out.zread operation timed outzConnection broken: T) SocketTimeoutrr^r r\rOSErrorrrrWcloser_isclosed release_conn)rk clean_exiterCrCrD_error_catchersD        z AsyncHTTPResponse._error_catcherc sJz |IdHWn tttfyYnw|jr#t|_d|_dSdS)z Read and discard any remaining HTTP response data in the response connection. Unread data in the HTTPResponse connection blocks the connection from being released back to the pool. N) _raw_readrrvr rLrrerSrkrCrCrD drain_conns zAsyncHTTPResponse.drain_connHTTPHeaderDict | NonecCs&|jdurdSt|jdr|jjSdS)z Retrieve post-response (trailing headers) if any. This WILL return None if no HTTP Trailer Headers have been received. Ntrailers)rVr`rr~rCrCrDrs  zAsyncHTTPResponse.trailersAsyncExtensionFromHTTP | NonecCs|jSr?)rOr~rCrCrD extensionszAsyncHTTPResponse.extensionitemrcs@|jdur tdt|jdst||IdH||_dS)Nzextension already plugged in_dsa)rOrvr`rVrstart)rkrrCrCrDstart_extensions   z!AsyncHTTPResponse.start_extension typing.Anycst|jIdHS)a Parses the body of the HTTP response as JSON. To use a custom JSON decoder pass the result of :attr:`HTTPResponse.data` to the decoder. This method can raise either `UnicodeDecodeError` or `json.JSONDecodeError`. Read more :ref:`here `. N)_jsonloadsdatar~rCrCrDjsons zAsyncHTTPResponse.jsonr]cs*|jr|jS|jr|jddIdHSdS)NT) cache_content)rTrVrIr~rCrCrDrs zAsyncHTTPResponse.dataamt int | Nonecs&|jsJ|r |tks|jru|jtkrutjdkrut}t|jt}|dus+|dkrq|dur9t |t }||8}nt }z|rH|j |IdH}n|j |}Wn t y[Y|Sw|sc |S| |~|dus+|dks+|St|jtr|j |IdHS|dur|j |S|j S)a_ Read a response with the thought that reading the number of bytes larger than can fit in a 32-bit int at a time via SSL in some known cases leads to an overflow error that has to be prevented if `amt` or `self.length_remaining` indicate that a problem may happen. The known cases: * 3.8 <= CPython < 3.9.7 because of a bug https://github.com/urllib3/urllib3/issues/2513#issuecomment-1152559900. * urllib3 injected with pyOpenSSL-backed SSL-support. * CPython < 3.10 only when `amt` does not fit 32-bit int. ) Nr)rVrrdsys version_infoioBytesIOrJr minr rIrKwritegetvalue)rkrbuffer is_async_ll chunk_amtrrCrCrD_fp_readsF          zAsyncHTTPResponse._fp_readc s\|jdurdSt|jdd}|4IdHa|s"||IdHnd}|duo/t|jdd }|dur:|dkr:|r<|rn|rUt|}|j|7_|jdurU|j|8_|j|jrn|jdurn|jdkrnt |j|jWdIdHn 1IdHs~wY|r|st|}|j|7_|jdur|j|8_t |t t fr|rt||_ |S)z7 Reads `amt` of bytes from the socket. NclosedFr)rVgetattrr|rrqrXrdrwr7rrJr]rprf)rkr fp_closedris_foreign_fp_uncloseddata_lenrCrCrDr}Ls<      (&  zAsyncHTTPResponse._raw_read bool | NonecsP|j|pd|dIdH}d|_|dur&t||kr&|j||j|S|S)a Similar to ``http.client.HTTPResponse.read1`` and documented in :meth:`io.BufferedReader.read1`, but with an additional parameter: ``decode_content``. :param amt: How much of the content to read. :param decode_content: If True, will attempt to decode the body based on the 'content-encoding' header. )rr,NT)rIrUrqreputrPrkrr,rrCrCrDread1s   zAsyncHTTPResponse.read1Frcs|j|||dIdHS)N)rr,r)_read)rkrr,rrCrCrDrIs  zAsyncHTTPResponse.read)partialrc sz||dur|j}|durd}|dkrt|jr8|jjr8t|jtt}|jd|d|d}|r8|j |t |j rt|j t |j W|j rpt|j ddrq|jdurrt|j drf|j jdusf|j jjrs|j|d|_SSSSS|dkr|jr|jjrt |j |kr|jd|d|t |j d}|j ||t |j kr|j |W|j rt|j ddr|jdurt|j dr|j jdus|j jjr|j|d|_SSSSS|jdur|j|4IdH||IdH}WdIdHn 1IdHswYn||IdH}|sd|_|r|dkrt |}d}|dur(d}n |dkr2|s2d}|s|t |j dkr||jrF|jjs||W|j rxt|j ddry|jdurzt|j drn|j jdusn|j jjr{|j|d|_SSSSS|dur||||}|r|js||_n|s|jrtd|W|j rt|j ddr|jdurt|j dr|j jdus|j jjr|j|d|_SSSSS|j||||t |j d }|j ||ot|j d} t |j |krY|rY| sY|jdur/|j|4IdH||IdH}WdIdHn 1IdHs)wYn||IdH}|j||||t |j d }|j |t |j |krY|rY| r|j |}|W|j rt|j ddr|jdurt|j dr|j jdus|j jjr|j|d|_SSSSS|j rt|j ddr|jdurt|j dr|j jdus|j jjr|j|d|_w) NFrr) flush_decoder max_length_eotrTz_Calling read(decode_content=False) is not supported after read(decode_content=True) was called.)r) _init_decoderr,rShas_unconsumed_tailmaxrfr r _decodererrqrPrVrrgr`rrforgetborrowr}rUrTrL RuntimeError) rkrr,rrcap decoded_datarrsurface_per_framerCrCrDrsv    `        J      ( 1     #       *           zAsyncHTTPResponse._read"typing.AsyncGenerator[bytes, None]cCs|dkrdS|jdurdSt|jr|js|jr>|jjr@|j||ddIdH}|r-|Vt|jr|js|jrB|jjsdSdSdSdS)NrT)rr,r)rVrrerSrrrrCrCrDstreamOs4  zAsyncHTTPResponse.streamcsj|jdur|jjs|jIdH|js|jr|j|jr(|jIdH|js3tj|dSdSr?)rrrwrVr_r:rIOBaser~rCrCrDrwbs  zAsyncHTTPResponse.closetyping.AsyncIterator[bytes]cCsg}|jddd2z=3dHW}d|vrB|d}dg||ddV|ddD]}|dVq-|dr?|dg}q g}q ||q 6|rSd|VdSdS)NrT)r, rrr)rrRjoinappend)rkrchunkchunksxrCrCrD __aiter__os      zAsyncHTTPResponse.__aiter__cCs4|js|jr |j|jstj|dSdSdSr?)rrVrwr:rrr~rCrCrD__del__s zAsyncHTTPResponse.__del__)r!NrrNTTNNNNNTNNTN)$r"rr#r$r%r&r'r&r(r)r*r+r,r+r-r.r/r0r1r2r3r4r5r6r7r+r8r)r9r)r:r+r;r<r=r>)rorpr=r&)r=rt)r=r>)r=r)r=r)rrr=r>)r=r)r=r]r?)rrr=r])NN)rrr,rr=r])NNF)rrr,rrr+r=r]) rrr,rrr+rr+r=r])rN)rrr,rr=r)r=r)__name__ __module__ __qualname__rnrsrr|rpropertyrrrrrrr}rrIrrrwrrrCrCrCrDr +sj i  7      5 ?     r )5 __future__rrrrrtypingrY contextlibrsocketrru _collectionsr_typingr _constantrr r r backend._asyncr exceptionsr rrrrrrrresponserr util.responserr util.retryrr1r TYPE_CHECKING email.messager_async.connectionpoolrcontrib.webextensions._asyncrutil._async.traffic_policerr rCrCrCrDs.      (