fix type hints

Author: Archmonger

src/reactpy_django/pyscript/layout_handler.py

@@ -1,10 +1,12 @@
+# mypy: disable-error-code=attr-defined
 import asyncio
-from typing import Coroutine
+from typing import Callable
 
 import js
 from jsonpointer import set_pointer
 from pyodide.ffi.wrappers import add_event_listener
 from reactpy.core.layout import Layout
+from reactpy.types import ComponentType
 
 
 class ReactPyLayoutHandler:
@@ -100,9 +102,9 @@ def delete_old_workspaces():
                     f"Warning: Could not auto delete PyScript workspace {workspace_name}"
                 )
 
-    async def run(self, workspace_function: Coroutine):
+    async def run(self, workspace_function: Callable[[], ComponentType]):
         self.delete_old_workspaces()
-        root_model = {}
+        root_model: dict = {}
 
         async with Layout(workspace_function()) as layout:
             while True:

src/reactpy_django/utils.py

@@ -11,7 +11,7 @@
 from fnmatch import fnmatch
 from importlib import import_module
 from pathlib import Path
-from typing import Any, Callable, Sequence
+from typing import Any, Callable, Mapping, Sequence
 from uuid import UUID, uuid4
 
 import jsonpointer
@@ -407,7 +407,11 @@ def strtobool(val):
 
 
 def prerender_component(
-    user_component: ComponentConstructor, args, kwargs, uuid, request: HttpRequest
+    user_component: ComponentConstructor,
+    args: Sequence,
+    kwargs: Mapping,
+    uuid: str | UUID,
+    request: HttpRequest,
 ) -> str:
     """Prerenders a ReactPy component and returns the HTML string."""
     search = request.GET.urlencode()
@@ -432,12 +436,12 @@ def prerender_component(
 
 
 def vdom_or_component_to_string(
-    vdom_or_component: Any, request: HttpRequest | None = None, uuid: UUID | None = None
+    vdom_or_component: Any, request: HttpRequest | None = None, uuid: str | None = None
 ) -> str:
     """Converts a VdomDict or component to an HTML string. If a string is provided instead, it will be
     automatically returned."""
     if isinstance(vdom_or_component, dict):
-        return vdom_to_html(vdom_or_component)
+        return vdom_to_html(vdom_or_component)  # type: ignore
 
     if hasattr(vdom_or_component, "render"):
         if not request: