U d&@sddlmZddlmZddlmZmZmZddlm Z m Z ddl m Z ddZdd d Zdd dZGdddeZGdddeZd S) OrderedDictwraps)request current_apphas_app_context)Maskapply)unpackcCst|tr|S|SN) isinstancetype)clsr;/tmp/pip-unpacked-wheel-dt_sn2ih/flask_restx/marshalling.pymake s rNFcst|||||\}}|rddlm}gg} |D]\} } | } t| trbt|| |d} nt| }t||}|r|| r|j t | O_ g} |j | ||d} |rfdd}|j p| } || | |j | ||d} | dks| |j |jkrq8|j } || | qq8| | rB| dks8| tks8| ikrBq8| | fq8t|rhtnt}|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_noneorderedrcs4r"|dks|tks|ikr"dS||fdSr )rappend)kvitemsrrr_appendNszmarshal.._appendN)_marshalfieldsrrrdictmarshalrresetexcludesetoutputkey containerformatdefaultrrtuple)datar"envelopermaskrout has_wildcardsrkeysZdkeyvalr)valuefieldZ is_wildcardr rrrr$sN&       "r$cs ddlm|ptdd}td|r:t|ddtttfrfdd D}|r|rtt||fgn||i}|d fSd d ifd d fddD}rdd|D}rt|nt |}|rrt||fgn||i}|d fS)rr rZ__mask__NZresolvedTskipcsg|]}t|dqS)r)r$).0d)r"rrrr sz_marshal..Fpresentcs2t|}t|rdd<|j|d}||fS)NTr<r)rrr()r)r4r6r5)rr.r2rrr__format_fields  z _marshal..__format_fieldc3s:|]2\}}t|tr(|t|dfn||VqdS)rN)rr#r$r9rr)r=r.rrrr sz_marshal..css4|],\}}|dk r|tkr|ikr||fVqdSr rr>rrrr?s  ) r"rgetattr apply_maskrlistr-rrr#)r.r"r/rr0rr1rr)rr=r.r"r2rrrr!ps.'  r!c@s"eZdZdZdddZddZdS) 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` NFcCs*||_||_||_||_t|dd|_dS)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 Tr7N)r"r/rrr r0)selfr"r/rr0rrrr__init__s zmarshal_with.__init__cstfdd}|S)Ncs||}j}tr0tjd}tj|p.|}t|trht |\}}}t |j j j |j||fSt |j j j |jSdS)NZRESTX_MASK_HEADER)r0rrconfigrheadersgetrr-r r$r"r/rr)argskwargsrespr0Z mask_headerr.coderGfrDrrwrappers4    z&marshal_with.__call__..wrapperrrDrNrOrrMr__call__szmarshal_with.__call__)NFNF__name__ __module__ __qualname____doc__rErQrrrrrCs! rCc@s eZdZdZddZddZdS)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` cCst|tr||_n||_dS)zP :param field: a single field with which to marshal the output. N)rrr6)rDr6rrrrEs  zmarshal_with_field.__init__cstfdd}|S)Ncs@||}t|tr4t|\}}}j|||fSj|Sr )rr-r r6r+)rIrJrKr.rLrGrMrrrO(s   z,marshal_with_field.__call__..wrapperrrPrrMrrQ'szmarshal_with_field.__call__NrRrrrrrWs rW)NFNF)NFNF) collectionsr functoolsrZflaskrrrr0r r rAutilsr rr$r!objectrCrWrrrrs    ` QN