Adjust typing for PageMethod

scrapy_playwright/page.py

@@ -1,4 +1,6 @@
-from typing import Any, Callable
+from typing import Any, Callable, Union
+
+from playwright.async_api import Page
 
 
 __all__ = ["PageMethod"]
@@ -8,10 +10,12 @@ class PageMethod:
     """
     Represents a method to be called (and awaited if necessary) on a
     Playwright page, such as "click", "screenshot", "evaluate", etc.
+
+    If a callable is received, it will be called with the page as its only argument.
     """
 
-    def __init__(self, method: str | Callable, *args, **kwargs) -> None:
-        self.method: str | Callable = method
+    def __init__(self, method: Union[str, Callable[[Page], Any]], *args, **kwargs) -> None:
+        self.method: Union[str, Callable[[Page], Any]] = method
         self.args: tuple = args
         self.kwargs: dict = kwargs
         self.result: Any = None