Merge pull request #13 from mithwick93/add-option-to-restart

Add option to restart

Author: mithwick93

core/m27q.py

@@ -10,7 +10,7 @@
 import usb.core
 import usb.util
 
-from util.alert import show_error_message
+from util.alert import show_alert
 from util.logger import get_logger
 
 
@@ -41,7 +41,7 @@ def clamp(self, v: int):
 
 
 def display_alert(e: Exception):
-    show_error_message(
+    show_alert(
         "Error",
         repr(e) + ".\nTry restarting the app",
         "Dismiss",

main.py

@@ -2,7 +2,7 @@
 
 from core.m27q import MonitorControl
 from ui.brightness_control_app import BrightnessControlApp
-from util.alert import show_error_message
+from util.alert import show_alert
 from util.logger import get_logger
 
 if __name__ in {"__main__", "__mp_main__"}:
@@ -17,6 +17,5 @@
         with MonitorControl() as monitor_Control:
             BrightnessControlApp(monitor_Control, main_logger).run()
     except Exception as e:
-        show_error_message("Critical Error", repr(e), "Exit", "critical")
-
+        show_alert("Critical Error", repr(e), "Exit", "critical")
         raise e

resources/sample.png

@@ -1,3 +1,5 @@
+import os
+import sys
 import threading
 import time
 from logging import Logger
@@ -11,6 +13,11 @@ def get_capitalized_setting(setting: str):
     return setting.replace('_', ' ').capitalize()
 
 
+def restart_app(_: rumps.SliderMenuItem):
+    executable = sys.executable
+    os.execl(executable, executable, *sys.argv)
+
+
 class BrightnessControlApp(rumps.App):
     def __init__(self, monitor_control: MonitorControl, logger: Logger):
         super(BrightnessControlApp, self).__init__(name="Gigabyte M27Q Settings Controller", icon="resources/icon.png")
@@ -40,6 +47,13 @@ def __init__(self, monitor_control: MonitorControl, logger: Logger):
         self._auto_refresh_toggle_menu_item = auto_refresh_toggle_menu_item
         self.menu.add(auto_refresh_toggle_menu_item)
 
+        restart_app__menu_item = rumps.MenuItem(
+            title="Restart",
+            callback=restart_app
+        )
+        self._restart_app__menu_item = restart_app__menu_item
+        self.menu.add(restart_app__menu_item)
+
         self._polling_thread = threading.Thread(target=self._poll_monitor_settings, daemon=True)
         self._polling_thread.start()
         self._update_ui_timer = rumps.Timer(self._update_ui_from_polling, self._auto_setting_refresh_interval_seconds)

ui/brightness_control_app.py

@@ -7,7 +7,7 @@
 }
 
 
-def show_error_message(
+def show_alert(
         title: str,
         message: str,
         button_text: str = "OK",