Merge pull request #4 from XiaoliChan/wmi-improv-admin-check

[wmi] bug fix in 'check_admin' function
Pennyw0rth · Sep 17, 2023 · 04a6ba7 · 04a6ba7
2 parents 2f0b74a + e3837d9
commit 04a6ba7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/cme/protocols/wmi.py b/cme/protocols/wmi.py
@@ -170,7 +170,7 @@ def check_if_admin(self):
             if "dcom" in locals():
                 dcom.disconnect()
 
-            if not str(e).find("access_denied") > 0:
+            if "access_denied" not in str(e).lower():
                 self.logger.fail(str(e))
         else:
             if not flag or not self.stringBinding:
@@ -188,7 +188,7 @@ def check_if_admin(self):
                 except Exception as e:
                     dcom.disconnect()
 
-                    if not str(e).find("access_denied") > 0:
+                    if "access_denied" not in str(e).lower():
                         self.logger.fail(str(e))
                 else:
                     dcom.disconnect()

diff --git a/poetry.lock b/poetry.lock