playlist table updates (ongoing)

ui/qt/widgets/playlist_table.py

@@ -1,6 +1,5 @@
 from PyQt5.QtWidgets import QTableWidget, QTableWidgetItem
 
-from core import db
 from core.repositories import TrackRepository
 
 
@@ -8,8 +7,10 @@ class PlaylistTable(QTableWidget):
     def __init__(self, parent):
         super(QTableWidget, self).__init__(parent)
         self.playlist = None
+        self.count = -1
 
     def populate(self, playlist):
+        self.playlist = playlist
         track_repo = TrackRepository()
 
         self.clear()
@@ -30,3 +31,9 @@ class PlaylistTable(QTableWidget):
             self.setItem(i, 0, item)
             self.setItem(i, 1, QTableWidgetItem(track.title))
             i += 1
+        self.count = i
+
+    def clear(self):
+        while self.rowCount():
+            self.removeRow(0)
+        super().clear()

ui/window.py

@@ -176,15 +176,17 @@ class MainWindow(QMainWindow):
         track = self.ui.explorerTable.selected_track()
         if track is None:
             return
-        playlist = self.ui.explorerTable.selected_track()
-        if track is None:
+        playlist = self.ui.explorerPlaylist.playlist
+        if playlist is None:
             return
+        order = self.ui.explorerPlaylist.count + 1
         track_repo = TrackRepository()
         track_repo.add_to_playlist(
             track.id,
-            None,
-            None
+            playlist.id,
+            order
         )
+        self.ui.explorerPlaylist.populate(playlist)
 
     def run(self):
         pass