98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
import db
|
|
import audio
|
|
from gi.repository import Gtk
|
|
from . import column
|
|
from . import model
|
|
from . import sort
|
|
|
|
class PlaylistView(Gtk.ColumnView):
|
|
def __init__(self):
|
|
Gtk.ColumnView.__init__(self)
|
|
self.add_css_class("data-table")
|
|
self.set_model(model.PlaylistSelection())
|
|
self.set_enable_rubberband(True)
|
|
self.set_hexpand(True)
|
|
self.set_vexpand(True)
|
|
self.connect("activate", self.activated)
|
|
audio.Player.connect("track-changed", self.track_changed)
|
|
|
|
self.append_column(column.TracknoColumn())
|
|
self.append_column(column.TitleColumn())
|
|
self.append_column(column.LengthColumn())
|
|
self.append_column(column.ArtistColumn())
|
|
self.append_column(column.AlbumColumn())
|
|
self.append_column(column.SubtitleColumn())
|
|
self.append_column(column.YearColumn())
|
|
self.append_column(column.PlayCountColumn())
|
|
self.append_column(column.LastPlayedColumn())
|
|
|
|
def activated(self, view, position):
|
|
track = self.get_model().get_item(position)
|
|
if audio.play_track(track):
|
|
plist = self.get_playlist()
|
|
if index := plist.get_track_index(track):
|
|
plist.current = index
|
|
if plist == db.user.Table.find("Queued Tracks"):
|
|
plist.remove_track(track)
|
|
|
|
def selected_tracks(self):
|
|
model = self.get_model()
|
|
selection = model.get_selection()
|
|
for n in range(selection.get_size()):
|
|
yield model.get_item(selection.get_nth(n))
|
|
model.unselect_all()
|
|
|
|
def clear_selection(self):
|
|
self.get_model().unselect_all()
|
|
|
|
def scroll_to(self, track):
|
|
selection = self.get_model()
|
|
adjustment = self.get_vadjustment()
|
|
for i in range(selection.get_n_items()):
|
|
if selection.get_item(i) == track:
|
|
selection.select_item(i, True)
|
|
pos = max(i - 5, 0) * adjustment.get_upper()
|
|
adjustment.set_value(pos / selection.get_n_items())
|
|
return True
|
|
return False
|
|
|
|
def track_changed(self, player, old, new):
|
|
if not self.scroll_to(new):
|
|
self.scroll_to(self.get_playlist().get_current_track())
|
|
|
|
def get_filter_model(self): return self.get_model().get_filter_model()
|
|
def get_playlist(self): return self.get_model().get_playlist()
|
|
def set_playlist(self, plist): self.get_model().set_playlist(plist)
|
|
|
|
|
|
class PlaylistWindow(Gtk.ScrolledWindow):
|
|
def __init__(self):
|
|
Gtk.ScrolledWindow.__init__(self)
|
|
self.set_child(PlaylistView())
|
|
|
|
def get_filter_model(self): return self.get_child().get_filter_model()
|
|
def get_playlist(self): return self.get_child().get_playlist()
|
|
def set_playlist(self, plist): self.get_child().set_playlist(plist)
|
|
def selected_tracks(self): return self.get_child().selected_tracks()
|
|
def clear_selection(self): self.get_child().clear_selection()
|
|
|
|
|
|
class SortOrderView(Gtk.ListView):
|
|
def __init__(self):
|
|
Gtk.ListView.__init__(self)
|
|
self.set_model(Gtk.NoSelection.new(model.FlatSortModel()))
|
|
self.set_factory(sort.Factory(self.get_model().get_model()))
|
|
|
|
def set_playlist(self, plist):
|
|
self.get_model().get_model().set_playlist(plist)
|
|
|
|
|
|
class SortOrderPopover(Gtk.Popover):
|
|
def __init__(self):
|
|
Gtk.Popover.__init__(self)
|
|
self.set_child(SortOrderView())
|
|
|
|
def set_playlist(self, plist):
|
|
self.get_child().set_playlist(plist)
|