Add annotations for some redis client methods (#5559)

matthewhughes934 · web-flow · commit 2a9f081abbf0 · 2021-05-31T22:01:11.000+02:00
Add annotation for move(), persist(), pexpire(), and expireat().
diff --git a/stubs/redis/redis/client.pyi b/stubs/redis/redis/client.pyi
@@ -1,4 +1,4 @@
-from datetime import timedelta
+from datetime import datetime, timedelta
 from typing import (
     Any,
     Callable,
@@ -477,10 +477,10 @@ class Redis(Generic[_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): ...
-    def pexpireat(self, name, when): ...
+    def move(self, name: _Key, db: int) -> bool: ...
+    def persist(self, name: _Key) -> bool: ...
+    def pexpire(self, name: _Key, time: Union[int, timedelta]) -> Literal[1, 0]: ...
+    def pexpireat(self, name: _Key, when: Union[int, datetime]) -> Literal[1, 0]: ...
     def psetex(self, name, time_ms, value): ...
     def pttl(self, name): ...
     def randomkey(self): ...
@@ -938,10 +938,10 @@ class Pipeline(Redis):
     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]
-    def pexpireat(self, name, when) -> Pipeline: ...  # type: ignore [override]
+    def move(self, name: _Key, db: int) -> Pipeline: ...  # type: ignore [override]
+    def persist(self, name: _Key) -> Pipeline: ...  # type: ignore [override]
+    def pexpire(self, name: _Key, time: Union[int, timedelta]) -> Pipeline: ...  # type: ignore [override]
+    def pexpireat(self, name: _Key, when: Union[int, datetime]) -> Pipeline: ...  # type: ignore [override]
     def psetex(self, name, time_ms, value) -> Pipeline: ...  # type: ignore [override]
     def pttl(self, name) -> Pipeline: ...  # type: ignore [override]
     def randomkey(self) -> Pipeline: ...  # type: ignore [override]
