muffin-rest 13.0.1__tar.gz → 13.0.2__tar.gz

{muffin_rest-13.0.1 → muffin_rest-13.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: muffin-rest
-Version: 13.0.1
+Version: 13.0.2
 Summary: The package provides enhanced support for writing REST APIs with Muffin framework
 License: MIT
 Keywords: rest,api,muffin,asgi,asyncio,trio

{muffin_rest-13.0.1 → muffin_rest-13.0.2}/muffin_rest/limits.py

@@ -1,4 +1,5 @@
 import abc
+from asyncio import Lock
 from time import time
 from typing import Any
 
@@ -13,13 +14,19 @@ class RateLimiter(abc.ABC):
             limit (int): The limit of requests.
             period (int): The period of time in seconds.
         """
+        self.lock = Lock()
         self.limit = limit
         self.period = period
 
-    @abc.abstractmethod
     async def check(self, key: str) -> bool:
         """Check the request."""
-        raise NotImplementedError
+        async with self.lock:
+            return await self._check(key)
+
+    @abc.abstractmethod
+    async def _check(self, key: str) -> bool:
+        """Check the request."""
+        raise NotImplementedError("Subclasses must implement this method.")
 
 
 class MemoryRateLimiter(RateLimiter):
@@ -29,7 +36,7 @@ class MemoryRateLimiter(RateLimiter):
         super().__init__(limit, **opts)
         self.storage: dict[Any, tuple[float, int]] = {}
 
-    async def check(self, key: str) -> bool:
+    async def _check(self, key: str) -> bool:
         """Check the request."""
         now = time()
         storage = self.storage
@@ -53,8 +60,6 @@ class MemoryRateLimiter(RateLimiter):
 class RedisRateLimiter(RateLimiter):
     """Redis rate limiter."""
 
-    # TODO: Asyncio lock
-
     def __init__(self, limit: int, *, redis, **opts):
         """Initialize the rate limiter.
 
@@ -66,7 +71,7 @@ class RedisRateLimiter(RateLimiter):
         super().__init__(limit, **opts)
         self.redis = redis
 
-    async def check(self, key: str) -> bool:
+    async def _check(self, key: str) -> bool:
         """Check the request."""
         value = await self.redis.get(key)
         if value is None:

{muffin_rest-13.0.1 → muffin_rest-13.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "muffin-rest"
-version = "13.0.1"
+version = "13.0.2"
 description = "The package provides enhanced support for writing REST APIs with Muffin framework"
 readme = "README.md"
 homepage = "https://github.com/klen/muffin-rest"