Add exception handler for wmi module

Lekuruu · Jun 25, 2024 · ac8e5df · ac8e5df
1 parent 13950a7
commit ac8e5df
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/osu/objects/client.py b/osu/objects/client.py
@@ -111,11 +111,14 @@ def disk_signature(self) -> str:
         if platform.system() != "Windows":
             return hashlib.md5(b"unknown").hexdigest()
 
-        from wmi import WMI
+        try:
+            from wmi import WMI
 
-        # Get serial number of first item
-        for item in WMI().query("SELECT * FROM Win32_DiskDrive"):
-            return hashlib.md5(item.SerialNumber.encode()).hexdigest()
+            # Get serial number of first item
+            for item in WMI().query("SELECT * FROM Win32_DiskDrive"):
+                return hashlib.md5(item.SerialNumber.encode()).hexdigest()
+        except Exception:
+            pass
 
         # Fallback
         return hashlib.md5(b"unknown").hexdigest()