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
|
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")
|
||||||
|
|
||||||
|
|||||||
@@ -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(
|
||||||
|
|||||||
Reference in new issue
Block a user