from apollo_utils.core.constants.dsp import DSP
from apollo_utils.service.schemas.fields.amazon_id import AmazonPlaylistIdField
from apollo_utils.service.schemas.mixins.market import MarketUsMixin
from marshmallow import fields

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


class PlaylistAmazonTracklistGraph:
    class Request(RequestSchema, MarketUsMixin):
        """Request schema to get apple playlist tracklist frontline and local tracks graphs."""

        playlist_id = AmazonPlaylistIdField(required=True, dsp=DSP.AMAZON)

    class Response(Schema):
        """Response schema for apple playlist tracklist frontline and local tracks graphs."""

        frontline_tracks = fields.Int(missing=None)
        local_tracks = fields.Int(missing=None)
        tracks_num = fields.Int(missing=None)