Skip to content

Commit 481027e

Browse files
authored
redis: Improve ClusterPubSub types (#8627)
1 parent d1d9b98 commit 481027e

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

stubs/redis/redis/cluster.pyi

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -161,15 +161,22 @@ class NodesManager:
161161
def reset(self) -> None: ...
162162

163163
class ClusterPubSub(PubSub):
164-
node: Any
165-
cluster: Any
164+
node: ClusterNode | None
165+
cluster: RedisCluster[Any]
166166
def __init__(
167-
self, redis_cluster, node: Any | None = ..., host: Any | None = ..., port: Any | None = ..., **kwargs
167+
self,
168+
redis_cluster: RedisCluster[Any],
169+
node: ClusterNode | None = ...,
170+
host: str | None = ...,
171+
port: int | None = ...,
172+
**kwargs,
173+
) -> None: ...
174+
def set_pubsub_node(
175+
self, cluster: RedisCluster[Any], node: ClusterNode | None = ..., host: str | None = ..., port: int | None = ...
168176
) -> None: ...
169-
def set_pubsub_node(self, cluster, node: Any | None = ..., host: Any | None = ..., port: Any | None = ...) -> None: ...
170-
def get_pubsub_node(self): ...
177+
def get_pubsub_node(self) -> ClusterNode | None: ...
171178
def execute_command(self, *args, **kwargs) -> None: ...
172-
def get_redis_connection(self): ...
179+
def get_redis_connection(self) -> Redis[Any] | None: ...
173180

174181
class ClusterPipeline(RedisCluster[_StrType], Generic[_StrType]):
175182
command_stack: list[Incomplete]

0 commit comments

Comments
 (0)