o ߥi@sddlZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZddlmZddlmZddlmZmZmZdd lmZd d l m!Z!m"Z"Gd d d e#Z$Gdddee%efeZ&Gddde&Z'dS)N)Any AsyncIteratorDictIteratorMapping) parse_qsl)URLAddressFormDataHeaders QueryParams UploadFile) MalformedJSONMalformedMultipartUnsupportedMediaType)parse_async_stream)MoreInfoFromHeaderMixin)ReceiveScopeSend)cached_property) empty_receive empty_sendc@seZdZdZdS)ClientDisconnectz' HTTP connection disconnected. N)__name__ __module__ __qualname____doc__rrG/home/ubuntu/.local/lib/python3.10/site-packages/baize/asgi/requests.pyrsrc @seZdZdZeefdedededdfddZ d e de fd d Z de e fd d ZdefddZde defddZedefddZedefddZedee e ffddZedefddZedefddZdS)HTTPConnectionz A base class for incoming HTTP connections. It is a valid Mapping type that allows you to directly access the values in any ASGI `scope` dictionary. scopereceivesendreturnNcCs||_||_||_dSN)_scope_send_receiveselfr"r#r$rrr __init__%s zHTTPConnection.__init__keycCs |j|Sr&)r')r+r-rrr __getitem__, zHTTPConnection.__getitem__cC t|jSr&)iterr'r+rrr __iter__/r/zHTTPConnection.__iter__cCr0r&)lenr'r2rrr __len__2r/zHTTPConnection.__len__othercCs4t||jstS|j|jko|j|jko|j|jkSr&) isinstance __class__NotImplementedr'r(r))r+r6rrr __eq__5s    zHTTPConnection.__eq__cCs|dpd\}}t||dS)z Client's IP and Port. Note that this depends on the "client" value given by the ASGI Server, and is not necessarily accurate. client)NN)hostport)getr )r+r<r=rrr r;>s zHTTPConnection.clientcCs t|jdS)z/ The full URL of this request. )r")rr'r2rrr urlI zHTTPConnection.urlcCs |diS)z> The path parameters parsed by the framework. path_params)r>r2rrr rAPr@zHTTPConnection.path_paramscCs t|dS)z? Query parameter. It is a multi-value mapping. query_string)r r2rrr query_paramsWr@zHTTPConnection.query_paramscCstdd|jdDS)z A read-only case-independent mapping. Note that in its internal storage, all keys are in lower case. css(|]\}}|d|dfVqdS)latin-1N)decode).0r-valuerrr es  z)HTTPConnection.headers..headers)r r'r2rrr rI^szHTTPConnection.headers)rrrrrrrrrr,strrr.rr3intr5boolr:rr r;rr?rrAr rCr rIrrrr r!s4   r!c seZdZeefdedededdffdd Ze de fdd Z de e fd d Zede fd d ZedefddZde de defddZedefddZdddZdefddZZS)Requestr"r#r$r%Ncs0|ddksJt|||d|_d|_dS)NtypehttpF)superr,_stream_consumed_is_disconnectedr*r8rr r,ls zRequest.__init__cCs|dS)z0 HTTP method. Uppercase string. methodrr2rrr rTtszRequest.methodcCsd|jvr|jdr|jIdHVdVdS|jr tdd|_ |IdH}|ddkrD|dd}|r<|V|dd sCnn |dd krPd|_tq$dVdS) a( Streaming read request body. e.g. `async for chunk in request.stream(): ...` If you access `.stream()` then the byte chunks are provided without storing the entire body to memory. Any subsequent calls to `.body`, `.form`, or `.json` will raise an error. bodyNzStream consumedTrNz http.request more_bodyFhttp.disconnect) __dict__donerUrQ RuntimeErrorr)r>rRr)r+messagerUrrr stream{s,     zRequest.streamcs ddd|2IdHS)zZ Read all the contents of the request body into the memory and return it. rVcsg|z3dHW}|q6Sr&r)rFchunkrrr sz Request.body..N)joinr]r2rrr rUsz Request.bodyc sh|jdkr0|jIdH}zt||jjddWStjy/}ztt |dd}~wwt d)z Call `await self.body` and use `json.loads` parse it. If `content_type` is not equal to `application/json`, an HTTPExcption exception will be thrown. zapplication/jsonNcharsetutf8) content_typerUjsonloadsrEoptionsr>JSONDecodeErrorrrJr)r+dataexcrrr rds  z Request.jsonboundaryracs tt|||tdIdHS)N) file_factory)r parse_multipartr]r )r+rjrarrr _parse_multiparts   zRequest._parse_multipartcs|jdkr*|jjdd}d|jjvrtd|jjdd}|||IdHS|jdkrG|jIdHj|jjddd }tt |d d St d ) a Parse the data in the form format and return it as a multi-value mapping. If `content_type` is equal to `multipart/form-data`, it will directly perform streaming analysis, and subsequent calls to `self.body` or `self.json` will raise errors. If `content_type` is not equal to `multipart/form-data` or `application/x-www-form-urlencoded`, an HTTPExcption exception will be thrown. zmultipart/form-datararbrjz'Missing boundary in header content-typerDNz!application/x-www-form-urlencoded)encodingT)keep_blank_valuesz6multipart/form-data, application/x-www-form-urlencoded) rcrfr>rencodermrUrEr rr)r+rarjrUrrr forms    z Request.formcs<d|jvr|jdr|jIdHIdHdSdSdS)z Close all temporary files in the `self.form`. This can always be called, regardless of whether you use form or not. rqN)rYrZrqacloser2rrr closesz Request.closecsX|js)ztj|ddIdH}|ddk|_W|jStjy(Y|jSw|jS)zU The method used to determine whether the connection is interrupted. gHz>)timeoutNrNrX)rRasynciowait_forr)r> TimeoutError)r+r\rrr is_disconnectedszRequest.is_disconnected)r%N)rrrrrrrrr,propertyrJrTrbytesr]rrUrrdr rmrqrsrLrx __classcell__rrrSr rMks.  rM)(rurdtypingrrrrr urllib.parserbaize.datastructuresrr r r r r baize.exceptionsrrrbaize.multipart_helperrrlbaize.requestsr baize.typingrrr baize.utilsrhelperrr ExceptionrrJr!rMrrrr s     N