Rhythmbox-Random-Album-Player/Rhythmbox-Random-Album-Player.py at master · UMLAUTaxl/Rhythmbox-Random-Album-Player · GitHub

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
'''
Plugin for Rhythmbox that random plays songs sorted by album (in correct disc order), track-number randomly
Copyright (C) 2012  Javon Harper <javon.d.harper@gmail.com>

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 2
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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
'''

import random

from gi.repository import GObject
from gi.repository import Peas
from gi.repository import RB
from gi.repository import Gtk
from gi.repository import Gio

from RandomAlbumConfigDialog import ConfigDialog

#~ GLib.threads_init()
from random_rb3compat import ActionGroup
from random_rb3compat import Action
from random_rb3compat import ApplicationShell

ui_str = '''
  <ui>
    <menubar name="MenuBar">
      <menu name="ControlMenu" action="Control">
        <menuitem name="RandomAlbumItem" action="RandomAlbum"/>
      </menu>
    </menubar>
  </ui>
'''


class RandomAlbumPlugin(GObject.Object, Peas.Activatable):
  __gtype_name__ = 'RandomAlbumPlugin'
  object = GObject.property(type=GObject.Object)

  def __init__(self):
    super(RandomAlbumPlugin, self).__init__()

  def do_activate(self):
    #~ shell = self.object
    print "Activating Random Album Plugin"
    #~ action = Gtk.Action ('RandomAlbum', _('Random Album'), _('Play a Random Album'), "")
    #~ action.connect ('activate', self.random_album, shell)
    #~ action_group = Gtk.ActionGroup('RandomAlbumActionGroup')
    #~ action_group.add_action_with_accel (action, "<alt>Z")

    self.shell = self.object

    self.action_group = ActionGroup(self.shell, 'RandomAlbumActionGroup')

    action = self.action_group.add_action_with_accel(func=self.random_album,
    action_name='RandomAlbum', label='RandomAlbum',
    action_type='app', action_state=ActionGroup.STANDARD,
    accel="<alt>Z")

    self._appshell = ApplicationShell(self.shell)
    self._appshell.insert_action_group(self.action_group)
    self._appshell.add_app_menuitems(ui_str, 'RandomAlbumActionGroup')

    self.settings = Gio.Settings('org.gnome.rhythmbox.plugins.randomalbumplayer')

  def do_deactivate(self):
    print 'Deactivating Random Album Plugin'
    shell = self.object
    #~ ui_manager = shell.props.ui_manager
    #~ ui_manager.remove_ui(self.ui_id)

  def random_album(self, *args):
    # Get URIs of all the songs in the queue and remove them
    play_queue = self.shell.props.queue_source
    for row in play_queue.props.query_model:
      entry = row[0]
      play_queue.remove_entry(entry)

    # Queue a random album
    for _ in range(self.settings['albums-to-play']):
        self.queue_random_album()

    # Start the music!(well, first stop it, but it'll start up again.)
    print 'Playing Album'
    player = self.shell.props.shell_player
    player.stop()
    player.set_playing_source(self.shell.props.queue_source)
    player.playpause(True)

  def queue_random_album(self):
    shell = self.object
    library = shell.props.library_source
    albums = {}

    ignore_albums = [ item.strip() for item in self.settings['ignored-albums'].split(',') ]

    for row in library.props.query_model:
      entry = row[0]
      album_name = entry.get_string(RB.RhythmDBPropType.ALBUM)

      if album_name in ignore_albums:
        continue

      album_struct = albums.get(album_name, { "songs" : [], "count": 0 })
      album_struct["count"] = album_struct["count"] + 1
      album_struct["songs"].append(entry)
      albums[album_name] = album_struct

    # Choose a random album with more than 5 songs
    while True:
        album_names = albums.keys()
        num_albums = len(albums)
        selected_album = album_names[random.randint(0, num_albums - 1)]

        print 'Queuing ' + selected_album+ '.'

        # Find all the songs from that album
        songs = albums[selected_album]["songs"]

        if len(songs) > 5:
            # album is long enough
            break

    # Sort the songs by disc number, track number
    songs = sorted(songs, key=lambda song: song.get_ulong(RB.RhythmDBPropType.TRACK_NUMBER))
    songs = sorted(songs, key=lambda song: song.get_ulong(RB.RhythmDBPropType.DISC_NUMBER))

    # Add the songs to the play queue
    for song in songs:
      shell.props.queue_source.add_entry(song, -1)