o !i&@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\/var/www/html/crm_dreinet/venv_linux/lib/python3.10/site-packages/flask_restx/marshalling.pymake s rNFcst|||||\}}|rddlm}gg} |D]\} } | } t| tr/t|| |d} nXt| }t||}|rK|| rK|j t | O_ g} |j | ||d} |rfdd}|j p`| } || |  |j | ||d} | dus|| |j |jkr}n |j } || | qgq| | r| dus| tks| ikrq| | fqt|rtnt}|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|dus|tks|ikrdS||fdSr )rappend)kvitemsrrr_appendNszmarshal.._appendTN)_marshalfieldsrrrdictmarshalrresetexcludesetoutputkey containerformatdefaultrrtuple)datar"envelopermaskrout has_wildcardsrkeysdkeyvalr)valuefield is_wildcardr rrrr$sR&       r$cs ddlm|p tdd}td|rt|ddtttfrBfdd D}|r>r:t||fgn||i}|d fSd d ifd d fddD}rfdd|D}rlt|nt |}|rr{t||fgn||i}|d fS)rr r__mask__NresolvedTskipcsg|] }t|dqS)r)r$).0d)r"rrrr sz_marshal..Fpresentcs2t|}t|r dd<|j|d}||fS)NTr@r)rrr()r)r5r7r6)rr.r2rrr__format_fields  z _marshal..__format_fieldc3s<|]\}}t|tr|t|dfn||VqdS)rN)rr#r$r=rr)rAr.rrrr s z_marshal..css6|]\}}|dur|tkr|ikr||fVqdSr rrBrrrrCs,) r"rgetattr apply_maskrlistr-rrr#)r.r"r/rr0rr1rr)rrAr.r"r2rrrr!ps. '  r!c@s$eZdZdZ dddZddZdS) 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 Tr;N)r"r/rrr r0)selfr"r/rr0rrrr__init__s  zmarshal_with.__init__ctfdd}|S)Ncs|i|}j}trtjd}tj|p|}t|tr6t |\}}}t |j j j |j||fSt |j j j |jS)NRESTX_MASK_HEADER)r0rrconfigrheadersgetrr-r r$r"r/rr)argskwargsrespr0 mask_headerr.coderMfrHrrwrappers*   z&marshal_with.__call__..wrapperrrHrUrVrrTr__call__szmarshal_with.__call__NFNF__name__ __module__ __qualname____doc__rIrXrrrrrGs ! rGc@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` cCs t|tr ||_dS||_dS)zP :param field: a single field with which to marshal the output. N)rrr7)rHr7rrrrIs   zmarshal_with_field.__init__crJ)NcsD|i|}t|trt|\}}}j|||fSj|Sr )rr-r r7r+)rOrPrQr.rSrMrTrrrV(s   z,marshal_with_field.__call__..wrapperrrWrrTrrX'szmarshal_with_field.__call__NrZrrrrr_s r_rY) collectionsr functoolsrflaskrrrr0r r rEutilsr rr$r!objectrGr_rrrrs    `QN