U
|
Þ=®d ã @ s d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl
|
m Z dd lm Z dd
|
l mZ ejrdd lmZ dd lmZ dd lmZ G dd dZdS )zt
|
Basic HTTP Proxy
|
================
|
|
.. autoclass:: ProxyMiddleware
|
|
:copyright: 2007 Pallets
|
:license: BSD-3-Clause
|
é )Ú annotationsN)Úclient)Úquote)Úurlsplité )ÚEnvironHeaders)Úis_hop_by_hop_header)Úget_input_stream)Ú StartResponse)ÚWSGIApplication)ÚWSGIEnvironmentc @ sN e Zd ZdZdddddddd d
|
Zd d d dd ddZddddddZdS )ÚProxyMiddlewareaÁ Proxy requests under a path to an external server, routing other
|
requests to the app.
|
|
This middleware can only proxy HTTP requests, as HTTP is the only
|
protocol handled by the WSGI server. Other protocols, such as
|
WebSocket requests, cannot be proxied at this layer. This should
|
only be used for development, in production a real proxy server
|
should be used.
|
|
The middleware takes a dict mapping a path prefix to a dict
|
describing the host to be proxied to::
|
|
app = ProxyMiddleware(app, {
|
"/static/": {
|
"target": "http://127.0.0.1:5001/",
|
}
|
})
|
|
Each host has the following options:
|
|
``target``:
|
The target URL to dispatch to. This is required.
|
``remove_prefix``:
|
Whether to remove the prefix from the URL before dispatching it
|
to the target. The default is ``False``.
|
``host``:
|
``"<auto>"`` (default):
|
The host header is automatically rewritten to the URL of the
|
target.
|
``None``:
|
The host header is unmodified from the client request.
|
Any other value:
|
The host header is overwritten with the value.
|
``headers``:
|
A dictionary of headers to be sent with the request to the
|
target. The default is ``{}``.
|
``ssl_context``:
|
A :class:`ssl.SSLContext` defining how to verify requests if the
|
target is HTTPS. The default is ``None``.
|
|
In the example above, everything under ``"/static/"`` is proxied to
|
the server on port 5001. The host header is rewritten to the target,
|
and the ``"/static/"`` prefix is removed from the URLs.
|
|
:param app: The WSGI application to wrap.
|
:param targets: Proxy target configurations. See description above.
|
:param chunk_size: Size of chunks to read from input stream and
|
write to target.
|
:param timeout: Seconds before an operation to a target fails.
|
|
.. versionadded:: 0.14
|
é @ é
|
r z t.Mapping[str, dict[str, t.Any]]ÚintÚNone)ÚappÚtargetsÚ
|
chunk_sizeÚtimeoutÚreturnc s> ddddd || _ fdd| ¡ D | _|| _|| _d S )Núdict[str, t.Any])Úoptsr c S s4 | dd¡ | dd¡ | di ¡ | dd ¡ | S )NÚ remove_prefixFÚhostú<auto>ÚheadersÚ ssl_context)Ú
|
setdefault)r © r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\werkzeug/middleware/http_proxy.pyÚ _set_defaultsX s
|
z/ProxyMiddleware.__init__.<locals>._set_defaultsc s( i | ] \}}d | d ¡ d |qS )ú/)Ústrip©Ú.0ÚkÚv©r! r r Ú
|
<dictcomp>` s z,ProxyMiddleware.__init__.<locals>.<dictcomp>)r Úitemsr r r )Úselfr r r r r r( r Ú__init__Q s
|
ÿzProxyMiddleware.__init__r Ústr)r ÚpathÚprefixr c sB t d j d¡ d¡ dddd fdd }|S )
|
NÚtargetÚidnaÚasciir r
|
út.Iterable[bytes]©ÚenvironÚstart_responser c s° t t| ¡ }dd |D |d d
< | d¡ d dkrL| df¡ n2d d krl| d| d f¡ n| dd f¡ | d ¡ ¡ }d rÈ|td
d
|
¡}j d
|
¡ d
|
| }| d ¡}d }|d krî| d|f¡ n|d k r| d¡ d}z8j
|
dkr2t j j p&djd}n@j
|
dkr`t jj pNdjd d}ntdj
|
d| ¡ t|dd}| d }|r¢| d | }|j| d |dd |D ](\} }
|
| ¡ d krÔd!}
|
| | |
|
¡ qº| ¡ t| } | j¡} | s
|
q6|r(| d"t| | f ¡ n
|
| | ¡ qô| ¡ W n. t k
|
rn d#d$l m } | | | Y S X | j d% j d&d !¡ D d'd( fd)d* }| S )+Nc S s, g | ]$\}}t |s| ¡ d kr||fqS ))zcontent-lengthr )r Úlowerr$ r r r Ú
|
<listcomp>q s ýzAProxyMiddleware.proxy_to.<locals>.application.<locals>.<listcomp>)Ú
|
ConnectionÚcloser r ÚHostZ HTTP_HOSTr r r" ÚCONTENT_LENGTHF)Ú NzContent-Length)zTransfer-EncodingÚchunkedTÚhttpéP )r Úhttpsi» r )r Úcontextz-Target scheme must be 'http' or 'https', got Ú.z!$&'()*+,/:;=@%)ÚsafeÚ QUERY_STRINGú?ÚREQUEST_METHOD)Ú skip_hostÚ
|
connectionr: s %x
|
%s
|
r )Ú
|
BadGatewayú c S s$ g | ] \}}t |s| ¡ |fqS r )r Útitler$ r r r r8 É s þzt.Iterator[bytes])r c 3 s< z j¡} W n tk
|
r( Y q8Y nX | s0q8| V q d S )N)Úreadr ÚOSError)Údata)Úrespr+ r r rM Ð s
|
z;ProxyMiddleware.proxy_to.<locals>.application.<locals>.read)"Úlistr r* ÚappendÚextendÚlenÚlstripr. ÚrstripÚgetÚschemer ÚHTTPConnectionÚportr ÚHTTPSConnectionÚ RuntimeErrorÚconnectr Ú
|
putrequestr7 Ú putheaderÚ
|
endheadersr rM r ÚsendÚ getresponserN Ú
|
exceptionsrJ ÚstatusÚreasonÚ
|
getheaders)r5 r6 r Z remote_pathÚcontent_lengthr> ÚconÚ
|
remote_urlZ querystringr& r' ÚstreamrO rJ rM ©r r r. r/ r+ r0 )rP r Ú applicationm s þ
|
|
|
|
|
ÿ
|
ü ÿ þþ z-ProxyMiddleware.proxy_to.<locals>.application)r ÚhostnameÚencodeÚdecode)r+ r r. r/ rl r rk r Úproxy_tof s qzProxyMiddleware.proxy_tor r
|
r3 r4 c C sH |d }| j }| j ¡ D ]$\}}| |¡r| |||¡} q>q|||S )NZ PATH_INFO)r r r* Ú
|
startswithrp )r+ r5 r6 r. r r/ r r r r Ú__call__à s
|
zProxyMiddleware.__call__N)r r )Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r, rp rr r r r r r s 9 ûzr )rv Ú
|
__future__r ÚtypingÚtr? r Ú urllib.parser r Zdatastructuresr r Zwsgir Ú TYPE_CHECKINGZ_typeshed.wsgir
|
r r r r r r r Ú<module> s
|
