def list_notes(update: Update, context: CallbackContext):
bot = context.bot
user = update.effective_user
chat_id = update.effective_chat.id
note_list = sql.get_all_chat_notes(chat_id)
notes = len(note_list) + 1
msg = "Get note by `/notenumber` or `#notename` \n\n *ID* *Note* \n"
msg_pm = f"*Notes from {update.effective_chat.title}* \nGet note by `/notenumber` or `#notename` in group \n\n *ID* *Note* \n"
for note_id, note in zip(range(1, notes), note_list):
if note_id < 10:
note_name = f"`{note_id:2}.` `{(note.name.lower())}`\n"
else:
note_name = f"`{note_id}.` `{(note.name.lower())}`\n"
if len(msg) + len(note_name) > MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(msg, parse_mode=ParseMode.MARKDOWN)
msg = ""
msg_pm = ""
msg += note_name
msg_pm += note_name
if not note_list:
try:
update.effective_message.reply_text("No notes in this chat!")
except BadRequest:
update.effective_message.reply_text("No notes in this chat!", quote=False)
elif len(msg) != 0:
setting = getprivatenotes(chat_id)
if setting == True:
bot.send_message(user.id, msg_pm, parse_mode=ParseMode.MARKDOWN)
else:
update.effective_message.reply_text(msg, parse_mode=ParseMode.MARKDOWN)
def get(update: Update, context: CallbackContext, notename, show_none=True, no_format=False):
bot = context.bot
user = update.effective_user
chat_id = update.effective_message.chat.id
note_chat_id = update.effective_chat.id
note = sql.get_note(note_chat_id, notename)
message = update.effective_message # type: Optional[Message]
if note:
if MessageHandlerChecker.check_user(update.effective_user.id):
return
# If we're replying to a message, reply to that message (unless it's an error)
if message.reply_to_message:
reply_id = message.reply_to_message.message_id
else:
reply_id = message.message_id
if note.is_reply:
if JOIN_LOGGER:
try:
bot.forward_message(
chat_id=chat_id, from_chat_id=JOIN_LOGGER, message_id=note.value
)
except BadRequest as excp:
if excp.message == "Message to forward not found":
message.reply_text(
"This message seems to have been lost - I'll remove it "
"from your notes list."
)
sql.rm_note(note_chat_id, notename)
else:
raise
else:
try:
bot.forward_message(
chat_id=chat_id, from_chat_id=chat_id, message_id=note.value
)
except BadRequest as excp:
if excp.message == "Message to forward not found":
message.reply_text(
"Looks like the original sender of this note has deleted "
"their message - sorry! Get your bot admin to start using a "
"message dump to avoid this. I'll remove this note from "
"your saved notes."
)
sql.rm_note(note_chat_id, notename)
else:
raise
else:
VALID_NOTE_FORMATTERS = [
"first",
"last",
"fullname",
"username",
"id",
"chatname",
"mention",
]
valid_format = escape_invalid_curly_brackets(
note.value, VALID_NOTE_FORMATTERS
)
if valid_format:
if not no_format:
if "%%%" in valid_format:
split = valid_format.split("%%%")
if all(split):
text = random.choice(split)
else:
text = valid_format
else:
text = valid_format
else:
text = valid_format
text = text.format(
first=escape_markdown(message.from_user.first_name),
last=escape_markdown(
message.from_user.last_name or message.from_user.first_name
),
fullname=escape_markdown(
" ".join(
[message.from_user.first_name, message.from_user.last_name]
if message.from_user.last_name
else [message.from_user.first_name]
)
),
username="******" + message.from_user.username
if message.from_user.username
else mention_markdown(
message.from_user.id, message.from_user.first_name
),
mention=mention_markdown(
message.from_user.id, message.from_user.first_name
),
chatname=escape_markdown(
message.chat.title
if message.chat.type != "private"
else message.from_user.first_name
),
id=message.from_user.id,
)
else:
text = ""
keyb = []
parseMode = ParseMode.MARKDOWN
buttons = sql.get_buttons(note_chat_id, notename)
if no_format:
parseMode = None
text += revert_buttons(buttons)
else:
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
try:
setting = getprivatenotes(chat_id)
if note.msgtype in (sql.Types.BUTTON_TEXT, sql.Types.TEXT):
if setting:
bot.send_message(
user.id,
text,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard,
)
else:
bot.send_message(
chat_id,
text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard,
)
else:
if setting:
ENUM_FUNC_MAP[note.msgtype](
user.id,
note.file,
caption=text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
reply_markup=keyboard,
)
else:
ENUM_FUNC_MAP[note.msgtype](
chat_id,
note.file,
caption=text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
reply_markup=keyboard,
)
except BadRequest as excp:
if excp.message == "Entity_mention_user_invalid":
message.reply_text(
"Looks like you tried to mention someone I've never seen before. If you really "
"want to mention them, forward one of their messages to me, and I'll be able "
"to tag them!"
)
elif FILE_MATCHER.match(note.value):
message.reply_text(
"This note was an incorrectly imported file from another bot - I can't use "
"it. If you really need it, you'll have to save it again. In "
"the meantime, I'll remove it from your notes list."
)
sql.rm_note(note_chat_id, notename)
else:
message.reply_text(
"This note could not be sent, as it is incorrectly formatted. Ask in "
f"@{SUPPORT_CHAT} if you can't figure out why!"
)
LOGGER.exception(
"Could not parse message #%s in chat %s",
notename,
str(note_chat_id),
)
LOGGER.warning("Message was: %s", str(note.value))
return
elif show_none:
message.reply_text("This note doesn't exist")