def connected(bot, update, chat, user_id, need_admin=True): user = update.effective_user msg = update.effective_message if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id): conn_id = sql.get_connected_chat(user_id).chat_id getstatusadmin = bot.get_chat_member(conn_id, msg.from_user.id) isadmin = getstatusadmin.status in ADMIN_STATUS ismember = getstatusadmin.status in MEMBER_STAUS isallow = sql.allow_connect_to_chat(conn_id) if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS): if need_admin is True: if getstatusadmin.status in ADMIN_STATUS or user_id in SUDO_USERS or user.id in DEV_USERS: return conn_id else: send_message(msg, "You must be an admin in the connected group!") raise Exception("Not admin!") else: return conn_id else: send_message(msg, "The group changed the connection rights or you are no longer an admin.\n" "I've disconnected you.") disconnect_chat(bot, update) raise Exception("Not admin!") else: return False
def blackliststicker(bot: Bot, update: Update, args: List[str]): msg = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] conn = connected(bot, update, chat, user.id, need_admin=False) if conn: chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: if chat.type == "private": return else: chat_id = update.effective_chat.id chat_name = chat.title sticker_list = "<b>List blacklisted stickers currently in {}:</b>\n".format(chat_name) all_stickerlist = sql.get_chat_stickers(chat_id) if len(args) > 0 and args[0].lower() == 'copy': for trigger in all_stickerlist: sticker_list += "<code>{}</code>\n".format(html.escape(trigger)) elif len(args) == 0: for trigger in all_stickerlist: sticker_list += " - <code>{}</code>\n".format(html.escape(trigger)) split_text = split_message(sticker_list) for text in split_text: if sticker_list == "<b>List blacklisted stickers currently in {}:</b>\n".format(chat_name).format(chat_name): send_message(update.effective_message, "There are no blacklist stickers in <b>{}</b>!".format(chat_name), parse_mode=ParseMode.HTML) return send_message(update.effective_message, text, parse_mode=ParseMode.HTML)
def help_connect_chat(bot: Bot, update: Update): msg = update.effective_message if msg.chat.type != "private": send_message(msg, "PM me with that command to get help.") return else: send_message(msg, "All commands", parse_mode="markdown")
def disconnect_chat(bot: Bot, update: Update): chat = update.effective_chat msg = update.effective_message if chat.type == 'private': disconnection_status = sql.disconnect(msg.from_user.id) if disconnection_status: sql.disconnected_chat = send_message(msg, "Disconnected from chat!") else: send_message(msg, "You're not connected!") else: send_message(msg, "This command is only available in PM.")
def connection_chat(bot: Bot, update: Update): chat = update.effective_chat user = update.effective_user msg = update.effective_message conn = connected(bot, update, chat, user.id, need_admin=True) if conn: chat_name = dispatcher.bot.getChat(conn).title else: if msg.chat.type != "private": return chat_name = chat.title if conn: message = "You are currently connected with {}.\n".format(chat_name) else: message = "You are currently not connected in any group.\n" send_message(msg, message, parse_mode="markdown")
def connect_chat(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat user = update.effective_user msg = update.effective_message if chat.type == 'private': if len(args) >= 1: try: connect_chat = int(args[0]) getstatusadmin = bot.get_chat_member(connect_chat, msg.from_user.id) except ValueError: try: connect_chat = str(args[0]) get_chat = bot.getChat(connect_chat) connect_chat = get_chat.id getstatusadmin = bot.get_chat_member(connect_chat, msg.from_user.id) except BadRequest: send_message(msg, "Invalid Chat ID!") return except BadRequest: send_message(msg, "Invalid Chat ID!") return isadmin = getstatusadmin.status in ADMIN_STATUS ismember = getstatusadmin.status in MEMBER_STAUS isallow = sql.allow_connect_to_chat(connect_chat) if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS): connection_status = sql.connect(msg.from_user.id, connect_chat) if connection_status: conn_chat = dispatcher.bot.getChat(connected(bot, update, chat, user.id, need_admin=False)) chat_name = conn_chat.title send_message(msg, "Successfully connected to *{chat_name}*." " Use /connection for see current available commands.", parse_mode=ParseMode.MARKDOWN) sql.add_history_conn(user.id, str(conn_chat.id), chat_name) else: send_message(msg, "Connection failed!") else: send_message(msg, "Connection to this chat is not allowed!") else: gethistory = sql.get_history_conn(user.id) if gethistory: buttons = [ InlineKeyboardButton(text="❎ Close button", callback_data="connect_close"), InlineKeyboardButton(text="🧹 Clear history", callback_data="connect_clear") ] else: buttons = [] conn = connected(bot, update, chat, user.id, need_admin=False) if conn: connectedchat = dispatcher.bot.getChat(conn) text = "You are connected to *{}* (`{}`)".format(connectedchat.title, conn) buttons.append(InlineKeyboardButton(text="🔌 Disconnect", callback_data="connect_disconnect")) else: text = "Write the chat ID or tag to connect!" if gethistory: text += "\n\n*Connection history:*\n" text += "╒═══「 *Info* 」\n" text += "│ Sorted: `Newest`\n" text += "│\n" buttons = [buttons] for x in sorted(gethistory.keys(), reverse=True): htime = time.strftime("%d/%m/%Y", time.localtime(x)) text += "╞═「 *{}* 」\n│ `{}`\n│ `{}`\n".format(gethistory[x]['chat_name'], gethistory[x]['chat_id'], htime) text += "│\n" buttons.append([InlineKeyboardButton(text=gethistory[x]['chat_name'], callback_data="connect({})".format(gethistory[x]['chat_id']))]) text += "╘══「 Total {} Chats 」".format( str(len(gethistory)) + " (max)" if len(gethistory) == 5 else str(len(gethistory))) conn_hist = InlineKeyboardMarkup(buttons) elif buttons: conn_hist = InlineKeyboardMarkup([buttons]) else: conn_hist = None send_message(msg, text, parse_mode="markdown", reply_markup=conn_hist) else: getstatusadmin = bot.get_chat_member(chat.id, msg.from_user.id) isadmin = getstatusadmin.status in ADMIN_STATUS ismember = getstatusadmin.status in MEMBER_STAUS isallow = sql.allow_connect_to_chat(chat.id) if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS): connection_status = sql.connect(msg.from_user.id, chat.id) if connection_status: chat_name = dispatcher.bot.getChat(chat.id).title send_message(msg, "Successfully connected to *{}*.".format(chat_name), parse_mode=ParseMode.MARKDOWN) try: sql.add_history_conn(user.id, str(chat.id), chat_name) bot.send_message(msg.from_user.id, f"You have connected with *{chat_name}*." f" Use /connection for see current available commands.", parse_mode="markdown") except BadRequest: pass except Unauthorized: pass else: send_message(msg, "Connection failed!") else: send_message(msg, "Connection to this chat is not allowed!")
def allow_connections(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat if chat.type != chat.PRIVATE: if len(args) >= 1: var = args[0] if var == "no": sql.set_allow_connect_to_chat(chat.id, False) send_message(update.effective_message, "Connection has been disabled for this chat") elif var == "yes": sql.set_allow_connect_to_chat(chat.id, True) send_message(update.effective_message, "Connection has been enabled for this chat") else: send_message(update.effective_message, "Please enter `yes` or `no`!", parse_mode=ParseMode.MARKDOWN) else: get_settings = sql.allow_connect_to_chat(chat.id) if get_settings: send_message(update.effective_message, "Connections to this group are *Allowed* for members!", parse_mode=ParseMode.MARKDOWN) else: send_message(update.effective_message, "Connection to this group are *Not Allowed* for members!", parse_mode=ParseMode.MARKDOWN) else: send_message(update.effective_message, "This command is for group only. Not in PM!")
def add_blackliststicker(bot: Bot, update: Update): msg = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] words = msg.text.split(None, 1) conn = connected(bot, update, chat, user.id) if conn: chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: chat_id = update.effective_chat.id if chat.type == "private": return else: chat_name = chat.title if len(words) > 1: text = words[1].replace('https://t.me/addstickers/', '') to_blacklist = list(set(trigger.strip() for trigger in text.split("\n") if trigger.strip())) added = 0 for trigger in to_blacklist: try: get = bot.getStickerSet(trigger) sql.add_to_stickers(chat_id, trigger.lower()) added += 1 except BadRequest: send_message(update.effective_message, "Sticker `{}` can not be found!".format(trigger), parse_mode="markdown") if added == 0: return if len(to_blacklist) == 1: send_message(update.effective_message, "Sticker <code>{}</code> added to blacklist stickers in <b>{}</b>!".format(html.escape(to_blacklist[0]), chat_name), parse_mode=ParseMode.HTML) else: send_message(update.effective_message, "<code>{}</code> stickers added to blacklist sticker in <b>{}</b>!".format(added, chat_name), parse_mode=ParseMode.HTML) elif msg.reply_to_message: added = 0 trigger = msg.reply_to_message.sticker.set_name if trigger == None: send_message(update.effective_message, "Sticker is invalid!") return try: get = bot.getStickerSet(trigger) sql.add_to_stickers(chat_id, trigger.lower()) added += 1 except BadRequest: send_message(update.effective_message, "Sticker `{}` can not be found!".format(trigger), parse_mode="markdown") if added == 0: return send_message(update.effective_message, "Sticker <code>{}</code> added to blacklist stickers in <b>{}</b>!".format(trigger, chat_name), parse_mode=ParseMode.HTML) else: send_message(update.effective_message, "Tell me what stickers you want to add to the blacklist.")
def blacklist_mode(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] msg = update.effective_message # type: Optional[Message] conn = connected(bot, update, chat, user.id, need_admin=True) if conn: chat = dispatcher.bot.getChat(conn) chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: if update.effective_message.chat.type == "private": send_message(update.effective_message, "You can do this command in groups, not PM") return "" chat = update.effective_chat chat_id = update.effective_chat.id chat_name = update.effective_message.chat.title if args: if args[0].lower() == 'off' or args[0].lower() == 'nothing' or args[0].lower() == 'no': settypeblacklist = 'turn off' sql.set_blacklist_strength(chat_id, 0, "0") elif args[0].lower() == 'del' or args[0].lower() == 'delete': settypeblacklist = 'left, the message will be deleted' sql.set_blacklist_strength(chat_id, 1, "0") elif args[0].lower() == 'warn': settypeblacklist = 'warned' sql.set_blacklist_strength(chat_id, 2, "0") elif args[0].lower() == 'mute': settypeblacklist = 'muted' sql.set_blacklist_strength(chat_id, 3, "0") elif args[0].lower() == 'kick': settypeblacklist = 'kicked' sql.set_blacklist_strength(chat_id, 4, "0") elif args[0].lower() == 'ban': settypeblacklist = 'banned' sql.set_blacklist_strength(chat_id, 5, "0") elif args[0].lower() == 'tban': if len(args) == 1: teks = """It looks like you are trying to set a temporary value to blacklist, but has not determined the time; use `/blstickermode tban <timevalue>`. Examples of time values: 4m = 4 minute, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" send_message(update.effective_message, teks, parse_mode="markdown") return settypeblacklist = 'temporary banned for {}'.format(args[1]) sql.set_blacklist_strength(chat_id, 6, str(args[1])) elif args[0].lower() == 'tmute': if len(args) == 1: teks = """It looks like you are trying to set a temporary value to blacklist, but has not determined the time; use `/blstickermode tmute <timevalue>`. Examples of time values: 4m = 4 minute, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""" send_message(update.effective_message, teks, parse_mode="markdown") return settypeblacklist = tl(update.effective_message, 'temporary muted for {}').format(args[1]) sql.set_blacklist_strength(chat_id, 7, str(args[1])) else: send_message(update.effective_message, "I only understand off/del/warn/ban/kick/mute/tban/tmute!") return if conn: text = "Blacklist sticker mode changed, users will be `{}` at *{}*!".format(settypeblacklist, chat_name) else: text = "Blacklist sticker mode changed, users will be `{}`!".format(settypeblacklist) send_message(update.effective_message, text, parse_mode="markdown") return "<b>{}:</b>\n" \ "<b>Admin:</b> {}\n" \ "Changed sticker blacklist mode. users will be {}.".format(html.escape(chat.title), mention_html(user.id, user.first_name), settypeblacklist) else: getmode, getvalue = sql.get_blacklist_setting(chat.id) if getmode == 0: settypeblacklist = "not active" elif getmode == 1: settypeblacklist = "hapus" elif getmode == 2: settypeblacklist = "warn" elif getmode == 3: settypeblacklist = "mute" elif getmode == 4: settypeblacklist = "kick" elif getmode == 5: settypeblacklist = "ban" elif getmode == 6: settypeblacklist = "temporarily banned for {}".format(getvalue) elif getmode == 7: settypeblacklist = "temporarily muted for {}".format(getvalue) if conn: text = "Blacklist sticker mode is currently set to *{}* in *{}*.".format(settypeblacklist, chat_name) else: text = "Blacklist sticker mode is currently set to *{}*.".format(settypeblacklist) send_message(update.effective_message, text, parse_mode=ParseMode.MARKDOWN) return ""
def unblackliststicker(bot: Bot, update: Update): msg = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] words = msg.text.split(None, 1) conn = connected(bot, update, chat, user.id) if conn: chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: chat_id = update.effective_chat.id if chat.type == "private": return else: chat_name = chat.title if len(words) > 1: text = words[1].replace('https://t.me/addstickers/', '') to_unblacklist = list(set(trigger.strip() for trigger in text.split("\n") if trigger.strip())) successful = 0 for trigger in to_unblacklist: success = sql.rm_from_stickers(chat_id, trigger.lower()) if success: successful += 1 if len(to_unblacklist) == 1: if successful: send_message(update.effective_message, "Sticker <code>{}</code> deleted from blacklist in <b>{}</b>!".format(html.escape(to_unblacklist[0]), chat_name), parse_mode=ParseMode.HTML) else: send_message(update.effective_message, "This sticker is not on the blacklist...!") elif successful == len(to_unblacklist): send_message(update.effective_message, "Sticker <code>{}</code> deleted from blacklist in <b>{}</b>!".format( successful, chat_name), parse_mode=ParseMode.HTML) elif not successful: send_message(update.effective_message, "None of these stickers exist, so they cannot be removed.".format( successful, len(to_unblacklist) - successful), parse_mode=ParseMode.HTML) else: send_message(update.effective_message, "Sticker <code>{}</code> deleted from blacklist. {} did not exist, so it's not deleted.".format(successful, len(to_unblacklist) - successful), parse_mode=ParseMode.HTML) elif msg.reply_to_message: trigger = msg.reply_to_message.sticker.set_name if trigger == None: send_message(update.effective_message, "Sticker is invalid!") return success = sql.rm_from_stickers(chat_id, trigger.lower()) if success: send_message(update.effective_message, "Sticker <code>{}</code> deleted from blacklist in <b>{}</b>!".format(trigger, chat_name), parse_mode=ParseMode.HTML) else: send_message(update.effective_message, "{} not found on blacklisted stickers...!".format(trigger)) else: send_message(update.effective_message, "Tell me what stickers you want to add to the blacklist.")