You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
306 lines
9.6 KiB
306 lines
9.6 KiB
from collections import OrderedDict
|
|
from functools import wraps
|
|
|
|
from flask import request, current_app, has_app_context
|
|
|
|
from .mask import Mask, apply as apply_mask
|
|
from .utils import unpack
|
|
|
|
|
|
def make(cls):
|
|
if isinstance(cls, type):
|
|
return cls()
|
|
return cls
|
|
|
|
|
|
def marshal(data, fields, envelope=None, skip_none=False, mask=None, ordered=False):
|
|
"""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)])
|
|
|
|
"""
|
|
out, has_wildcards = _marshal(data, fields, envelope, skip_none, mask, ordered)
|
|
|
|
if has_wildcards:
|
|
# ugly local import to avoid dependency loop
|
|
from .fields import Wildcard
|
|
|
|
items = []
|
|
keys = []
|
|
for dkey, val in fields.items():
|
|
key = dkey
|
|
if isinstance(val, dict):
|
|
value = marshal(data, val, skip_none=skip_none, ordered=ordered)
|
|
else:
|
|
field = make(val)
|
|
is_wildcard = isinstance(field, Wildcard)
|
|
# exclude already parsed keys from the wildcard
|
|
if is_wildcard:
|
|
field.reset()
|
|
if keys:
|
|
field.exclude |= set(keys)
|
|
keys = []
|
|
value = field.output(dkey, data, ordered=ordered)
|
|
if is_wildcard:
|
|
|
|
def _append(k, v):
|
|
if skip_none and (v is None or v == OrderedDict() or v == {}):
|
|
return
|
|
items.append((k, v))
|
|
|
|
key = field.key or dkey
|
|
_append(key, value)
|
|
while True:
|
|
value = field.output(dkey, data, ordered=ordered)
|
|
if value is None or value == field.container.format(
|
|
field.default
|
|
):
|
|
break
|
|
key = field.key
|
|
_append(key, value)
|
|
continue
|
|
|
|
keys.append(key)
|
|
if skip_none and (value is None or value == OrderedDict() or value == {}):
|
|
continue
|
|
items.append((key, value))
|
|
|
|
items = tuple(items)
|
|
|
|
out = OrderedDict(items) if ordered else dict(items)
|
|
|
|
if envelope:
|
|
out = OrderedDict([(envelope, out)]) if ordered else {envelope: out}
|
|
|
|
return out
|
|
|
|
return out
|
|
|
|
|
|
def _marshal(data, fields, envelope=None, skip_none=False, mask=None, ordered=False):
|
|
"""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)])
|
|
|
|
"""
|
|
# ugly local import to avoid dependency loop
|
|
from .fields import Wildcard
|
|
|
|
mask = mask or getattr(fields, "__mask__", None)
|
|
fields = getattr(fields, "resolved", fields)
|
|
if mask:
|
|
fields = apply_mask(fields, mask, skip=True)
|
|
|
|
if isinstance(data, (list, tuple)):
|
|
out = [marshal(d, fields, skip_none=skip_none, ordered=ordered) for d in data]
|
|
if envelope:
|
|
out = OrderedDict([(envelope, out)]) if ordered else {envelope: out}
|
|
return out, False
|
|
|
|
has_wildcards = {"present": False}
|
|
|
|
def __format_field(key, val):
|
|
field = make(val)
|
|
if isinstance(field, Wildcard):
|
|
has_wildcards["present"] = True
|
|
value = field.output(key, data, ordered=ordered)
|
|
return (key, value)
|
|
|
|
items = (
|
|
(k, marshal(data, v, skip_none=skip_none, ordered=ordered))
|
|
if isinstance(v, dict)
|
|
else __format_field(k, v)
|
|
for k, v in fields.items()
|
|
)
|
|
|
|
if skip_none:
|
|
items = (
|
|
(k, v) for k, v in items if v is not None and v != OrderedDict() and v != {}
|
|
)
|
|
|
|
out = OrderedDict(items) if ordered else dict(items)
|
|
|
|
if envelope:
|
|
out = OrderedDict([(envelope, out)]) if ordered else {envelope: out}
|
|
|
|
return out, has_wildcards["present"]
|
|
|
|
|
|
class marshal_with(object):
|
|
"""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`
|
|
"""
|
|
|
|
def __init__(
|
|
self, fields, envelope=None, skip_none=False, mask=None, ordered=False
|
|
):
|
|
"""
|
|
: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
|
|
"""
|
|
self.fields = fields
|
|
self.envelope = envelope
|
|
self.skip_none = skip_none
|
|
self.ordered = ordered
|
|
self.mask = Mask(mask, skip=True)
|
|
|
|
def __call__(self, f):
|
|
@wraps(f)
|
|
def wrapper(*args, **kwargs):
|
|
resp = f(*args, **kwargs)
|
|
mask = self.mask
|
|
if has_app_context():
|
|
mask_header = current_app.config["RESTX_MASK_HEADER"]
|
|
mask = request.headers.get(mask_header) or mask
|
|
if isinstance(resp, tuple):
|
|
data, code, headers = unpack(resp)
|
|
return (
|
|
marshal(
|
|
data,
|
|
self.fields,
|
|
self.envelope,
|
|
self.skip_none,
|
|
mask,
|
|
self.ordered,
|
|
),
|
|
code,
|
|
headers,
|
|
)
|
|
else:
|
|
return marshal(
|
|
resp, self.fields, self.envelope, self.skip_none, mask, self.ordered
|
|
)
|
|
|
|
return wrapper
|
|
|
|
|
|
class marshal_with_field(object):
|
|
"""
|
|
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`
|
|
"""
|
|
|
|
def __init__(self, field):
|
|
"""
|
|
:param field: a single field with which to marshal the output.
|
|
"""
|
|
if isinstance(field, type):
|
|
self.field = field()
|
|
else:
|
|
self.field = field
|
|
|
|
def __call__(self, f):
|
|
@wraps(f)
|
|
def wrapper(*args, **kwargs):
|
|
resp = f(*args, **kwargs)
|
|
|
|
if isinstance(resp, tuple):
|
|
data, code, headers = unpack(resp)
|
|
return self.field.format(data), code, headers
|
|
return self.field.format(resp)
|
|
|
|
return wrapper
|