Parcourir la source

fix(minor): code clean up

version-14
shadrak gurupnor il y a 3 ans
Parent
révision
b057a07259
1 fichiers modifiés avec 11 ajouts et 9 suppressions
  1. +11
    -9
      frappe/rate_limiter.py

+ 11
- 9
frappe/rate_limiter.py Voir le fichier

@@ -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


Chargement…
Annuler
Enregistrer