emmental/sidebar/row.py

# Copyright 2021 (c) Anna Schumaker.
from gi.repository import Gtk
from gi.repository import Pango
from . import widgets

class Label(Gtk.Label):
    def __init__(self):
        Gtk.Label.__init__(self)
        self.set_ellipsize(Pango.EllipsizeMode.MIDDLE)
        self.set_halign(Gtk.Align.START)
        self.set_hexpand(True)


class Grid(Gtk.Grid):
    def __init__(self):
        Gtk.Grid.__init__(self)
        self.icon  = self.attach(Gtk.Image(), 0, 0, 1, 2)
        self.name  = self.attach(Label(),     1, 0, 1, 1)
        self.count = self.attach(Label(),     1, 1, 1, 1)
        self.set_column_spacing(5)

    def attach(self, widget, x, y, w, h):
        super().attach(widget, x, y, w, h)
        return widget

    def set_count(self, item, *args):
        n = item.get_n_tracks()
        self.count.set_text(f"{n} Track{'s' if n != 1 else ''}")

    def set_item(self, item):
        self.icon.set_from_icon_name(item.get_property("icon-name"))
        self.name.set_text(item.get_property("name"))

        item.connect("track-added", self.set_count)
        item.connect("track-removed", self.set_count)
        item.connect("refreshed", self.set_count)
        self.set_count(item)

    def unset_item(self, item):
        if item:
            item.disconnect_by_func(self.set_count)
            item.disconnect_by_func(self.set_count)
            item.disconnect_by_func(self.set_count)


class UserGrid(Grid):
    def __init__(self):
        Grid.__init__(self)
        self.remove = self.attach(Gtk.Button(), 2, 0, 1, 2)
        self.remove.set_icon_name("list-remove")
        self.remove.add_css_class("flat")
        self.remove.set_halign(Gtk.Align.CENTER)
        self.remove.set_valign(Gtk.Align.CENTER)

    def set_item(self, item):
        system = [ "Collection", "Favorites", "New Tracks",
                   "Previous", "Queued Tracks" ]
        super().set_item(item)
        self.remove.set_visible(item.name not in system)
        self.remove.connect("clicked", self.clicked, item)

    def unset_item(self, item):
        self.remove.disconnect_by_func(self.clicked)

    def clicked(self, button, item):
        item.delete()


class LibraryGrid(Grid):
    def __init__(self):
        Grid.__init__(self)
        self.menu = self.attach(Gtk.MenuButton(), 2, 0, 1, 2)
        self.menu.add_css_class("flat")
        self.menu.get_first_child().add_css_class("flat")
        self.menu.set_direction(Gtk.ArrowType.LEFT)
        self.menu.set_halign(Gtk.Align.CENTER)
        self.menu.set_valign(Gtk.Align.CENTER)

    def set_item(self, item):
        super().set_item(item)
        self.menu.set_popover(widgets.LibraryPopover(item))

    def unset_item(self, item):
        super().unset_item(item)
        self.menu.set_popover(None)