add user to db on any action

2 files changed, 22 insertions, 2 deletions

diff --git a/main.py b/main.py
index 19114be..be7161a 100644
--- a/main.py
+++ b/main.py
@@ -124,6 +124,8 @@ async def guest_middleware(handler, event, data):
                 )
         
                 return
+
+            ensure_user(user_id)
             
             last_command_time[user_id] = current_time
             user_text = message.text or message.caption or ""
@@ -236,7 +238,8 @@ async def text_handler(message: Message):
             return
 
         last_command_time[user_id] = current_time
-    
+        ensure_user(user_id)
+        
     replied = message.reply_to_message
     prompt = ""
 
@@ -395,7 +398,8 @@ async def text_handler(message: Message):
 @router.inline_query()
 async def inline_handler(inline_query: InlineQuery):
     user_id = inline_query.from_user.id
-
+    ensure_user(user_id)
+    
     button = InlineKeyboardButton(text="ало жди 📞", callback_data="pasholnaxxuy")
 
     if inline_query.query == "clear":
diff --git a/supergenerator.py b/supergenerator.py
index 4b5dbf9..e3827f5 100644
--- a/supergenerator.py
+++ b/supergenerator.py
@@ -133,6 +133,22 @@ def set_user_config(
         }
 
 
+def ensure_user(
+    user_id,
+    model_name=DEFAULT_MODEL,
+    stream_mode=DEFAULT_STREAM,
+    call_mode=DEFAULT_CALL
+):
+    with shelve.open("users") as db:
+        key = str(user_id)
+
+        if key not in db:
+            db[key] = {
+                "model": model_name,
+                "stream": stream_mode,
+                "call": call_mode,
+            }
+
 def get_user_config(user_id):
     with shelve.open("users") as db:
         return db.get(