fix antiflood megaozbeksystem

1 files changed, 14 insertions, 1 deletions

diff --git a/helpers.py b/helpers.py
index fde738b..d97236c 100644
--- a/helpers.py
+++ b/helpers.py
@@ -11,6 +11,12 @@ def is_blocked(user_id):
     current_time = time.time()
     if user_id and user_id in last_command_time:
         time_diff = current_time - last_command_time[user_id][0]
+        if user_id in last_error_time:
+            time_error_diff = current_time - last_error_time[user_id]
+        else:
+            last_error_time[user_id] = current_time
+            time_error_diff = 1984
+        
         if time_diff < 3:
             alo = "🚫 *файл не вошёл.* хватит так быстро слать свои сообщения.\n" \
                 f"подожди {round(3 - time_diff, 3)} секунд, брат😡😡"
@@ -18,9 +24,16 @@ def is_blocked(user_id):
             alo = "🚫 *файл не вошёл.* я ещё не сгенерировал прошлое сообщение.\n" \
                 f"подожди {round(10 - time_diff, 3)} секунд, чтобы я закрыл глаза " \
                 "на это da."
+        elif len(last_command_time[user_id][2]) > 1:
+            alo = "🚫 *файл не вошёл.* хватит так быстро слать свои сообщения."
         else:
             return "", False
-        return alo, True
+            
+        if time_error_diff < 2:
+            return "", True
+        else:
+            last_error_time[user_id] = current_time
+            return alo, True
     return "", False