from sqlalchemy.orm import Query

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


class TiktokRepo:
    videos_model = ArtistTiktokVideos

    @classmethod
    async def get_artist_videos(cls, _id: str, limit: int = 6) -> list:
        query = Query(cls.videos_model.value).where(cls.videos_model.id == _id).limit(limit)

        return list(await DataQueryWrapper.select(query))