o ;ivU @sdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddlm Z m Z m Z mZmZmZmZmZmZmZmZddlmZmZejdkrTddlmZnddlmZGd d d eZGd d d eZGd ddeZGdddeZeeefZ eeeefZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%GdddeZ&GdddeZ'GdddeZ(Gdd d eZ)Gd!d"d"eZ*Gd#d$d$eZ+Gd%d&d&eZ,Gd'd(d(eZ-Gd)d*d*eZ.Gd+d,d,eZ/Gd-d.d.eZ0Gd/d0d0eZ1ee"e%e&e(e*e,e-fZ2ee#e$e%e'e)e+e.e/e0e1f Z3e gee2fZ4e e3gedfZ5e e!e4e5gedfZ6Gd1d2d2eZ7Gd3d4d4ed5d6Z8Gd7d8d8eZ9Gd9d:d:eZ:ed;e;ee>e>d<Z?Gd=d>d>e7e8e?Z@e;eAeBeBee fZCe eDgdfZEGd?d@d@eZFeeDZGe e@eFgeGfZHGdAdBdBZIeJdCZKZLdDdEZMdFedGeIdHe@fdIdJZNGdKdLdLZOGdMdNdNZPdS)Oa Vendored version of a2wsgi v1.10.2. We vendor only a2wsgi/wsgi.py, plus type annotations, to convert WSGI apps into ASGI protocol versions using the `WSGIMiddleware` class. This is a well-tested library marked as an optional dependency of uvicorn. It doesn't have the issues with buffering request streams that asgiref has, and it also is simpler, only requiring a standard `concurrent.futures.ThreadPoolExecutor`. --- Copyright 2022 abersheeran Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. N)ThreadPoolExecutor) TracebackType) AnyCallableDictListLiteralOptionalProtocolTupleType TypedDictUnion) AwaitableIterable) ) NotRequiredc@s"eZdZUeed<eded<dS) ASGIVersions spec_versionz3.0versionN)__name__ __module__ __qualname__str__annotations__rrrM/home/ubuntu/.local/lib/python3.10/site-packages/modal/_vendor/a2wsgi_wsgi.pyr<s rc@seZdZUeded<eed<eed<eed<eed<eed<eeed<eed <eed <e e eefed <ee ee fed <ee ee e fed <ee eefed<ee ee eeffed<dS) HTTPScopehttptypeasgi http_versionmethodschemepathraw_path query_string root_pathheadersclientserverstate extensionsN)rrrrrrrrbytesrtupleintr dictrobjectrrrrrAs   rc@seZdZUeded<eed<eed<eed<eed<eed<eed<eed <ee eefed <e e ee fed <e e ee e fed <eeed <e e eefed<e e ee eeffed<dS)WebSocketScope websocketr r!r"r$r%r&r'r(r)r*r+ subprotocolsr,r-N)rrrrrrrr.rr/rr0r r1rr2rrrrr3Rs   r3c@s6eZdZUeded<eed<eeee fed<dS) LifespanScopelifespanr r!r,N) rrrrrrrr1rrrrrrr6cs r6c@s.eZdZUeded<eed<eeed<dS)HTTPRequestEventz http.requestr body more_bodyN)rrrrrr.rboolrrrrr8ms r8c@sFeZdZUeded<eed<eeee e fed<ee ed<dS)HTTPResponseStartEventhttp.response.startr statusr)trailersN) rrrrrr0rrr/r.r;rrrrr<ss r<c@2eZdZUeded<eeed<eeed<dS)HTTPResponseBodyEventhttp.response.bodyr r9r:N)rrrrrrr.r;rrrrrAz  rAc@eZdZUeded<dS)HTTPDisconnectEventzhttp.disconnectr NrrrrrrrrrrE rEc@rD)WebSocketConnectEventzwebsocket.connectr NrFrrrrrHrGrHc@s>eZdZUeded<eeed<eeee e fed<dS)WebSocketAcceptEventzwebsocket.acceptr subprotocolr)N) rrrrrrrrr/r.rrrrrIs  rIc@r@)WebSocketReceiveEventzwebsocket.receiver r.textNrrrrrrr.rrrrrrKrCrKc@r@)WebSocketSendEventzwebsocket.sendr r.rLNrMrrrrrNrCrNc@"eZdZUeded<eed<dS)WebSocketDisconnectEventzwebsocket.disconnectr codeN)rrrrrr0rrrrrP  rPc@r@)WebSocketCloseEventwebsocket.closer rQreasonN)rrrrrrr0rrrrrrSrCrSc@rD)LifespanStartupEventlifespan.startupr NrFrrrrrVrGrVc@rD)LifespanShutdownEventlifespan.shutdownr NrFrrrrrXrGrXc@rD)LifespanStartupCompleteEventlifespan.startup.completer NrFrrrrrZrGrZc@rO)LifespanStartupFailedEventzlifespan.startup.failedr messageNrrrrrrrrrrr\rRr\c@rD)LifespanShutdownCompleteEventlifespan.shutdown.completer NrFrrrrr_rGr_c@rO)LifespanShutdownFailedEventzlifespan.shutdown.failedr r]Nr^rrrrrarRrac@s.eZdZUeed<eed<eed<eed<dS)CGIRequiredDefinedREQUEST_METHOD SERVER_NAME SERVER_PORTSERVER_PROTOCOLNrrrrrrrrrrbs  rbc@sNeZdZUeed<eed<eed<eed<eed<eed<eed<eed<d S) CGIOptionalDefined REQUEST_URI REMOTE_ADDR REMOTE_PORT SCRIPT_NAME PATH_INFO QUERY_STRING CONTENT_TYPECONTENT_LENGTHNrgrrrrrhs  rhF)totalc@sPeZdZdZddedefddZddedefdd Zdd edeefd d Z d S) InputStreamas An input stream (file-like object) from which the HTTP request body bytes can be read. (The server or gateway may perform reads on-demand as requested by the application, or it may pre- read the client's request body and buffer it in-memory or on disk, or use any other technique for providing such an input stream, according to its preference.) sizereturncCt)aB The server is not required to read past the client's specified Content-Length, and should simulate an end-of-file condition if the application attempts to read past that point. The application should not attempt to read more data than is specified by the CONTENT_LENGTH variable. A server should allow read() to be called without an argument, and return the remainder of the client's input stream. A server should return empty bytestrings from any attempt to read from an empty or exhausted input stream. NotImplementedError)selfrtrrrreads zInputStream.readlimitcCrv)a Servers should support the optional "size" argument to readline(), but as in WSGI 1.0, they are allowed to omit support for it. (In WSGI 1.0, the size argument was not supported, on the grounds that it might have been complex to implement, and was not often used in practice... but then the cgi module started using it, and so practical servers had to start supporting it anyway!) rw)ryr{rrrreadline"s zInputStream.readlinehintcCrv)z Note that the hint argument to readlines() is optional for both caller and implementer. The application is free not to supply it, and the server or gateway is free to ignore it. rw)ryr}rrr readlines-szInputStream.readlinesNrs) rrr__doc__r0r.rzr|listr~rrrrrr s   rrc@sBeZdZdZd ddZdedefddZd eedefd d Z dS) ErrorStreama An output stream (file-like object) to which error output can be written, for the purpose of recording program or other errors in a standardized and possibly centralized location. This should be a "text mode" stream; i.e., applications should use " " as a line ending, and assume that it will be converted to the correct line ending by the server/gateway. (On platforms where the str type is unicode, the error stream should accept and log arbitrary unicode without raising an error; it is allowed, however, to substitute characters that cannot be rendered in the stream's encoding.) For many servers, wsgi.errors will be the server's main error log. Alternatively, this may be sys.stderr, or a log file of some sort. The server's documentation should include an explanation of how to configure this or where to find the recorded output. A server or gateway may supply different error streams to different applications, if this is desired. ruNcCrv)a Since the errors stream may not be rewound, servers and gateways are free to forward write operations immediately, without buffering. In this case, the flush() method may be a no-op. Portable applications, however, cannot assume that output is unbuffered or that flush() is a no-op. They must call flush() if they need to ensure that output has in fact been written. (For example, to minimize intermingling of data from multiple processes writing to the same error log.) rwryrrrflushGs zErrorStream.flushscCrvNrw)ryrrrrwriteSzErrorStream.writeseqcCrvrrw)ryrrrr writelinesVrzErrorStream.writelines)ruN) rrrrrrrrrrrrrrr6s   r WSGIDefined) wsgi.versionwsgi.url_scheme wsgi.input wsgi.errorswsgi.multithreadwsgi.multiprocess wsgi.run_oncec@seZdZdZdS)Environz WSGI Environ N)rrrrrrrrrpsrc @s:eZdZ ddedeeeefdeedefddZ dS) StartResponseNr>response_headersexc_inforucCrvrrw)ryr>rrrrr__call__}szStartResponse.__call__r) rrrrrr/r ExceptionInfo WriteCallablerrrrrr|src@seZdZdejdeddfddZedefddZ de fd d Z dd e de fd dZ dde de fddZdde dee fddZdeje ddffddZdS)BodyloopreceiveruNcCst|_||_||_d|_dS)NT) bytearraybufferrr _has_more)ryrrrrr__init__s z Body.__init__cCs|js|jrdSdS)NTF)rrrrrrhas_mores z Body.has_morecCs@|jsdStj||jd}|}|dd|_|ddS)Nrr:Fr9)rasynciorun_coroutine_threadsaferrresultget)ryfuturer]rrr_receive_more_datas  zBody._receive_more_datarsrtcCs|dks |t|jkr"|j||jsn |dks |t|jks |dkr2t|j}|j|St|jd|}|jd|=|S)Nrs)lenrextendrrr.clear)ryrtrrrrrzs   z Body.readr{cCs |jdd|dkr |nd}|dkr(t|jd|d}|jd|d=|S|dkr=t|jd|}|jd|=|S|jsAn |j|qt|j}|j|S)NT rrs)rfindr.rrrr)ryr{lf_indexrrrrr|s"   z Body.readliner}cshjsgS|dkr)d}|d}|ddkr"|t|ddd|DSfddt|DS)Nrsr rcSsg|]}|dqS)rr).0linerrr z"Body.readlines..csg|]}qSr)r|)r_rrrrr)rrzsplitpoprrange)ryr}raw_databytelistrrrr~s   zBody.readlinesccs |jr|V|jsdSdSr)rr|rrrr__iter__s z Body.__iter__r)rrrrAbstractEventLoopReceiverpropertyr;rr.rr0rzr|rr~typing Generatorrrrrrrs  rsurrogateescapecCs|ttdS)zCConvert an environment variable to a WSGI "bytes-as-unicode" stringz iso-8859-1)encodeENCESCdecode)urrrunicode_to_wsgisrscoper9rucCsz|dddd}|ddd}||r#|t|d}tjdd}|r0t|}||d|||d d d |d d |dd|tj dddd }|dpWd\}}||d<t |pbd|d<|d}|dur}|\} } | |d<t | |d<|dgD]7\} } | d} | dkrd} n| dkrd} n d |  d!d"} | d} | |vr|| d#| } | || <q|S)$zE Builds a scope and request body into a WSGI environ object. r(utf8latin1r%Nrlr#r'asciizHTTP/r")rrr$rTF) z asgi.scopercrlrmrnrfrrrrrrrr+) localhostPrdrrer*rjrkr)zcontent-lengthrpz content-typeroHTTP_-r,) rrr startswithrosenvironrsysstdoutrupperreplace)rr9 script_name path_infoscript_name_environ_varr server_addr server_portr*addrportnamevaluecorrected_namerrr build_environsR         rc @sHeZdZdZ ddedededdfdd Zd ed ed e ddfd dZ dS)WSGIMiddlewarez% Convert WSGIApp to ASGIApp. rappworkerssend_queue_sizeruNcCs||_||_td|d|_dS)NWSGI)thread_name_prefix max_workers)rrrexecutor)ryrrrrrrrs  zWSGIMiddleware.__init__rrsendcs|ddkrt|j|j|j}||||IdHS|ddkr+|dddIdHdS|ddkra|IdH}|ddks?J|dd iIdH|IdH}|dd ksVJ|dd iIdHdSdS) Nr rr4rTi)r rQr7rWr[rYr`) WSGIResponderrrr)ryrrr responderr]rrrr%s      zWSGIMiddleware.__call__)rr) rrrrWSGIAppr0rScoperSendrrrrrrs  rc @seZdZdedededdfddZded ed e ddfd d Z d e j e ddfddZd e ddfddZ ddedeeeefde j edefddZdededdfddZdS)rrrrruNcCs2||_||_t|_t||_d|_d|_dS)NF) rrrget_event_looprQueue send_queueresponse_startedr)ryrrrrrrr9s    zWSGIResponder.__init__rrrc st|j|}t||}d}zb|j||}t}t|j |j }|j |j |||j IdH|jdIdH|jIdHt|dIdH|jdur`|jd|jd|jdW|rm|so|dSdSdS|r||s}|www)Nrr)rrr create_tasksender contextvars copy_context functoolspartialrunwsgirun_in_executorrstart_responserputjoinrwait_forrwith_tracebackdonecancel) ryrrrr9rrcontextfuncrrrrCs0         zWSGIResponder.__call__r]cCs$tj|j||jd}|dS)Nr)rrrrrr)ryr]rrrrrYs   zWSGIResponder.sendcs: |jIdH}|j|durdS||IdHqr)rr task_done)ryrr]rrrr`s zWSGIResponder.senderr>rrcsV|_js%d_|dd\}}t|}dd|D}d||dfdd S) NT rcSs0g|]\}}|d|dfqS)r)striprlower)rrrrrrrssz0WSGIResponder.start_response..r=)r r>r)csd|ddS)NrBTr r9r:)r)chunkrrr~s z.WSGIResponder.start_response..)rrrr0r)ryr>rrstatus_code_stringr status_coder)rrrrhs zWSGIResponder.start_responserrc Csf|||}z"|D] }|d|ddq |dddWt|ddddSt|dddw) NrBTrr)r r9closecSsdSrrrrrrrsz$WSGIResponder.wsgi..)rrgetattr)ryrriterablerrrrrs  *zWSGIResponder.wsgir)rrrrrr0rrrrrrr SendEventrrrrr/rrrrrrrrrrr8s0    r)Qrrrrrrrconcurrent.futuresrtypesrrrrrrr r r r r rcollections.abcrr version_infortyping_extensionsrrr3r6WWWScoperr8r<rArErHrIrKrNrPrSrVrXrZr\r_ra ReceiveEventrrrASGIApprbrhrrrr/r0rr;rrr BaseExceptionrr.rrIterableChunksrrgetfilesystemencodingrrrrrrrrrrs  4     *$  E:!