Skip to content

Commit cd153d9

Browse files
committed
passed typecheck
1 parent 68aef10 commit cd153d9

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

src/paste/main.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,17 @@
5454
)
5555
app.state.limiter = limiter
5656

57-
async def rate_limit_exceeded_handler(request: Request, exc: Exception) -> Union[Response, Awaitable[Response]]:
58-
return _rate_limit_exceeded_handler(request, exc)
57+
58+
def rate_limit_exceeded_handler(request: Request, exc: Exception) -> Union[Response, Awaitable[Response]]:
59+
if isinstance(exc, RateLimitExceeded):
60+
return Response(
61+
content="Rate limit exceeded",
62+
status_code=429
63+
)
64+
return Response(
65+
content="An error occurred",
66+
status_code=500
67+
)
5968

6069
app.add_exception_handler(RateLimitExceeded, rate_limit_exceeded_handler)
6170

0 commit comments

Comments
 (0)