2 # Copyright (C) 2009 Anton Khirnov <wyskas@gmail.com>
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.
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.
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/>.
18 from PyQt4
import QtGui
, QtCore
20 from ..plugin
import Plugin
21 from ..common
import MIMETYPES
, SongsMimeData
22 from ..song
import PlaylistEntryRef
24 class Playlist(Plugin
):
26 info
= 'Shows the playlist.'
32 DEFAULTS
= {'columns': ['track', 'title', 'artist', 'date', 'album', 'length'], 'header_state' : QtCore
.QByteArray()}
35 self
.o
= PlaylistWidget(self
)
40 def _get_dock_widget(self
):
41 return self
._create
_dock
(self
.o
)
43 class PlaylistWidget(QtGui
.QWidget
):
48 def __init__(self
, plugin
):
49 QtGui
.QWidget
.__init
__(self
)
52 self
.playlist
= PlaylistTree(self
.plugin
)
54 self
.setLayout(QtGui
.QVBoxLayout())
55 self
.layout().setSpacing(0)
56 self
.layout().setMargin(0)
57 self
.layout().addWidget(self
.playlist
)
59 self
.plugin
.mpclient
.playlist(self
.playlist
.fill
)
61 class PlaylistTree(QtGui
.QTreeWidget
):
69 # currently playing song item (highlighted
72 def __init__(self
, plugin
):
73 QtGui
.QTreeWidget
.__init
__(self
)
76 self
.setSelectionMode(QtGui
.QTreeWidget
.ExtendedSelection
)
77 self
.setAlternatingRowColors(True)
78 self
.setRootIsDecorated(False)
81 self
.viewport().setAcceptDrops(True)
82 self
.setDropIndicatorShown(True)
83 self
.setDragDropMode(QtGui
.QAbstractItemView
.DragDrop
)
85 columns
= self
.plugin
.settings
.value(self
.plugin
.name
+ '/columns')
86 self
.setColumnCount(len(columns
))
87 self
.setHeaderLabels(columns
)
88 self
.header().restoreState(self
.plugin
.settings
.value(self
.plugin
.name
+ '/header_state'))
91 self
._menu
= QtGui
.QMenu()
92 self
._same
_menu
= self
._menu
.addMenu('Add same...')
93 self
.setContextMenuPolicy(QtCore
.Qt
.CustomContextMenu
)
94 self
.customContextMenuRequested
.connect(self
._show
_context
_menu
)
96 self
.itemActivated
.connect(self
._song
_activated
)
97 self
.header().geometriesChanged
.connect(self
._save
_state
)
98 self
.plugin
.mpclient
.playlist_changed
.connect(lambda :self
.plugin
.mpclient
.playlist(self
.fill
))
99 self
.plugin
.mpclient
.connect_changed
.connect(self
._update
_menu
)
100 self
.plugin
.mpclient
.song_changed
.connect(self
._update
_cur
_song
)
102 def _save_state(self
):
103 self
.plugin
.settings
.setValue(self
.plugin
.name
+ '/header_state', self
.header().saveState())
105 def _song_activated(self
, item
):
106 self
.plugin
.mpclient
.play(item
.song
['id'])
108 def _update_cur_song(self
, song
):
110 self
._cur
_song
.set_current(False)
113 self
._cur
_song
= None
116 self
._cur
_song
= self
.topLevelItem(int(song
['pos']))
118 self
._cur
_song
.set_current(True)
120 def fill(self
, songs
):
121 columns
= self
.plugin
.settings
.value(self
.plugin
.name
+ '/columns')
122 self
._cur
_song
= None
125 item
= PlaylistSongItem(PlaylistEntryRef(self
.plugin
.mpclient
, song
['id']))
126 for i
in range(len(columns
)):
127 item
.setText(i
, song
['?' + columns
[i
]])
128 self
.addTopLevelItem(item
)
129 self
._update
_cur
_song
(self
.plugin
.mpclient
.cur_song
)
131 def keyPressEvent(self
, event
):
132 if event
.matches(QtGui
.QKeySequence
.Delete
):
134 for item
in self
.selectedItems():
135 ids
.append(item
.song
['id'])
137 self
.plugin
.mpclient
.delete(ids
)
139 QtGui
.QTreeWidget
.keyPressEvent(self
, event
)
141 def mimeData(self
, items
):
142 data
= SongsMimeData()
143 data
.set_plistsongs([items
[0].song
['id']])
146 def dropMimeData(self
, parent
, index
, data
, action
):
147 if data
.hasFormat(MIMETYPES
['plistsongs']):
149 index
= self
.indexOfTopLevelItem(parent
)
150 elif index
>= self
.topLevelItemCount():
151 index
= self
.topLevelItemCount() - 1
152 self
.plugin
.mpclient
.move(data
.plistsongs()[0], index
)
154 elif data
.hasFormat(MIMETYPES
['songs']):
156 index
= self
.indexOfTopLevelItem(parent
)
157 self
.plugin
.mpclient
.add(data
.songs(), index
)
161 def supportedDropActions(self
):
162 return QtCore
.Qt
.CopyAction | QtCore
.Qt
.MoveAction
165 return [MIMETYPES
['songs'], MIMETYPES
['plistsongs']]
167 def _update_menu(self
):
168 """Update popup menu. Invoked on (dis)connect."""
169 self
._same
_menu
.clear()
170 for tag
in self
.plugin
.mpclient
.tagtypes
:
171 self
._same
_menu
.addAction(tag
, lambda tag
= tag
: self
._add
_selected
_same
(tag
))
173 def _add_selected_same(self
, tag
):
174 """Adds all tracks in DB with tag 'tag' same as selected tracks."""
175 for it
in self
.selectedItems():
176 self
.plugin
.mpclient
.findadd(tag
, it
.song
['?' + tag
])
178 def _show_context_menu(self
, pos
):
179 if not self
.indexAt(pos
).isValid():
181 self
._menu
.popup(self
.mapToGlobal(pos
))
183 class PlaylistSongItem(QtGui
.QTreeWidgetItem
):
187 def __init__(self
, song
):
188 QtGui
.QTreeWidgetItem
.__init
__(self
)
191 def set_current(self
, val
):
197 for i
in xrange(self
.columnCount()):
198 self
.setFont(i
, font
)