from marshmallow import fields

from server.publishers.schemas.base import BaseMessageBodySchema
from server.schemas.messages.mixins import DeviceIdNoneMixin
from server.schemas.messages.push import PushMessageMixin


class PushMessageBodySchema(PushMessageMixin, DeviceIdNoneMixin, BaseMessageBodySchema):
    data = fields.Dict(default=None)
    priority = fields.Function(lambda obj: obj.priority.value)