LibreTranslate/libretranslate/flood.py

48 lines
1.1 KiB
Python
Raw Permalink Normal View History

2023-03-09 21:09:04 +00:00
from libretranslate.storage import get_storage
2023-03-09 18:59:25 +00:00
2021-05-16 15:50:22 +00:00
active = False
threshold = -1
2021-05-18 03:41:02 +00:00
2021-11-24 17:41:12 +00:00
def forgive_banned():
global threshold
2021-05-16 15:50:22 +00:00
2021-11-24 17:41:12 +00:00
clear_list = []
2023-03-09 21:09:04 +00:00
s = get_storage()
banned = s.get_all_hash_int("banned")
2021-11-24 17:41:12 +00:00
for ip in banned:
if banned[ip] <= 0:
clear_list.append(ip)
else:
2023-03-10 04:07:12 +00:00
s.set_hash_int("banned", ip, min(threshold, banned[ip]) - 1)
2021-11-24 17:41:12 +00:00
for ip in clear_list:
2023-03-09 21:09:04 +00:00
s.del_hash("banned", ip)
2023-03-10 04:07:12 +00:00
def setup(args):
2021-05-16 15:50:22 +00:00
global active
global threshold
2023-03-10 04:07:12 +00:00
if args.req_flood_threshold > 0:
active = True
threshold = args.req_flood_threshold
2021-05-16 15:50:22 +00:00
def report(request_ip):
2021-05-16 15:52:39 +00:00
if active:
2023-03-09 21:09:04 +00:00
get_storage().inc_hash_int("banned", request_ip)
def decrease(request_ip):
2023-03-09 21:09:04 +00:00
s = get_storage()
if s.get_hash_int("banned", request_ip) > 0:
s.dec_hash_int("banned", request_ip)
def has_violation(request_ip):
2023-03-09 21:09:04 +00:00
s = get_storage()
return s.get_hash_int("banned", request_ip) > 0
2021-05-18 03:41:02 +00:00
2021-05-16 15:50:22 +00:00
def is_banned(request_ip):
2023-03-09 21:09:04 +00:00
s = get_storage()
2021-05-16 15:50:22 +00:00
# More than X offences?
2023-03-09 21:09:04 +00:00
return active and s.get_hash_int("banned", request_ip) >= threshold