OOPed the way playlist views are created.

This commit is contained in:
The Wobbler 2025-01-25 17:21:43 +01:00
parent 88b846f3b6
commit 028c38b1b6
7 changed files with 61 additions and 67 deletions

View file

@ -22,72 +22,13 @@ class Library:
self.temporary_playlist = Playlist(self.app, "Temporary Playlist")
self.playlists = [self.temporary_playlist]
self.play_icon = QIcon.fromTheme(QIcon.ThemeIcon.MediaPlaybackStart)
def create_playlist_views(self):
def load_playlist_views(self):
for library_dock in self.library_docks:
playlist_tabs: QTabWidget = library_dock.library.playlist_tabs
playlist_tabs.playlists = {}
for playlist in self.playlists:
playlist_view = self.create_playlist_view(playlist)
playlist_tabs.playlists[playlist.title] = playlist_view
playlist_view = PlaylistView(playlist)
playlist_tabs.addTab(playlist_view, playlist.title)
def create_playlist_view(self, playlist: Playlist):
view = PlaylistView(playlist)
view.itemActivated.connect(self.on_track_activation)
num_tracks = len(playlist.tracks)
i = 0
while i < num_tracks:
track = playlist.tracks[i]
track_item = QTreeWidgetItem(view)
track_item.setFlags(
Qt.ItemFlag.ItemIsEnabled |
Qt.ItemFlag.ItemIsSelectable |
Qt.ItemFlag.ItemIsDragEnabled
)
track_item.track = track
track_item.setText(2, track.tags.title)
track_item.setText(3, track.tags.artist)
track_item.setText(4, track.tags.album)
track_item.setText(5, str(i + 1))
i += 1
return view
def on_track_activation(self, item, column):
index = item.treeWidget().indexOfTopLevelItem(item)
self.app.player.play_track_in_playlist(index)
def get_track_tree_item(self, track, playlist_view: PlaylistView):
index = self.app.player.current_playlist.tracks.index(track)
item = playlist_view.topLevelItem(index)
return item
def get_playing_playlist_view(self):
view = self.main_library_dock.library.playlist_tabs.playlists[self.app.player.current_playlist.title]
return view
def on_track_change(self, previous_track, track):
playing_playlist_view = self.get_playing_playlist_view()
if previous_track:
item = self.get_track_tree_item(previous_track, playing_playlist_view)
item.setIcon(0, QIcon(None))
if track:
item = self.get_track_tree_item(track, playing_playlist_view)
item.setIcon(0, self.play_icon)