"""Schema for Instant Grats."""

import marshmallow
from marshmallow import fields
from marshmallow.validate import Range


class InstantGratBaseSchema(marshmallow.Schema):
    """Base Schema used for Instant Grat data."""

    tuid = fields.Int(required=True, validate=Range(min=1))
    date = fields.DateTime(required=True, format='%Y-%m-%d')
    store_id = fields.Int(required=True, validate=Range(min=1))


class InstantGratsUpdateSchema(marshmallow.Schema):
    """Schema used for creating/updating instant grats."""

    items = fields.Nested(InstantGratBaseSchema, many=True)