E6i&ddlmZddlmZddlmZmZmZddlm Z m Z ddl m Z dZdd Zdd ZGd d eZGd deZy) OrderedDictwraps)request current_apphas_app_context)Maskapply)unpackc4t|tr|S|SN) isinstancetype)clss Z/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/flask_restx/marshalling.pymaker s#tu JNc`t|||||\}}|rddlm}gg} |jD]<\} } | } t | t rt || |} nt| }t ||}|r2|j| r |xjt| zc_ g} |j| ||} |rwfd}|jxs| } || |  |j| ||} | (| |jj|jk(rn|j} || | U| j!| r| | t#k(s| ik(r*j!| | f?t%|r t#n t }|r|rt#||fgn||i}|S|S)Takes raw data (in the form of a dict, list, object) and a dict of fields to output and filters the data based on those fields. :param data: the actual object(s) from which the fields are taken from :param fields: a dict of whose keys will make up the final serialized response output :param envelope: optional key that will be used to envelop the serialized response :param bool skip_none: optional key will be used to eliminate fields which value is None or the field's key not exist in data :param bool ordered: Wether or not to preserve order >>> from flask_restx import fields, marshal >>> data = { 'a': 100, 'b': 'foo', 'c': None } >>> mfields = { 'a': fields.Raw, 'c': fields.Raw, 'd': fields.Raw } >>> marshal(data, mfields) {'a': 100, 'c': None, 'd': None} >>> marshal(data, mfields, envelope='data') {'data': {'a': 100, 'c': None, 'd': None}} >>> marshal(data, mfields, skip_none=True) {'a': 100} >>> marshal(data, mfields, ordered=True) OrderedDict([('a', 100), ('c', None), ('d', None)]) >>> marshal(data, mfields, envelope='data', ordered=True) OrderedDict([('data', OrderedDict([('a', 100), ('c', None), ('d', None)]))]) >>> marshal(data, mfields, skip_none=True, ordered=True) OrderedDict([('a', 100)]) r Wildcard skip_noneorderedrcZr||tk(s|ik(ryj||fyr)rappend)kvitemsrs r_appendzmarshal.._appendNs-$!)qKM7IQRTW" aV,r)_marshalfieldsrr"rdictmarshalrresetexcludesetoutputkey containerformatdefaultrrtuple)datar%envelopermaskrout has_wildcardsrkeysdkeyvalr,valuefield is_wildcardr#r"s ` @rr'r'sL"$)T7SC$$ 'ID#C#t$cYPS (9 KKM T2 ! T4 A-  ))+tCC' % T4 I =EU__5K5K!MM6-"#iiU+ KK emu /ERT LL#u &I$ 'Le $+k% e 4;+#/0(CC Jrc  ddlm |xs t|dd}t|d|}|rt||d}t t t fr6Dcgc]}t||}}|rrt||fgn||i}|d fSd d i fd  fd |jD}r d |D}r t|n t|}|rrt||fgn||i}| d fScc}w)rr r__mask__NresolvedTskiprFpresentclt|}t|rdd<|j|}||fS)NTrAr)rrr+)r,r8r:r9rr1r5rs r__format_fieldz _marshal..__format_fields>S  eX &'+M) $ S$ 8U|rc3xK|]1\}}t|tr|t|fn||3yw)rN)rr&r').0r r!rCr1rrs r z_marshal..sJ  Aq a  GD!y' BC Aq ! " s7:c3XK|]"\}}| |tk7s|ik7s||f$ywrr)rEr r!s rrFz_marshal..s3 q! !{}:LQRVXQXQF s * ** *) r%rgetattr apply_maskrlistr0r'rr"r&) r1r%r2rr3rdr4r"rrCr5s ` ` ` @@@rr$r$ps N!  476:t4D VZ 0F FDt4$u &QUVAwq&IwGVV 4;+#/0(CCEz&M LLN  E $ !(+e T%[C07kHc?+,h_  i( ((?WsC/c eZdZdZ ddZdZy) marshal_witha+A decorator that apply marshalling to the return values of your methods. >>> from flask_restx import fields, marshal_with >>> mfields = { 'a': fields.Raw } >>> @marshal_with(mfields) ... def get(): ... return { 'a': 100, 'b': 'foo' } ... ... >>> get() OrderedDict([('a', 100)]) >>> @marshal_with(mfields, envelope='data') ... def get(): ... return { 'a': 100, 'b': 'foo' } ... ... >>> get() OrderedDict([('data', OrderedDict([('a', 100)]))]) >>> mfields = { 'a': fields.Raw, 'c': fields.Raw, 'd': fields.Raw } >>> @marshal_with(mfields, skip_none=True) ... def get(): ... return { 'a': 100, 'b': 'foo', 'c': None } ... ... >>> get() OrderedDict([('a', 100)]) see :meth:`flask_restx.marshal` Nc`||_||_||_||_t |d|_y)z :param fields: a dict of whose keys will make up the final serialized response output :param envelope: optional key that will be used to envelop the serialized response Tr?N)r%r2rrr r3)selfr%r2rr3rs r__init__zmarshal_with.__init__s.   " D) rc2tfd}|S)Nc|i|} j}tr6tjd}tj j |xs|}t|trJt|\}}}t| j j j| j||fSt| j j j| jS)NRESTX_MASK_HEADER)r3r rconfigrheadersgetrr0r r'r%r2rr) argskwargsrespr3 mask_headerr1coderUfrOs rwrapperz&marshal_with.__call__..wrappersd%f%D99D )001DE **;7?4$&&,Tl#dG     $++t}}dnndDLLrrrOr\r]s`` r__call__zmarshal_with.__call__s q   2rNFNF__name__ __module__ __qualname____doc__rPr_rrrMrMsBJO *rrMceZdZdZdZdZy)marshal_with_fieldaP A decorator that formats the return values of your methods with a single field. >>> from flask_restx import marshal_with_field, fields >>> @marshal_with_field(fields.List(fields.Integer)) ... def get(): ... return ['1', 2, 3.0] ... >>> get() [1, 2, 3] see :meth:`flask_restx.marshal_with` cLt|tr ||_y||_y)zP :param field: a single field with which to marshal the output. N)rrr:)rOr:s rrPzmarshal_with_field.__init__s eT "DJDJrc2tfd}|S)Nc|i|}t|tr-t|\}}}jj |||fSjj |Sr)rr0r r:r.)rWrXrYr1r[rUr\rOs rr]z,marshal_with_field.__call__..wrapper(s]d%f%D$&&,Tl#dGzz((.g==::$$T* *rrr^s`` rr_zmarshal_with_field.__call__'s q +  +rNrarfrrrhrhs  rrhr`) collectionsr functoolsrflaskrrr r3r r rIutilsr rr'r$objectrMrhrfrrrqsE#77+ ]@N)bK6K\""r