hz ne pomny

1 files changed, 9 insertions, 5 deletions

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