add typing for baseproperty

Kaliiiiiiiiii-Vinyzu · Mar 25, 2024 · 67cdaef · 67cdaef
1 parent 3dc6241
commit 67cdaef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cdp_patches/input/async_input.py b/cdp_patches/input/async_input.py
@@ -6,6 +6,7 @@
 import re
 import threading
 import time
+import typing
 from typing import Any, Generator, Literal, Optional, Union
 
 from pywinauto.base_wrapper import ElementNotEnabled, ElementNotVisible
@@ -48,13 +49,14 @@ async def __ainit__(self) -> None:
         if os.name == "nt":
             self._base = WindowsBase(self.pid, self.scale_factor)
         else:
+            # mind to change typing of `self.base` property when implementing
             raise NotImplementedError(f"pyinput not implemented yet for {os.name}")
         await self._wait_for_window()
 
         self._base._offset_toolbar_height()
 
     @property
-    def base(self):
+    def base(self) -> WindowsBase:
         return self._base
 
     async def _wait_for_window(self) -> None: