Add stubs for redis Redis and Pipeline methods (#5415)

matthewhughes934 · web-flow · commit 7a9b4f93baa0 · 2021-05-13T09:53:26.000+02:00
diff --git a/stubs/redis/redis/client.pyi b/stubs/redis/redis/client.pyi
@@ -473,10 +473,10 @@ class Redis(Generic[_StrType]):
     def incr(self, name: _Key, amount: int = ...) -> int: ...
     def incrby(self, name: _Key, amount: int = ...) -> int: ...
     def incrbyfloat(self, name: _Key, amount: float = ...) -> float: ...
-    def keys(self, pattern=...): ...
-    def mget(self, keys, *args): ...
-    def mset(self, *args, **kwargs): ...
-    def msetnx(self, *args, **kwargs): ...
+    def keys(self, pattern: _Key = ...) -> List[_StrType]: ...
+    def mget(self, keys: Union[_Key, Iterable[_Key]], *args: _Key) -> List[Optional[_StrType]]: ...
+    def mset(self, mapping: Mapping[_Key, _Value]) -> Literal[True]: ...
+    def msetnx(self, mapping: Mapping[_Key, _Value]) -> bool: ...
     def move(self, name, db): ...
     def persist(self, name): ...
     def pexpire(self, name, time): ...
@@ -931,10 +931,10 @@ class Pipeline(Redis):
     def incr(self, name, amount=...) -> Pipeline: ...  # type: ignore [override]
     def incrby(self, name, amount=...) -> Pipeline: ...  # type: ignore [override]
     def incrbyfloat(self, name, amount=...) -> Pipeline: ...  # type: ignore [override]
-    def keys(self, pattern=...) -> Pipeline: ...  # type: ignore [override]
-    def mget(self, keys, *args) -> Pipeline: ...  # type: ignore [override]
-    def mset(self, *args, **kwargs) -> Pipeline: ...  # type: ignore [override]
-    def msetnx(self, *args, **kwargs) -> Pipeline: ...  # type: ignore [override]
+    def keys(self, pattern: _Key = ...) -> Pipeline: ...  # type: ignore [override]
+    def mget(self, keys: Union[_Key, Iterable[_Key]], *args: _Key) -> Pipeline: ...  # type: ignore [override]
+    def mset(self, mapping: Mapping[_Key, _Value]) -> Pipeline: ...  # type: ignore [override]
+    def msetnx(self, mapping: Mapping[_Key, _Value]) -> Pipeline: ...  # type: ignore [override]
     def move(self, name, db) -> Pipeline: ...  # type: ignore [override]
     def persist(self, name) -> Pipeline: ...  # type: ignore [override]
     def pexpire(self, name, time) -> Pipeline: ...  # type: ignore [override]
