from PyQt5.QtWidgets import QTableWidget

from core import db
from core.repositories import SessionRepository


class PlaylistTable(QTableWidget):
    def __init__(self, parent):
        super(QTableWidget, self).__init__(parent)

        self.playlist = None

    def load_playlist(self, playlist):
        session = db.Session()
        playlist_repo = SessionRepository(session)

        self.clear()

        playlists = playlist_repo.get_all()