Megamichi/Wobuzz
1
0
Fork 0
forked from Wobbl/Wobuzz
Code Pull requests Activity

Did some memory optimisation, moved some files and created a completely not tested gui class that will list an artist's tracks.

Browse source 
Made tracks return an already existing object when they get created with a path of an already existing track object.
This commit is contained in:
The Wobbler 2025-03-06 16:35:13 +01:00
parent 9ae1704e4a
commit 7edaebc3c3
7 changed files with 77 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

188
wobuzz/ui/playlist.py
View file

@ -1,188 +0,0 @@
#!/usr/bin/python3
from PyQt6.QtCore import Qt, pyqtSignal
from PyQt6.QtGui import QDropEvent, QIcon
from PyQt6.QtWidgets import QTreeWidget, QAbstractItemView
from .track import TrackItem
class PlaylistView(QTreeWidget):
itemDropped = pyqtSignal(QTreeWidget, list)
sort_signal = pyqtSignal(int, Qt.SortOrder)
playing_mark = QIcon.fromTheme(QIcon.ThemeIcon.MediaPlaybackStart)
def __init__(self, playlist, library_widget, parent=None):
super().__init__(parent)
self.playlist = playlist
self.library_widget = library_widget
self.app = playlist.app
self.header = self.header()
self.header.setSectionsClickable(True)
self.header.setSortIndicatorShown(True)
playlist.views[id(self.library_widget)] = self # let the playlist know that this view exists
self.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
self.setColumnCount(4)
headers = [
"#",
"Title",
"Artist",
"Album",
"# Custom Sorting"
]
self.setHeaderLabels(headers)
self.itemActivated.connect(self.on_track_activation)
self.header.sectionClicked.connect(self.on_header_click)
self.sort_signal.connect(self.sortItems)
def on_header_click(self, section_index: int):
if section_index == 0: # this would just invert the current sorting
return
sorting = self.playlist.sorting
last_sort_section_index, order = sorting[4]
if last_sort_section_index == section_index:
order = not order # invert order
self.playlist.sorting[4] = (section_index, order) # set sorting
# convert True/False to Qt.SortOrder.AscendingOrder/Qt.SortOrder.DescendingOrder
qorder = Qt.SortOrder.AscendingOrder if order else Qt.SortOrder.DescendingOrder
self.header.setSortIndicator(section_index, qorder)
else:
del sorting[0] # remove first sort
sorting.append((section_index, True)) # last sort is this section index, ascending
self.header.setSortIndicator(section_index, Qt.SortOrder.AscendingOrder)
self.sort()
def sort(self):
for index, order in self.playlist.sorting:
# convert True/False to Qt.SortOrder.AscendingOrder/Qt.SortOrder.DescendingOrder
qorder = Qt.SortOrder.AscendingOrder if order else Qt.SortOrder.DescendingOrder
# somehow, QTreeWidget.sortItems() cant be called from a thread, so we have to use a signal to execute it
# in the main thread
self.sort_signal.emit(index, qorder)
# self.sortItems(index, qorder)
self.on_sort()
def on_sort(self, user_sort: bool=False):
num_tracks = self.topLevelItemCount()
i = 0
while i < num_tracks:
track = self.topLevelItem(i)
i += 1
track.setText(0, str(i)) # 0 = index
if user_sort:
track.setText(4, str(i)) # 4 = user sort index
if user_sort:
if not self.playlist.sorting[4][0] == 4: # set last sort to user sort
del self.playlist.sorting[0]
self.playlist.sorting.append((4, True))
self.header.setSortIndicator(4, Qt.SortOrder.AscendingOrder)
self.playlist.sync(self, user_sort) # sync playlist to this view
def dropEvent(self, event: QDropEvent):
# receive items that were dropped and create new items from its tracks (new items bc. widgets can only have
# one parent)
if event.source() == self:
items = self.selectedItems() # dragged items are always selected items
self.itemDropped.emit(self, items)
else:
items = self.app.gui.dropped
i = 0
for item in items:
track = item.track
self.playlist.tracks.append(track)
track_item = TrackItem(track, i, self)
i += 1
super().dropEvent(event)
event.accept()
self.on_sort(True)
def dragEnterEvent(self, event):
# store dragged items in gui.dropped, so the other playlist can receive it
if event.source() == self:
items = self.selectedItems()
self.app.gui.dropped = items
super().dragEnterEvent(event)
event.accept()
def load_tracks(self):
i = 0
for track in self.playlist.tracks:
track_item = TrackItem(track, i, self)
i += 1
def on_track_activation(self, item, column):
if not self.app.player.current_playlist == self.playlist:
self.app.player.current_playlist = self.playlist
index = self.indexOfTopLevelItem(item)
self.app.player.play_track_in_playlist(index)
def on_track_change(self, previous_track, track):
# unmark the previous track and playlist and mark the current track and playlist as playing
playlist_tabs = self.parent().parent()
index = playlist_tabs.indexOf(self) # tab index of this playlist
if previous_track:
# unmark all playlists by looping through the tabs
for i in range(playlist_tabs.count()):
playlist_tabs.setTabIcon(i, QIcon(None))
# unmark the previous track in all playlists
for item in previous_track.items:
item.unmark()
if track:
playlist_tabs.setTabIcon(index, self.playing_mark) # mark this playlist
# mark the current track in this playlist
item = self.topLevelItem(self.app.player.current_playlist.current_track_index)
item.mark()
def append_track(self, track):
TrackItem(track, self.topLevelItemCount(), self)