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

Apply suggestions from code review

Browse Source 
Co-authored-by: Reuben Thomas-Davis <reubenestd@gmail.com>
This commit is contained in:
Laurent Savaete
2023-08-21 16:44:19 +02:00
committed by GitHub
parent 90f2ed3a32
commit 93046a00c9
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
slowapi/wrappers.py
View File

@@ -29,6 +29,7 @@ class Limit(object):
self.methods = methods
self.error_message = error_message
self.exempt_when = exempt_when
self._exempt_when_takes_request = len(inspect.signature(self.exempt_when).parameters) == 1
self.cost = cost
self.override_defaults = override_defaults
@@ -43,9 +44,7 @@ class Limit(object):
"""
if self.exempt_when is None:
return False
params = inspect.signature(self.exempt_when).parameters
param_len = len(params)
if param_len == 1 and request:
if self._exempt_when_takes_request and request:
return self.exempt_when(request)
return self.exempt_when()