mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-06 22:54:16 +02:00
137 lines
4.3 KiB
Python
Executable File
137 lines
4.3 KiB
Python
Executable File
import time
|
|
from typing import List
|
|
|
|
from PyQt5.QtCore import QThread, pyqtSignal, QCoreApplication
|
|
from PyQt5.QtGui import QIcon
|
|
from PyQt5.QtWidgets import QSystemTrayIcon, QMenu
|
|
|
|
from fpakman.core import resource, system
|
|
from fpakman.core.controller import FlatpakManager
|
|
from fpakman.view.qt.about import AboutDialog
|
|
from fpakman.view.qt.window import ManageWindow
|
|
|
|
|
|
class UpdateCheck(QThread):
|
|
|
|
signal = pyqtSignal(list)
|
|
|
|
def __init__(self, manager: FlatpakManager, check_interval: int, parent=None):
|
|
super(UpdateCheck, self).__init__(parent)
|
|
self.check_interval = check_interval
|
|
self.manager = manager
|
|
|
|
def run(self):
|
|
|
|
while True:
|
|
|
|
apps = self.manager.read_installed()
|
|
|
|
updates = [app for app in apps if app['update']]
|
|
|
|
if updates:
|
|
self.signal.emit(updates)
|
|
|
|
time.sleep(self.check_interval)
|
|
|
|
|
|
class LoadDatabase(QThread):
|
|
|
|
signal_finished = pyqtSignal()
|
|
|
|
def __init__(self, manager: FlatpakManager, parent=None):
|
|
super(LoadDatabase, self).__init__(parent)
|
|
self.manager = manager
|
|
|
|
def run(self):
|
|
self.manager.load_full_database()
|
|
self.signal_finished.emit()
|
|
|
|
|
|
class TrayIcon(QSystemTrayIcon):
|
|
|
|
def __init__(self, locale_keys: dict, manager: FlatpakManager, check_interval: int = 60, update_notification: bool = True):
|
|
super(TrayIcon, self).__init__()
|
|
self.locale_keys = locale_keys
|
|
self.manager = manager
|
|
|
|
self.icon_default = QIcon(resource.get_path('img/logo.svg'))
|
|
self.icon_update = QIcon(resource.get_path('img/logo_update.svg'))
|
|
self.setIcon(self.icon_default)
|
|
|
|
self.menu = QMenu()
|
|
|
|
self.action_refreshing = self.menu.addAction(self.locale_keys['tray.action.refreshing'] + '...')
|
|
self.action_refreshing.setEnabled(False)
|
|
|
|
self.action_manage = self.menu.addAction(self.locale_keys['tray.action.manage'])
|
|
self.action_manage.triggered.connect(self.show_manage_window)
|
|
self.action_manage.setVisible(False)
|
|
|
|
self.action_about = self.menu.addAction(self.locale_keys['tray.action.about'])
|
|
self.action_about.triggered.connect(self.show_about)
|
|
|
|
self.action_exit = self.menu.addAction(self.locale_keys['tray.action.exit'])
|
|
self.action_exit.triggered.connect(lambda: QCoreApplication.exit())
|
|
|
|
self.setContextMenu(self.menu)
|
|
|
|
self.manage_window = None
|
|
self.check_thread = UpdateCheck(check_interval=check_interval, manager=self.manager)
|
|
self.check_thread.signal.connect(self.notify_updates)
|
|
self.check_thread.start()
|
|
|
|
self.dialog_about = None
|
|
|
|
self.thread_database = LoadDatabase(manager)
|
|
self.thread_database.signal_finished.connect(self._update_menu)
|
|
self.last_updates = set()
|
|
self.update_notification = update_notification
|
|
|
|
def load_database(self):
|
|
self.thread_database.start()
|
|
|
|
def _update_menu(self):
|
|
self.action_refreshing.setVisible(False)
|
|
self.action_manage.setVisible(True)
|
|
|
|
def notify_updates(self, updates: List[dict]):
|
|
|
|
if len(updates) > 0:
|
|
|
|
update_keys = {'{}:{}'.format(app['id'], app['latest_version']) for app in updates}
|
|
|
|
new_icon = self.icon_update
|
|
|
|
if update_keys.difference(self.last_updates):
|
|
self.last_updates = update_keys
|
|
msg = '{}: {}'.format(self.locale_keys['notification.new_updates'].format('Flatpak'), len(updates))
|
|
self.setToolTip(msg)
|
|
|
|
if self.update_notification:
|
|
system.notify_user(msg)
|
|
|
|
else:
|
|
new_icon = self.icon_default
|
|
self.setToolTip(None)
|
|
|
|
if self.icon().cacheKey() != new_icon.cacheKey(): # changes the icon if needed
|
|
self.setIcon(new_icon)
|
|
|
|
def show_manage_window(self):
|
|
|
|
if self.manage_window is None:
|
|
self.manage_window = ManageWindow(locale_keys=self.locale_keys,
|
|
manager=self.manager,
|
|
tray_icon=self)
|
|
|
|
self.manage_window.refresh()
|
|
self.manage_window.show()
|
|
|
|
def show_about(self):
|
|
|
|
if self.dialog_about is None:
|
|
self.dialog_about = AboutDialog(self.locale_keys)
|
|
|
|
if self.dialog_about.isHidden():
|
|
self.dialog_about.show()
|