python/slowapi
1
0
Fork 0
mirror of https://github.com/laurentS/slowapi.git synced 2026-03-13 09:10:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace assert with raise ...

Browse Source
This commit is contained in:
Marat Sarbasov
2022-01-19 00:51:52 +03:00
parent 59d2cdd9de
commit 5e78adde2f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
slowapi/wrappers.py
View File

@@ -82,8 +82,9 @@ class LimitGroup(object):
if "key" in inspect.signature(self.__limit_provider).parameters.keys():
assert (
"request" in inspect.signature(self.key_function).parameters.keys()
)
assert self.request
), "If limit provider function depends on `key` argument, key function must accept `request`"
if self.request is None:
raise Exception("`request` object can't be None")
limit_raw = self.__limit_provider(self.key_function(self.request))
else:
limit_raw = self.__limit_provider()