added a type hint

amirreza8002 · amirreza8002 · commit 80bd01fc420f · 2024-09-16T02:57:15.000+03:30
diff --git a/django_valkey/async_cache/client/default.py b/django_valkey/async_cache/client/default.py
@@ -19,6 +19,7 @@
 
 if TYPE_CHECKING:
     from django_valkey.async_cache.cache import AsyncValkeyCache
+    from django_valkey.async_cache.pool import AsyncConnectionFactory, AsyncSentinelConnectionFactory
 
 
 async def glove_escape(s: str) -> str:
@@ -55,7 +56,7 @@ def __init__(
             "DJANGO_VALKEY_CONNECTION_FACTORY",
             "django_valkey.async_cache.pool.AsyncConnectionFactory",
         )
-        self.connection_factory = pool.get_connection_factory(
+        self.connection_factory: AsyncConnectionFactory | AsyncSentinelConnectionFactory | Any = pool.get_connection_factory(
             options=self._options, path=self._connection_factory
         )
 

Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,7 @@`
`19`	`19`
`20`	`20`	`if TYPE_CHECKING:`
`21`	`21`	`from django_valkey.async_cache.cache import AsyncValkeyCache`
	`22`	`+ from django_valkey.async_cache.pool import AsyncConnectionFactory, AsyncSentinelConnectionFactory`
`22`	`23`
`23`	`24`
`24`	`25`	`async def glove_escape(s: str) -> str:`
`@@ -55,7 +56,7 @@ def __init__(`
`55`	`56`	`"DJANGO_VALKEY_CONNECTION_FACTORY",`
`56`	`57`	`"django_valkey.async_cache.pool.AsyncConnectionFactory",`
`57`	`58`	`)`
`58`		`- self.connection_factory = pool.get_connection_factory(`
	`59`	`+ self.connection_factory: AsyncConnectionFactory \| AsyncSentinelConnectionFactory \| Any = pool.get_connection_factory(`
`59`	`60`	`options=self._options, path=self._connection_factory`
`60`	`61`	`)`
`61`	`62`