From 82cd7d22c629a26d9d595045a373aebf688bb2e1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=81rni=20M=C3=A1r=20J=C3=B3nsson?= <arnimarj@gmail.com>
Date: Tue, 23 Mar 2021 02:08:12 +0000
Subject: [PATCH] add stubs for redis.lock (#5105)

---
 stubs/redis/redis/lock.pyi | 41 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 stubs/redis/redis/lock.pyi

diff --git a/stubs/redis/redis/lock.pyi b/stubs/redis/redis/lock.pyi
new file mode 100644
index 000000000..0e6090a16
--- /dev/null
+++ b/stubs/redis/redis/lock.pyi
@@ -0,0 +1,41 @@
+from types import TracebackType
+from typing import Optional, Text, Type, Union
+
+from redis.client import Redis
+
+_TokenValue = Union[bytes, Text]
+
+class Lock:
+    def __init__(
+        self,
+        redis: Redis,
+        name: str,
+        timeout: Union[None, int, float] = ...,
+        sleep: float = ...,
+        blocking: bool = ...,
+        blocking_timeout: Optional[bool] = ...,
+        thread_local: bool = ...,
+    ) -> None: ...
+    def register_scripts(self) -> None: ...
+    def __enter__(self) -> Lock: ...
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_value: Optional[BaseException],
+        traceback: Optional[TracebackType],
+    ) -> Optional[bool]: ...
+    def acquire(
+        self,
+        blocking: Optional[bool] = ...,
+        blocking_timeout: Union[None, int, float] = ...,
+        token: Optional[_TokenValue] = ...,
+    ) -> bool: ...
+    def do_acquire(self, token: _TokenValue) -> bool: ...
+    def locked(self) -> bool: ...
+    def owned(self) -> bool: ...
+    def release(self) -> None: ...
+    def do_release(self, expected_token: _TokenValue) -> None: ...
+    def extend(self, additional_time: Union[int, float], replace_ttl: bool = ...) -> bool: ...
+    def do_extend(self, additional_time: Union[int, float], replace_ttl: bool) -> bool: ...
+    def reacquire(self) -> bool: ...
+    def do_reacquire(self) -> bool: ...