from delphi_api.bigtable import BigTableModel
from delphi_api.v2.data_models.amazon import (
    AmazonPlaylistStream,
    AmazonPlaylistTrackStream,
    AmazonTrackStream,
)
from delphi_api.v2.view_models.dsp import DspViewModel


class AmazonTrackStreamViewModel(DspViewModel):
    """See :class:`DspViewModel`"""

    @property
    def data_model(self) -> BigTableModel:
        return AmazonTrackStream()


class AmazonPlaylistTrackStreamViewModel(DspViewModel):
    """See :class:`DspViewModel`"""

    @property
    def data_model(self) -> BigTableModel:
        return AmazonPlaylistTrackStream()


class AmazonPlaylistStreamViewModel(DspViewModel):
    """See :class:`DspViewModel`"""

    @property
    def data_model(self) -> BigTableModel:
        return AmazonPlaylistStream()