def set_leave(bot, update):
chat_id = update.effective_chat.id
msg = update.effective_message
raw_text = msg.text
args = raw_text.split(None,
1) # use python's maxsplit to separate cmd and args
buttons = []
# determine what the contents of the filter are - text, image, sticker, etc
if len(args) >= 2:
offset = len(args[1]) - len(
msg.text) # set correct offset relative to command + notename
content, buttons = button_markdown_parser(
args[1], entities=msg.parse_entities(), offset=offset)
if buttons:
data_type = sql.Types.BUTTON_TEXT
else:
data_type = sql.Types.TEXT
elif msg.reply_to_message and msg.reply_to_message.sticker:
content = msg.reply_to_message.sticker.file_id
data_type = sql.Types.STICKER
elif msg.reply_to_message and msg.reply_to_message.document:
content = msg.reply_to_message.document.file_id
data_type = sql.Types.DOCUMENT
elif msg.reply_to_message and msg.reply_to_message.photo:
content = msg.reply_to_message.photo[
-1].file_id # last elem = best quality
data_type = sql.Types.PHOTO
elif msg.reply_to_message and msg.reply_to_message.audio:
content = msg.reply_to_message.audio.file_id
data_type = sql.Types.AUDIO
elif msg.reply_to_message and msg.reply_to_message.voice:
content = msg.reply_to_message.voice.file_id
data_type = sql.Types.VOICE
elif msg.reply_to_message and msg.reply_to_message.video:
content = msg.reply_to_message.video.file_id
data_type = sql.Types.VIDEO
else:
msg.reply_text("You didn't specify what to reply with!")
return
sql.set_custom_leave(chat_id, content, data_type, buttons)
update.effective_message.reply_text(
"Successfully set custom leave message!")
def reset_leave(bot, update):
chat_id = update.effective_chat.id
sql.set_custom_leave(chat_id, sql.DEFAULT_LEAVE, sql.Types.TEXT)
update.effective_message.reply_text(
"Successfully reset leave message to default!")