from marshmallow import fields, validate

from server.constants.devices import Platform
from server.schemas.base.schemas import RequestSchema


class DeviceRegister:
    """Register user device."""

    class Request(RequestSchema):
        expo_token = fields.String(required=True, validate=validate.Length(min=1, max=255))
        device_inner_id = fields.String(required=True)
        os = fields.String(required=True, validate=validate.OneOf(Platform.values()))


class DeviceUnregister:
    """Unregister user device."""

    class Request(RequestSchema):
        expo_token = fields.String(required=True, validate=validate.Length(min=1, max=255))
        device_inner_id = fields.String(required=True)
        os = fields.String(required=True, validate=validate.OneOf(Platform.values()))