o
.j8 @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z ddl
m Z ddl m
Z
dd lmZmZ dd
lmZmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddl m Z ddl m Z ddl!m"Z" ej#ruddlm$Z$m%Z%m&Z&m'Z' G dd dZ(G dd de
Z)dZ*dS ) ) annotationsN) timedelta)run_sync)pyfetch )DEFAULT_RETRIES) BaseAdapter)ConnectionErrorConnectTimeout)PreparedRequestResponse)
MaxRetryError)BytesQueueBuffer) HTTPResponse)Timeout)Retry)CaseInsensitiveDict)get_encoding_from_headers) ProxyType RetryTypeTLSClientCertType
TLSVerifyTypec @ sn e Zd ZdZ d"d#d d
Zd$d d Zd%d
dZ d&d'ddZd(d)ddZd$dd Z d*d d Z
d+d d!Z dS ),
_PyodideRawIOaN File-like wrapper around a Pyodide Fetch response with true streaming via JSPI.
When constructed with a JS Response object, reads chunks incrementally from the
JavaScript ReadableStream using ``run_sync(reader.read())`` per chunk.
When constructed with preloaded content (non-streaming), serves from a memory buffer.
N js_response
typing.Anypreloaded_content bytes | NonereturnNonec C sT || _ t | _d| _|d up|d u | _d | _i | _d | _|d ur(| j| d S d S )NF) _js_responser _buffer_closed _finished_readerheaders extensionput)selfr r r( Y/home/thesage/.local/lib/python3.10/site-packages/niquests/extensions/pyodide/__init__.py__init__" s z_PyodideRawIO.__init__c C sX | j du r(| jdur*z| jj}|dur| | _ W dS W dS ty' Y dS w dS dS )z9Initialize the ReadableStream reader if not already done.N)r# r body getReader Exception)r' r+ r( r( r) _ensure_reader2 s z _PyodideRawIO._ensure_readerc C sh | | jdu r dS z t| j }|jrW dS |j}|dur't| W S W dS ty3 Y dS w )zBRead the next chunk from the JS ReadableStream, blocking via JSPI.N) r. r# r readdonevaluebytesto_pyr- )r' resultr1 r( r( r) _get_next_chunk< s
z _PyodideRawIO._get_next_chunkTamt
int | Nonedecode_contentboolr2 c C sN | j rt| jdkr| jt| jS dS | jr>t| jdkr!dS |d u s)|dk r2| jt| jS | jt|t| jS |d u sF|dk rl | }|d u rPn| j| qGd| _t| jdkrj| jt| jS dS t| j|k r| js| }|d u rd| _n| j| t| j|k r| jrvt| jdkrdS | jt|t| jS )Nr T)r! lenr getr" minr5 r& r' r6 r8 chunkr( r( r) r/ P s6 $ $ z_PyodideRawIO.readint#typing.Generator[bytes, None, None]c c s | |}|s dS |V q)z$Iterate over chunks of the response.TN)r/ r> r( r( r) streamv s
z_PyodideRawIO.streamc C sF d| _ | jd urz t| j W n ty Y nw d | _d | _d S NT)r! r# r cancelr- r r' r( r( r) close~ s
z_PyodideRawIO.closetyping.Iterator[bytes]c C s | S Nr( rE r( r( r) __iter__ z_PyodideRawIO.__iter__c C s | d}|s t|S )Ni )r/
StopIteration)r' r? r( r( r) __next__ s
z_PyodideRawIO.__next__)NN)r r r r r r r r )r r rC )r6 r7 r8 r9 r r2 )T)r6 r@ r8 r9 r rA )r rG )r r2 ) __name__
__module__ __qualname____doc__r* r. r5 r/ rB rF rI rL r( r( r( r) r s
&
r c sv e Zd ZdZefd/ fdd
Zd0d d
Z
d1d2d"d#Zd3d%d&Zd4d(d)Z d5d+d,Z
d6d-d.Z Z S )7PyodideAdapterzMSynchronous adapter for making HTTP requests in Pyodide using JSPI + pyfetch. max_retriesr r r c s. t t|tr|| _d S t|| _d S rH )superr*
isinstancer rS from_int)r' rS __class__r( r) r* s
zPyodideAdapter.__init__strc C dS )Nzr( rE r( r( r) __repr__ rJ zPyodideAdapter.__repr__FNTrequestr rB r9 timeout)int | float | tuple | TimeoutSauce | Noneverifyr certTLSClientCertType | NoneproxiesProxyType | Noneon_post_connection*typing.Callable[[typing.Any], None] | Noneon_upload_body;typing.Callable[[int, int | None, bool, bool], None] | Noneon_early_response(typing.Callable[[Response], None] | None multiplexedr c
C sH t |trt|dkr|d p|d }n|d }n t |tr$|jp#|j}| j} |jp+d} t }
z | |||}W n t
yY } z| j | |j |d} |
W Y d}~q0d}~ww td|j|j|j|j d }t|jd
}| | |j|rz | j | |j |d } W n ty | jr | Y S w |
| q0tt |
d |_|S )
zFSend a PreparedRequest using Pyodide's pyfetch (synchronous via JSPI).r r GETT)errorNr: )r+ r$ statusrequest_method request_urlz Retry-After)response)seconds)rU tupler; TimeoutSaucetotalconnect_timeoutrS methodtime_do_sendr- incrementurlsleepBaseHTTPResponser$ status_coder9 r< is_retryr
raise_on_statusr elapsed)r' r\ rB r] r_ r` rb rd rf rh rj retriesrw startrq err
base_responsehas_retry_afterr( r( r) send sL
zPyodideAdapter.sendint | float | Nonec
C s |j pd}d|v r|dd nd}|dv r | ||S |dv r)| |||S i }|jrV|j D ]"\}}| dvrUt|trD|n| d|t|trP|n| d< q3|j
} | durt| trh| d } n1t| t j
rt| ttfsg }
| D ]} t| tr|
| d qyt| tr|
| qyd
|
} |jpd |d } | r| | d
< d}
|durddlm} |t|d }
ztt|j fd|
i| }W n2 ty } z&t| }d|v sd|v sd|v rtd|j dtd|j d| d}~ww i }zLt |drK|j}t |dr5| D ] \}}t|tr |n| d|t|tr-|n| d< qnt |drK| D ] }|d ||d < q?W n
tyV Y nw t }|j |_ t!||_||_"|j pl|j |_ t#||_$z |j%pyd|_&W n
ty d|_&Y nw |rt'|j(d }||_||_)d |_*d |_+|S zt| }W n ty d
}Y nw t'|d }||_||_)||_*|S ) zCPerform the actual request using pyfetch made synchronous via JSPI. z://r )wswss)ssepsse)hostcontent-length
connectionztransfer-encodinglatin-1Nzutf-8r: rl )rw r$ r+ ) AbortSignali signalabortr] z timed outzConnection to z
timed outzFailed to fetch z: r$ itemsentries )r F)r ),r{ splitlower _do_send_ws _do_send_sser$ r rU rY decoder+ encodetypingIterabler2 bytearrayappendjoinrw jsr r] r@ r r r- r
r hasattrr r rn r~ r r\ r encoding status_textreasonr r raw_content_content_consumed)r' r\ rB r] r{ scheme headers_dictkeyr1 r+ chunksr?
fetch_optionsr r r eerr_strresponse_headers
js_headersentryrq raw_io
response_bodyr( r( r) ry s
zPyodideAdapter._do_sendr{ c
C s ddl m} z||}W n ty" } z
td| d| d}~ww t }d|_tddd |_||_||_ d
|_
t }||_ ||_
d |_|S ) z>Handle WebSocket connections via browser native WebSocket API.r )PyodideWebSocketExtensionzWebSocket connection to failed: Ne websocketupgrade)r r zSwitching Protocolsr: )_wsr r- r r r~ r r$ r\ r{ r r r% r r )r' r\ r{ r extr rq r r( r( r) r Q s$ zPyodideAdapter._do_send_wsr c
C s ddl m} |dkr|dddn|ddd}i }|jrE|j D ]"\}}| dvrDt|tr3|n|d |t|tr?|n|d < q"z|||d
} W n t yc }
z
t
d | d |
d
}
~
ww t } d| _ t
ddi| _|| _|| _d| _t } | | _| | _d| _d| _| S )z>Handle SSE connections via pyfetch streaming + manual parsing.r )PyodideSSEExtensionr zsse://zhttps://zpsse://zhttp://)r r r r )r$ zSSE connection to r N z content-typeztext/event-streamOKF)_sser replacer$ r r rU rY r r- r r r~ r r\ r{ r r r% r r r )
r' r\ r{ r r http_urlr r r1 r r rq r r( r( r) r h s6 $ zPyodideAdapter._do_send_ssec C rZ )zClean up adapter resources.Nr( rE r( r( r) rF s zPyodideAdapter.close)rS r r r )r rY ) FNTNNNNNF)r\ r rB r9 r] r^ r_ r r` ra rb rc rd re rf rg rh ri rj r9 r r )r\ r rB r9 r] r r r )r\ r r{ rY r r )r\ r r{ rY r rY r r rM )
rN rO rP rQ r r* r[ r ry r r rF
__classcell__r( r( rW r) rR s$
;
v
#rR )rR )+
__future__r rx r datetimer pyodide.ffir pyodide.httpr _constantr adaptersr
exceptionsr r
modelsr r packages.urllib3.exceptionsr
packages.urllib3.responser r r} packages.urllib3.utilr rt packages.urllib3.util.retryr
structuresr utilsr
TYPE_CHECKINGr r r r r rR __all__r( r( r( r) s. x