add user to db on any action

This commit is contained in:
huker667 committed 2026-05-23 17:26:56 +03:00
1 parent 958ee69ab7
commit f9207e1f99
2 files changed
+20

No files matched your search

+4
View File
@@ -125,6 +125,8 @@ async def guest_middleware(handler, event, data):
return return
ensure_user(user_id)
last_command_time[user_id] = current_time last_command_time[user_id] = current_time
user_text = message.text or message.caption or "" user_text = message.text or message.caption or ""
replied = message.reply_to_message or None replied = message.reply_to_message or None
@@ -236,6 +238,7 @@ async def text_handler(message: Message):
return return
last_command_time[user_id] = current_time last_command_time[user_id] = current_time
ensure_user(user_id)
replied = message.reply_to_message replied = message.reply_to_message
prompt = "" prompt = ""
@@ -395,6 +398,7 @@ async def text_handler(message: Message):
@router.inline_query() @router.inline_query()
async def inline_handler(inline_query: InlineQuery): async def inline_handler(inline_query: InlineQuery):
user_id = inline_query.from_user.id user_id = inline_query.from_user.id
ensure_user(user_id)
button = InlineKeyboardButton(text="ало жди 📞", callback_data="pasholnaxxuy") button = InlineKeyboardButton(text="ало жди 📞", callback_data="pasholnaxxuy")
+16
View File
@@ -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): def get_user_config(user_id):
with shelve.open("users") as db: with shelve.open("users") as db:
return db.get( return db.get(