From a4166f271c5727b51dc696bfafd045d9e8fcea92 Mon Sep 17 00:00:00 2001
From: Aevann <randomname42029@gmail.com>
Date: Fri, 17 May 2024 01:51:07 +0300
Subject: [PATCH] restore auto under_attack

---
 files/routes/wrappers.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/files/routes/wrappers.py b/files/routes/wrappers.py
index bdc17d5d1..beac1cb75 100644
--- a/files/routes/wrappers.py
+++ b/files/routes/wrappers.py
@@ -56,6 +56,20 @@ def calc_users():
 		g.loggedin_counter = len(loggedin)
 		g.loggedout_counter = len(loggedout)
 
+		if SITE == 'watchpeopledie.tv':
+			ddos_threshold = 3500
+		else:
+			ddos_threshold = 1000
+
+		if g.loggedin_counter + g.loggedout_counter > ddos_threshold or (SITE == 'watchpeopledie.tv' and get_setting('login_required')):
+			if not get_setting('under_attack'):
+				set_setting('under_attack', True)
+				set_security_level('under_attack')
+		else:
+			if get_setting('under_attack'):
+				set_setting('under_attack', False)
+				set_security_level('high')
+
 	return ''
 
 def get_logged_in_user():