o ߥi2@sddlZddlZddlmZmZmZmZmZmZddl m Z ddlm Z m Z m Z mZmZddlmZddlmZmZGd d d eeZGd d d eeeefZGd ddeZdeeeegeefgeefdee ge ffddZdS)N)AnyAsyncGenerator AsyncIterator AwaitableCallableMutableMapping)Headers)ASGIAppScopeReceiveSendMessage)Request)ResponseStreamingResponsec@s@eZdZd ddZdeddfddZd dd Zdefd d ZdS) PipeStreamreturnNcCstjdd|_d|_dS)Nr)maxsizeF)asyncioQueue_queue _pushed_eofselfrI/home/ubuntu/.local/lib/python3.10/site-packages/baize/asgi/middleware.py__init__s zPipeStream.__init__chunkcs&|jrtd|j|IdHdS)Nz!Cannot push chunk after push EOF.)r RuntimeErrorrputrrrrrpushszPipeStream.pushcs d|_dS)NT)rrrrrpush_eofs zPipeStream.push_eofcs&|jr |jr t|jIdHSN)rremptyStopAsyncIterationgetrrrr __anext__szPipeStream.__anext__)rN)__name__ __module__ __qualname__rbytesr#r$r)rrrrrs   rc@sFeZdZdededdfddZdeddfddZdeefd d Z dS) NextRequestnamevaluerNcCs||j|<dSr%_scope)rr/r0rrr __setitem__&szNextRequest.__setitem__cCs |j|=dSr%r1)rr/rrr __delitem__)s zNextRequest.__delitem__cCstd)Nz'Cannot read request body in middleware.)r rrrrstream,szNextRequest.stream) r*r+r,strrr3r4rr-r5rrrrr.%sr.c@s@eZdZdZdeedffddZedede ddfdd Z dS) NextResponsez3 This is a response object for middleware. rNcCs"|j2z 3dHW}|Vq6dSr%)iterabler"rrr render_stream5szNextResponse.render_streamapprequestcsLdttdtddffdd }|||j|IdHtS)zc This is a helper method to convert a ASGI application into a NextResponse object. messagerNcs~|ddkr|dtdd|dgDdS|ddkr;|dd IdH|d d s=IdHdSdSdS) Ntypezhttp.response.startstatuscSs$g|]\}}|d|dfqS)zlatin-1)decode).0kvrrr Hsz7NextResponse.from_app..send..headerszhttp.response.bodybody more_bodyF)r r(r#r$)r=rFrE status_coderrsendBs    z#NextResponse.from_app..send)r rr_receiver7)clsr:r;rKrrIrfrom_app9s zNextResponse.from_app) r*r+r,__doc__rr-r9 classmethodr r.rNrrrrr70s r7handlerrcs$tdtdtffdd }|S)a This can turn a callable object into a middleware for ASGI application. ```python @middleware async def m( request: NextRequest, next_call: Callable[[NextRequest], Awaitable[NextResponse]] ) -> Response: ... response = await next_call(request) ... return response @m @request_response async def v(request: Request) -> Response: ... # OR @m async def asgi(scope: Scope, receive: Receive, send: Send) -> None: ... ``` r:rc s.tdtdtdtddffdd }|S)z0 This is the actual middleware. scopereceiverKrNcsJt|||}dtdtffdd }||IdH}||||IdHdS)Nr;rcst|IdHSr%)r7rN)r;r:rr next_callsz6middleware..d..asgi..next_call)r.r7)rRrSrKr;rUresponse)r:rQrrasgi|s  z#middleware..d..asgi) functoolswrapsr r r )r:rWrQrTrdvs" zmiddleware..d)rXrYr )rQr[rrZr middlewareVs r\)rrXtypingrrrrrrdatastructuresr r r r r rrequestsr responsesrrr-rr6r.r7r\rrrrs&    &