def save(bot, update):
chat_id = update.effective_chat.id
msg = update.effective_message
raw_text = msg.text
args = raw_text.split(
None, 2) # use python's maxsplit to separate Cmd, note_name, and data
if len(args) >= 3:
note_name = args[1]
note = args[2]
offset = len(note) - len(
raw_text) # set correct offset relative to command + notename
markdown_note, buttons = button_markdown_parser(
note, entities=msg.parse_entities(), offset=offset)
note_data = markdown_note.strip()
if not note_data:
update.effective_message.reply_text(
"You can't save an empty message! If you added a button, you MUST "
"have some text in the message too.")
return
sql.add_note_to_db(chat_id,
note_name,
note_data,
is_reply=False,
buttons=buttons)
update.effective_message.reply_text("Yas! Added " + note_name)
else:
update.effective_message.reply_text("Dude, there's no note")
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 filters(bot, update):
chat = update.effective_chat
msg = update.effective_message
args = msg.text.split(
None,
1) # use python's maxsplit to separate Cmd, keyword, and reply_text
if len(args) < 2:
return
extracted = split_quotes(args[1])
if len(extracted) < 1:
return
# set trigger -> lower, so as to avoid adding duplicate filters with different cases
keyword = extracted[0].lower()
is_sticker = False
is_document = False
is_image = False
is_voice = False
is_audio = False
is_video = False
buttons = []
# determine what the contents of the filter are - text, image, sticker, etc
if len(extracted) >= 2:
offset = len(extracted[1]) - len(
msg.text) # set correct offset relative to command + notename
content, buttons = button_markdown_parser(
extracted[1], entities=msg.parse_entities(), offset=offset)
elif msg.reply_to_message and msg.reply_to_message.sticker:
content = msg.reply_to_message.sticker.file_id
is_sticker = True
elif msg.reply_to_message and msg.reply_to_message.document:
content = msg.reply_to_message.document.file_id
is_document = True
elif msg.reply_to_message and msg.reply_to_message.photo:
content = msg.reply_to_message.photo[
-1].file_id # last elem = best quality
is_image = True
elif msg.reply_to_message and msg.reply_to_message.audio:
content = msg.reply_to_message.audio.file_id
is_audio = True
elif msg.reply_to_message and msg.reply_to_message.voice:
content = msg.reply_to_message.voice.file_id
is_voice = True
elif msg.reply_to_message and msg.reply_to_message.video:
content = msg.reply_to_message.video.file_id
is_video = True
else:
msg.reply_text("You didn't specify what to reply with!")
return
# Add the filter
# Note: perhaps handlers can be removed somehow using sql.get_chat_filters
for handler in dispatcher.handlers.get(HANDLER_GROUP, []):
if handler.filters == (keyword, chat.id):
dispatcher.remove_handler(handler, HANDLER_GROUP)
sql.add_filter(chat.id, keyword, content, is_sticker, is_document,
is_image, is_audio, is_voice, is_video, buttons)
msg.reply_text("Handler {} added!".format(keyword))
raise DispatcherHandlerStop