From 95f5bcdde7e44de7bc5cba5b8a1011b2002ae4b1 Mon Sep 17 00:00:00 2001
From: Shohan Dutta Roy <shohanduttaroy99@gmail.com>
Date: Wed, 6 Nov 2024 23:59:52 +0530
Subject: [PATCH] fix: Fix recording not working because of bug in read window
 data

---
 openadapt/window/_windows.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/openadapt/window/_windows.py b/openadapt/window/_windows.py
index 9a49d6b7d..e08b89c25 100644
--- a/openadapt/window/_windows.py
+++ b/openadapt/window/_windows.py
@@ -1,8 +1,10 @@
 from pprint import pprint
+from typing import TYPE_CHECKING
 import pickle
 import time
 
-import pywinauto
+if TYPE_CHECKING:
+    import pywinauto
 
 from openadapt.custom_logger import logger
 
@@ -53,7 +55,7 @@ def get_active_window_state(read_window_data: bool) -> dict:
 
 
 def get_active_window_meta(
-    active_window: pywinauto.application.WindowSpecification,
+    active_window: "pywinauto.application.WindowSpecification",
 ) -> dict:
     """Get the meta information of the active window.
 
@@ -88,18 +90,22 @@ def get_active_element_state(x: int, y: int) -> dict:
     return properties
 
 
-def get_active_window() -> pywinauto.application.WindowSpecification:
+def get_active_window() -> "pywinauto.application.WindowSpecification":
     """Get the active window object.
 
     Returns:
         pywinauto.application.WindowSpecification: The active window object.
     """
+    import pywinauto
+
     app = pywinauto.application.Application(backend="uia").connect(active_only=True)
     window = app.top_window()
     return window.wrapper_object()
 
 
-def get_element_properties(element: pywinauto.application.WindowSpecification) -> dict:
+def get_element_properties(
+    element: "pywinauto.application.WindowSpecification",
+) -> dict:
     """Recursively retrieves the properties of each element and its children.
 
     Args:
@@ -132,7 +138,7 @@ def get_element_properties(element: pywinauto.application.WindowSpecification) -
     return properties
 
 
-def dictify_rect(rect: pywinauto.win32structures.RECT) -> dict:
+def dictify_rect(rect: "pywinauto.win32structures.RECT") -> dict:
     """Convert a rectangle object to a dictionary.
 
     Args:
@@ -150,7 +156,7 @@ def dictify_rect(rect: pywinauto.win32structures.RECT) -> dict:
     return rect_dict
 
 
-def get_properties(element: pywinauto.application.WindowSpecification) -> dict:
+def get_properties(element: "pywinauto.application.WindowSpecification") -> dict:
     """Retrieves specific writable properties of an element.
 
     This function retrieves a dictionary of writable properties for a given element.
@@ -168,6 +174,7 @@ class of the element may have a `get_properties()` function that raises errors.
 
     """
     _element_class = element.__class__
+    import pywinauto
 
     class TempElement(element.__class__):
         writable_props = pywinauto.base_wrapper.BaseWrapper.writable_props