"""Schema for track placements."""

from marshmallow import fields

from sound_recordings.schemas.common import BaseSchema, SourceSchema


class DemographicAgeBreakdownSchema(BaseSchema):
    """Marshmallow schema for demographic detail object."""

    _17 = fields.Integer()
    _18_22 = fields.Integer()
    _23_27 = fields.Integer()
    _28_34 = fields.Integer()
    _35_44 = fields.Integer()
    _45_59 = fields.Integer()
    _60_ = fields.Integer()
    UA = fields.Integer()


class DemographicGenderBreakdownSchema(BaseSchema):
    """Marshmallow schema for demographic detail by gender object."""

    M = fields.Integer()
    F = fields.Integer()
    UG = fields.Integer()


class DemographicSchema(BaseSchema):
    """Marshmallow schema for demographic by store response object."""

    age = fields.Nested(nested=DemographicAgeBreakdownSchema)
    gender = fields.Nested(nested=DemographicGenderBreakdownSchema)


class DemographicsSchema(BaseSchema):
    """Marshmallow schema for demographics response object."""

    isrc = fields.String()
    demographics = fields.Nested(nested=DemographicSchema)
    sources = fields.Nested(SourceSchema, many=True)