Fix Linux AssertionError

Kaliiiiiiiiii-Vinyzu · Apr 18, 2024 · 708f40f · 708f40f
1 parent e9717c7
commit 708f40f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cdp_patches/input/os_base/linux.py b/cdp_patches/input/os_base/linux.py
@@ -117,7 +117,8 @@ def search_windows_by_pid(query_tree, pid: int):
                     search_windows_by_pid(window.query_tree(), pid)
 
         search_windows_by_pid(self.display.screen().root.query_tree(), self.pid)
-        assert res_windows, WindowClosedException(f"No windows found for PID: {self.pid}")
+        if not res_windows:
+            raise WindowClosedException(f"No windows found for PID: {self.pid}")
 
         for window in res_windows:
             # Getting necessary window properties
@@ -158,7 +159,8 @@ def search_windows_by_pid(query_tree, pid: int):
                     search_windows_by_pid(window.query_tree(), pid)
 
         await self._loop.run_in_executor(None, lambda: search_windows_by_pid(self.display.screen().root.query_tree(), self.pid))
-        assert res_windows, WindowClosedException(f"No windows found for PID: {self.pid}")
+        if not res_windows:
+            raise WindowClosedException(f"No windows found for PID: {self.pid}")
 
         for window in res_windows:
             # Getting necessary window properties