summaryrefslogtreecommitdiff
path: root/nephilim/plugins/Playlist.py
blob: 9d32e8c18499377447a3437eb3396d818589db7f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from PyQt4 import QtGui, QtCore
from PyQt4.QtCore import QVariant

from ..plugin import Plugin

class Playlist(Plugin):
    o        = None
    DEFAULTS  = {'columns': ['track', 'title', 'artist',
                             'date', 'album', 'length']}
    LISTENERS = {'onPlaylistChange' : 'on_playlist_change',
                 'onDisconnect' : 'on_playlist_change', 'onReady' : 'on_playlist_change'}

    def _load(self):
        self.o = PlaylistWidget(self)

    def _unload(self):
        self.o = None
    def getInfo(self):
        return "The playlist showing the songs that will be played."

    def _get_dock_widget(self):
        return self._create_dock(self.o)

    def on_playlist_change(self, params = None):
        self.o.fill_playlist()


class PlaylistWidget(QtGui.QWidget):
    plugin   = None
    playlist = None

    def __init__(self, plugin):
        QtGui.QWidget.__init__(self)
        self.plugin = plugin

        self.playlist = self.Playlist(self.plugin)

        self.setLayout(QtGui.QVBoxLayout())
        self.layout().setSpacing(0)
        self.layout().setMargin(0)
        self.layout().addWidget(self.playlist)

    class Playlist(QtGui.QTreeWidget):
        song   = None
        plugin = None

        def __init__(self, plugin):
            QtGui.QTreeWidget.__init__(self)
            self.plugin = plugin

            self.setSelectionMode(QtGui.QTreeWidget.ExtendedSelection)
            self.setAlternatingRowColors(True)
            self.setRootIsDecorated(False)
            columns = self.plugin.settings().value(self.plugin.name() + '/columns').toStringList()
            self.setColumnCount(len(columns))
            self.setHeaderLabels(columns)
            self.header().restoreState(self.plugin.settings().value(self.plugin.name() + '/header_state').toByteArray())
            self.connect(self, QtCore.SIGNAL('itemActivated(QTreeWidgetItem*, int)'), self._song_activated)
            self.connect(self.header(), QtCore.SIGNAL('geometriesChanged()'), self._save_state)

        def _save_state(self):
            self.plugin.settings().setValue(self.plugin.name() + '/header_state', QVariant(self.header().saveState()))

        def _song_activated(self, item):
            self.plugin.mpclient().play(item.data(0, QtCore.Qt.UserRole).toPyObject().getID())

        def fill(self):
            columns = self.plugin.settings().value(self.plugin.name() + '/columns').toStringList()
            self.clear()
            for song in self.plugin.mpclient().playlist():
                item = QtGui.QTreeWidgetItem()
                for i in range(len(columns)):
                    item.setText(i, unicode(song.getTag(str(columns[i]))))
                item.setData(0, QtCore.Qt.UserRole, QVariant(song))
                self.addTopLevelItem(item)

        def keyPressEvent(self, event):
            if event.matches(QtGui.QKeySequence.Delete):
                ids = []
                for item in self.selectedItems():
                    ids.append(item.data(0, QtCore.Qt.UserRole).toPyObject().getID())

                self.plugin.mpclient().delete(ids)
            else:
                QtGui.QTreeWidget.keyPressEvent(self, event)

    def fill_playlist(self):
        self.playlist.fill()