-
Notifications
You must be signed in to change notification settings - Fork 0
/
playlistsizetruncaterandom.py
123 lines (100 loc) · 4.9 KB
/
playlistsizetruncaterandom.py
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# Copyright (C) 2018 Xavier Dalem
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# This plugin allows the user to filter a playlist randomly so that it fits on
# e.g. a removable device or media player
import random, sys
from gi.repository import Gtk
from quodlibet import _, app
from quodlibet.plugins.playlist import PlaylistPlugin
from quodlibet.qltk import Icons
from quodlibet.util.dprint import print_d
import threading
import time
class PlaylistSize(PlaylistPlugin):
PLUGIN_ID = 'playlistsizetruncaterandom'
PLUGIN_NAME = _('Truncate playlist to size')
PLUGIN_DESC = _('Truncates a playlist to a given size in MB by removing songs at random')
PLUGIN_ICON = Icons.EDIT_CLEAR
REQUIRES_ACTION = True
def plugin_handles(self, playlists):
return len(playlists) == 1
def truncatePlaylist(self, playlist, currentSize, desiredSize):
print_d('Going to truncate the playlist')
# Work on a copy that we can edit without all the callbacks and saves on playlist.songs, which take a very long time
copiedSongList = [] + playlist.songs
self.progressLabel.set_text('Songs remaining: ' + str(len(copiedSongList)) + ' (' + str(self.toMB(currentSize)) + ' MB)')
songsToRemove = []
initialSizeDifference = desiredSize - currentSize
self.progressBar.set_fraction(0)
while desiredSize < currentSize:
song = random.choice(copiedSongList)
songsToRemove.append(song)
copiedSongList.remove(song)
currentSize -= song['~#filesize']
print_d('Removing song: ' + str(song))
self.progressLabel.set_text('Songs remaining: ' + str(len(copiedSongList)) + ' (' + str(self.toMB(currentSize)) + ' MB)')
currentSizeDifference = desiredSize - currentSize
self.progressBar.set_fraction((initialSizeDifference - currentSizeDifference) / initialSizeDifference)
print_d('Removing ' + str(len(songsToRemove)) + ' songs from the playlist')
self.progressBar.pulse()
time.sleep(10)
playlist.remove_songs(songsToRemove, True)
print_d('Removed all requested songs from the playlist')
self.progressDialog.destroy()
def toMB(self, size):
return int(size/1024/1024)
def plugin_playlist(self, playlist):
size = 0
for s in playlist.songs:
size += s['~#filesize']
# Create a dialog.
sizeSelectDlg = Gtk.Dialog(title=_('Select a size'),
parent=app.window,
flags=(Gtk.DialogFlags.MODAL |
Gtk.DialogFlags.DESTROY_WITH_PARENT))
sizeSelectDlg.add_button(_('_Cancel'), Gtk.ResponseType.REJECT)
sizeSelectDlg.add_button(_('_Apply'), Gtk.ResponseType.APPLY)
sizeSelectDlg.set_default_response(Gtk.ResponseType.APPLY)
sizeSelectDlg.vbox.set_spacing(5)
label = Gtk.Label(_('Enter a maximum size in MB'))
sizeSelectDlg.vbox.add(label)
maxSize = Gtk.SpinButton()
maxSize.set_numeric(True)
maxSize.set_adjustment(Gtk.Adjustment(self.toMB(size), 1, sys.maxsize, 1, 1))
sizeSelectDlg.vbox.add(maxSize)
sizeSelectDlg.show_all()
# Only operate if apply is pressed.
if sizeSelectDlg.run() == Gtk.ResponseType.APPLY:
sizeSelectDlg.destroy()
currentSize = size;
desiredSize = int(maxSize.get_value() * 1024 * 1024)
self.progressDialog = Gtk.Dialog(title=_('Progress'),
parent=app.window,
flags=(Gtk.DialogFlags.MODAL |
Gtk.DialogFlags.DESTROY_WITH_PARENT))
self.progressDialog.vbox.set_spacing(5)
self.progressLabel = Gtk.Label()
self.progressDialog.vbox.add(self.progressLabel)
self.progressBar = Gtk.ProgressBar()
self.progressDialog.vbox.add(self.progressBar)
self.progressDialog.show_all()
# Work in a thread to avoid locking up the UI
thread = threading.Thread(target=self.truncatePlaylist,
args=(playlist, currentSize, desiredSize))
thread.start()
return True
else:
sizeSelectDlg.destroy()
return False