def rem_user(client, message):
if not message.reply_to_message:
message.edit_text("You've gotta reply to someone!")
return
user_id = message.reply_to_message.from_user.id
is_user = db.is_user(user_id)
if not is_user:
message.edit_text("AI isn't enabled for this user in the first place!")
else:
db.rem_user(user_id)
message.edit_text("AI disabled for this user successfully!")
LOGGER.info(f"AI disabled for user - {user_id}")
async def rem_user(_, message: Message) -> None:
"""Remove AI for a user."""
if not message.reply_to_message:
message.edit_text("You've gotta reply to someone!")
return
user_id = message.reply_to_message.from_user.id
is_user = db.is_user(user_id)
if not is_user:
await message.edit_text(
"AI isn't enabled for this user in the first place!")
else:
db.rem_user(user_id)
await message.edit_text("AI disabled for this user successfully!")
LOGGER.info(f"AI disabled for user - {user_id}")
def add_user(client, message):
if not message.reply_to_message:
message.edit_text(
"Reply to someone to enable chatbot for that person!")
return
user_id = message.reply_to_message.from_user.id
is_user = db.is_user(user_id)
if not is_user:
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
db.set_ses(user_id, ses_id, expires)
message.edit_text("AI enabled for user successfully!")
LOGGER.info(f"AI enabled for user - {user_id}")
else:
message.edit_text("AI is already enabled for this user!")