"""Schema for product track metrics."""

from marshmallow import fields

from analytics.schemas.common import BaseSchema, SoundRecordingCoreMetadataSchema


class ProductTrackMetricsItemSchema(SoundRecordingCoreMetadataSchema):
    """Marshmallow schema for items in product track metrics."""

    track_id = fields.String()
    tuid = fields.String()
    streams_1_day = fields.Integer()
    growth_percentage_1_day = fields.Float()
    streams_7_days = fields.Integer()
    growth_percentage_7_days = fields.Float()
    streams_28_days = fields.Integer()
    growth_percentage_28_days = fields.Float()
    streams_all_time = fields.Integer()


class ProductTrackMetricsSchema(BaseSchema):
    """Marshmallow schema for the product metrics by track."""

    product_id = fields.String()
    tracks = fields.Nested(nested=ProductTrackMetricsItemSchema, many=True)