mirror/vorta
1
0
Fork 0
mirror of https://github.com/borgbase/vorta synced 2025-01-03 05:36:19 +00:00
Code Issues Releases Wiki Activity

No submenu in system tray, when only one profile exists. Fixes #57

Browse source
This commit is contained in:
Manu 2018-11-26 15:02:07 +08:00
parent c24f70ead5
commit aa9ef7642f
1 changed files with 39 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

71
src/vorta/tray_menu.py
View file

@ -1,4 +1,4 @@
from PyQt5.QtWidgets import QMenu, QSystemTrayIcon
from PyQt5.QtWidgets import QMenu, QSystemTrayIcon, QAction
from PyQt5.QtGui import QIcon
from .utils import get_asset
@ -13,23 +13,6 @@ def __init__(self, parent=None):
self.app = parent
menu = QMenu()
self.status = menu.addAction(self.app.scheduler.next_job)
self.status.setEnabled(False)
self.profile_menu = menu.addMenu('Backup Now')
self.cancel_action = menu.addAction("Cancel Backup")
self.cancel_action.triggered.connect(self.app.backup_cancelled_event.emit)
settings_action = menu.addAction("Settings")
settings_action.triggered.connect(self.app.open_main_window_action)
menu.addSeparator()
exit_action = menu.addAction("Exit")
exit_action.triggered.connect(self.app.quit)
self.on_user_click()
# https://stackoverflow.com/questions/43657890/pyqt5-qsystemtrayicon-activated-signal-not-working
menu.aboutToShow.connect(self.on_user_click)
@ -38,18 +21,42 @@ def __init__(self, parent=None):
self.show()
def on_user_click(self):
"""Adjust labels to reflect current status."""
if BorgThread.is_running():
self.status.setText('Backup in Progress')
self.profile_menu.setEnabled(False)
self.cancel_action.setVisible(True)
else:
self.status.setText(f'Next Task: {self.app.scheduler.next_job}')
self.profile_menu.setEnabled(True)
self.cancel_action.setVisible(False)
"""
Build system tray menu based on current status.
self.profile_menu.clear()
for profile in BackupProfileModel.select():
new_item = self.profile_menu.addAction(profile.name)
new_item.setData(profile.id)
new_item.triggered.connect(lambda profile_id=profile.id: self.app.create_backup_action(profile_id))
"""
menu = self.contextMenu()
menu.clear()
status = menu.addAction(self.app.scheduler.next_job)
status.setEnabled(False)
profiles = BackupProfileModel.select()
if profiles.count() > 1:
profile_menu = menu.addMenu('Backup Now')
for profile in profiles:
new_item = profile_menu.addAction(profile.name)
new_item.setData(profile.id)
new_item.triggered.connect(lambda profile_id=profile.id: self.app.create_backup_action(profile_id))
else:
profile = profiles.first()
profile_menu = menu.addAction('Backup Now')
profile_menu.triggered.connect((lambda profile_id=profile.id: self.app.create_backup_action(profile_id)))
if BorgThread.is_running():
status.setText('Backup in Progress')
profile_menu.setEnabled(False)
cancel_action = menu.addAction("Cancel Backup")
cancel_action.triggered.connect(self.app.backup_cancelled_event.emit)
else:
status.setText(f'Next Task: {self.app.scheduler.next_job}')
profile_menu.setEnabled(True)
settings_action = menu.addAction("Settings")
settings_action.triggered.connect(self.app.open_main_window_action)
menu.addSeparator()
exit_action = menu.addAction("Exit")
exit_action.triggered.connect(self.app.quit)