from apollo_utils.service.schemas.fields.datetime import DateStringField
from apollo_utils.service.schemas.mixins.market import MarketListOptionalMixin
from marshmallow import fields

from server.schemas.base.schemas import RequestSchema, Schema


class YoutubeDemographics:
    """Visited tracks schema."""

    class Request(RequestSchema, MarketListOptionalMixin):
        video_id = fields.String(required=True)
        start_date = DateStringField(required=True)
        end_date = DateStringField(required=True)

    class Response(Schema):
        age_band_most_popular = fields.String()
        age_band_percentage = fields.String()
        gender_percentage = fields.String()
        gender_most_popular = fields.String()