def __init__(self, config, default_artist, default_title, default_player, connect_to_player, tray_action):
super().__init__()
self.config = config
self.mpris_connect = MprisConnector()
self.worker_search = None
self.worker_mpris = None
self.mpris_player = None
self.main_window = None
self.tray = None
self.start_app(default_artist, default_title, default_player, connect_to_player, tray_action)
class MainController(object):
def __init__(self, config, default_artist, default_title, default_player, connect_to_player, tray_action):
super().__init__()
self.config = config
self.mpris_connect = MprisConnector()
self.worker_search = None
self.worker_mpris = None
self.mpris_player = None
self.main_window = None
self.tray = None
self.start_app(default_artist, default_title, default_player, connect_to_player, tray_action)
def start_app(self, default_artist, default_title, default_player, connect_to_player, tray_action):
app = QApplication(sys.argv)
app.setWindowIcon(QIcon(os.path.join(path, 'pixmaps', 'prismriver.png')))
app.setApplicationName('Prismriver Lyrics')
self.main_window = MainWindow()
self.main_window.edit_artist.setText(default_artist)
self.main_window.edit_title.setText(default_title)
self.main_window.btn_refresh.clicked.connect(self.refresh_players)
self.init_layout(State.waiting)
self.refresh_players(default_player)
if connect_to_player:
player = self.get_current_player()
if player and (default_player is None or default_player == player.name):
self.start_mpris()
elif default_artist and default_title:
self.start_search()
show_tray = (tray_action != 'hide')
show_main = (tray_action != 'minimize')
if show_main:
self.main_window.show()
if show_tray:
self.tray = TrayIcon()
self.tray.activated.connect(self.toggle_main_window)
self.tray.right_menu.aboutToShow.connect(self.update_tray_menu)
self.tray.show()
sys.exit(app.exec_())
def init_layout(self, state):
try:
self.main_window.btn_search.clicked.disconnect()
except TypeError:
pass
try:
self.main_window.btn_connect.clicked.disconnect()
except TypeError:
pass
if state == State.waiting:
self.main_window.btn_search.setIcon(
QIcon.fromTheme('edit-find', self.main_window.style().standardIcon(QStyle.SP_DialogApplyButton)))
self.main_window.btn_search.setText('Search...')
self.main_window.btn_search.clicked.connect(self.start_search)
self.main_window.btn_connect.setText('Connect...')
self.main_window.btn_connect.clicked.connect(self.start_mpris)
self.main_window.btn_search.setEnabled(True)
self.main_window.btn_connect.setEnabled(True)
self.main_window.btn_refresh.setEnabled(True)
self.main_window.edit_artist.setReadOnly(False)
self.main_window.edit_title.setReadOnly(False)
self.main_window.edit_player.setEnabled(True)
elif state == State.searching:
self.main_window.btn_search.setIcon(
QIcon.fromTheme('process-stop', self.main_window.style().standardIcon(QStyle.SP_DialogDiscardButton)))
self.main_window.btn_search.setText('Stop')
self.main_window.btn_search.clicked.connect(self.interrupt_search)
self.main_window.btn_connect.setText('Connect...')
self.main_window.btn_connect.clicked.connect(self.start_mpris)
self.main_window.btn_search.setEnabled(True)
self.main_window.btn_connect.setEnabled(False)
self.main_window.btn_refresh.setEnabled(False)
self.main_window.edit_artist.setReadOnly(True)
self.main_window.edit_title.setReadOnly(True)
self.main_window.edit_player.setEnabled(False)
elif state == State.listening:
self.main_window.btn_search.setIcon(
QIcon.fromTheme('edit-find', self.main_window.style().standardIcon(QStyle.SP_DialogApplyButton)))
self.main_window.btn_search.setText('Search...')
self.main_window.btn_search.clicked.connect(self.start_search)
self.main_window.btn_connect.setText('Disconnect')
self.main_window.btn_connect.clicked.connect(self.stop_mpris)
self.main_window.btn_search.setEnabled(False)
self.main_window.btn_connect.setEnabled(True)
self.main_window.btn_refresh.setEnabled(False)
self.main_window.edit_artist.setReadOnly(True)
self.main_window.edit_title.setReadOnly(True)
self.main_window.edit_player.setEnabled(False)
else:
return
def get_current_player(self):
if use_pyqt5:
return self.main_window.edit_player.currentData(PlayerListModel.DataRole)
else:
index = self.main_window.edit_player.currentIndex()
return self.main_window.edit_player.itemData(index, PlayerListModel.DataRole)
def set_status_message(self, message):
self.main_window.statusBar().showMessage(message)
def show_tray_notification(self, artist, title, songs):
if self.tray:
self.tray.showMessage(title + '\n' + artist,
'Found {} results'.format(len(songs)),
QSystemTrayIcon.NoIcon, 7000)
def start_search(self, background=False, meta=None):
if not background:
self.init_layout(State.searching)
self.set_status_message('Searching...')
if meta:
self.worker_search = SearchThread(meta, self.config, background)
else:
self.worker_search = SearchThread(
[[self.main_window.edit_artist.text(), self.main_window.edit_title.text()]],
self.config, background)
self.worker_search.resultReady.connect(self.finish_search)
self.worker_search.start()
def finish_search(self, worker_id, artist, title, songs, total_time, background):
if background:
self.set_status_message('Listening to the player...')
self.main_window.lyrics_table_model.update_data(songs)
else:
if not self.worker_search or (self.worker_search.worker_id != worker_id):
return
self.init_layout(State.waiting)
self.set_status_message('Search completed in {}'.format(util.format_time_ms(total_time)))
self.main_window.lyrics_table_model.update_data(songs)
self.show_tray_notification(artist, title, songs)
def interrupt_search(self):
self.init_layout(State.waiting)
if self.worker_search:
self.set_status_message('Search stopped')
self.worker_search.quit()
self.worker_search = None
def refresh_players(self, default_player):
players = self.mpris_connect.get_players()
self.main_window.edit_player.clear()
self.main_window.edit_player_model.update_data(players)
if players:
if default_player:
for pl in players:
if pl.name == default_player:
self.main_window.edit_player.setCurrentIndex(players.index(pl))
break
else:
self.main_window.edit_player.setCurrentIndex(0)
else:
self.main_window.edit_player.setCurrentIndex(0)
def start_mpris(self, selected_player=None):
if selected_player:
players = self.mpris_connect.get_players()
self.main_window.edit_player.setCurrentIndex(players.index(selected_player))
player = self.get_current_player()
if player:
self.init_layout(State.listening)
self.set_status_message('Listening to the player...')
self.mpris_player = player
self.worker_mpris = MprisThread(self.mpris_connect, player)
self.worker_mpris.meta_ready.connect(self.update_mpris_results)
self.worker_mpris.connection_closed.connect(self.stop_mpris)
self.worker_mpris.start()
def start_mpris_from_tray(self, selected_player):
if selected_player == self.mpris_player:
return
else:
self.stop_mpris()
self.start_mpris(selected_player)
def stop_mpris(self):
self.init_layout(State.waiting)
self.set_status_message('Player listener stopped')
self.mpris_player = None
self.worker_mpris.active = False
def update_mpris_results(self, meta):
current_artist = self.main_window.edit_artist.text()
current_title = self.main_window.edit_title.text()
if meta:
new_artist = meta[0][0]
new_title = meta[0][1]
if (current_artist != new_artist or current_title != new_title) and (new_artist and new_title):
self.main_window.edit_artist.setText(new_artist)
self.main_window.edit_title.setText(new_title)
self.start_search(True, meta)
def toggle_main_window(self, reason):
if reason == QSystemTrayIcon.Trigger:
if self.main_window.isHidden():
self.main_window.show()
else:
self.main_window.hide()
def update_tray_menu(self):
self.tray.right_menu.clear()
show_action = QAction('Show &Main Window', self.tray.right_menu)
show_action.triggered.connect(self.main_window.show)
self.tray.right_menu.addAction(show_action)
self.tray.right_menu.addSeparator()
players = self.main_window.edit_player_model.players
if players:
for player in players:
if player == self.mpris_player:
player_action = QAction(player.identity + ' [connected]', self.tray.right_menu)
font = player_action.font()
font.setBold(True)
player_action.setFont(font)
else:
player_action = QAction(player.identity, self.tray.right_menu)
player_action.setIcon(QIcon(mpris.get_player_icon_path(player.name)))
player_action.triggered.connect(partial(self.start_mpris_from_tray, selected_player=player))
self.tray.right_menu.addAction(player_action)
self.tray.right_menu.addSeparator()
quit_action = QAction('&Quit', self.tray.right_menu)
quit_action.triggered.connect(QApplication.quit)
self.tray.right_menu.addAction(quit_action)