Source code for marshmallow.fields

# -*- coding: utf-8 -*-
"""Field classes for various types of data."""

from __future__ import absolute_import, unicode_literals

import collections
import copy
import datetime as dt
import numbers
import uuid
import decimal
import math

from marshmallow import validate, utils, class_registry
from marshmallow.base import FieldABC, SchemaABC
from marshmallow.utils import is_collection, missing as missing_
from marshmallow.compat import basestring, text_type, Mapping
from marshmallow.exceptions import ValidationError, StringNotCollectionError
from marshmallow.validate import Validator

__all__ = [
    'Field',
    'Raw',
    'Nested',
    'Dict',
    'List',
    'String',
    'UUID',
    'Number',
    'Integer',
    'Decimal',
    'Boolean',
    'FormattedString',
    'Float',
    'DateTime',
    'LocalDateTime',
    'Time',
    'Date',
    'TimeDelta',
    'Url',
    'URL',
    'Email',
    'Method',
    'Function',
    'Str',
    'Bool',
    'Int',
    'Constant',
]

MISSING_ERROR_MESSAGE = (
    'ValidationError raised by `{class_name}`, but error key `{key}` does '
    'not exist in the `error_messages` dictionary.'
)


[docs]class Field(FieldABC): """Basic field from which other fields should extend. It applies no formatting by default, and should only be used in cases where data does not need to be formatted before being serialized or deserialized. On error, the name of the field will be returned. :param default: If set, this value will be used during serialization if the input value is missing. If not set, the field will be excluded from the serialized output if the input value is missing. May be a value or a callable. :param str attribute: The name of the attribute to get the value from when serializing. If `None`, assumes the attribute has the same name as the field. :param str data_key: The name of the key to get the value from when deserializing. If `None`, assumes the key has the same name as the field. :param callable validate: Validator or collection of validators that are called during deserialization. Validator takes a field's input value as its only parameter and returns a boolean. If it returns `False`, an :exc:`ValidationError` is raised. :param required: Raise a :exc:`ValidationError` if the field value is not supplied during deserialization. :param allow_none: Set this to `True` if `None` should be considered a valid value during validation/deserialization. If ``missing=None`` and ``allow_none`` is unset, will default to ``True``. Otherwise, the default is ``False``. :param bool load_only: If `True` skip this field during serialization, otherwise its value will be present in the serialized data. :param bool dump_only: If `True` skip this field during deserialization, otherwise its value will be present in the deserialized object. In the context of an HTTP API, this effectively marks the field as "read-only". :param missing: Default deserialization value for the field if the field is not found in the input data. May be a value or a callable. :param dict error_messages: Overrides for `Field.default_error_messages`. :param metadata: Extra arguments to be stored as metadata. .. versionchanged:: 2.0.0 Removed `error` parameter. Use ``error_messages`` instead. .. versionchanged:: 2.0.0 Added `allow_none` parameter, which makes validation/deserialization of `None` consistent across fields. .. versionchanged:: 2.0.0 Added `load_only` and `dump_only` parameters, which allow field skipping during the (de)serialization process. .. versionchanged:: 2.0.0 Added `missing` parameter, which indicates the value for a field if the field is not found during deserialization. .. versionchanged:: 2.0.0 ``default`` value is only used if explicitly set. Otherwise, missing values inputs are excluded from serialized output. .. versionchanged:: 3.0.0b8 Add ``data_key`` parameter for the specifying the key in the input and output data. This parameter replaced both ``load_from`` and ``dump_to``. """ # Some fields, such as Method fields and Function fields, are not expected # to exist as attributes on the objects to serialize. Set this to False # for those fields _CHECK_ATTRIBUTE = True _creation_index = 0 # Used for sorting #: Default error messages for various kinds of errors. The keys in this dictionary #: are passed to `Field.fail`. The values are error messages passed to #: :exc:`marshmallow.ValidationError`. default_error_messages = { 'required': 'Missing data for required field.', 'null': 'Field may not be null.', 'validator_failed': 'Invalid value.', } def __init__( self, default=missing_, attribute=None, data_key=None, error=None, validate=None, required=False, allow_none=None, load_only=False, dump_only=False, missing=missing_, error_messages=None, **metadata ): self.default = default self.attribute = attribute self.data_key = data_key self.validate = validate if utils.is_iterable_but_not_string(validate): if not utils.is_generator(validate): self.validators = validate else: self.validators = list(validate) elif callable(validate): self.validators = [validate] elif validate is None: self.validators = [] else: raise ValueError( "The 'validate' parameter must be a callable " 'or a collection of callables.', ) self.required = required # If missing=None, None should be considered valid by default if allow_none is None: if missing is None: self.allow_none = True else: self.allow_none = False else: self.allow_none = allow_none self.load_only = load_only self.dump_only = dump_only self.missing = missing self.metadata = metadata self._creation_index = Field._creation_index Field._creation_index += 1 # Collect default error message from self and parent classes messages = {} for cls in reversed(self.__class__.__mro__): messages.update(getattr(cls, 'default_error_messages', {})) messages.update(error_messages or {}) self.error_messages = messages def __repr__(self): return ('<fields.{ClassName}(default={self.default!r}, ' 'attribute={self.attribute!r}, ' 'validate={self.validate}, required={self.required}, ' 'load_only={self.load_only}, dump_only={self.dump_only}, ' 'missing={self.missing}, allow_none={self.allow_none}, ' 'error_messages={self.error_messages})>' .format(ClassName=self.__class__.__name__, self=self))
[docs] def get_value(self, obj, attr, accessor=None, default=missing_): """Return the value for a given key from an object. :param object obj: The object to get the value from :param str attr: The attribute/key in `obj` to get the value from. :param callable accessor: A callable used to retrieve the value of `attr` from the object `obj`. Defaults to `marshmallow.utils.get_value`. """ # NOTE: Use getattr instead of direct attribute access here so that # subclasses aren't required to define `attribute` member attribute = getattr(self, 'attribute', None) accessor_func = accessor or utils.get_value check_key = attr if attribute is None else attribute return accessor_func(obj, check_key, default)
[docs] def _validate(self, value): """Perform validation on ``value``. Raise a :exc:`ValidationError` if validation does not succeed. """ errors = [] kwargs = {} for validator in self.validators: try: r = validator(value) if not isinstance(validator, Validator) and r is False: self.fail('validator_failed') except ValidationError as err: kwargs.update(err.kwargs) if isinstance(err.messages, dict): errors.append(err.messages) else: errors.extend(err.messages) if errors: raise ValidationError(errors, **kwargs)
# Hat tip to django-rest-framework.
[docs] def fail(self, key, **kwargs): """A helper method that simply raises a `ValidationError`. """ try: msg = self.error_messages[key] except KeyError: class_name = self.__class__.__name__ msg = MISSING_ERROR_MESSAGE.format(class_name=class_name, key=key) raise AssertionError(msg) if isinstance(msg, basestring): msg = msg.format(**kwargs) raise ValidationError(msg)
[docs] def _validate_missing(self, value): """Validate missing values. Raise a :exc:`ValidationError` if `value` should be considered missing. """ if value is missing_: if hasattr(self, 'required') and self.required: self.fail('required') if value is None: if hasattr(self, 'allow_none') and self.allow_none is not True: self.fail('null')
[docs] def serialize(self, attr, obj, accessor=None, **kwargs): """Pulls the value for the given key from the object, applies the field's formatting and returns the result. :param str attr: The attribute or key to get from the object. :param str obj: The object to pull the key from. :param callable accessor: Function used to pull values from ``obj``. :param dict kwargs': Field-specific keyword arguments. :raise ValidationError: In case of formatting problem """ if self._CHECK_ATTRIBUTE: value = self.get_value(obj, attr, accessor=accessor) if value is missing_ and hasattr(self, 'default'): default = self.default value = default() if callable(default) else default if value is missing_: return value else: value = None return self._serialize(value, attr, obj, **kwargs)
[docs] def deserialize(self, value, attr=None, data=None, **kwargs): """Deserialize ``value``. :param value: The value to be deserialized. :param str attr: The attribute/key in `data` to be deserialized. :param dict data: The raw input data passed to the `Schema.load`. :param dict kwargs': Field-specific keyword arguments. :raise ValidationError: If an invalid value is passed or if a required value is missing. """ # Validate required fields, deserialize, then validate # deserialized value self._validate_missing(value) if value is missing_: _miss = self.missing return _miss() if callable(_miss) else _miss if getattr(self, 'allow_none', False) is True and value is None: return None output = self._deserialize(value, attr, data, **kwargs) self._validate(output) return output
# Methods for concrete classes to override.
[docs] def _bind_to_schema(self, field_name, schema): """Update field with values from its parent schema. Called by :meth:`_bind_field<marshmallow.Schema._bind_field>`. :param str field_name: Field name set in schema. :param Schema schema: Parent schema. """ self.parent = self.parent or schema self.name = self.name or field_name
[docs] def _serialize(self, value, attr, obj, **kwargs): """Serializes ``value`` to a basic Python datatype. Noop by default. Concrete :class:`Field` classes should implement this method. Example: :: class TitleCase(Field): def _serialize(self, value, attr, obj, **kwargs): if not value: return '' return unicode(value).title() :param value: The value to be serialized. :param str attr: The attribute or key on the object to be serialized. :param object obj: The object the value was pulled from. :param dict kwargs': Field-specific keyword arguments. :raise ValidationError: In case of formatting or validation failure. :return: The serialized value """ return value
[docs] def _deserialize(self, value, attr, data, **kwargs): """Deserialize value. Concrete :class:`Field` classes should implement this method. :param value: The value to be deserialized. :param str attr: The attribute/key in `data` to be deserialized. :param dict data: The raw input data passed to the `Schema.load`. :param dict kwargs': Field-specific keyword arguments. :raise ValidationError: In case of formatting or validation failure. :return: The deserialized value. .. versionchanged:: 3.0.0 Add ``**kwargs`` parameters .. versionchanged:: 2.0.0 Added ``attr`` and ``data`` parameters. """ return value
# Properties @property def context(self): """The context dictionary for the parent :class:`Schema`.""" return self.parent.context @property def root(self): """Reference to the `Schema` that this field belongs to even if it is buried in a `List`. Return `None` for unbound fields. """ ret = self while hasattr(ret, 'parent') and ret.parent: ret = ret.parent return ret if isinstance(ret, SchemaABC) else None
[docs]class Raw(Field): """Field that applies no formatting or validation.""" pass
[docs]class Nested(Field): """Allows you to nest a :class:`Schema <marshmallow.Schema>` inside a field. Examples: :: user = fields.Nested(UserSchema) user2 = fields.Nested('UserSchema') # Equivalent to above collaborators = fields.Nested(UserSchema, many=True, only=('id',)) parent = fields.Nested('self') When passing a `Schema <marshmallow.Schema>` instance as the first argument, the instance's ``exclude``, ``only``, and ``many`` attributes will be respected. Therefore, when passing the ``exclude``, ``only``, or ``many`` arguments to `fields.Nested`, you should pass a `Schema <marshmallow.Schema>` class (not an instance) as the first argument. :: # Yes author = fields.Nested(UserSchema, only=('id', 'name')) # No author = fields.Nested(UserSchema(), only=('id', 'name')) :param Schema nested: The Schema class or class name (string) to nest, or ``"self"`` to nest the :class:`Schema` within itself. :param tuple exclude: A list or tuple of fields to exclude. :param only: A tuple or string of the field(s) to marshal. If `None`, all fields will be marshalled. If a field name (string) is given, only a single value will be returned as output instead of a dictionary. This parameter takes precedence over ``exclude``. :param bool many: Whether the field is a collection of objects. :param unknown: Whether to exclude, include, or raise an error for unknown fields in the data. Use `EXCLUDE`, `INCLUDE` or `RAISE`. :param kwargs: The same keyword arguments that :class:`Field` receives. """ default_error_messages = { 'type': 'Invalid type.', } def __init__(self, nested, default=missing_, exclude=tuple(), only=None, **kwargs): # Raise error if only or exclude is passed as string, not list of strings if only is not None and not is_collection(only): raise StringNotCollectionError('"only" should be a list of strings') if exclude is not None and not is_collection(exclude): raise StringNotCollectionError('"exclude" should be a list of strings') self.nested = nested self.only = only self.exclude = exclude self.many = kwargs.get('many', False) self.unknown = kwargs.get('unknown') self.__schema = None # Cached Schema instance super(Nested, self).__init__(default=default, **kwargs) @property def schema(self): """The nested Schema object. .. versionchanged:: 1.0.0 Renamed from `serializer` to `schema` """ if not self.__schema: # Inherit context from parent. context = getattr(self.parent, 'context', {}) if isinstance(self.nested, SchemaABC): self.__schema = self.nested self.__schema.context.update(context) else: if isinstance(self.nested, type) and issubclass(self.nested, SchemaABC): schema_class = self.nested elif not isinstance(self.nested, basestring): raise ValueError( 'Nested fields must be passed a ' 'Schema, not {0}.'.format(self.nested.__class__), ) elif self.nested == 'self': schema_class = self.parent.__class__ else: schema_class = class_registry.get_class(self.nested) self.__schema = schema_class( many=self.many, only=self.only, exclude=self.exclude, context=context, load_only=self._nested_normalized_option('load_only'), dump_only=self._nested_normalized_option('dump_only'), ) self.__schema.ordered = getattr(self.parent, 'ordered', False) return self.__schema def _nested_normalized_option(self, option_name): nested_field = '%s.' % self.name return [field.split(nested_field, 1)[1] for field in getattr(self.root, option_name, set()) if field.startswith(nested_field)]
[docs] def _serialize(self, nested_obj, attr, obj, **kwargs): # Load up the schema first. This allows a RegistryError to be raised # if an invalid schema name was passed schema = self.schema if nested_obj is None: return None try: return schema.dump(nested_obj, many=self.many) except ValidationError as exc: raise ValidationError(exc.messages, valid_data=exc.valid_data)
def _test_collection(self, value): if self.many and not utils.is_collection(value): self.fail('type', input=value, type=value.__class__.__name__) def _load(self, value, data, partial=None): try: valid_data = self.schema.load( value, unknown=self.unknown, partial=partial, ) except ValidationError as exc: raise ValidationError(exc.messages, valid_data=exc.valid_data) return valid_data
[docs] def _deserialize(self, value, attr, data, partial=None, **kwargs): """Same as :meth:`Field._deserialize` with additional ``partial`` argument. :param bool|tuple partial: For nested schemas, the ``partial`` parameter passed to `Schema.load`. .. versionchanged:: 3.0.0 Add ``partial`` parameter """ self._test_collection(value) return self._load(value, data, partial=partial)
class Pluck(Nested): """Allows you to replace nested data with one of the data's fields. Examples: :: user = fields.Pluck(UserSchema, 'name') collaborators = fields.Pluck(UserSchema, 'id', many=True) parent = fields.Pluck('self', 'name') :param Schema nested: The Schema class or class name (string) to nest, or ``"self"`` to nest the :class:`Schema` within itself. :param str field_name: :param kwargs: The same keyword arguments that :class:`Nested` receives. """ def __init__(self, nested, field_name, **kwargs): super(Pluck, self).__init__(nested, only=(field_name,), **kwargs) self.field_name = field_name @property def _field_data_key(self): only_field = self.schema.fields[self.field_name] return only_field.data_key or self.field_name def _serialize(self, nested_obj, attr, obj, **kwargs): ret = super(Pluck, self)._serialize(nested_obj, attr, obj, **kwargs) if ret is None: return None if self.many: return utils.pluck(ret, key=self._field_data_key) return ret[self._field_data_key] def _deserialize(self, value, attr, data, partial=None, **kwargs): self._test_collection(value) if self.many: value = [{self._field_data_key: v} for v in value] else: value = {self._field_data_key: value} return self._load(value, data, partial=partial)
[docs]class List(Field): """A list field, composed with another `Field` class or instance. Example: :: numbers = fields.List(fields.Float()) :param Field cls_or_instance: A field class or instance. :param bool default: Default value for serialization. :param kwargs: The same keyword arguments that :class:`Field` receives. .. versionchanged:: 2.0.0 The ``allow_none`` parameter now applies to deserialization and has the same semantics as the other fields. """ default_error_messages = { 'invalid': 'Not a valid list.', } def __init__(self, cls_or_instance, **kwargs): super(List, self).__init__(**kwargs) if isinstance(cls_or_instance, type): if not issubclass(cls_or_instance, FieldABC): raise ValueError( 'The type of the list elements ' 'must be a subclass of ' 'marshmallow.base.FieldABC', ) self.container = cls_or_instance() else: if not isinstance(cls_or_instance, FieldABC): raise ValueError( 'The instances of the list ' 'elements must be of type ' 'marshmallow.base.FieldABC', ) self.container = cls_or_instance
[docs] def get_value(self, obj, attr, accessor=None): """Return the value for a given key from an object.""" value = super(List, self).get_value(obj, attr, accessor=accessor) if self.container.attribute: if utils.is_collection(value): return [ self.container.get_value(each, self.container.attribute) for each in value ] return self.container.get_value(value, self.container.attribute) return value
[docs] def _bind_to_schema(self, field_name, schema): super(List, self)._bind_to_schema(field_name, schema) self.container = copy.deepcopy(self.container) self.container.parent = self self.container.name = field_name
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None if utils.is_collection(value): return [self.container._serialize(each, attr, obj, **kwargs) for each in value] return [self.container._serialize(value, attr, obj, **kwargs)]
[docs] def _deserialize(self, value, attr, data, **kwargs): if not utils.is_collection(value): self.fail('invalid') result = [] errors = {} for idx, each in enumerate(value): try: result.append(self.container.deserialize(each)) except ValidationError as e: result.append(e.data) errors.update({idx: e.messages}) if errors: raise ValidationError(errors, data=result) return result
[docs]class String(Field): """A string field. :param kwargs: The same keyword arguments that :class:`Field` receives. """ default_error_messages = { 'invalid': 'Not a valid string.', 'invalid_utf8': 'Not a valid utf-8 string.', }
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None return utils.ensure_text_type(value)
[docs] def _deserialize(self, value, attr, data, **kwargs): if not isinstance(value, basestring): self.fail('invalid') try: return utils.ensure_text_type(value) except UnicodeDecodeError: self.fail('invalid_utf8')
[docs]class UUID(String): """A UUID field.""" default_error_messages = { 'invalid_uuid': 'Not a valid UUID.', }
[docs] def _validated(self, value): """Format the value or raise a :exc:`ValidationError` if an error occurs.""" if value is None: return None if isinstance(value, uuid.UUID): return value try: if isinstance(value, bytes) and len(value) == 16: return uuid.UUID(bytes=value) else: return uuid.UUID(value) except (ValueError, AttributeError, TypeError): self.fail('invalid_uuid')
[docs] def _serialize(self, value, attr, obj, **kwargs): validated = str(self._validated(value)) if value is not None else None return super(String, self)._serialize(validated, attr, obj, **kwargs)
[docs] def _deserialize(self, value, attr, data, **kwargs): return self._validated(value)
[docs]class Number(Field): """Base class for number fields. :param bool as_string: If True, format the serialized value as a string. :param kwargs: The same keyword arguments that :class:`Field` receives. """ num_type = float default_error_messages = { 'invalid': 'Not a valid number.', } def __init__(self, as_string=False, **kwargs): self.as_string = as_string super(Number, self).__init__(**kwargs)
[docs] def _format_num(self, value): """Return the number value for value, given this field's `num_type`.""" # (value is True or value is False) is ~5x faster than isinstance(value, bool) if value is True or value is False: raise TypeError('value must be a Number, not a boolean.') return self.num_type(value)
[docs] def _validated(self, value): """Format the value or raise a :exc:`ValidationError` if an error occurs.""" if value is None: return None try: return self._format_num(value) except (TypeError, ValueError): self.fail('invalid')
def _to_string(self, value): return str(value)
[docs] def _serialize(self, value, attr, obj, **kwargs): """Return a string if `self.as_string=True`, otherwise return this field's `num_type`.""" ret = self._validated(value) return self._to_string(ret) if (self.as_string and ret not in (None, missing_)) else ret
[docs] def _deserialize(self, value, attr, data, **kwargs): return self._validated(value)
[docs]class Integer(Number): """An integer field. :param kwargs: The same keyword arguments that :class:`Number` receives. """ num_type = int default_error_messages = { 'invalid': 'Not a valid integer.', } # override Number def __init__(self, strict=False, **kwargs): self.strict = strict super(Integer, self).__init__(**kwargs) # override Number
[docs] def _format_num(self, value): if self.strict: if isinstance(value, numbers.Number) and isinstance(value, numbers.Integral): return super(Integer, self)._format_num(value) self.fail('invalid') return super(Integer, self)._format_num(value)
[docs]class Float(Number): """ A double as IEEE-754 double precision string. :param bool allow_nan: If `True`, `NaN`, `Infinity` and `-Infinity` are allowed, even though they are illegal according to the JSON specification. :param bool as_string: If True, format the value as a string. :param kwargs: The same keyword arguments that :class:`Number` receives. """ num_type = float default_error_messages = { 'special': 'Special numeric values (nan or infinity) are not permitted.', } def __init__(self, allow_nan=False, as_string=False, **kwargs): self.allow_nan = allow_nan super(Float, self).__init__(as_string=as_string, **kwargs)
[docs] def _format_num(self, value): num = super(Float, self)._format_num(value) if self.allow_nan is False: if math.isnan(num) or num == float('inf') or num == float('-inf'): self.fail('special') return num
[docs]class Decimal(Number): """A field that (de)serializes to the Python ``decimal.Decimal`` type. It's safe to use when dealing with money values, percentages, ratios or other numbers where precision is critical. .. warning:: This field serializes to a `decimal.Decimal` object by default. If you need to render your data as JSON, keep in mind that the `json` module from the standard library does not encode `decimal.Decimal`. Therefore, you must use a JSON library that can handle decimals, such as `simplejson`, or serialize to a string by passing ``as_string=True``. .. warning:: If a JSON `float` value is passed to this field for deserialization it will first be cast to its corresponding `string` value before being deserialized to a `decimal.Decimal` object. The default `__str__` implementation of the built-in Python `float` type may apply a destructive transformation upon its input data and therefore cannot be relied upon to preserve precision. To avoid this, you can instead pass a JSON `string` to be deserialized directly. :param int places: How many decimal places to quantize the value. If `None`, does not quantize the value. :param rounding: How to round the value during quantize, for example `decimal.ROUND_UP`. If None, uses the rounding value from the current thread's context. :param bool allow_nan: If `True`, `NaN`, `Infinity` and `-Infinity` are allowed, even though they are illegal according to the JSON specification. :param bool as_string: If True, serialize to a string instead of a Python `decimal.Decimal` type. :param kwargs: The same keyword arguments that :class:`Number` receives. .. versionadded:: 1.2.0 """ num_type = decimal.Decimal default_error_messages = { 'special': 'Special numeric values (nan or infinity) are not permitted.', } def __init__(self, places=None, rounding=None, allow_nan=False, as_string=False, **kwargs): self.places = decimal.Decimal((0, (1,), -places)) if places is not None else None self.rounding = rounding self.allow_nan = allow_nan super(Decimal, self).__init__(as_string=as_string, **kwargs) # override Number
[docs] def _format_num(self, value): num = decimal.Decimal(str(value)) if self.allow_nan: if num.is_nan(): return decimal.Decimal('NaN') # avoid sNaN, -sNaN and -NaN else: if num.is_nan() or num.is_infinite(): self.fail('special') if self.places is not None and num.is_finite(): num = num.quantize(self.places, rounding=self.rounding) return num
# override Number
[docs] def _validated(self, value): try: return super(Decimal, self)._validated(value) except decimal.InvalidOperation: self.fail('invalid')
# override Number def _to_string(self, value): return format(value, 'f')
[docs]class Boolean(Field): """A boolean field. :param set truthy: Values that will (de)serialize to `True`. If an empty set, any non-falsy value will deserialize to `True`. If `None`, `marshmallow.fields.Boolean.truthy` will be used. :param set falsy: Values that will (de)serialize to `False`. If `None`, `marshmallow.fields.Boolean.falsy` will be used. :param kwargs: The same keyword arguments that :class:`Field` receives. """ #: Default truthy values. truthy = { 't', 'T', 'true', 'True', 'TRUE', 'on', 'On', 'ON', '1', 1, True, } #: Default falsy values. falsy = { 'f', 'F', 'false', 'False', 'FALSE', 'off', 'Off', 'OFF', '0', 0, 0.0, False, } default_error_messages = { 'invalid': 'Not a valid boolean.', } def __init__(self, truthy=None, falsy=None, **kwargs): super(Boolean, self).__init__(**kwargs) if truthy is not None: self.truthy = set(truthy) if falsy is not None: self.falsy = set(falsy)
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None elif value in self.truthy: return True elif value in self.falsy: return False return bool(value)
[docs] def _deserialize(self, value, attr, data, **kwargs): if not self.truthy: return bool(value) else: try: if value in self.truthy: return True elif value in self.falsy: return False except TypeError: pass self.fail('invalid')
[docs]class FormattedString(Field): """Interpolate other values from the object into this field. The syntax for the source string is the same as the string `str.format` method from the python stdlib. :: class UserSchema(Schema): name = fields.String() greeting = fields.FormattedString('Hello {name}') ser = UserSchema() res = ser.dump(user) res.data # => {'name': 'Monty', 'greeting': 'Hello Monty'} """ default_error_messages = { 'format': 'Cannot format string with given data.', } _CHECK_ATTRIBUTE = False def __init__(self, src_str, *args, **kwargs): Field.__init__(self, *args, **kwargs) self.src_str = text_type(src_str)
[docs] def _serialize(self, value, attr, obj, **kwargs): try: data = utils.to_marshallable_type(obj) return self.src_str.format(**data) except (TypeError, IndexError): self.fail('format')
[docs]class DateTime(Field): """A formatted datetime string in UTC. Example: ``'2014-12-22T03:12:58.019077+00:00'`` Timezone-naive `datetime` objects are converted to UTC (+00:00) by :meth:`Schema.dump <marshmallow.Schema.dump>`. :meth:`Schema.load <marshmallow.Schema.load>` returns `datetime` objects that are timezone-aware. :param str format: Either ``"rfc"`` (for RFC822), ``"iso"`` (for ISO8601), or a date format string. If `None`, defaults to "iso". :param kwargs: The same keyword arguments that :class:`Field` receives. """ SERIALIZATION_FUNCS = { 'iso': utils.isoformat, 'iso8601': utils.isoformat, 'rfc': utils.rfcformat, 'rfc822': utils.rfcformat, } DESERIALIZATION_FUNCS = { 'iso': utils.from_iso_datetime, 'iso8601': utils.from_iso_datetime, 'rfc': utils.from_rfc, 'rfc822': utils.from_rfc, } DEFAULT_FORMAT = 'iso' OBJ_TYPE = 'datetime' SCHEMA_OPTS_VAR_NAME = 'datetimeformat' localtime = False default_error_messages = { 'invalid': 'Not a valid {obj_type}.', 'format': '"{input}" cannot be formatted as a {obj_type}.', } def __init__(self, format=None, **kwargs): super(DateTime, self).__init__(**kwargs) # Allow this to be None. It may be set later in the ``_serialize`` # or ``_deserialize`` methods This allows a Schema to dynamically set the # format, e.g. from a Meta option self.format = format
[docs] def _bind_to_schema(self, field_name, schema): super(DateTime, self)._bind_to_schema(field_name, schema) self.format = ( self.format or getattr(schema.opts, self.SCHEMA_OPTS_VAR_NAME) or self.DEFAULT_FORMAT )
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None data_format = self.format or self.DEFAULT_FORMAT format_func = self.SERIALIZATION_FUNCS.get(data_format) if format_func: try: return format_func(value, localtime=self.localtime) except (TypeError, AttributeError, ValueError): self.fail('format', input=value, obj_type=self.OBJ_TYPE) else: return value.strftime(data_format)
[docs] def _deserialize(self, value, attr, data, **kwargs): if not value: # Falsy values, e.g. '', None, [] are not valid raise self.fail('invalid', obj_type=self.OBJ_TYPE) data_format = self.format or self.DEFAULT_FORMAT func = self.DESERIALIZATION_FUNCS.get(data_format) if func: try: return func(value) except (TypeError, AttributeError, ValueError): raise self.fail('invalid', obj_type=self.OBJ_TYPE) else: try: return self._make_object_from_format(value, data_format) except (TypeError, AttributeError, ValueError): raise self.fail('invalid', obj_type=self.OBJ_TYPE)
@staticmethod def _make_object_from_format(value, data_format): return dt.datetime.strptime(value, data_format)
[docs]class LocalDateTime(DateTime): """A formatted datetime string in localized time, relative to UTC. ex. ``"Sun, 10 Nov 2013 08:23:45 -0600"`` Takes the same arguments as :class:`DateTime <marshmallow.fields.DateTime>`. """ localtime = True
[docs]class Time(Field): """ISO8601-formatted time string. :param kwargs: The same keyword arguments that :class:`Field` receives. """ default_error_messages = { 'invalid': 'Not a valid time.', 'format': '"{input}" cannot be formatted as a time.', }
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None try: ret = value.isoformat() except AttributeError: self.fail('format', input=value) if value.microsecond: return ret[:15] return ret
[docs] def _deserialize(self, value, attr, data, **kwargs): """Deserialize an ISO8601-formatted time to a :class:`datetime.time` object.""" if not value: # falsy values are invalid self.fail('invalid') try: return utils.from_iso_time(value) except (AttributeError, TypeError, ValueError): self.fail('invalid')
[docs]class Date(DateTime): """ISO8601-formatted date string. :param format: Either ``"iso"`` (for ISO8601) or a date format string. If `None`, defaults to "iso". :param kwargs: The same keyword arguments that :class:`Field` receives. """ default_error_messages = { 'invalid': 'Not a valid date.', 'format': '"{input}" cannot be formatted as a date.', } SERIALIZATION_FUNCS = { 'iso': utils.to_iso_date, 'iso8601': utils.to_iso_date, } DESERIALIZATION_FUNCS = { 'iso': utils.from_iso_date, 'iso8601': utils.from_iso_date, } DEFAULT_FORMAT = 'iso' OBJ_TYPE = 'date' SCHEMA_OPTS_VAR_NAME = 'dateformat' @staticmethod def _make_object_from_format(value, data_format): return dt.datetime.strptime(value, data_format).date()
[docs]class TimeDelta(Field): """A field that (de)serializes a :class:`datetime.timedelta` object to an integer and vice versa. The integer can represent the number of days, seconds or microseconds. :param str precision: Influences how the integer is interpreted during (de)serialization. Must be 'days', 'seconds', 'microseconds', 'milliseconds', 'minutes', 'hours' or 'weeks'. :param str error: Error message stored upon validation failure. :param kwargs: The same keyword arguments that :class:`Field` receives. .. versionchanged:: 2.0.0 Always serializes to an integer value to avoid rounding errors. Add `precision` parameter. """ DAYS = 'days' SECONDS = 'seconds' MICROSECONDS = 'microseconds' MILLISECONDS = 'milliseconds' MINUTES = 'minutes' HOURS = 'hours' WEEKS = 'weeks' default_error_messages = { 'invalid': 'Not a valid period of time.', 'format': '{input!r} cannot be formatted as a timedelta.', } def __init__(self, precision='seconds', error=None, **kwargs): precision = precision.lower() units = ( self.DAYS, self.SECONDS, self.MICROSECONDS, self.MILLISECONDS, self.MINUTES, self.HOURS, self.WEEKS, ) if precision not in units: msg = 'The precision must be {0} or "{1}".'.format( ', '.join(['"{}"'.format(each) for each in units[:-1]]), units[-1], ) raise ValueError(msg) self.precision = precision super(TimeDelta, self).__init__(error=error, **kwargs)
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None try: base_unit = dt.timedelta(**{self.precision: 1}) return int(value.total_seconds() / base_unit.total_seconds()) except AttributeError: self.fail('format', input=value)
[docs] def _deserialize(self, value, attr, data, **kwargs): try: value = int(value) except (TypeError, ValueError): self.fail('invalid') kwargs = {self.precision: value} try: return dt.timedelta(**kwargs) except OverflowError: self.fail('invalid')
[docs]class Dict(Field): """A dict field. Supports dicts and dict-like objects. Optionally composed with another `Field` class or instance. Example: :: numbers = fields.Dict(values=fields.Float(), keys=fields.Str()) :param Field values: A field class or instance for dict values. :param Field keys: A field class or instance for dict keys. :param kwargs: The same keyword arguments that :class:`Field` receives. .. note:: When the structure of nested data is not known, you may omit the `values` and `keys` arguments to prevent content validation. .. versionadded:: 2.1.0 """ default_error_messages = { 'invalid': 'Not a valid mapping type.', } def __init__(self, values=None, keys=None, **kwargs): super(Dict, self).__init__(**kwargs) if values is None: self.value_container = None elif isinstance(values, type): if not issubclass(values, FieldABC): raise ValueError( '"values" must be a subclass of ' 'marshmallow.base.FieldABC', ) self.value_container = values() else: if not isinstance(values, FieldABC): raise ValueError( '"values" must be of type ' 'marshmallow.base.FieldABC', ) self.value_container = values if keys is None: self.key_container = None elif isinstance(keys, type): if not issubclass(keys, FieldABC): raise ValueError( '"keys" must be a subclass of ' 'marshmallow.base.FieldABC', ) self.key_container = keys() else: if not isinstance(keys, FieldABC): raise ValueError( '"keys" must be of type ' 'marshmallow.base.FieldABC', ) self.key_container = keys
[docs] def _bind_to_schema(self, field_name, schema): super(Dict, self)._bind_to_schema(field_name, schema) if self.value_container: self.value_container = copy.deepcopy(self.value_container) self.value_container.parent = self self.value_container.name = field_name if self.key_container: self.key_container = copy.deepcopy(self.key_container) self.key_container.parent = self self.key_container.name = field_name
[docs] def _serialize(self, value, attr, obj, **kwargs): if value is None: return None if not self.value_container and not self.key_container: return value if isinstance(value, Mapping): values = value.values() if self.value_container: values = [ self.value_container._serialize(item, attr, obj, **kwargs) for item in values ] keys = value.keys() if self.key_container: keys = [self.key_container._serialize(key, attr, obj, **kwargs) for key in keys] return dict(zip(keys, values)) self.fail('invalid')
[docs] def _deserialize(self, value, attr, data, **kwargs): if not isinstance(value, Mapping): self.fail('invalid') if not self.value_container and not self.key_container: return value errors = collections.defaultdict(dict) values = list(value.values()) keys = list(value.keys()) if self.key_container: for idx, key in enumerate(keys): try: keys[idx] = self.key_container.deserialize(key) except ValidationError as e: errors[key]['key'] = e.messages if self.value_container: for idx, item in enumerate(values): try: values[idx] = self.value_container.deserialize(item) except ValidationError as e: values[idx] = e.data key = keys[idx] errors[key]['value'] = e.messages result = dict(zip(keys, values)) if errors: raise ValidationError(errors, data=result) return result
[docs]class Url(String): """A validated URL field. Validation occurs during both serialization and deserialization. :param default: Default value for the field if the attribute is not set. :param str attribute: The name of the attribute to get the value from. If `None`, assumes the attribute has the same name as the field. :param bool relative: Allow relative URLs. :param kwargs: The same keyword arguments that :class:`String` receives. """ default_error_messages = {'invalid': 'Not a valid URL.'} def __init__(self, relative=False, schemes=None, require_tld=True, **kwargs): String.__init__(self, **kwargs) self.relative = relative self.require_tld = require_tld # Insert validation into self.validators so that multiple errors can be # stored. self.validators.insert( 0, validate.URL( relative=self.relative, schemes=schemes, require_tld=self.require_tld, error=self.error_messages['invalid'], ), ) def _validated(self, value): if value is None: return None return validate.URL( relative=self.relative, require_tld=self.require_tld, error=self.error_messages['invalid'], )(value)
[docs]class Email(String): """A validated email field. Validation occurs during both serialization and deserialization. :param args: The same positional arguments that :class:`String` receives. :param kwargs: The same keyword arguments that :class:`String` receives. """ default_error_messages = {'invalid': 'Not a valid email address.'} def __init__(self, *args, **kwargs): String.__init__(self, *args, **kwargs) # Insert validation into self.validators so that multiple errors can be # stored. self.validators.insert(0, validate.Email(error=self.error_messages['invalid'])) def _validated(self, value): if value is None: return None return validate.Email( error=self.error_messages['invalid'], )(value)
[docs]class Method(Field): """A field that takes the value returned by a `Schema` method. :param str serialize: The name of the Schema method from which to retrieve the value. The method must take an argument ``obj`` (in addition to self) that is the object to be serialized. :param str deserialize: Optional name of the Schema method for deserializing a value The method must take a single argument ``value``, which is the value to deserialize. .. versionchanged:: 2.0.0 Removed optional ``context`` parameter on methods. Use ``self.context`` instead. .. versionchanged:: 2.3.0 Deprecated ``method_name`` parameter in favor of ``serialize`` and allow ``serialize`` to not be passed at all. .. versionchanged:: 3.0.0 Removed ``method_name`` parameter. """ _CHECK_ATTRIBUTE = False def __init__(self, serialize=None, deserialize=None, **kwargs): # Set dump_only and load_only based on arguments kwargs['dump_only'] = bool(serialize) and not bool(deserialize) kwargs['load_only'] = bool(deserialize) and not bool(serialize) super(Method, self).__init__(**kwargs) self.serialize_method_name = serialize self.deserialize_method_name = deserialize
[docs] def _serialize(self, value, attr, obj, **kwargs): if not self.serialize_method_name: return missing_ method = utils.callable_or_raise( getattr(self.parent, self.serialize_method_name, None), ) return method(obj)
[docs] def _deserialize(self, value, attr, data, **kwargs): if self.deserialize_method_name: method = utils.callable_or_raise( getattr(self.parent, self.deserialize_method_name, None), ) return method(value) return value
[docs]class Function(Field): """A field that takes the value returned by a function. :param callable serialize: A callable from which to retrieve the value. The function must take a single argument ``obj`` which is the object to be serialized. It can also optionally take a ``context`` argument, which is a dictionary of context variables passed to the serializer. If no callable is provided then the ```load_only``` flag will be set to True. :param callable deserialize: A callable from which to retrieve the value. The function must take a single argument ``value`` which is the value to be deserialized. It can also optionally take a ``context`` argument, which is a dictionary of context variables passed to the deserializer. If no callable is provided then ```value``` will be passed through unchanged. .. versionchanged:: 2.3.0 Deprecated ``func`` parameter in favor of ``serialize``. .. versionchanged:: 3.0.0a1 Removed ``func`` parameter. """ _CHECK_ATTRIBUTE = False def __init__(self, serialize=None, deserialize=None, func=None, **kwargs): # Set dump_only and load_only based on arguments kwargs['dump_only'] = bool(serialize) and not bool(deserialize) kwargs['load_only'] = bool(deserialize) and not bool(serialize) super(Function, self).__init__(**kwargs) self.serialize_func = serialize and utils.callable_or_raise(serialize) self.deserialize_func = deserialize and utils.callable_or_raise(deserialize)
[docs] def _serialize(self, value, attr, obj, **kwargs): return self._call_or_raise(self.serialize_func, obj, attr)
[docs] def _deserialize(self, value, attr, data, **kwargs): if self.deserialize_func: return self._call_or_raise(self.deserialize_func, value, attr) return value
def _call_or_raise(self, func, value, attr): if len(utils.get_func_args(func)) > 1: if self.parent.context is None: msg = 'No context available for Function field {0!r}'.format(attr) raise ValidationError(msg) return func(value, self.parent.context) else: return func(value)
[docs]class Constant(Field): """A field that (de)serializes to a preset constant. If you only want the constant added for serialization or deserialization, you should use ``dump_only=True`` or ``load_only=True`` respectively. :param constant: The constant to return for the field attribute. .. versionadded:: 2.0.0 """ _CHECK_ATTRIBUTE = False def __init__(self, constant, **kwargs): super(Constant, self).__init__(**kwargs) self.constant = constant self.missing = constant self.default = constant
[docs] def _serialize(self, value, *args, **kwargs): return self.constant
[docs] def _deserialize(self, value, *args, **kwargs): return self.constant
class Inferred(Field): """A field that infers how to serialize, based on the value type. .. warning:: This class is treated as private API. Users should not need to use this class directly. """ def __init__(self): super(Inferred, self).__init__() # We memoize the fields to avoid creating and binding new fields # every time on serialization. self._field_cache = {} def _serialize(self, value, attr, obj, **kwargs): field_cls = self.root.TYPE_MAPPING.get(type(value)) if field_cls is None: field = super(Inferred, self) else: field = self._field_cache.get(field_cls) if field is None: field = field_cls() field._bind_to_schema(self.name, self.parent) self._field_cache[field_cls] = field return field._serialize(value, attr, obj, **kwargs) # Aliases URL = Url Str = String Bool = Boolean Int = Integer