from marshmallow import Schema, fields
from marshmallow.validate import Regexp


class ChartPlacementPrivateDataSchema(Schema):
    is_part_of_catalog = fields.Boolean(required=True, allow_none=False)
    brand_ids = fields.List(fields.String(required=True, allow_none=False), required=False, allow_none=True)


class ChartsPlacementDataloaderPrivate:
    """Request & Response schema for getting /charts/placement-dataloader-private API"""

    class Request(Schema):
        chart_id = fields.UUID(required=True)
        chart_date = fields.Date(required=True)
        uuid = fields.UUID(required=True)

    class Response(Schema):
        data = fields.Nested(ChartPlacementPrivateDataSchema, required=True)