velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

ratelimits.handler: add check for when rule is none (404s)

This commit is contained in:
Luna Mendes 2018-11-05 03:15:46 -03:00
parent c200ab8707
commit 8ffa14d934
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
litecord/ratelimits/handler.py
View File

@ -58,6 +58,11 @@ async def ratelimit_handler():
""" """
rule = request.url_rule rule = request.url_rule
if rule is None:
return await _handle_global(
app.ratelimiter.global_bucket
)
# rule.endpoint is composed of '<blueprint>.<function>' # rule.endpoint is composed of '<blueprint>.<function>'
# and so we can use that to make routes with different # and so we can use that to make routes with different
# methods have different ratelimits # methods have different ratelimits
@ -73,5 +78,6 @@ async def ratelimit_handler():
ratelimit = app.ratelimiter.get_ratelimit(rule_path) ratelimit = app.ratelimiter.get_ratelimit(rule_path)
await _handle_specific(ratelimit) await _handle_specific(ratelimit)
except KeyError: except KeyError:
ratelimit = app.ratelimiter.global_bucket await _handle_global(
await _handle_global(ratelimit) app.ratelimiter.global_bucket
)