Handle when wmi is not present

USBToolBox · Feb 27, 2022 · 05d983a · 05d983a
1 parent 18833a9
commit 05d983a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/Scripts/shared.py b/Scripts/shared.py
@@ -7,7 +7,7 @@
 
 from Scripts._build import BUILD
 
-VERSION = "0.0.9"
+VERSION = "0.1.1"
 
 if getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS"):
     current_dir = Path(sys.executable).parent

diff --git a/Windows.py b/Windows.py
@@ -2,9 +2,17 @@
 import json
 import time
 from enum import Enum
-from pathlib import Path
 
-import wmi  # pylint: disable=import-error
+try:
+    import wmi  # pylint: disable=import-error
+except Exception:  # pylint: disable=broad-except
+    # Dummy WMI
+    class wmi:  # pylint: disable=invalid-name
+        class WMI:
+            @staticmethod
+            def query(*args, **kwargs):  # pylint: disable=unused-argument
+                return []
+
 
 from base import BaseUSBMap
 from Scripts import shared, usbdump