mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-11-15 12:41:35 +00:00
75 lines
1.6 KiB
Python
75 lines
1.6 KiB
Python
import atexit
|
|
import random
|
|
import string
|
|
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
|
|
def generate_secret():
|
|
return ''.join(random.choices(string.ascii_uppercase + string.digits, k=7))
|
|
|
|
banned = {}
|
|
active = False
|
|
threshold = -1
|
|
secrets = [generate_secret(), generate_secret()]
|
|
|
|
def forgive_banned():
|
|
global banned
|
|
global threshold
|
|
|
|
clear_list = []
|
|
|
|
for ip in banned:
|
|
if banned[ip] <= 0:
|
|
clear_list.append(ip)
|
|
else:
|
|
banned[ip] = min(threshold, banned[ip]) - 1
|
|
|
|
for ip in clear_list:
|
|
del banned[ip]
|
|
|
|
def rotate_secrets():
|
|
global secrets
|
|
secrets[0] = secrets[1]
|
|
secrets[1] = generate_secret()
|
|
|
|
def secret_match(s):
|
|
return s in secrets
|
|
|
|
def get_current_secret():
|
|
return secrets[1]
|
|
|
|
def setup(violations_threshold=100):
|
|
global active
|
|
global threshold
|
|
|
|
active = True
|
|
threshold = violations_threshold
|
|
|
|
scheduler = BackgroundScheduler()
|
|
scheduler.add_job(func=forgive_banned, trigger="interval", minutes=30)
|
|
scheduler.add_job(func=rotate_secrets, trigger="interval", minutes=30)
|
|
|
|
scheduler.start()
|
|
|
|
# Shut down the scheduler when exiting the app
|
|
atexit.register(lambda: scheduler.shutdown())
|
|
|
|
|
|
def report(request_ip):
|
|
if active:
|
|
banned[request_ip] = banned.get(request_ip, 0)
|
|
banned[request_ip] += 1
|
|
|
|
|
|
def decrease(request_ip):
|
|
if banned[request_ip] > 0:
|
|
banned[request_ip] -= 1
|
|
|
|
|
|
def has_violation(request_ip):
|
|
return request_ip in banned and banned[request_ip] > 0
|
|
|
|
|
|
def is_banned(request_ip):
|
|
# More than X offences?
|
|
return active and banned.get(request_ip, 0) >= threshold
|