"""Schemas for UGC video metrics endpoint."""

from marshmallow import fields

from analytics.schemas.common import BaseSchema, SourceSchema


class UgcVideoMetricsItemSchema(BaseSchema):
    """Marshmallow schema for a items in a list of UGC metrics."""

    video_id = fields.String()
    asset_id = fields.String()
    channel_id = fields.String()
    track_id = fields.Integer()
    track_isrc = fields.String()
    views = fields.Integer()
    premium_views = fields.Integer()
    percentage = fields.Float()
    claim_type = fields.String()
    claim_created_date = fields.String()
    claim_policy = fields.String()
    claim_status = fields.String()
    content_type = fields.String()
    video_name = fields.String()
    channel_name = fields.String()
    release_date = fields.String()


class UgcVideoMetricsSchema(BaseSchema):
    """Marshmallow schema for the list of UGC metrics."""

    total_results = fields.Integer()
    items = fields.Nested(nested=UgcVideoMetricsItemSchema, many=True)
    sources = fields.Nested(SourceSchema, many=True)