vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix(minor): code clean up

Browse source
This commit is contained in:
shadrak gurupnor 2021-09-15 16:07:03 +05:30
parent 851778e561
commit b057a07259
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
frappe/rate_limiter.py
View file

@ -109,18 +109,20 @@ def rate_limit(key: str = None, limit: Union[int, Callable] = 5, seconds: int =
_limit = limit() if callable(limit) else limit
ip = frappe.local.request_ip
ip = frappe.local.request_ip if ip_based is True else None
if key is None and ip_based is False:
frappe.throw(_('Either key or IP flag is required.'))
elif key is None:
identity = ip
elif ip_based is False:
identity = frappe.form_dict[key]
else:
user_key=frappe.form_dict[key]
user_key = frappe.form_dict[key] if key else None
identity = None
if key and ip_based:
identity = ":".join([ip, user_key])
identity = identity or ip or user_key
if not identity:
frappe.throw(_('Either key or IP flag is required.'))
cache_key = f"rl:{frappe.form_dict.cmd}:{identity}"
value = frappe.cache().get(cache_key) or 0