o ߥi3@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZddlmZddlmZmZddlmZmZdd l m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(ed d d deDZ)Gddde!Z*edZ+Gddde*ej,ee+Z-Gddde-ee.e/fZ0Gddde0Z1Gddde-e Z2Gddde*Z3Gddde*ej,ee+Z4Gddde4e.Z5Gd d!d!e4e'Z6Gd"d#d#e*e"Z7dS)$N) HTTPStatus) guess_type)choices) AnyDict GeneratorGenericIterableMappingOptionalSequenceTupleTypeVarUnion)ThreadPoolExecutor)URL defaultdict)MalformedRangeHeaderRangeNotSatisfiable) BaseResponseFileResponseMixinbuild_bytes_from_sse iri_to_uri)EnvironServerSentEvent StartResponsecCs |dS)Nz Unknown Status Code)statusrrH/home/ubuntu/.local/lib/python3.10/site-packages/baize/wsgi/responses.py#s rcCs"i|] }t||d|jqS) )intphrase).0rrrr $s"r$c@s*eZdZdZdededeefddZdS)Responsezh The parent class of all responses, whose objects can be used directly as WSGI application. environstart_responsereturncCs&d|jd<|t|j|jdddS)N0content-lengthFas_bytes))headersStatusStringMapping status_code list_headersselfr&r'rrr__call__.s zResponse.__call__N) __name__ __module__ __qualname____doc__rrr bytesr4rrrrr%(sr% _ContentTypec seZdZUdZdZeed<dZeed<    dded e d e e eefde ede ed df fd d Z e jded efddZdeded eefddZZS) SmallResponsez9 Abstract base class for small response objects. media_typeutf-8charsetNcontentr0r.r(cs0t||||_|p|j|_|p|j|_dSN)super__init__rAr=r?)r3rAr0r.r=r? __class__rrrDCs zSmallResponse.__init__cCstrBNotImplementedErrorr3rArrrrenderPszSmallResponse.renderr&r'ccs||j}|rd|jvrtt|}||jd<|j}|r4d|jvr4|dr/|d|j7}||jd<|t|j |j dd|VdS)Nr* content-typeztext/ ; charset=Fr+) rJrAr.strlenr= startswithr?r/r0r1)r3r&r'bodycontent_length content_typerrrr4Ts      zSmallResponse.__call__)r@NNN)r5r6r7r8r=rM__annotations__r?r:r!r r rDabcabstractmethodr9rJrrr r4 __classcell__rrrErr;;s<    r;c@s*eZdZdZdeeefdefddZdS)PlainTextResponsez text/plainrAr(cCst|tr ||jS|SrB) isinstancerMencoder?rIrrrrJiszPlainTextResponse.renderN)r5r6r7r=rr9rMrJrrrrrWfsrWc@seZdZdZdS) HTMLResponsez text/htmlN)r5r6r7r=rrrrrZmsrZc s^eZdZdZdZ  ddededeee e fded df fd d Z ded e fd d Z Z S) JSONResponsezW `**kwargs` is used to accept all the parameters that `json.loads` can accept. zapplication/jsonr@NrAr0r.kwargsr(c s:dddddd|_|jjdi|tj|||ddS)NF),:) ensure_ascii allow_nanindent separatorsdefaultr0r.r) json_kwargsupdaterCrD)r3rAr0r.r\rErrrDxszJSONResponse.__init__cCstj|fi|j|jSrB)jsondumpsrerYr?rIrrrrJszJSONResponse.renderr@N)r5r6r7r8r=rr!r r rMrDr9rJrVrrrErr[qs"r[c sHeZdZ  d deeefdedeeeefddffdd Z Z S) RedirectResponse3Nurlr0r.r(cs&tj||dtt||jd<dS)Nrdlocation)rCrDrrMr.)r3rlr0r.rErrrDszRedirectResponse.__init__)rkN) r5r6r7rrMrr!r r rDrVrrrErrjs rjc s|eZdZ  ddeededeeeefddffdd Z e j de e ddffd d Zd ed edee fd dZZS)StreamingResponser@Niterabler0r.r(cst||||_d|_dS)NF)rCrDro_client_closed)r3ror0r.rErrrDs zStreamingResponse.__init__ccstrBrGr3rrr render_streamszStreamingResponse.render_streamr&r'ccs,|t|j|jdd|EdHdS)NFr+)r/r0r1rrr2rrrr4s zStreamingResponse.__call__ri)r5r6r7r r:r!r r rMrDrTrUrr9rrrrr4rVrrrErrns* rnc sbeZdZ   d deededeeeefdeddf fd d Z de eddffd d Z Z S)StreamResponser@Napplication/octet-streamror0r.rRr(cst|||||jd<dS)N Content-Type)rCrDr.)r3ror0r.rRrErrrDszStreamResponse.__init__ccs|jEdHdSrB)rorqrrrrrszStreamResponse.render_stream)r@Nrt) r5r6r7r r9r!r r rMrDrrrrVrrrErrss  rscseZdZdZedddZdddZ  dd d d d eede de e e e fde de dd f fddZdeed d ffddZZS)SendEventResponsea| Server-sent events response. When the cilent closes the connection, the generator will be closed. Use `try-finally` to clean up resources. ```python def generator(): try: while True: yield ServerSentEvent() finally: print("generator closed") response = SendEventResponse(generator()) ``` SendEvent_) max_workersthread_name_prefixzno-cacheztext/event-stream)z Cache-Controlrur@Nr>) ping_intervalr?ror0r.r|r?r(csT|r i|j|}nt|j}|dd|7<t|||||_||_dS)NrurL)required_headersdictrCrDr|r?)r3ror0r.r|r?rErrrDs   zSendEventResponse.__init__c#stjdddd fdd }j|}zU|r!sMzjjd}|dur/Wnt|j VWn tj yDdVYnw|r!r!Wd s\ rT| sj| }|durl|dSdSd s| rt| s| }|dur|w) N)maxsizeFr(c sz4tj}sz t|Wn tydYnwrWdj}t|dr3|dSdSdj}t|drG|ww)NTclose)iterroputnext StopIterationhasattrr)igqr3 should_stoprrpushs(        z-SendEventResponse.render_stream..push)timeouts: ping T)r(N)queueQueue thread_poolsubmitdoneemptygetr|rr?Empty get_nowaitcancel exception)r3r push_futureeventexcrrrrrsF    zSendEventResponse.render_streamri)r5r6r7r8rrr}r rr!r r rMfloatrDrr9rrrVrrrErrvs2   rvcseZdZdZ     ddedeeeefdeedeedeejd e d dffd d Z d e de de d e eddffddZd e de de de de d e eddff ddZd e de de deee e fd e eddff ddZdede d eefddZZS) FileResponsez File response. It will automatically determine whether to send only headers and the range of files that need to be sent. Nfilepathr.rR download_name stat_result chunk_sizer(cstj|d||_|pt|ptj|dpd|_||_|p$t ||_ t |j j r4t |d||_|j||j|j|j|j dS)N)r.rrtz is a directory)rCrDrrospathbasenamerRrstatrS_ISDIRst_modeIsADirectoryErrorrr.rfgenerate_common_headers)r3rr.rRrrrrErrrD"s" zFileResponse.__init__send_header_only file_sizer'ccst|j|jd<t||jd<|td|jdd|r"dVdSt|jd}td||jD] }| |jVq0WddS1sEwYdS) NrKr*r@Fr+r-rbr) rMrRr.r/r1openrrangerread)r3rrr'file_rrr handle_all=s"zFileResponse.handle_allstartendccsd|d|dd||jd<t|j|jd<t|||jd<|td|jd d |r4d VdSt|jd $}||t|||j D]}| t |j ||VqGWddS1sawYdS) Nzbytes -r/z content-rangerKr*Fr+r-r) r.rMrRr/r1rrseekrrrmin)r3rrr'rrrhererrrhandle_single_rangeOs  "z FileResponse.handle_single_rangerangesc csdtddd}d||jd<|||||j\}}t||jd<|td|jd d |r6d VdSt|j d >}|D](\} } | | || | Vt | | |j D]} | t|j | | VqUd Vq?d|ddVWddS1s}wYdS)Nr<$abcdefghijklmnopqrstuvwxyz0123456789 )kzmultipart/byteranges; boundary=rKr*rFr+r-r z--z-- ascii)joinrandom_choicesr.generate_multipartrRrMr/r1rrrrrrrrY) r3rrr'rboundaryrQgenerate_headersrrrrrrrhandle_several_rangesds(    "z"FileResponse.handle_several_rangesr&c cs|ddk}|j}|j}d|vsd|vr)||d|s)||||EdHdSz ||d|}Wn2ttfye}z$|t|jg|j pGi |j durSdn|j dVWYd}~dSd}~wwt |dkr|d\}} |||||| EdHdS|||||EdHdS) NREQUEST_METHODHEAD HTTP_RANGE HTTP_IF_RANGEr-utf8rr)rst_sizejudge_if_ranger parse_rangerrr/r0r.itemsrArYrNrr) r3r&r'rrrrrrrrrrr4s:    zFileResponse.__call__)NNNNr)r5r6r7r8rMr r rrr!rDboolrrr9rrr r rrr r4rVrrrErrs|       r)8rTrgrrrhttpr mimetypesrrandomrrtypingrrrrr r r r r rrbaize.concurrencyrbaize.datastructuresrrbaize.exceptionsrrbaize.responsesrrrr baize.typingrrrr/r%r:ABCr;r9rMrWrZr[rjrnrsrvrrrrrs:   4  + Y