33import asyncio
44import logging
55from collections import defaultdict
6- from collections .abc import Awaitable
76from typing import (
87 TYPE_CHECKING ,
98 Any ,
3938from reactpy_django .utils import django_query_postprocessor , ensure_async , generate_obj_name , get_pk
4039
4140if TYPE_CHECKING :
42- from collections .abc import Sequence
41+ from collections .abc import Awaitable , Sequence
4342
4443 from channels_redis .core import RedisChannelLayer
4544 from django .contrib .auth .models import AbstractUser
@@ -124,9 +123,9 @@ def use_query(
124123 """
125124
126125 should_execute , set_should_execute = use_state (True )
127- data , set_data = use_state (cast (Inferred , None ))
126+ data , set_data = use_state (cast (" Inferred" , None ))
128127 loading , set_loading = use_state (True )
129- error , set_error = use_state (cast (Union [Exception , None ], None ))
128+ error , set_error = use_state (cast (" Union[Exception, None]" , None ))
130129 query_ref = use_ref (query )
131130 kwargs = kwargs or {}
132131 postprocessor_kwargs = postprocessor_kwargs or {}
@@ -140,14 +139,14 @@ async def execute_query() -> None:
140139 try :
141140 # Run the query
142141 query_async = cast (
143- Callable [..., Awaitable [Inferred ]], ensure_async (query , thread_sensitive = thread_sensitive )
142+ " Callable[..., Awaitable[Inferred]]" , ensure_async (query , thread_sensitive = thread_sensitive )
144143 )
145144 new_data = await query_async (** kwargs )
146145
147146 # Run the postprocessor
148147 if postprocessor :
149148 async_postprocessor = cast (
150- Callable [..., Awaitable [Any ]], ensure_async (postprocessor , thread_sensitive = thread_sensitive )
149+ " Callable[..., Awaitable[Any]]" , ensure_async (postprocessor , thread_sensitive = thread_sensitive )
151150 )
152151 new_data = await async_postprocessor (new_data , ** postprocessor_kwargs )
153152
@@ -227,7 +226,7 @@ def use_mutation(
227226 """
228227
229228 loading , set_loading = use_state (False )
230- error , set_error = use_state (cast (Union [Exception , None ], None ))
229+ error , set_error = use_state (cast (" Union[Exception, None]" , None ))
231230 async_task_refs = use_ref (set ())
232231
233232 # The main "running" function for `use_mutation`
@@ -289,7 +288,7 @@ def use_user() -> AbstractUser:
289288
290289
291290def use_user_data (
292- default_data : (None | dict [str , Callable [[], Any ] | Callable [[], Awaitable [Any ]] | Any ]) = None ,
291+ default_data : (dict [str , Callable [[], Any ] | Callable [[], Awaitable [Any ]] | Any ] | None ) = None ,
293292 save_default_data : bool = False ,
294293) -> UserData :
295294 """Get or set user data stored within the REACTPY_DATABASE.
@@ -441,7 +440,7 @@ async def logout(rerender: bool = True) -> None:
441440 return UseAuthTuple (login = login , logout = logout )
442441
443442
444- async def _get_user_data (user : AbstractUser , default_data : None | dict , save_default_data : bool ) -> dict | None :
443+ async def _get_user_data (user : AbstractUser , default_data : dict | None , save_default_data : bool ) -> dict | None :
445444 """The mutation function for `use_user_data`"""
446445 from reactpy_django .models import UserDataModel
447446
0 commit comments