mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 01:14:15 +02:00
84 lines
2.8 KiB
Python
Executable File
84 lines
2.8 KiB
Python
Executable File
import os
|
|
import time
|
|
|
|
from PyQt5.QtCore import QThread, pyqtSignal, QCoreApplication
|
|
from PyQt5.QtGui import QIcon
|
|
from PyQt5.QtWidgets import QSystemTrayIcon, QMenu
|
|
|
|
from core import resource
|
|
|
|
# to update
|
|
from core.controller import FlatpakController
|
|
from view.qt.window import ManageWindow
|
|
|
|
|
|
class UpdateCheck(QThread):
|
|
|
|
signal = pyqtSignal(int)
|
|
|
|
def __init__(self, check_interval: int, controller: FlatpakController, parent=None):
|
|
super(UpdateCheck, self).__init__(parent)
|
|
self.controller = controller
|
|
self.check_interval = check_interval
|
|
|
|
def run(self):
|
|
|
|
while True:
|
|
|
|
apps = self.controller.refresh()
|
|
|
|
updates = len([app for app in apps if app['update']])
|
|
|
|
self.signal.emit(updates)
|
|
|
|
time.sleep(self.check_interval)
|
|
|
|
|
|
class TrayIcon(QSystemTrayIcon):
|
|
|
|
def __init__(self, locale_keys: dict, controller: FlatpakController, check_interval: int = 60, parent=None):
|
|
self.locale_keys = locale_keys
|
|
self.controller = controller
|
|
|
|
self.icon_default = QIcon(resource.get_path('img/flathub_45.svg'))
|
|
self.icon_update = QIcon(resource.get_path('img/update_logo.svg'))
|
|
QSystemTrayIcon.__init__(self, self.icon_default, parent)
|
|
|
|
self.menu = QMenu(parent)
|
|
self.action_manage = self.menu.addAction(self.locale_keys['tray.action.manage'])
|
|
self.action_manage.triggered.connect(self.show_manage_window)
|
|
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 = ManageWindow(locale_keys=self.locale_keys, controller=controller, tray_icon=self)
|
|
self.check_thread = UpdateCheck(check_interval=check_interval, controller=self.controller)
|
|
self.check_thread.signal.connect(self.notify_updates)
|
|
self.check_thread.start()
|
|
|
|
def notify_updates(self, updates: int):
|
|
if updates > 0:
|
|
if self.icon().cacheKey() != self.icon_update.cacheKey():
|
|
self.setIcon(self.icon_update)
|
|
|
|
msg = '{}: {}'.format(self.locale_keys['notification.new_updates'], updates)
|
|
self.setToolTip(msg)
|
|
|
|
if bool(os.getenv('FPAKMAN_UPDATE_NOTIFICATION', 1)):
|
|
os.system("notify-send -i {} '{}'".format(resource.get_path('img/flathub_10.svg'), msg))
|
|
|
|
if self.manage_window and not self.manage_window.isHidden():
|
|
self.manage_window.refresh()
|
|
|
|
else:
|
|
self.setIcon(self.icon_default)
|
|
self.setToolTip(None)
|
|
|
|
def show_manage_window(self):
|
|
|
|
if not self.manage_window:
|
|
self.manage_window = ManageWindow(controller=self.controller)
|
|
|
|
self.manage_window.refresh()
|
|
self.manage_window.show()
|