Add icon to PyInstaller exe

Author: bitoffdev

kasatk.spec

@@ -1,5 +1,6 @@
 # -*- mode: python ; coding: utf-8 -*-
 
+RELATIVE_ICON_PATH = "./extra/icon.ico"
 block_cipher = None
 
 def kasa_dist_info_datas():
@@ -20,7 +21,8 @@ def kasa_dist_info_datas():
 a = Analysis(['kasatk/__main__.py'],
              pathex=['.\\env\\Lib\\site-packages\\', '.'],
              binaries=[],
-             datas=[kasa_dist_info_datas()],
+             # Note: the second item in each tuple MUST be a directory.
+             datas=[kasa_dist_info_datas(), (RELATIVE_ICON_PATH, "./extra")],
              hiddenimports=[],
              hookspath=[],
              runtime_hooks=[],
@@ -44,5 +46,7 @@ exe = EXE(pyz,
           upx=True,
           upx_exclude=[],
           runtime_tmpdir=None,
-          icon='extra/icon.ico',
+          icon=RELATIVE_ICON_PATH,
           console=False )
+
+# vi:syntax=python

kasatk/__main__.py

@@ -18,6 +18,16 @@
 logger = logging.getLogger(__name__)
 
 
+def resource_path(relative_path):
+    """Find a resource in a PyInstaller executable, or in the local directory"""
+    if hasattr(sys, "_MEIPASS"):
+        logger.info("Finding resource inside PyInstaller executable")
+        p = os.path.join(sys._MEIPASS, relative_path)
+        assert os.path.exists(p)
+        return p
+    return os.path.join(os.path.abspath("."), relative_path)
+
+
 class LightState(TypedDict):
     on_off: int
     mode: str
@@ -370,7 +380,11 @@ def main():
     root = tkinter.Tk()
     root.title("Kasa Devices")
     root.geometry("500x400")
-    root.iconbitmap("extra/icon.ico")
+    try:
+        root.iconbitmap(resource_path("extra/icon.ico"))
+    except Exception as e:
+        logger.warning("Failed to load application icon.")
+        logger.error(e)
     scroll_frame = ScrollableFrame(root)
     device_frame = KasaDevices(scroll_frame.scrollable_frame)
     device_frame.pack()