From 481027e194a5bc979f0ec03484a251aaafd14908 Mon Sep 17 00:00:00 2001 From: Nikita Sobolev Date: Sat, 27 Aug 2022 18:13:46 +0300 Subject: [PATCH] redis: Improve `ClusterPubSub` types (#8627) --- stubs/redis/redis/cluster.pyi | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/stubs/redis/redis/cluster.pyi b/stubs/redis/redis/cluster.pyi index a21251f77..f5510eb9e 100644 --- a/stubs/redis/redis/cluster.pyi +++ b/stubs/redis/redis/cluster.pyi @@ -161,15 +161,22 @@ class NodesManager: def reset(self) -> None: ... class ClusterPubSub(PubSub): - node: Any - cluster: Any + node: ClusterNode | None + cluster: RedisCluster[Any] def __init__( - self, redis_cluster, node: Any | None = ..., host: Any | None = ..., port: Any | None = ..., **kwargs + self, + redis_cluster: RedisCluster[Any], + node: ClusterNode | None = ..., + host: str | None = ..., + port: int | None = ..., + **kwargs, ) -> None: ... - def set_pubsub_node(self, cluster, node: Any | None = ..., host: Any | None = ..., port: Any | None = ...) -> None: ... - def get_pubsub_node(self): ... + def set_pubsub_node( + self, cluster: RedisCluster[Any], node: ClusterNode | None = ..., host: str | None = ..., port: int | None = ... + ) -> None: ... + def get_pubsub_node(self) -> ClusterNode | None: ... def execute_command(self, *args, **kwargs) -> None: ... - def get_redis_connection(self): ... + def get_redis_connection(self) -> Redis[Any] | None: ... class ClusterPipeline(RedisCluster[_StrType], Generic[_StrType]): command_stack: list[Incomplete]