fix: get rid of private method

Author: drainpixie

app.py

@@ -1,7 +1,5 @@
 """A simple FastAPI app that serves a REST API for proselint."""
 
-from typing import TYPE_CHECKING, cast
-
 from fastapi import FastAPI, HTTPException, Request, Response, status
 from fastapi.middleware.cors import CORSMiddleware
 from slowapi import Limiter
@@ -13,9 +11,6 @@
 from proselint.registry import CheckRegistry
 from proselint.tools import LintFile, LintResult
 
-if TYPE_CHECKING:
-    from slowapi import ViewRateLimit
-
 
 def _lint(input_text: str) -> list[LintResult]:
     return LintFile(content=input_text, source="<api>").lint()
@@ -35,21 +30,20 @@ def _lint(input_text: str) -> list[LintResult]:
 CheckRegistry().register_many(__register__)
 
 
-# NOTE: We don't use slowapi's handler due to the error key name.
 @app.exception_handler(RateLimitExceeded)
 def rate_limit_exceeded_handler(
-    request: Request,
+    _: Request,
     exc: RateLimitExceeded,
 ) -> Response:
     """Middleware to handle exceeded ratelimits."""
-    response = JSONResponse(
-        {"detail": f"rate limit exceeded: {exc.detail}"},
+    return JSONResponse(
+        {
+            "status": "error",
+            "message": "rate limit exceeded",
+            "limit": str(exc.detail),
+        },
         status_code=status.HTTP_429_TOO_MANY_REQUESTS,
-    )
-
-    rate_limit = cast("ViewRateLimit", request.state.view_rate_limit)
-    return limiter._inject_headers(  # noqa: SLF001
-        response, rate_limit
+        headers=getattr(exc, "headers", None),
     )
 
 

typings/slowapi/__init__.pyi

@@ -1,11 +1,5 @@
-from typing import Any, Callable, Protocol, TypeVar
+from typing import Any, Callable, TypeVar
 from starlette.requests import Request
-from starlette.responses import Response
-
-class ViewRateLimit(Protocol):
-    limit: str
-    reset: int
-    remaining: int
 
 F = TypeVar("F", bound=Callable[..., Any])
 KeyFunc = Callable[[Request], str]
@@ -14,9 +8,3 @@ class Limiter:
     def __init__(self, key_func: KeyFunc) -> None: ...
 
     def limit(self, limit: str) -> Callable[[F], F]: ...
-
-    def _inject_headers(
-        self,
-        response: Response,
-        rate_limit: ViewRateLimit,
-    ) -> Response: ...