Merge pull request #184 from mathoudebine/feature/181-turn-the-screen-off-before-computer-sleeps-and-back-on-after-resume

Turn screen off/on when computer goes to/resume from sleep (Windows)

Author: mathoudebine

configure.py

@@ -109,8 +109,8 @@ def __init__(self):
         self.wl_cb.place(x=500, y=150, width=210)
 
         self.lhm_admin_warning = ttk.Label(self.window,
-                                           text="Admin rights needed, or select another Hardware monitoring",
-                                           foreground='#00f')
+                                           text="❌ Restart as admin. or select another Hardware monitoring",
+                                           foreground='#f00')
         self.lhm_admin_warning.place(x=320, y=190)
 
         sysmon_label = ttk.Label(self.window, text='Display configuration', font='bold')
@@ -142,7 +142,7 @@ def __init__(self):
         self.brightness_val_label.place(x=500, y=385)
         self.brightness_warning_label = ttk.Label(self.window,
                                                   text="⚠ Turing / rev. A displays can get hot at high brightness!",
-                                                  foreground='#f00')
+                                                  foreground='#ff8c00')
         self.brightness_warning_label.place(x=320, y=420)
 
         self.edit_theme_btn = ttk.Button(self.window, text="Edit theme", command=lambda: self.on_theme_editor_click())
@@ -295,12 +295,15 @@ def on_hwlib_change(self, e=None):
             self.eth_cb.configure(state="readonly", foreground="#000")
             self.wl_cb.configure(state="readonly", foreground="#000")
 
-        if hwlib == "LHM":
-            self.lhm_admin_warning.place(x=320, y=190)
-        elif hwlib == "AUTO" and sys.platform == "win32":
-            self.lhm_admin_warning.place(x=320, y=190)
-        else:
-            self.lhm_admin_warning.place_forget()
+        if sys.platform == "win32":
+            import ctypes
+            is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
+            if (hwlib == "LHM" or hwlib == "AUTO") and not is_admin:
+                self.lhm_admin_warning.place(x=320, y=190)
+                self.save_run_btn.state(["disabled"])
+            else:
+                self.lhm_admin_warning.place_forget()
+                self.save_run_btn.state(["!disabled"])
 
     def show_hide_brightness_warning(self, e=None):
         if int(self.brightness_slider.get()) > 50 and [k for k, v in revision_map.items() if v == self.model_cb.get()][

library/display.py

@@ -82,6 +82,13 @@ def initialize_display(self):
         # Send initialization commands
         self.lcd.InitializeComm()
 
+        # Turn on display, set brightness and LEDs for supported HW
+        self.turn_on()
+
+        # Set orientation
+        self.lcd.SetOrientation(_get_theme_orientation())
+
+    def turn_on(self):
         # Turn screen on in case it was turned off previously
         self.lcd.ScreenOn()
 
@@ -91,8 +98,12 @@ def initialize_display(self):
         # Set backplate RGB LED color (for supported HW only)
         self.lcd.SetBackplateLedColor(config.THEME_DATA['display'].get("DISPLAY_RGB_LED", (255, 255, 255)))
 
-        # Set orientation
-        self.lcd.SetOrientation(_get_theme_orientation())
+    def turn_off(self):
+        # Turn screen off
+        self.lcd.ScreenOff()
+
+        # Turn off backplate RGB LED
+        self.lcd.SetBackplateLedColor(led_color=(0, 0, 0))
 
     def display_static_images(self):
         if config.THEME_DATA.get('static_images', False):

main.py

@@ -65,11 +65,8 @@
 
 
     def clean_stop(tray_icon=None):
-        # Turn screen off before stopping
-        display.lcd.ScreenOff()
-
-        # Turn backplate LED off for supported devices
-        display.lcd.SetBackplateLedColor(led_color=(0, 0, 0))
+        # Turn screen and LEDs off before stopping
+        display.turn_off()
 
         # Do not stop the program now in case data transmission was in progress
         # Instead, ask the scheduler to empty the action queue before stopping
@@ -121,15 +118,26 @@ def on_clean_exit(*args):
         def on_win32_ctrl_event(event):
             """Handle Windows console control events (like Ctrl-C)."""
             if event in (win32con.CTRL_C_EVENT, win32con.CTRL_BREAK_EVENT, win32con.CTRL_CLOSE_EVENT):
-                logger.info("Caught Windows control event %s, exiting" % event)
+                logger.debug("Caught Windows control event %s, exiting" % event)
                 clean_stop()
             return 0
 
 
         def on_win32_wm_event(hWnd, msg, wParam, lParam):
             """Handle Windows window message events (like ENDSESSION, CLOSE, DESTROY)."""
-            logger.debug("Caught Windows window message event %s, exiting" % msg)
-            clean_stop()
+            logger.debug("Caught Windows window message event %s" % msg)
+            if msg == win32con.WM_POWERBROADCAST:
+                # WM_POWERBROADCAST is used to detect computer going to/resuming from sleep
+                if wParam == win32con.PBT_APMSUSPEND:
+                    logger.info("Computer is going to sleep, display will turn off")
+                    display.turn_off()
+                elif wParam == win32con.PBT_APMRESUMEAUTOMATIC:
+                    logger.info("Computer is resuming from sleep, display will turn on")
+                    display.turn_on()
+            else:
+                # For any other events, the program will stop
+                logger.info("Program will now exit")
+                clean_stop()
 
     # Create a tray icon for the program, with an Exit entry in menu
     try:
@@ -214,7 +222,8 @@ def on_win32_wm_event(hWnd, msg, wParam, lParam):
                       win32con.WM_ENDSESSION: on_win32_wm_event,
                       win32con.WM_QUIT: on_win32_wm_event,
                       win32con.WM_DESTROY: on_win32_wm_event,
-                      win32con.WM_CLOSE: on_win32_wm_event}
+                      win32con.WM_CLOSE: on_win32_wm_event,
+                      win32con.WM_POWERBROADCAST: on_win32_wm_event}
 
         wndclass.lpfnWndProc = messageMap