o ߥi!@s>ddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZz ddlmZmZWney[dd Zd d ZYnwd d lmZmZd dlmZm Z d dl m!Z!m"Z"eddGdddZ#eGdddZ$de"de%de&fddZ'de%de%fddZ(dS)N) formatdate)sha1)chain) CallableDictIterableListMappingOptionalSequenceTupleUnionoverloadquote) mypyc_attrtraitcC|SN)clsrrC/home/ubuntu/.local/lib/python3.10/site-packages/baize/responses.pyrrcOsddS)NcSrrr)xrrrszmypyc_attr..r)attrskwattrsrrrrsr)CookieMutableHeaders)MalformedRangeHeaderRangeNotSatisfiable)LiteralServerSentEventT)allow_interpreted_subclassesc@seZdZ d!dedeeeefddfddZ    d"d edededeededeededede dddfddZ   d#d edededeededede dddfddZ e de dde eeeffddZe de d de eeeffddZd dZdS)$ BaseResponseN status_codeheadersreturncCs||_t||_g|_dSr)r'rr(cookies)selfr'r(rrr__init__(s  zBaseResponse.__init__/Flaxkeyvaluemax_ageexpirespathdomainsecurehttponlysamesite)strictr0nonec CsFd} |durtjt|} |jt||| |||||| d dS)Nr4r3r5r6r7r8r9)datetime fromtimestamptimer*appendr) r+r1r2r3r4r5r6r7r8r9expires_datetimerrr set_cookie/s  zBaseResponse.set_cookiec Cs|j|dd|||||ddS)Nrr<)rB)r+r1r2r5r6r7r8r9rrr delete_cookieMs  zBaseResponse.delete_cookieas_bytesTcCdSrrr+rDrrr list_headersbrzBaseResponse.list_headerscCrErrrFrrrrGercCsJ|rgdd|jDdd|jDSg|jdd|jDS)zR Merge `self.headers` and `self.cookies` then returned as a list. css(|]\}}|d|dfVqdS)latin-1Nencode).0r1r2rrr ns  z,BaseResponse.list_headers..cs|] }dt|fVqdS)s set-cookieN)bytesrKcookierrrrLrcsrM)z set-cookieN)strrOrrrrLwrQ)r(itemsr*rFrrrrGhs)r&N)r-r.Nr/NFFr0)r-r/NFFr0)__name__ __module__ __qualname__intr r rRr,boolr"rBrCrrr rNrGrrrrr%&s     ! $$ r%c@seZdZdededeedejdeeeff ddZe dejdefdd Z e d edejde fd d Z e d ededeeeeeefffddZdeeeefdedededeeeeegefff ddZdS)FileResponseMixinfilepath content_type download_name stat_resultr)cCs`dt|jddd||dd}|s|dkr.|ptj|}d|dt|}||d <|S) NrNTusegmt")z accept-rangesz last-modifiedetagzapplication/octet-streamzattachment; filename="z"; filename*=utf-8''zcontent-disposition)rst_mtime generate_etagosr5basenamer)r+rZr[r\r]r(content_dispositionrrrgenerate_common_headers}s  z)FileResponseMixin.generate_common_headerscCs$|jd|j}t|dS)N-ascii)rbst_sizerrJ hexdigest)r]datarrrrcszFileResponseMixin.generate_etagif_range_raw_linecCs(|d||dkp|t|jddkS)z^ Judge whether if-range is consistent with the value of etag or last-modified r`Tr^)rcrrb)rrmr]rrrjudge_if_rangesz FileResponseMixin.judge_if_rangerange_raw_linemax_sizec s4z |jddd\}}Wn tytw|dkrtdfddtd|D}t|d kr5td tfd d |DrDttd d |DrQtdt|dkrY|Sg}|D]:\}}tt|D](}||\} } || krtqg|| kr| |||fnt || t || f||<n| ||fq]|S)zU Parse the Range header and make appropriate merge or cut processing =r)maxsplitrNzOnly support bytes rangecslg|]2}|dkr|drt|dnt|d|dr1|dr1t|dkr1t|ddnfqS))r-r-rr)rWrK_rprr s"2z1FileResponseMixin.parse_range..z (\d*)-(\d*)rz%Range header: range must be requestedc3s.|]\}}d|kokn VqdS)rNr)rKstartrtrurrrLs,z0FileResponseMixin.parse_range..css|] \}}||kVqdSrrrKrwendrrrrLrQz)Range header: start must be less than end) split ValueErrorr refindalllenanyr!rangeinsertminmaxr@) rorpunit ranges_strrangesresultrwrypp_startp_endrrur parse_ranges@       zFileResponseMixin.parse_rangerboundarycsVt}d|ttttfdd|Dd|}|fddfS)a Multipart response headers generator. ``` --{boundary}\n Content-Type: {content_type}\n Content-Range: bytes {start}-{end-1}/{max_size}\n \n ..........content...........\n --{boundary}\n Content-Type: {content_type}\n Content-Range: bytes {start}-{end-1}/{max_size}\n \n ..........content...........\n --{boundary}--\n ``` ,c3s<|]\}}tt|tt|d||VqdS)rN)r~rRrx)static_header_part_lenrrrLs z7FileResponseMixin.generate_multipart..c s.ddd|d|ddd dS) Nz--z Content-Type: z Content-Range: bytes rhrr/z rHrI)rwry)rr[rprrrsz6FileResponseMixin.generate_multipart..)r~rRsum)r+rrrpr[ boundary_lencontent_lengthr)rr[rprrgenerate_multiparts z$FileResponseMixin.generate_multipartN)rTrUrVrRr rdr]rrg staticmethodrc classmethodrXrnrWr r r rrrNrrrrrrY{sR     2rYeventcharsetr)csTd|vrfdd|dD}nd}dttfdd|||dS) z/ helper function for SendEventResponse rlc3s |] }d|VqdS)zdata: NrIrsrrrrL sz'build_bytes_from_sse..r cs|d|S)Nz: rI)kvrrrrsz&build_bytes_from_sse..)r)pop splitlinesjoinrmapkeysvalues)rrrlrrrbuild_bytes_from_ssesriricCs t|ddS)z Convert an Internationalized Resource Identifier (IRI) portion to a URI portion that is suitable for inclusion in a URL. z/#%[]=:;$&()+,!?*@'~)safer)rrrr iri_to_uris r))r=rdr|r? email.utilsrhashlibr itertoolsrtypingrrrrr r r r r r urllib.parsermypy_extensionsrr ImportErrordatastructuresrr exceptionsr r!r"r#r%rYrRrNrrrrrrs2   0    T