Add timer for switching theme

Author: chase9

yin_yang/__main__.py

@@ -1,22 +1,23 @@
 #!/bin/env python3
 
-import sys
 import logging
+import sys
 from argparse import ArgumentParser
 from logging.handlers import RotatingFileHandler
 from pathlib import Path
 
 from PySide6 import QtWidgets
-from PySide6.QtCore import QTranslator, QLibraryInfo, QLocale, QObject
+from PySide6.QtCore import QLibraryInfo, QLocale, QTranslator
 from PySide6.QtGui import QIcon
-from PySide6.QtWidgets import QSystemTrayIcon, QMenu
+from PySide6.QtWidgets import QMenu, QSystemTrayIcon
 from systemd import journal
 
-from yin_yang.notification_handler import NotificationHandler
-from yin_yang import daemon_handler
+from yin_yang import daemon_handler, theme_switcher
+from yin_yang.config import Modes, config
+from yin_yang.helpers import is_flatpak
 from yin_yang.meta import ConfigEvent
-from yin_yang import theme_switcher
-from yin_yang.config import config, Modes
+from yin_yang.notification_handler import NotificationHandler
+from yin_yang.scheduler import Scheduler
 from yin_yang.ui import main_window_connector
 
 logger = logging.getLogger()
@@ -36,7 +37,7 @@ def setup_logger(use_systemd_journal: bool):
         # noinspection SpellCheckingInspection
         logging.basicConfig(
             level=logging.DEBUG,
-            format='%(asctime)s %(levelname)s - %(name)s: %(message)s'
+            format='%(asctime)s %(levelname)s - %(name)s: %(message)s',
         )
     else:
         # if you run it with "python -O __main__.py" instead, debug is false
@@ -45,12 +46,13 @@ def setup_logger(use_systemd_journal: bool):
         # noinspection SpellCheckingInspection
         logging.basicConfig(
             level=logging.WARNING,
-            format='%(asctime)s %(levelname)s - %(name)s: %(message)s'
+            format='%(asctime)s %(levelname)s - %(name)s: %(message)s',
         )
         # and add a handler that limits the size to 1 GB
         file_handler = RotatingFileHandler(
             str(Path.home()) + '/.local/share/yin_yang.log',
-            maxBytes=10**9, backupCount=1
+            maxBytes=10**9,
+            backupCount=1,
         )
         logging.root.addHandler(file_handler)
 
@@ -65,11 +67,15 @@ def systray_icon_clicked(reason: QSystemTrayIcon.ActivationReason):
 
 # using ArgumentParser for parsing arguments
 parser = ArgumentParser()
-parser.add_argument('-t', '--toggle',
-                    help='toggles Yin-Yang',
-                    action='store_true')
-parser.add_argument('--systemd', help='uses systemd journal handler and applies desired theme', action='store_true')
-parser.add_argument('--minimized', help='starts the program to tray bar', action='store_true')
+parser.add_argument('-t', '--toggle', help='toggles Yin-Yang', action='store_true')
+parser.add_argument(
+    '--systemd',
+    help='uses systemd journal handler and applies desired theme',
+    action='store_true',
+)
+parser.add_argument(
+    '--minimized', help='starts the program to tray bar', action='store_true'
+)
 arguments = parser.parse_args()
 setup_logger(arguments.systemd)
 
@@ -126,20 +132,34 @@ def systray_icon_clicked(reason: QSystemTrayIcon.ActivationReason):
 
         menu = QMenu('Yin & Yang')
         menu.addAction(
-            app.translate('systray', 'Open Yin Yang', 'Context menu action in the systray'),
-            lambda: window.show())
+            app.translate(
+                'systray', 'Open Yin Yang', 'Context menu action in the systray'
+            ),
+            lambda: window.show(),
+        )
         menu.addAction(
-            app.translate('systray', 'Toggle theme', 'Context menu action in the systray'),
-            lambda: theme_switcher.set_mode(not config.dark_mode))
-        menu.addAction(QIcon.fromTheme('application-exit'),
-                       app.translate('systray', 'Quit', 'Context menu action in the systray'),
-                       app.quit)
+            app.translate(
+                'systray', 'Toggle theme', 'Context menu action in the systray'
+            ),
+            lambda: theme_switcher.set_mode(not config.dark_mode),
+        )
+        menu.addAction(
+            QIcon.fromTheme('application-exit'),
+            app.translate('systray', 'Quit', 'Context menu action in the systray'),
+            app.quit,
+        )
 
         icon.setContextMenu(menu)
         icon.show()
     else:
         logger.debug('System tray is unsupported')
 
+    # Start the timer to handle automatic theme switching. We only need this
+    # currently if running in a flatpak as systemd units are not supported.
+    if is_flatpak:
+        timer = Scheduler(45, theme_switcher.set_desired_theme)
+        timer.start()
+
     if arguments.minimized:
         sys.exit(app.exec())
     else:

yin_yang/scheduler.py

@@ -0,0 +1,11 @@
+import logging
+from threading import Timer
+
+logger = logging.getLogger(__name__)
+
+
+class Scheduler(Timer):
+    def run(self):
+        while not self.finished.wait(self.interval):
+            logger.debug("Timer expired, checking theme.")
+            self.function(*self.args, **self.kwargs)