nephilim/plugins/Playlist.py

   1 #
   2 #    Copyright (C) 2009 Anton Khirnov <wyskas@gmail.com>
   3 #
   4 #    Nephilim is free software: you can redistribute it and/or modify
   5 #    it under the terms of the GNU General Public License as published by
   6 #    the Free Software Foundation, either version 3 of the License, or
   7 #    (at your option) any later version.
   8 #
   9 #    Nephilim is distributed in the hope that it will be useful,
  10 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 #    GNU General Public License for more details.
  13 #
  14 #    You should have received a copy of the GNU General Public License
  15 #    along with Nephilim.  If not, see <http://www.gnu.org/licenses/>.
  16 #
  17
  18 from PyQt4 import QtGui, QtCore
  19
  20 from ..plugin import Plugin
  21 from ..common import MIMETYPES, SongsMimeData
  22 from ..song   import PlaylistEntryRef
  23
  24 class Playlist(Plugin):
  25     # public, const
  26     info = 'Shows the playlist.'
  27
  28     # public, read-only
  29     o        = None
  30
  31     # private
  32     DEFAULTS  = {'columns': ['track', 'title', 'artist', 'date', 'album', 'length'], 'header_state' : QtCore.QByteArray()}
  33
  34     def _load(self):
  35         self.o = PlaylistWidget(self)
  36
  37     def _unload(self):
  38         self.o = None
  39
  40     def _get_dock_widget(self):
  41         return self._create_dock(self.o)
  42
  43 class PlaylistWidget(QtGui.QWidget):
  44     plugin   = None
  45     playlist = None
  46     toolbar  = None
  47
  48     def __init__(self, plugin):
  49         QtGui.QWidget.__init__(self)
  50         self.plugin = plugin
  51
  52         self.playlist = PlaylistTree(self.plugin)
  53
  54         self.toolbar  = QtGui.QToolBar(self.plugin.name + ' toolbar', self)
  55         self.toolbar.addAction(QtGui.QIcon(':icons/shuffle.png'), 'Shuffle current playlist.', self.plugin.mpclient.shuffle)
  56         self.toolbar.addAction(QtGui.QIcon(':icons/delete.png'), 'Clear current playlist.', self.plugin.mpclient.clear)
  57         add_url = PlaylistAddURL(self.plugin.mpclient, self)
  58         add_url.setToolTip('Add an URL to current playlist.')
  59         self.toolbar.addWidget(add_url)
  60
  61         self.setLayout(QtGui.QVBoxLayout())
  62         self.layout().setSpacing(0)
  63         self.layout().setMargin(0)
  64         self.layout().addWidget(self.toolbar)
  65         self.layout().addWidget(self.playlist)
  66
  67         self.plugin.mpclient.playlist(self.playlist.fill)
  68
  69 class PlaylistAddURL(QtGui.QLineEdit):
  70
  71     ### PRIVATE ####
  72     _mpclient = None
  73
  74     def __init__(self, mpclient, parent = None):
  75         QtGui.QLineEdit.__init__(self, parent)
  76         self._mpclient = mpclient
  77         self.returnPressed.connect(self._return_pressed)
  78
  79     def _return_pressed(self):
  80         self._mpclient.add([self.text()])
  81         self.clear()
  82
  83 class PlaylistTree(QtGui.QTreeWidget):
  84     plugin = None
  85
  86     ### PRIVATE ###
  87     # popup menu
  88     _menu      = None
  89     # add same... menu
  90     _same_menu = None
  91     # currently playing song item (highlighted
  92     _cur_song = None
  93
  94     def __init__(self, plugin):
  95         QtGui.QTreeWidget.__init__(self)
  96         self.plugin = plugin
  97
  98         self.setSelectionMode(QtGui.QTreeWidget.ExtendedSelection)
  99         self.setAlternatingRowColors(True)
 100         self.setRootIsDecorated(False)
 101
 102         # drag&drop
 103         self.viewport().setAcceptDrops(True)
 104         self.setDropIndicatorShown(True)
 105         self.setDragDropMode(QtGui.QAbstractItemView.DragDrop)
 106
 107         columns = self.plugin.settings.value(self.plugin.name + '/columns')
 108         self.setColumnCount(len(columns))
 109         self.setHeaderLabels(columns)
 110         self.header().restoreState(self.plugin.settings.value(self.plugin.name + '/header_state'))
 111
 112         # menu
 113         self._menu      = QtGui.QMenu()
 114         self._same_menu = self._menu.addMenu('Add same...')
 115         self.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
 116         self.customContextMenuRequested.connect(self._show_context_menu)
 117
 118         self.itemActivated.connect(self._song_activated)
 119         self.header().geometriesChanged.connect(self._save_state)
 120         self.plugin.mpclient.playlist_changed.connect(lambda :self.plugin.mpclient.playlist(self.fill))
 121         self.plugin.mpclient.connect_changed.connect(self._update_menu)
 122         self.plugin.mpclient.song_changed.connect(self._update_cur_song)
 123         self.plugin.mpclient.songpos_changed.connect(self._update_cur_song)
 124
 125     def _save_state(self):
 126         self.plugin.settings.setValue(self.plugin.name + '/header_state', self.header().saveState())
 127
 128     def _song_activated(self, item):
 129         self.plugin.mpclient.play(item.song['id'])
 130
 131     def _update_cur_song(self, song):
 132         if self._cur_song:
 133             self._cur_song.set_current(False)
 134
 135         if not song:
 136             self._cur_song = None
 137             return
 138
 139         self._cur_song = self.topLevelItem(int(song['pos']))
 140         if self._cur_song:
 141             self._cur_song.set_current(True)
 142
 143     def fill(self, songs):
 144         columns = self.plugin.settings.value(self.plugin.name + '/columns')
 145         self._cur_song = None
 146         self.clear()
 147         for song in songs:
 148             item = PlaylistSongItem(PlaylistEntryRef(self.plugin.mpclient, song['id']))
 149             for i in range(len(columns)):
 150                 item.setText(i, song['?' + columns[i]])
 151             self.addTopLevelItem(item)
 152         self._update_cur_song(self.plugin.mpclient.cur_song)
 153
 154     def keyPressEvent(self, event):
 155         if event.matches(QtGui.QKeySequence.Delete):
 156             ids = []
 157             for item in self.selectedItems():
 158                 ids.append(item.song['id'])
 159
 160             self.plugin.mpclient.delete(ids)
 161         else:
 162             QtGui.QTreeWidget.keyPressEvent(self, event)
 163
 164     def mimeData(self, items):
 165         data = SongsMimeData()
 166         data.set_plistsongs([items[0].song['id']])
 167         return data
 168
 169     def dropMimeData(self, parent, index, data, action):
 170         if data.hasFormat(MIMETYPES['plistsongs']):
 171             if parent:
 172                 index = self.indexOfTopLevelItem(parent)
 173             elif index >= self.topLevelItemCount():
 174                 index = self.topLevelItemCount() - 1
 175             self.plugin.mpclient.move(data.plistsongs()[0], index)
 176             return True
 177         elif data.hasFormat(MIMETYPES['songs']):
 178             if parent:
 179                 index = self.indexOfTopLevelItem(parent)
 180             self.plugin.mpclient.add(data.songs(), index)
 181             return True
 182         return False
 183
 184     def supportedDropActions(self):
 185         return QtCore.Qt.CopyAction | QtCore.Qt.MoveAction
 186
 187     def mimeTypes(self):
 188         return [MIMETYPES['songs'], MIMETYPES['plistsongs']]
 189
 190     def _update_menu(self):
 191         """Update popup menu. Invoked on (dis)connect."""
 192         self._same_menu.clear()
 193         for tag in self.plugin.mpclient.tagtypes:
 194             self._same_menu.addAction(tag, lambda tag = tag: self._add_selected_same(tag))
 195
 196     def _add_selected_same(self, tag):
 197         """Adds all tracks in DB with tag 'tag' same as selected tracks."""
 198         for it in self.selectedItems():
 199             self.plugin.mpclient.findadd(tag, it.song['?' + tag])
 200
 201     def _show_context_menu(self, pos):
 202         if not self.indexAt(pos).isValid():
 203             return
 204         self._menu.popup(self.mapToGlobal(pos))
 205
 206 class PlaylistSongItem(QtGui.QTreeWidgetItem):
 207     ### PUBLIC ###
 208     song = None
 209
 210     def __init__(self, song):
 211         QtGui.QTreeWidgetItem.__init__(self)
 212         self.song = song
 213
 214     def set_current(self, val):
 215         font = QtGui.QFont()
 216         if val:
 217             font.setBold(True)
 218             font.setItalic(True)
 219
 220         for i in xrange(self.columnCount()):
 221             self.setFont(i, font)