summaryrefslogtreecommitdiff
path: root/nephilim/plugins/Playlist.py
blob: e57c9ac5f40386e577cdb949b41c0d8e37cf19a3 (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
89
90
91
92
from PyQt4 import QtGui, QtCore
from PyQt4.QtCore import QVariant

from ..clPlugin import Plugin

# Dependencies:
#   playcontrol
class Playlist(Plugin):
    o        = None
    DEFAULTS = {'columns': ['track', 'title', 'artist',
                            'date', 'album', 'length']}

    def __init__(self, winMain):
        Plugin.__init__(self, winMain, 'Playlist')
    def _load(self):
        self.o = PlaylistWidget(self)
        self.mpclient.add_listener('onPlaylistChange', self.on_playlist_change)
        self.mpclient.add_listener('onDisconnect', self.on_playlist_change)
        self.mpclient.add_listener('onReady', self.on_playlist_change)

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

    def _getDockWidget(self):
        return self._createDock(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.getName() + '/columns').toStringList()
            self.setColumnCount(len(columns))
            self.setHeaderLabels(columns)
            self.header().restoreState(self.plugin.settings.value(self.plugin.getName() + '/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.getName() + '/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.getName() + '/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()