Added ClusterNode type

redis · Nov 20, 2022 · f887618 · f887618
1 parent b69fa00
commit f887618
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/redis/cluster.py b/redis/cluster.py
@@ -4,7 +4,7 @@
 import threading
 import time
 from collections import OrderedDict
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union, TypeVar
 
 from redis.backoff import default_backoff
 from redis.client import CaseInsensitiveDict, PubSub, Redis, parse_scan
@@ -38,6 +38,10 @@
     str_if_bytes,
 )
 
+TargetNodesT = TypeVar(
+    "TargetNodesT", str, "ClusterNode", List["ClusterNode"], Dict[Any, "ClusterNode"]
+)
+
 
 def get_node_name(host: str, port: Union[str, int]) -> str:
     return f"{host}:{port}"