def welcome(update: Update, context: CallbackContext):
args = context.args
chat = update.effective_chat
# if no args, show current replies.
if not args or args[0].lower() == "noformat":
noformat = True
pref, welcome_m, cust_content, welcome_type = sql.get_welc_pref(
chat.id)
update.effective_message.reply_text(
f"This chat has it's welcome setting set to: `{pref}`.\n"
f"*The welcome message (not filling the {{}}) is:*",
parse_mode=ParseMode.MARKDOWN,
)
if welcome_type == sql.Types.BUTTON_TEXT or welcome_type == sql.Types.TEXT:
buttons = sql.get_welc_buttons(chat.id)
if noformat:
welcome_m += revert_buttons(buttons)
update.effective_message.reply_text(welcome_m)
else:
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
send(update, welcome_m, keyboard, sql.DEFAULT_WELCOME)
else:
buttons = sql.get_welc_buttons(chat.id)
if noformat:
welcome_m += revert_buttons(buttons)
ENUM_FUNC_MAP[welcome_type](chat.id,
cust_content,
caption=welcome_m)
else:
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
ENUM_FUNC_MAP[welcome_type](
chat.id,
cust_content,
caption=welcome_m,
reply_markup=keyboard,
parse_mode=ParseMode.MARKDOWN,
disable_web_page_preview=True,
)
elif len(args) >= 1:
if args[0].lower() in ("on", "yes"):
sql.set_welc_preference(str(chat.id), True)
update.effective_message.reply_text(
"Okay! I'll be welcoming all new members from now on!")
elif args[0].lower() in ("off", "no"):
sql.set_welc_preference(str(chat.id), False)
update.effective_message.reply_text(
"I'll stay quite when new members join.")
else:
update.effective_message.reply_text(
"I understand 'on/yes' or 'off/no' only!")
def goodbye(update: Update, context: CallbackContext):
args = context.args
chat = update.effective_chat
if not args or args[0] == "noformat":
noformat = True
pref, goodbye_m, goodbye_type = sql.get_gdbye_pref(chat.id)
update.effective_message.reply_text(
f"This chat has it's goodbye setting set to: `{pref}`.\n"
f"*The goodbye message (not filling the {{}}) is:*",
parse_mode=ParseMode.MARKDOWN,
)
if goodbye_type == sql.Types.BUTTON_TEXT:
buttons = sql.get_gdbye_buttons(chat.id)
if noformat:
goodbye_m += revert_buttons(buttons)
update.effective_message.reply_text(goodbye_m)
else:
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
send(update, goodbye_m, keyboard, sql.DEFAULT_GOODBYE)
else:
if noformat:
ENUM_FUNC_MAP[goodbye_type](chat.id, goodbye_m)
else:
ENUM_FUNC_MAP[goodbye_type](chat.id,
goodbye_m,
parse_mode=ParseMode.MARKDOWN)
elif len(args) >= 1:
if args[0].lower() in ("on", "yes"):
sql.set_gdbye_preference(str(chat.id), True)
update.effective_message.reply_text(
"I'll stay quiet when new members join.")
elif args[0].lower() in ("off", "no"):
sql.set_gdbye_preference(str(chat.id), False)
update.effective_message.reply_text("Ok!")
else:
# idek what you're writing, say yes or no
update.effective_message.reply_text(
"I understand 'on/yes' or 'off/no' only!")
def get(update, context, notename, show_none=True, no_format=False):
bot = context.bot
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:
if note.msgtype in (sql.Types.BUTTON_TEXT, sql.Types.TEXT):
bot.send_message(
chat_id,
text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard)
else:
ENUM_FUNC_MAP[note.msgtype](
chat_id,
note.file,
caption=text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
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")