Added ClusterNode type

redis · Nov 20, 2022 · 9cfa7ca · 9cfa7ca
1 parent b69fa00
commit 9cfa7ca
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, TypeVar, Union
 
 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}"