Skip to content

Commit bf5eb6c

Browse files
author
Chats
committed
Cleanup Rate Limit middleware
1 parent 00175e2 commit bf5eb6c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/utils/middleware/rate_limit.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,13 @@
44

55

66
async def rate_limit(request: Request, call_next):
7-
regex_patterns = [r"Helldivers%20Companion.*", r"Heckdiver.*"]
8-
client_id = request.headers.get("User-Agent", request.headers.get("X-Super-Client"))
7+
regex_patterns = [r"Helldivers%20Companion.*", r"Heckdivers.*"]
8+
headers = request.headers
9+
client_id = headers.get("User-Agent", headers.get("X-Super-Client", ""))
910

1011
for pattern in regex_patterns:
1112
if re.search(pattern, client_id):
1213
return PTR(status_code=429, content="", headers={"content-type": ""})
14+
# return None, return "", return - All Error Out
1315

1416
return await call_next(request)

0 commit comments

Comments
 (0)