Add a cross-platform tray icon with Exit menu entry

Author: mathoudebine

config.yaml

@@ -47,7 +47,7 @@ display:
   DISPLAY_REVERSE: false
 
   # Display resolution in portrait orientation
-  # Do not use this setting to rotate display! Use DISPLAY_REVERSE
+  # Do not use this setting to rotate display! The selected theme handles landscape/portrait orientation
   DISPLAY_WIDTH: 320  # Do not change unless you have a good reason
   DISPLAY_HEIGHT: 480  # Do not change unless you have a good reason
 

main.py

@@ -3,10 +3,13 @@
 # https://github.com/mathoudebine/turing-smart-screen-python
 import locale
 import os
+import pystray
 import signal
 import sys
 import time
 
+from PIL import Image
+
 MIN_PYTHON = (3, 7)
 if sys.version_info < MIN_PYTHON:
     print("[ERROR] Python %s.%s or later is required." % MIN_PYTHON)
@@ -24,9 +27,8 @@
     # Apply system locale to this program
     locale.setlocale(locale.LC_ALL, '')
 
-    def sighandler(signum, frame):
-        logger.info(" Caught signal %d, exiting" % signum)
 
+    def clean_stop(tray_icon=None):
         # Do not stop the program now in case data transmission was in progress
         # Instead, ask the scheduler to empty the action queue before stopping
         scheduler.STOPPING = True
@@ -41,13 +43,27 @@ def sighandler(signum, frame):
 
         logger.debug("(%.1fs)" % (5 - wait_time))
 
+        # Remove icon just before the exit
+        if tray_icon:
+            tray_icon.visible = False
+
         # We force the exit to avoid waiting for other scheduled tasks: they may have a long delay!
         try:
             sys.exit(0)
         except:
             os._exit(0)
 
 
+    def sighandler(signum, frame=None):
+        logger.info("Caught signal %d, exiting" % signum)
+        clean_stop()
+
+
+    def on_exit_tray(tray_icon, item):
+        logger.info("Exit from tray icon")
+        clean_stop(tray_icon)
+
+
     # Set the signal handlers, to send a complete frame to the LCD before exit
     signal.signal(signal.SIGINT, sighandler)
     signal.signal(signal.SIGTERM, sighandler)
@@ -81,3 +97,14 @@ def sighandler(signum, frame):
     scheduler.NetStats()
     scheduler.DateStats()
     scheduler.QueueHandler()
+
+    # Create a tray icon for the program, with an Exit entry in menu
+    tray_icon = pystray.Icon(
+        name='Turing System Monitor',
+        title='Turing System Monitor',
+        icon=Image.open("res/icons/724873501557740364-64.png"),
+        menu=pystray.Menu(
+            pystray.MenuItem(
+                'Exit',
+                on_exit_tray))
+    ).run()

requirements.txt

@@ -4,6 +4,7 @@ pyserial~=3.5     # Serial linl to communicate with the display
 PyYAML~=6.0       # For themes files
 psutil~=5.9.4     # CPU / disk / network metrics
 GPUtil~=1.4.0     # Nvidia GPU
+pystray~=0.19.4   # Tray icon (all OS)
 
 # Following packages are for AMD GPU on Linux
 # Note: you may need to manually install Cython first

res/icons/724873501557740364-128.png

@@ -0,0 +1,5 @@
+Icon by <a href="https://freeicons.io/profile/730">Anu Rocks</a> on <a href="https://freeicons.io">freeicons.io</a>
+
+https://freeicons.io/regular-life-icons/monitor-icon-17865
+
+This work is licensed under a <a rel="license" href="http://creativecommons.org/licenses/by/3.0/">Creative Commons Attribution 3.0 Unported License</a>.