Fixed format issues

Author: arsh-anwr

backend/app/api/routes/users.py

@@ -33,8 +33,12 @@
 
 @router.get(
     "/",
-    dependencies=[Depends(RateLimiter(limit=10, window_seconds=60, key_policy=KeyStrategyName.IP)),
-    Depends(get_current_active_superuser)],
+    dependencies=[
+        Depends(
+            RateLimiter(limit=10, window_seconds=60, key_policy=KeyStrategyName.IP)
+        ),
+        Depends(get_current_active_superuser),
+    ],
     response_model=UsersPublic,
 )
 def read_users(session: SessionDep, skip: int = 0, limit: int = 100) -> Any:

backend/app/core/config.py

@@ -43,9 +43,7 @@ class Settings(BaseSettings):
 
     # Correct Redis default inside Docker Compose
     REDIS_URL: str = ""
-    RATE_LIMITER_STRATEGY: Literal[
-        "none", "sliding_window"
-    ] = "none"
+    RATE_LIMITER_STRATEGY: Literal["none", "sliding_window"] = "none"
     RATE_LIMIT_FAIL_OPEN: bool = True
 
     @computed_field  # type: ignore[prop-decorator]

backend/app/core/rate_limiter/key_strategy/key_strategy_registry.py

@@ -4,7 +4,9 @@
 from app.core.rate_limiter.key_strategy.key_strategy_enum import KeyStrategyName
 
 
-def get_key_strategy(name: KeyStrategyName, header_name: str | None = None) -> KeyStrategy:
+def get_key_strategy(
+    name: KeyStrategyName, header_name: str | None = None
+) -> KeyStrategy:
     if name == KeyStrategyName.IP:
         return IPKeyStrategy()
 

backend/app/core/rate_limiter/rate_limiter.py

@@ -7,8 +7,14 @@
 
 logger = logging.getLogger(__name__)
 
+
 class RateLimiter:
-    def __init__(self, limit: int, window_seconds: int, key_policy: KeyStrategyName = KeyStrategyName.IP):
+    def __init__(
+        self,
+        limit: int,
+        window_seconds: int,
+        key_policy: KeyStrategyName = KeyStrategyName.IP,
+    ):
         self.limit = limit
         self.window_seconds = window_seconds
         self.key_policy = key_policy
@@ -28,19 +34,19 @@ async def __call__(self, request: Request) -> None:
         retry_after = None
         try:
             allowed, retry_after = await rate_limiter.allow_request(
-                        key, self.limit, self.window_seconds
-                )
+                key, self.limit, self.window_seconds
+            )
         except Exception:
-                logger.exception("Error invoking rate limiter")
-                if rate_limiter.get_fail_open():
-                    raise HTTPException(
-                        status_code=503,
-                        detail={"detail": "Rate limiter unavailable"},
-                    )
+            logger.exception("Error invoking rate limiter")
+            if rate_limiter.get_fail_open():
+                raise HTTPException(
+                    status_code=503,
+                    detail={"detail": "Rate limiter unavailable"},
+                )
 
         if not allowed:
             raise HTTPException(
                 status_code=429,
                 detail=f"Too Many Requests. Retry after {retry_after}s",
-                headers={"Retry-After": str(retry_after)}
+                headers={"Retry-After": str(retry_after)},
             )

backend/app/core/rate_limiter/rate_limiting_algorithm/base.py

@@ -7,7 +7,9 @@ class BaseRateLimiter(ABC):
     """Interface for pluggable rate limiter strategies."""
 
     @abstractmethod
-    async def allow_request(self, key: str, limit: int, window_seconds: int, member_id: str | None = None) -> tuple[bool, int | None]:
+    async def allow_request(
+        self, key: str, limit: int, window_seconds: int, member_id: str | None = None
+    ) -> tuple[bool, int | None]:
         """
         Return (allowed: bool, retry_after_seconds: Optional[int]).
         If allowed True -> retry_after_seconds is None.

backend/app/core/rate_limiter/rate_limiting_algorithm/registry.py

@@ -1,4 +1,3 @@
-
 import redis.asyncio as redis
 
 from app.core.rate_limiter.rate_limiting_algorithm.base import BaseRateLimiter
@@ -7,7 +6,9 @@
 )
 
 
-def get_rate_limiter(strategy: str | None, redis_url: str | None, fail_open: bool | None) -> BaseRateLimiter | None:
+def get_rate_limiter(
+    strategy: str | None, redis_url: str | None, fail_open: bool | None
+) -> BaseRateLimiter | None:
     """
     Factory: returns an instance of BaseRateLimiter or None (if disabled).
     """
@@ -17,7 +18,7 @@ def get_rate_limiter(strategy: str | None, redis_url: str | None, fail_open: boo
     if not redis_url:
         return None
 
-    rc: redis.Redis = redis.from_url(redis_url, encoding="utf-8", decode_responses=True) # type: ignore[no-untyped-call]
+    rc: redis.Redis = redis.from_url(redis_url, encoding="utf-8", decode_responses=True)  # type: ignore[no-untyped-call]
     st = strategy.lower()
     if st == "sliding_window" or st == "sliding-window":
         return SlidingWindowRateLimiter(rc, fail_open or False)

backend/app/core/rate_limiter/rate_limiting_algorithm/sliding_window.py

@@ -9,8 +9,7 @@
 logger = logging.getLogger(__name__)
 
 SCRIPT_PATH = (
-    Path(__file__).resolve()
-    .parents[3]
+    Path(__file__).resolve().parents[3]
     / "alembic"
     / "rate_limiting_algorithms"
     / "sliding_window.lua"
@@ -31,13 +30,8 @@ async def load_script(self) -> str | None:
         return self.lua_script
 
     async def allow_request(
-        self,
-        key: str,
-        limit: int,
-        window_seconds: int,
-        member_id: str | None = None
+        self, key: str, limit: int, window_seconds: int, member_id: str | None = None
     ) -> tuple[bool, int | None]:
-
         now_ms = int(time.time() * 1000)
         window_ms = window_seconds * 1000
         member = member_id or f"{now_ms}"
@@ -46,14 +40,8 @@ async def allow_request(
             sha = await self.load_script()
             if sha is None:
                 raise Exception
-            res = await self.redis.evalsha(# type: ignore[misc]
-                sha,
-                1,
-                key,
-                now_ms,
-                window_ms,
-                limit,
-                member
+            res = await self.redis.evalsha(  # type: ignore[misc]
+                sha, 1, key, now_ms, window_ms, limit, member
             )
         except Exception:
             logger.exception("Redis error; failing open")

backend/app/main.py

@@ -33,7 +33,11 @@ def custom_generate_unique_id(route: APIRoute) -> str:
 
 # Set rate Limiting
 if settings.rate_limit_enabled:
-    rate_limiter = get_rate_limiter(settings.RATE_LIMITER_STRATEGY, settings.REDIS_URL, settings.RATE_LIMIT_FAIL_OPEN)
+    rate_limiter = get_rate_limiter(
+        settings.RATE_LIMITER_STRATEGY,
+        settings.REDIS_URL,
+        settings.RATE_LIMIT_FAIL_OPEN,
+    )
     app.state.rate_limiter = rate_limiter
 
 app.include_router(api_router, prefix=settings.API_V1_STR)