add user to db on any action
This commit is contained in:
1 parent
958ee69ab7
commit
f9207e1f99
2 files changed
+20
No files matched your search
@@ -125,6 +125,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 ""
|
||||
replied = message.reply_to_message or None
|
||||
@@ -236,6 +238,7 @@ 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,6 +398,7 @@ 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")
|
||||
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in new issue
Block a user