diff --git a/third_party/2and3/redis/client.pyi b/third_party/2and3/redis/client.pyi index f41676177..cd6559644 100644 --- a/third_party/2and3/redis/client.pyi +++ b/third_party/2and3/redis/client.pyi @@ -1,4 +1,7 @@ -from typing import Any +from datetime import timedelta +from typing import Any, Text, Optional, Mapping, Union + +from .connection import ConnectionPool SYM_EMPTY: Any @@ -31,16 +34,37 @@ def parse_hscan(response, **options): ... def parse_zscan(response, **options): ... def parse_slowlog_get(response, **options): ... -class StrictRedis: +class Redis(object): RESPONSE_CALLBACKS: Any @classmethod def from_url(cls, url, db=..., **kwargs): ... connection_pool: Any response_callbacks: Any - def __init__(self, host=..., port=..., db=..., password=..., socket_timeout=..., socket_connect_timeout=..., - socket_keepalive=..., socket_keepalive_options=..., connection_pool=..., unix_socket_path=..., encoding=..., - encoding_errors=..., charset=..., errors=..., decode_responses=..., retry_on_timeout=..., ssl=..., - ssl_keyfile=..., ssl_certfile=..., ssl_cert_reqs=..., ssl_ca_certs=...) -> None: ... + def __init__( + self, + host: Text = ..., + port: int = ..., + db: int = ..., + password: Optional[Text] = ..., + socket_timeout: Optional[float] = ..., + socket_connect_timeout: Optional[float] = ..., + socket_keepalive: Optional[bool] = ..., + socket_keepalive_options: Optional[Mapping[str, Union[int, str]]] = ..., + connection_pool: Optional[ConnectionPool] = ..., + unix_socket_path: Optional[Text] = ..., + encoding: Text = ..., + encoding_errors: Text = ..., + charset: Optional[Text] = ..., + errors: Optional[Text] = ..., + decode_responses: bool = ..., + retry_on_timeout: bool = ..., + ssl: bool = ..., + ssl_keyfile: Optional[Text] = ..., + ssl_certfile: Optional[Text] = ..., + ssl_cert_reqs: Optional[Union[str, int]] = ..., + ssl_ca_certs: Optional[Text] = ..., + max_connections: Optional[int] = ..., + ) -> None: ... def set_response_callback(self, command, callback): ... def pipeline(self, transaction=..., shard_hint=...): ... def transaction(self, func, *watches, **kwargs): ... @@ -93,9 +117,9 @@ class StrictRedis: def dump(self, name): ... def exists(self, name): ... __contains__: Any - def expire(self, name, time): ... + def expire(self, name: Text, time: Union[int, timedelta]) -> bool: ... def expireat(self, name, when): ... - def get(self, name): ... + def get(self, name: Text) -> Optional[bytes]: ... def __getitem__(self, name): ... def getbit(self, name, offset): ... def getrange(self, key, start, end): ... @@ -117,7 +141,15 @@ class StrictRedis: def rename(self, src, dst): ... def renamenx(self, src, dst): ... def restore(self, name, ttl, value): ... - def set(self, name, value, ex=..., px=..., nx=..., xx=...): ... + def set( + self, + name: Text, + value: Any, + ex: Union[None, int, timedelta] = ..., + px: Union[None, int, timedelta] = ..., + nx: bool = ..., + xx: bool = ..., + ) -> Optional[bool]: ... def __setitem__(self, name, value): ... def setbit(self, name, offset, value): ... def setex(self, name, time, value): ... @@ -169,7 +201,7 @@ class StrictRedis: def srem(self, name, *values): ... def sunion(self, keys, *args): ... def sunionstore(self, dest, keys, *args): ... - def zadd(self, name, *args, **kwargs): ... + def zadd(self, name, mapping, nx: bool = ..., xx: bool = ..., ch: bool = ..., incr: bool = ...): ... def zcard(self, name): ... def zcount(self, name, min, max): ... def zincrby(self, name, value, amount=...): ... @@ -213,12 +245,7 @@ class StrictRedis: def script_load(self, script): ... def register_script(self, script): ... -class Redis(StrictRedis): - RESPONSE_CALLBACKS: Any - def pipeline(self, transaction=..., shard_hint=...): ... - def setex(self, name, value, time): ... - def lrem(self, name, value, num=...): ... - def zadd(self, name, *args, **kwargs): ... +StrictRedis = Redis class PubSub: PUBLISH_MESSAGE_TYPES: Any