simple-login/app
1
1
Fork 0
mirror of https://github.com/simple-login/app.git synced 2025-02-25 00:03:03 +08:00
Code Issues Projects Releases Packages Wiki Activity

Fix: if redis is not configured do not enable rate limit (#2027)

Browse source
This commit is contained in:
Adrià Casajús 2024-02-05 14:53:01 +01:00 committed by GitHub
parent 30aaf118e7
commit 4a7c0293f8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
app/rate_limiter.py
View file

@ -23,9 +23,11 @@ def check_bucket_limit(
# Calculate current bucket time
bucket_id = int(datetime.utcnow().timestamp()) % bucket_seconds
bucket_lock_name = f"bl:{lock_name}:{bucket_id}"
if not lock_redis:
return
try:
value = lock_redis.incr(bucket_lock_name, bucket_seconds)
if value > max_hits:
raise werkzeug.exceptions.TooManyRequests()
except redis.exceptions.RedisError:
except (redis.exceptions.RedisError, AttributeError):
log.e("Cannot connect to redis")