from sqlalchemy.orm import Query

from server.artist.models import ArtistSoundCloudTracks
from server.dna.utils import DataQueryWrapper


class SoundCloudRepo:
    tracks_model = ArtistSoundCloudTracks

    # TODO: develop one more universal method (get_data) that
    # will return track's data, albums data, external_links data
    @classmethod
    async def get_tracks(cls, _id: str, limit: int = 6) -> list:
        query = Query(cls.tracks_model.value).where(cls.tracks_model.id == _id)
        results: list = list(await DataQueryWrapper.select(query))

        return results[:limit]