def __init__(self):
self.songs = []
self.radios = []
self.search_results = []
self.most_recent_searches = []
self.title = ''
self.subtitle = ''
self.main_menu = True
self.page = 0
self.selected = 0
self.streamer = StreamerWrapper()
self.full_options = None
self.page_options = None
class ContentManager(object):
"""Main class for controlling what content is available to Gstreamer"""
def __init__(self):
self.songs = []
self.radios = []
self.search_results = []
self.most_recent_searches = []
self.title = ''
self.subtitle = ''
self.main_menu = True
self.page = 0
self.selected = 0
self.streamer = StreamerWrapper()
self.full_options = None
self.page_options = None
def load(self):
"""Load Songs, Stations, and Playlists"""
print "Logged into Google Music. Now loading songs and radios."
self.set_menu_options()
self.streamer.run()
self.songs = self.streamer.client.get_all_songs()
self.radios = self.streamer.client.get_all_stations()
def search_songs(self, artist="", album="", song=""):
"""Search for songs matching certain parameters"""
# Filter according to criteria supplied
search_results = [s for s in self.songs \
if album.lower() in s['album'].lower() \
and artist.lower() in s['artist'].lower() \
and song.lower() in s['title'].lower()]
# Check to see if we should assign results to current state
if len(search_results) > 0:
search_results = self.order_results(search_results)
self.assign_search_results(artist+album+song, search_results)
def order_results(self, search_results):
"""Orders the results by album, tracknumber, then title"""
return sorted(search_results, \
key = lambda y: (y['album'], y['trackNumber'],y['title']))
def assign_search_results(self, search_parameters, search_results):
"""Assigns state-based parameters according to search results"""
self.title = search_parameters
self.subtitle = "Search Results"
self.main_menu = False
self.search_results = search_results
self.set_menu_options()
def get_most_recent_searches(self):
"""Formats 'Most Recent Searches' for main menu"""
return self.most_recent_searches + ['Options', 'Exit']
def play_track(self, num):
"""Play a track from the search results"""
return self.page_options[num]
def queue(self, page_index):
"""Queue a set of songs for playback"""
queue = []
track_index = self.search_results.index(self.page_options[page_index])
if track_index < len(self.search_results):
queue += self.search_results[track_index+1:]
if track_index > 0:
queue += self.search_results[:track_index-1]
self.streamer.queue = queue
def set_menu_options(self):
'''Draws a menu with the given parameters'''
if self.main_menu:
self.selected = 0
self.full_options = self.get_most_recent_searches()
self.title = "Main Menu"
self.subtitle = "Options"
else:
self.full_options = self.search_results
# Notify the menu (if it exists)
if hasattr(self, 'menu'):
self.page_options = self.menu.notify(self.full_options)
def get_page_number(self):
"""Returns a string formatted like '(Page 2 or 3)'"""
num_pages = int(math.ceil(len(self.full_options) / len(self.page_options)))
if num_pages > 1:
return " (Page {0} of {1})".format(self.page, num_pages)
return ""
def back_to_main(self):
"""Returns to main menu"""
self.main_menu = True
self.set_menu_options()
def change_page(self, val):
"""Changes the page by <val>"""
num_pages = int(math.ceil(len(self.full_options) / len(self.page_options)))
self.page = min(max(0, self.page+val), num_pages)
# Notify the menu (if it exists)
if hasattr(self, 'menu'):
self.page_options = self.menu.notify(self.full_options, self.page)
def adjust_selection(self, val):
"""Adjusts the position of the selection cursor in the menu"""
self.selected += val
if self.selected > len(self.page_options):
self.change_page(1)
if self.selected < 0:
self.change_page(-1)
self.selected = self.selected % len(self.page_options)
# Notify the menu (if it exists)
if hasattr(self, 'menu'):
self.menu.draw()
def handle_execute(self):
"""Runs whenever the Enter key is called (execute command)"""
if self.selected == len(self.page_options)-1:
self.page = 0
return self.page_options[-1]
if self.main_menu:
return self.most_recent_searches[self.selected]
return 'play {0}'.format(self.selected)
def attach_to_streamer(self, now_playing=None):
"""Allows objects to attach to this object's streamer"""
self.streamer.attach(now_playing)
def attach(self, menu=None):
"""Allows objects to attach to this object"""
if menu is not None:
self.menu = menu
def exit(self):
"""Calls on the menu to exit the application"""
self.menu.exit()