Update TrayIcon.py

intersimone999 · Sep 28, 2024 · 7d2aaf0 · 7d2aaf0
1 parent 94dd5c5
commit 7d2aaf0
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/pyqtws/plugin/TrayIcon.py b/pyqtws/plugin/TrayIcon.py
@@ -27,16 +27,14 @@ def __init__(self, config: QTWSConfig):
             )
         )
 
-        self.tmp_icon_path = tempfile.NamedTemporaryFile(suffix=".png").name
-
-        QImage(config.icon).save(self.tmp_icon_path)
-
-        self.tray_icon = pystray.Icon(
-            config.name,
-            Image.open(self.tmp_icon_path),
-            menu=self.menu
-        )
-        self.tmp_icon_path.close()
+        with tempfile.NamedTemporaryFile(suffix=".png") as self.tmp_icon:
+            QImage(config.icon).save(self.tmp_icon.name)
+
+            self.tray_icon = pystray.Icon(
+                config.name,
+                Image.open(self.tmp_icon.name),
+                menu=self.menu
+            )
 
         self.tray_icon.visible = True
         self.tray_icon.run_detached()