tortle 0.1.0__tar.gz

tortle-0.1.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.3
+Name: tortle
+Version: 0.1.0
+Summary: Add your description here
+Author: Abdul Karim
+Author-email: Abdul Karim <karim.develops@gmail.com>
+Requires-Dist: fastapi[standard]>=0.141.1
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+

tortle-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[project]
+name = "tortle"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.14"
+dependencies = ["fastapi[standard]>=0.141.1"]
+
+[[project.authors]]
+name = "Abdul Karim"
+email = "karim.develops@gmail.com"
+
+[project.scripts]
+tortle = "tortle:main"
+
+[build-system]
+requires = ["uv_build>=0.12.5,<0.13.0"]
+build-backend = "uv_build"

tortle-0.1.0/pyproject.toml.orig

@@ -0,0 +1,15 @@
+[project]
+name = "tortle"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+authors = [{ name = "Abdul Karim", email = "karim.develops@gmail.com" }]
+requires-python = ">=3.14"
+dependencies = ["fastapi[standard]>=0.141.1"]
+
+[project.scripts]
+tortle = "tortle:main"
+
+[build-system]
+requires = ["uv_build>=0.12.5,<0.13.0"]
+build-backend = "uv_build"

tortle-0.1.0/src/tortle/__init__.py

@@ -0,0 +1,3 @@
+from .slidinglog import SlidingLogLimiter
+
+__all__ = ["SlidingLogLimiter"]

tortle-0.1.0/src/tortle/slidinglog.py

@@ -0,0 +1,58 @@
+import time
+
+from fastapi import Request, status
+from fastapi.responses import JSONResponse
+from starlette.middleware.base import BaseHTTPMiddleware
+
+
+class Counter:
+    def __init__(self) -> None:
+        self.value = 0
+        self.start_time = time.time()
+
+    def increment(self):
+        self.value += 1
+
+    def reset(self):
+        self.value = 0
+        self.start_time = time.time()
+
+
+class SlidingLogLimiter(BaseHTTPMiddleware):
+    def __init__(self, app, limit, window):
+        super().__init__(app)
+        self.limit = limit
+        self.window = window
+        self.counters: dict[str, Counter] = {}
+
+    async def dispatch(self, request: Request, call_next):
+
+        if not request.client:
+            return JSONResponse(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                content={"detail": "host doesn't exist"},
+            )
+
+        ip = request.client.host
+
+        if ip not in self.counters:
+            counter = Counter()
+            self.counters[ip] = counter
+
+        self.counters[ip].increment()
+
+        current_time = time.time()
+        elapsed_seconds = current_time - self.counters[ip].start_time
+
+        if elapsed_seconds > self.window:
+            self.counters[ip].reset()
+
+        if self.counters[ip].value > self.limit:
+            return JSONResponse(
+                status_code=status.HTTP_429_TOO_MANY_REQUESTS,
+                content={"detail": "Too many reqs"},
+                headers={"Retry-After": f"{60 - elapsed_seconds}"},
+            )
+
+        response = await call_next(request)
+        return response