темп бан да 2

3 files changed, 7 insertions, 5 deletions

diff --git a/commands.py b/commands.py
index 3afe666..ac34d1c 100644
--- a/commands.py
+++ b/commands.py
@@ -31,6 +31,7 @@ from vars import *
 from helpers import *
 from supergenerator import *
 
+
 router = Router()
 
 commands = [
@@ -48,7 +49,8 @@ commands = [
     "/aben",
     "/button",
     "/clean",
-    "/reset"
+    "/reset",
+    "/tben",
 ]
 
 @router.message(Command("stats"))
@@ -143,8 +145,8 @@ async def tben_handler(message: Message, command: CommandObject):
     args = command.args
     if not args or len(args.split(" ", 2)) < 2: 
         return await message.answer("dayn eblan suka")
-    key, val = args.split()[0], args.split()[1]
-    temp_benned = temp_benned | {str(key): val}
+    key, val = args.split(" ", 2)[0], args.split(" ", 2)[1]
+    temp_benned[str(key)] = val
     await message.answer(f"✅ хамха забенен")
     
 
diff --git a/supergenerator.py b/supergenerator.py
index e552ef3..fda127e 100644
--- a/supergenerator.py
+++ b/supergenerator.py
@@ -123,7 +123,7 @@ async def generate(
 
     text = ""
 
-    if user_id in temp_benned:
+    if str(user_id) in temp_benned:
         return temp_benned[str(user_id)]
 
     add_one_to("stats/gens")
diff --git a/vars.py b/vars.py
index b4658df..b0dad1f 100644
--- a/vars.py
+++ b/vars.py
@@ -3,4 +3,4 @@ last_error_time = {}
 last_block_time = {}
 alo_command = {}
 user_contexts = {}
-temp_benned = []
+temp_benned = {}