def warns(update, context): message = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] args = context.args user_id = extract_user(message, args) or update.effective_user.id result = sql.get_warns(user_id, chat.id) num = 1 if result and result[0] != 0: num_warns, reasons = result limit, soft_warn = sql.get_warn_setting(chat.id) if reasons: text = "This user has {}/{} warnings, for the following reasons:".format( num_warns, limit) for reason in reasons: text += "\n {}. {}".format(num, reason) num += 1 msgs = split_message(text) for msg in msgs: update.effective_message.reply_text(msg) else: update.effective_message.reply_text( "User has {}/{} warnings, but no reasons for any of them.". format(num_warns, limit)) else: update.effective_message.reply_text( "This user hasn't got any warnings!")
def demote(update, context): chat = update.effective_chat message = update.effective_message user = update.effective_user args = context.args if user_can_promote(chat, user, context.bot.id) is False: message.reply_text("You don't have enough rights to demote someone!") return "" user_id = extract_user(message, args) if not user_id: message.reply_text("mention one.... 🤷🏻♂.") return "" user_member = chat.get_member(user_id) if user_member.status == "creator": message.reply_text("I'm not gonna demote Creator this group.... 🙄") return "" if not user_member.status == "administrator": message.reply_text( "How I'm supposed to demote someone who is not even an admin!") return "" if user_id == context.bot.id: message.reply_text("Yeahhh... Not gonna demote myself!") return "" try: context.bot.promoteChatMember( int(chat.id), int(user_id), can_change_info=False, can_post_messages=False, can_edit_messages=False, can_delete_messages=False, can_invite_users=False, can_restrict_members=False, can_pin_messages=False, ) message.reply_text("Successfully demoted!") return ("<b>{}:</b>" "\n#DEMOTED" "\n<b>Admin:</b> {}" "\n<b>User:</b> {}".format( html.escape(chat.title), mention_html(user.id, user.first_name), mention_html(user_member.user.id, user_member.user.first_name), )) except BadRequest: message.reply_text( "Failed to demote. I might not be admin, or the admin status was appointed by another " "user, so I can't act upon them!") return ""
def unmute(update, context): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message # type: Optional[Message] args = context.args if user_can_ban(chat, user, context.bot.id) == False: message.reply_text("You don't have enough rights to unmute people") return "" user_id = extract_user(message, args) if not user_id: message.reply_text( "You'll need to either give me a username to unmute, or reply to someone to be unmuted." ) return "" member = chat.get_member(int(user_id)) if member.status != "kicked" and member.status != "left": if (member.can_send_messages and member.can_send_media_messages and member.can_send_other_messages and member.can_add_web_page_previews): message.reply_text("This user already has the right to speak.") else: context.bot.restrict_chat_member( chat.id, int(user_id), permissions=ChatPermissions( can_send_messages=True, can_invite_users=True, can_pin_messages=True, can_send_polls=True, can_change_info=True, can_send_media_messages=True, can_send_other_messages=True, can_add_web_page_previews=True, ), ) message.reply_text("Yep! this user can start talking again...") return ("<b>{}:</b>" "\n#UNMUTE" "\n<b>Admin:</b> {}" "\n<b>User:</b> {}".format( html.escape(chat.title), mention_html(user.id, user.first_name), mention_html(member.user.id, member.user.first_name), )) else: message.reply_text( "This user isn't even in the chat, unmuting them won't make them talk more than they " "already do!") return ""
def mute(update, context): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message # type: Optional[Message] args = context.args if user_can_ban(chat, user, context.bot.id) == False: message.reply_text( "You don't have enough rights to restrict someone from talking!") return "" user_id = extract_user(message, args) if not user_id: message.reply_text( "You'll need to either give me a username to mute, or reply to someone to be muted." ) return "" if user_id == context.bot.id: message.reply_text("Yeahh... I'm not muting myself!") return "" member = chat.get_member(int(user_id)) if member: if is_user_admin(chat, user_id, member=member): message.reply_text( "Well i'm not gonna stop an admin from talking!") elif member.can_send_messages is None or member.can_send_messages: context.bot.restrict_chat_member( chat.id, user_id, permissions=ChatPermissions(can_send_messages=False), ) message.reply_text("👍🏻 muted! 🤐") return ("<b>{}:</b>" "\n#MUTE" "\n<b>Admin:</b> {}" "\n<b>User:</b> {}".format( html.escape(chat.title), mention_html(user.id, user.first_name), mention_html(member.user.id, member.user.first_name), )) else: message.reply_text("This user is already taped 🤐") else: message.reply_text("This user isn't in the chat!") return ""
def promote(update, context): chat_id = update.effective_chat.id message = update.effective_message chat = update.effective_chat user = update.effective_user args = context.args if user_can_promote(chat, user, context.bot.id) is False: message.reply_text("You don't have enough rights to promote someone!") return "" user_id = extract_user(message, args) if not user_id: message.reply_text("mention one.... 🤷🏻♂.") return "" user_member = chat.get_member(user_id) if (user_member.status == "administrator" or user_member.status == "creator"): message.reply_text("This person is already an admin...!") return "" if user_id == context.bot.id: message.reply_text("I hope, if i could promote myself!") return "" # set same perms as bot - bot can't assign higher perms than itself! bot_member = chat.get_member(context.bot.id) context.bot.promoteChatMember( chat_id, user_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_invite_users=bot_member.can_invite_users, can_restrict_members=bot_member.can_restrict_members, can_pin_messages=bot_member.can_pin_messages, ) message.reply_text("Promoted🧡") return ("<b>{}:</b>" "\n#PROMOTED" "\n<b>Admin:</b> {}" "\n<b>User:</b> {}".format( html.escape(chat.title), mention_html(user.id, user.first_name), mention_html(user_member.user.id, user_member.user.first_name), ))
def slap(update, context): args = context.args msg = update.effective_message # reply to correct message reply_text = ( msg.reply_to_message.reply_text if msg.reply_to_message else msg.reply_text ) # get user who sent message if msg.from_user.username: curr_user = "******" + escape_markdown(msg.from_user.username) else: curr_user = "******".format( msg.from_user.first_name, msg.from_user.id ) user_id = extract_user(update.effective_message, args) if user_id: slapped_user = context.bot.get_chat(user_id) user1 = curr_user if slapped_user.username: user2 = "@" + escape_markdown(slapped_user.username) else: user2 = "[{}](tg://user?id={})".format( slapped_user.first_name, slapped_user.id ) # if no target found, bot targets the sender else: user1 = "[{}](tg://user?id={})".format( context.bot.first_name, context.bot.id ) user2 = curr_user temp = random.choice(fun.SLAP_TEMPLATES) item = random.choice(fun.ITEMS) hit = random.choice(fun.HIT) throw = random.choice(fun.THROW) repl = temp.format( user1=user1, user2=user2, item=item, hits=hit, throws=throw ) reply_text(repl, parse_mode=ParseMode.MARKDOWN)
def punch(update, context): args = context.args msg = update.effective_message # type: Optional[Message] # reply to correct message reply_text = ( msg.reply_to_message.reply_text if msg.reply_to_message else msg.reply_text ) # get user who sent message if msg.from_user.username: curr_user = "******" + escape_markdown(msg.from_user.username) else: curr_user = "******".format( msg.from_user.first_name, msg.from_user.id ) user_id = extract_user(update.effective_message, args) if user_id: punched_user = context.bot.get_chat(user_id) user1 = curr_user if punched_user.username: user2 = "@" + escape_markdown(punched_user.username) else: user2 = "[{}](tg://user?id={})".format( punched_user.first_name, punched_user.id ) # if no target found, bot targets the sender else: user1 = "[{}](tg://user?id={})".format( context.bot.first_name, context.bot.id ) user2 = curr_user temp = random.choice(fun.PUNCH_TEMPLATES) item = random.choice(fun.ITEMS) punch = random.choice(fun.PUNCH) repl = temp.format(user1=user1, user2=user2, item=item, punches=punch) reply_text(repl, parse_mode=ParseMode.MARKDOWN)
def get_id(update, context): args = context.args user_id = extract_user(update.effective_message, args) if user_id: if (update.effective_message.reply_to_message and update.effective_message.reply_to_message.forward_from): user1 = update.effective_message.reply_to_message.from_user user2 = update.effective_message.reply_to_message.forward_from update.effective_message.reply_text( "The original sender, {}, has an ID of `{}`.\nThe forwarder, {}, has an ID of `{}`." .format( escape_markdown(user2.first_name), user2.id, escape_markdown(user1.first_name), user1.id, ), parse_mode=ParseMode.MARKDOWN, ) else: user = context.bot.get_chat(user_id) update.effective_message.reply_text( "{}'s id is `{}`.".format(escape_markdown(user.first_name), user.id), parse_mode=ParseMode.MARKDOWN, ) else: chat = update.effective_chat # type: Optional[Chat] if chat.type == "private": update.effective_message.reply_text( "Your id is `{}`.".format(chat.id), parse_mode=ParseMode.MARKDOWN, ) else: update.effective_message.reply_text( "This group's id is `{}`.".format(chat.id), parse_mode=ParseMode.MARKDOWN, )
def reset_warns(update, context): message = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] args = context.args user_id = extract_user(message, args) if user_id: sql.reset_warns(user_id, chat.id) message.reply_text("Warnings have been reset!") warned = chat.get_member(user_id).user return ("<b>{}:</b>" "\n#RESETWARNS" "\n<b>Admin:</b> {}" "\n<b>User:</b> {} (<code>{}</code>)".format( html.escape(chat.title), mention_html(user.id, user.first_name), mention_html(warned.id, warned.first_name), warned.id, )) else: message.reply_text("No user has been designated!") return ""
def about_bio(update, context): message = update.effective_message # type: Optional[Message] args = context.args user_id = extract_user(message, args) if user_id: user = context.bot.get_chat(user_id) else: user = message.from_user info = sql.get_user_bio(user.id) if info: update.effective_message.reply_text( "*{}*:\n{}".format(user.first_name, escape_markdown(info)), parse_mode=ParseMode.MARKDOWN, ) elif message.reply_to_message: username = user.first_name update.effective_message.reply_text( "{} No details about him have been saved yet !".format(username)) else: update.effective_message.reply_text( " Your bio about you has been saved !")
def about_me(update, context): message = update.effective_message # type: Optional[Message] args = context.args user_id = extract_user(message, args) if user_id: user = bot.get_chat(user_id) else: user = message.from_user info = sql.get_user_me_info(user.id) if info: update.effective_message.reply_text( "*{}*:\n{}".format(user.first_name, escape_markdown(info)), parse_mode=ParseMode.MARKDOWN, ) elif message.reply_to_message: username = message.reply_to_message.from_user.first_name update.effective_message.reply_text( username + "Information about him is currently unavailable !") else: update.effective_message.reply_text( "You have not added any information about yourself yet !")
def info(update, context): args = context.args msg = update.effective_message # type: Optional[Message] user_id = extract_user(update.effective_message, args) chat = update.effective_chat if user_id: user = context.bot.get_chat(user_id) elif not msg.reply_to_message and not args: user = msg.from_user elif not msg.reply_to_message and ( not args or (len(args) >= 1 and not args[0].startswith("@") and not args[0].isdigit() and not msg.parse_entities([MessageEntity.TEXT_MENTION]))): msg.reply_text("I can't extract a user from this.") return else: return del_msg = msg.reply_text( "Hold tight while I steal some data from <b>FBI Database</b>...", parse_mode=ParseMode.HTML, ) text = ("<b>USER INFO</b>:" "\n\nID: <code>{}</code>" "\nFirst Name: {}".format(user.id, html.escape(user.first_name))) if user.last_name: text += "\nLast Name: {}".format(html.escape(user.last_name)) if user.username: text += "\nUsername: @{}".format(html.escape(user.username)) text += "\nPermanent user link: {}".format(mention_html(user.id, "link")) text += "\nNumber of profile pics: {}".format( context.bot.get_user_profile_photos(user.id).total_count) try: sw = spamwtc.get_ban(int(user.id)) if sw: text += "\n\n<b>This person is banned in Spamwatch!</b>" text += f"\nResason: <pre>{sw.reason}</pre>" else: pass except: pass # Don't break on exceptions like if api is down? if user.id == OWNER_ID: text += "\n\nAye this guy is my owner.\nI would never do anything against him!" elif user.id in SUDO_USERS: text += ("\n\nThis person is one of my sudo users! " "Nearly as powerful as my owner - so watch it.") elif user.id in SUPPORT_USERS: text += ("\n\nThis person is one of my support users! " "Not quite a sudo user, but can still gban you off the map.") elif user.id in WHITELIST_USERS: text += ("\n\nThis person has been whitelisted! " "That means I'm not allowed to ban/kick them.") try: memstatus = chat.get_member(user.id).status if memstatus == "administrator" or memstatus == "creator": result = context.bot.get_chat_member(chat.id, user.id) if result.custom_title: text += f"\n\nThis user has custom title <b>{result.custom_title}</b> in this chat." except BadRequest: pass for mod in USER_INFO: try: mod_info = mod.__user_info__(user.id).strip() except TypeError: mod_info = mod.__user_info__(user.id, chat.id).strip() if mod_info: text += "\n\n" + mod_info try: profile = context.bot.get_user_profile_photos(user.id).photos[0][-1] context.bot.sendChatAction(chat.id, "upload_photo") context.bot.send_photo( chat.id, photo=profile, caption=(text), parse_mode=ParseMode.HTML, disable_web_page_preview=True, ) except IndexError: context.bot.sendChatAction(chat.id, "typing") msg.reply_text(text, parse_mode=ParseMode.HTML, disable_web_page_preview=True) finally: del_msg.delete()