Add ratelimit stubs

Author: Tatsh

stubs/ratelimit/METADATA.toml

@@ -0,0 +1,2 @@
+version = "2.2.*"
+upstream_repository = "https://github.com/tomasbasham/ratelimit"

stubs/ratelimit/ratelimit/__init__.pyi

@@ -0,0 +1,7 @@
+from ratelimit.decorators import RateLimitDecorator, sleep_and_retry
+from ratelimit.exception import RateLimitException
+
+limits = RateLimitDecorator
+rate_limited = RateLimitDecorator
+
+__all__ = ["RateLimitException", "limits", "rate_limited", "sleep_and_retry"]

stubs/ratelimit/ratelimit/decorators.pyi

@@ -0,0 +1,16 @@
+from collections.abc import Callable
+from typing import TypeVar
+from typing_extensions import ParamSpec
+
+_P = ParamSpec("_P")
+_T = TypeVar("_T")
+
+now: Callable[[], float] = ...
+
+class RateLimitDecorator:
+    def __init__(
+        self, calls: int = ..., period: float = ..., clock: Callable[[], float] = ..., raise_on_limit: bool = ...
+    ) -> None: ...
+    def __call__(self, func: Callable[_P, _T]) -> Callable[_P, _T]: ...
+
+def sleep_and_retry(func: Callable[_P, _T]) -> Callable[_P, _T]: ...

stubs/ratelimit/ratelimit/exception.pyi

@@ -0,0 +1,3 @@
+class RateLimitException(Exception):
+    period_remaining: float
+    def __init__(self, message: str, period_remaining: float) -> None: ...