Files
Loren Eteval 00c86a91c3 Initial commit
Signed-off-by: Loren Eteval <loren.eteval@proton.me>
2024-01-26 23:52:46 +08:00

123 lines
3.9 KiB
Python

from Furious.PyFramework import *
from Furious.QtFramework import *
from Furious.QtFramework import gettext as _
from Furious.Utility import *
from Furious.TrayActions import *
from PySide6 import QtCore
from PySide6.QtWidgets import QSystemTrayIcon
import logging
logger = logging.getLogger(__name__)
__all__ = ['SystemTrayIcon']
class SystemTrayIcon(QTranslatable, SupportConnectedCallback, QSystemTrayIcon):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
actions = [
ConnectAction(),
RoutingAction(),
AppQSeperator(),
ImportAction(),
EditConfigurationAction(),
AppQSeperator(),
LanguageAction(),
SettingsAction(),
AppQSeperator(),
ExitAction(),
]
# Some old version PySide6 does not have setMenu method
# for QAction. Protect it. Currently only used in SystemTrayIcon
if hasattr(AppQAction, 'setMenu'):
logger.info('contextMenu uses setMenu implementation')
for action in actions:
if isinstance(action, AppQAction):
if hasattr(self, f'{action}'):
logger.warning(f'{self} already has action {action}')
setattr(self, f'{action}', action)
self._menu = AppQMenu(*actions)
self.setContextMenu(self._menu)
else:
logger.info('contextMenu uses addMenu implementation')
self._refs = []
self._menu = AppQMenu()
for action in actions:
if isinstance(action, AppQAction):
if hasattr(self, f'{action}'):
logger.warning(f'{self} already has action {action}')
setattr(self, f'{action}', action)
if action._menu is None:
self._menu.addAction(action)
else:
menu = AppQMenu(*action._menu._actions, title=action.text())
menu.setIcon(action.icon())
self._refs.append(menu)
self._menu.addMenu(menu)
else:
self._menu.addSeparator()
self.setContextMenu(self._menu)
self.setDisconnectedIcon()
self.activated.connect(self.handleActivated)
def bootstrap(self):
if AppSettings.isStateON_('StartupOnBoot'):
# Rrefresh startup application location
StartupOnBoot.on_()
if AppSettings.isStateON_('Connect'):
self.ConnectAction.trigger()
def showMessage(self, message: str, *args, **kwargs):
if message:
super().showMessage(_(APPLICATION_NAME), message, *args, **kwargs)
def setDisconnectedIcon(self):
if PLATFORM == 'Darwin' or isWindows7():
# Darker
self.setIcon(bootstrapIcon('rocket-takeoff-dark.svg'))
else:
self.setIcon(bootstrapIcon('rocket-takeoff.svg'))
def setConnectedIcon(self):
if isAdministrator():
self.setIcon(bootstrapIcon('rocket-takeoff-admin-connected.svg'))
else:
if PLATFORM == 'Darwin' or isWindows7():
# Darker
self.setIcon(bootstrapIcon('rocket-takeoff-connected-dark.svg'))
else:
self.setIcon(bootstrapIcon('rocket-takeoff-connected.svg'))
@QtCore.Slot(QSystemTrayIcon.ActivationReason)
def handleActivated(self, reason):
if reason == QSystemTrayIcon.ActivationReason.DoubleClick:
APP().mainWindow.show()
def setCustomToolTip(self):
self.setToolTip(f'{_(APPLICATION_NAME)} {APPLICATION_VERSION}')
def disconnectedCallback(self):
self.setDisconnectedIcon()
def connectedCallback(self):
self.setConnectedIcon()
def retranslate(self):
# Nothing to do
pass