o
.j+ @ s d dl mZ d dlZddlmZ ddlmZ ddlmZm Z ddl
m Z ddl m
Z
dd l mZ dd
lmZ dd lmZ dd lmZ ejrSdd
lmZmZmZmZm Z d dl m Z G dd dZ G dd deZ dS ) ) annotationsN )DEFAULT_RETRIES) BaseAdapter)PreparedRequestResponse)
MaxRetryError)BytesQueueBuffer) HTTPResponse)Timeout)Retry)CaseInsensitiveDict) ProxyType RetryTypeTLSClientCertType
TLSVerifyTypeWSGIApp)BytesIOc @ sT e Zd ZdZd dd Z
d d ddZd d!ddZd"ddZd#ddZd$dd Z d
S )%
_WSGIRawIOz@File-like wrapper around a WSGI response iterator for streaming. generator#typing.Generator[bytes, None, None]headerslist[tuple[str, str]]returnNonec C s$ || _ t | _d| _|| _d | _d S )NF)
_generatorr _buffer_closedr extension)selfr r r U/home/thesage/.local/lib/python3.10/site-packages/niquests/extensions/sgi/__init__.py__init__ s
z_WSGIRawIO.__init__NTamt
int | Nonedecode_contentboolbytesc C s | j rdS |d u s
|dk r"| jD ]}| j| q| jt| jS t| j|k rEz | jt| j W n ty= Y nw t| j|k s)t| jdkrNdS | jt|t| jS )N r ) r r r putgetlennext
StopIterationminr r# r% chunkr r r! read s
z_WSGIRawIO.readintc c s | |}|s dS |V q)z$Iterate over chunks of the response.TN)r1 r/ r r r! stream9 s
z_WSGIRawIO.streamc C s$ d| _ t| jdr| j d S d S )NTclose)r hasattrr r4 r r r r! r4 A s z_WSGIRawIO.closetyping.Iterator[bytes]c C s | S Nr r6 r r r! __iter__F z_WSGIRawIO.__iter__c C s | d}|s t|S )Ni )r1 r- )r r0 r r r! __next__I s
z_WSGIRawIO.__next__)r r r r r r )NT)r# r$ r% r& r r' )T)r# r2 r% r& r r r r )r r7 )r r' )
__name__
__module__ __qualname____doc__r" r1 r3 r4 r9 r; 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 )7WebServerGatewayInterfacez:Adapter for making requests to WSGI applications directly.appr max_retriesr r r c s4 t || _t|tr|| _dS t|| _dS )z
Initialize the WSGI adapter.
:param app: A WSGI application callable.
:param max_retries: Maximum number of retries for requests.
N)superr" rB
isinstancer rC from_int)r rB rC __class__r r! r" S s
z"WebServerGatewayInterface.__init__strc C dS )Nzr r6 r r r! __repr__b r: z"WebServerGatewayInterface.__repr__FNTrequestr r3 r& 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 s@ t |trt|dkr|d p|d }n|d }n t |tr$|jp#|j}| j} |jp+d} z| ||}
W n* t y_ } z z | j
| |j |d} W n t yP w |
W Y d}~q,d}~ww td|
j|
j|j|j d }t|
jd
}| | |
j|rz | j
| |j |d } W n t y | jr |
Y S w |
| q,|
S ) z/Send a PreparedRequest to the WSGI application.r r GETT)errorNr( )bodyr statusrequest_method request_urlz Retry-After)response)rE tupler+ TimeoutSaucetotalconnect_timeoutrC method_do_send Exception incrementurlr sleepBaseHTTPResponser status_coder& r* is_retryraise_on_status)r rL r3 rM rO rP rR rT rV rX rZ retriesrg rb err
base_responsehas_retry_afterr r r! sende sP
z WebServerGatewayInterface.sendc
s ddl m} ddlm} ||j}|jdkrd}nd}|j}|j|j d| dd|_| |}||_d g dd fdd
}| ||fdd} || }
t } | _
t | _ || _
|| _| j dd| _ttg } |
| _| | _d| _d| _| S )zCHandle SSE requests by wrapping the WSGI response with SSE parsing.r urlparse )WSGISSEExtensionssehttpshttpz://Nr_ rI r r c t | ddd | d S N rx r r2 splitr_ r exc_inforesponse_headersrn r r! start_response z>WebServerGatewayInterface._do_send_sse..start_responsec 3 @ z E d H W t dr d S d S t dr w w Nr4 r5 r4 r resultr r! generate
z8WebServerGatewayInterface._do_send_sse..generate content-typeutf-8Fr8 r_ rI r r ) urllib.parserw _ssery rk schemereplace_create_environrB r rn r
r rL r* encodingr iterr raw_content_content_consumed)
r rL rw ry parsed http_scheme original_urlenvironr r extrb raw_ior r r rn r! _do_send_sse s8
z&WebServerGatewayInterface._do_send_ssec s$ ddl m} ||j}|jdv rtd|jdv r | |S | |}dg dd fd d
}| ||t }|_ t
|_ ||_ |j|_|j
d
d|_fdd}t| |_|rgd|_d|_|S g } zD ]}
| |
qlW tdr~ n tdr w w d| |_|S )z Perform the actual WSGI request.r rv )wswssz$WebSocket is not supported over WSGI)rz psseNr_ rI r r c r} r~ r r r r r! r r z:WebServerGatewayInterface._do_send..start_responser r c 3 r r r r r r r! r r z4WebServerGatewayInterface._do_send..generateFr4 r( r8 r )r rw rk r NotImplementedErrorr r rB r rn r
r rL r* r r r r r appendr5 r4 join) r rL r3 rw r r r rb r body_chunksr0 r r r! rh sH
z"WebServerGatewayInterface._do_senddictc C sR ddl m}m} ||j}|jpd}t|tr |d}n t|tj r9t|tt
t t fs9d}|D ]}||7 }q0|}|j
d||jpBd|jpFd|jpJdt|jpV|jdkrUd nd
d d |jp]d
t|t dddtt|d}|jr|j D ]1\} }
t| tr| n| d} | dd} | dkr|
|d< qu| dkr|
|d< qu|
|d| < qu|S )z2Create a WSGI environ dict from a PreparedRequest.r )unquoterw r( r / localhostr{ i P zHTTP/1.1)rx r r| TF)REQUEST_METHOD SCRIPT_NAME PATH_INFO QUERY_STRING SERVER_NAME SERVER_PORTSERVER_PROTOCOLz wsgi.versionzwsgi.url_schemez
wsgi.inputz wsgi.errorszwsgi.multithreadzwsgi.multiprocessz
wsgi.run_onceCONTENT_LENGTHzlatin-1-_ CONTENT_TYPEr HTTP_)r r rw rk r^ rE rI encodetypingIterabler' bytearrayrc rg pathqueryhostnameportr r r+ r itemsdecodeupperr ) r rL r rw r r^ tmpr0 r keyvalue key_upperr r r! r sH
z)WebServerGatewayInterface._create_environc C rJ )zClean up adapter resources.Nr r6 r r r! r4 D s z WebServerGatewayInterface.close)rB r rC r r r )r rI ) FNTNNNNNF)rL r r3 r& rM rN rO r rP rQ rR rS rT rU rV rW rX rY rZ r& r r )rL r r r )rL r r3 r& r r )rL r r r r< )
r= r> r? r@ r r" rK ru r rh r r4
__classcell__r r rG r! rA P s$
?
5
:1rA )!
__future__r r _constantr adaptersr modelsr r packages.urllib3.exceptionsr packages.urllib3.responser r
rm packages.urllib3.utilr rd packages.urllib3.util.retryr
structuresr
TYPE_CHECKINGr r r r r ior r rA r r r r! s ;