def clean_welcome(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
if not args:
clean_pref = sql.get_clean_pref(chat.id)
if clean_pref:
update.effective_message.reply_text(
"I should be deleting welcome messages up to two days old.")
else:
update.effective_message.reply_text(
"I'm currently not deleting old welcome messages!")
return ""
if args[0].lower() in ("on", "yes"):
sql.set_clean_welcome(str(chat.id), True)
update.effective_message.reply_text(
"I'll try to delete old welcome messages!")
return "<b>{}:</b>" \
"\n#CLEAN_WELCOME" \
"\n<b>Admin:</b> {}" \
"\nHas toggled clean welcomes to <code>ON</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name))
elif args[0].lower() in ("off", "no"):
sql.set_clean_welcome(str(chat.id), False)
update.effective_message.reply_text(
"I won't delete old welcome messages.")
return "<b>{}:</b>" \
"\n#CLEAN_WELCOME" \
"\n<b>Admin:</b> {}" \
"\nHas toggled clean welcomes to <code>OFF</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name))
else:
# idek what you're writing, say yes or no
update.effective_message.reply_text(
"I understand 'on/yes' or 'off/no' only!")
return ""
def new_member(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
new_members = update.effective_message.new_chat_members
for mems in new_members:
if is_user_ban_protected(chat, mems.id, chat.get_member(mems.id)):
continue
if is_safemoded(chat.id).safemode_status:
try:
bot.restrict_chat_member(chat.id,
mems.id,
can_send_messages=True,
can_send_media_messages=False,
can_send_other_messages=False,
can_add_web_page_previews=False,
until_date=(int(time.time() +
24 * 60 * 60)))
except BadRequest as excp:
LOGGER.warning(update)
LOGGER.exception(
"ERROR muting user %s in chat %s (%s) due to %s", mems.id,
chat.title, chat.id, excp.message)
should_welc, cust_welcome, welc_type = sql.get_welc_pref(chat.id)
if should_welc:
sent = None
for new_mem in new_members:
# Give the owner a special welcome
if new_mem.id == OWNER_ID:
update.effective_message.reply_text(
"Master is in the houseeee, let's get this party started!")
bot_member = chat.get_member(bot.id)
bot.promoteChatMember(
chat_id,
new_mem.id,
can_change_info=bot_member.can_change_info,
can_post_messages=bot_member.can_post_messages,
can_edit_messages=bot_member.can_edit_messages,
can_delete_messages=bot_member.can_delete_messages,
can_restrict_members=bot_member.can_restrict_members,
can_pin_messages=bot_member.can_pin_messages,
can_promote_members=bot_member.can_promote_members)
continue
# Don't welcome yourself
elif new_mem.id == bot.id:
continue
else:
# If welcome message is media, send with appropriate function
if welc_type != sql.Types.TEXT and welc_type != sql.Types.BUTTON_TEXT:
ENUM_FUNC_MAP[welc_type](chat.id, cust_welcome)
return
# else, move on
first_name = new_mem.first_name or "PersonWithNoName" # edge case of empty name - occurs for some bugs.
if cust_welcome:
if new_mem.last_name:
fullname = "{} {}".format(first_name,
new_mem.last_name)
else:
fullname = first_name
count = chat.get_members_count()
mention = mention_markdown(new_mem.id, first_name)
if new_mem.username:
username = "******" + escape_markdown(new_mem.username)
else:
username = mention
valid_format = escape_invalid_curly_brackets(
cust_welcome, VALID_WELCOME_FORMATTERS)
res = valid_format.format(
first=escape_markdown(first_name),
last=escape_markdown(new_mem.last_name or first_name),
fullname=escape_markdown(fullname),
username=username,
mention=mention,
count=count,
chatname=escape_markdown(chat.title),
id=new_mem.id)
buttons = sql.get_welc_buttons(chat.id)
keyb = build_keyboard(buttons)
else:
res = sql.DEFAULT_WELCOME.format(first=first_name)
keyb = []
keyboard = InlineKeyboardMarkup(keyb)
sent = send(update, res, keyboard,
sql.DEFAULT_WELCOME.format(
first=first_name)) # type: Optional[Message]
delete_join(bot, update)
prev_welc = sql.get_clean_pref(chat.id)
if prev_welc:
try:
bot.delete_message(chat.id, prev_welc)
except BadRequest as excp:
pass
if sent:
sql.set_clean_welcome(chat.id, sent.message_id)