diff options
| -rw-r--r-- | main.py | 8 | ||||
| -rw-r--r-- | supergenerator.py | 16 |
2 files changed, 22 insertions, 2 deletions
@@ -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( |