def leave_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) fed_info = sql.get_fed_info(fed_id) # administrators = chat.get_administrators().status getuser = bot.get_chat_member(chat.id, user.id).status if getuser in 'creator' or user.id in SUDO_USERS: if sql.chat_leave_fed(chat.id) == True: update.effective_message.reply_text("This chat has left the federation: {}!".format(fed_info['fname'])) else: update.effective_message.reply_text("How can you leave a federation that you never joined?!") else: update.effective_message.reply_text("Only group creators can use this command!")
def del_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text("Only fed owner can do this!") return if len(args) >= 1: fed_id = args[0] sql.del_fed(fed_id, chat.id) update.effective_message.reply_text("Deleted!") else: update.effective_message.reply_text("Please write federation id to remove!")
def welcome_fed(bot, update): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) fban = fban = sql.get_fban_user(fed_id, user.id) if not fban == False: update.effective_message.reply_text( tld( chat.id, "This user is banned in current federation! I will remove him." )) bot.kick_chat_member(chat.id, user.id) return True else: return False
def del_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text(tld(chat.id, "At the moment, We only support deleting federation on the group that joined it.")) return if not is_user_fed_owner(fed_id, user.id): update.effective_message.reply_text(tld(chat.id, "Only fed owner can do this!")) return sql.del_fed(fed_id, chat.id) update.effective_message.reply_text(tld(chat.id, "Deleted!"))
def broadcast(bot: Bot, update: Update, args: List[str]): to_send = update.effective_message.text.split(None, 1) if len(to_send) >= 2: chat = update.effective_chat # type: Optional[Chat] fed_id = sql.get_fed_id(chat.id) chats = sql.all_fed_chats(fed_id) failed = 0 for Q in chats: try: bot.sendMessage(Q, to_send[1]) sleep(0.1) except TelegramError: failed += 1 LOGGER.warning("Couldn't send broadcast to %s, group name %s", str(chat.chat_id), str(chat.chat_name)) update.effective_message.reply_text("Federations Broadcast complete. {} groups failed to receive the message, probably " "due to left federation.").format(failed)
def __user_info__(user_id, chat_id): fed_id = sql.get_fed_id(chat_id) if fed_id: fban = sql.get_fban_user(fed_id, user_id) info = sql.get_fed_info(fed_id) infoname = info.fed_name if is_user_fed_admin(fed_id, user_id) == True: text = f"This user is a fed admin in the current federation, <code>{infoname}</code>." elif not fban == False: text = "Banned in current federation - <b>Yes</b>" text += "\n<b>Reason:</b> {}".format(fban) else: text = "Banned in current federation - <b>No</b>" else: text = "" return text
def get_frules(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text(tld(chat.id, "This chat is not in any federation!")) return ruless = sql.get_frules(fed_id) try: rules = ruless.rules print(rules) text = "*Rules in this fed:*\n" text += rules update.effective_message.reply_text(tld(chat.id, text), parse_mode=ParseMode.MARKDOWN) return except AttributeError: update.effective_message.reply_text(tld(chat.id, "There are no rules in this federation!")) return
def join_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message administrators = chat.get_administrators() fed_id = sql.get_fed_id(chat.id) if user.id in SUDO_USERS: pass else: for admin in administrators: status = admin.status if status == "creator": print(admin) if str(admin.user.id) == str(user.id): pass else: update.effective_message.reply_text( tld(chat.id, "Only group creator can do it!")) return if fed_id: message.reply_text( tld(chat.id, "Uh, Are you gonna join two federations at one chat?")) return if len(args) >= 1: fedd = args[0] print(fedd) if sql.search_fed_by_id(fedd) == False: message.reply_text( tld(chat.id, "Please enter valid federation id.")) return x = sql.chat_join_fed(fedd, chat.id) if not x: message.reply_text( tld( chat.id, "Failed to join to federation! Due to some errors that basically I have no idea, try reporting it in support group!" )) return message.reply_text(tld(chat.id, "Chat joined to federation!"))
def fed_info(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) if not fed_id: update.effective_message.reply_text( tld(chat.id, "This group is not in any federation!")) return if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed admins can do this!")) return OW = bot.get_chat(info.owner_id) HAHA = OW.id FEDADMIN = sql.all_fed_users(fed_id) FEDADMIN.append(int(HAHA)) ACTUALADMIN = len(FEDADMIN) print(fed_id) user = update.effective_user # type: Optional[Chat] chat = update.effective_chat # type: Optional[Chat] info = sql.get_fed_info(fed_id) text = "<b>Fed info:</b>" text += "\nFedID: <code>{}</code>".format(fed_id) text += "\nName: {}".format(info.fed_name) text += "\nCreator: {}".format(mention_html(HAHA, "this guy")) text += "\nNumber of admins: <code>{}</code>".format(ACTUALADMIN) R = 0 for O in sql.get_all_fban_users(fed_id): R = R + 1 text += "\nNumber of bans: <code>{}</code>".format(R) h = sql.all_fed_chats(fed_id) asdf = len(h) text += "\nNumber of connected chats: <code>{}</code>".format(asdf) update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def user_demote_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed owner can do this!")) return msg = update.effective_message # type: Optional[Message] user_id = extract_user(msg, args) if user_id: user = 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(tld(chat.id, "I can't extract a user from this.")) return else: return if sql.search_user_in_fed(fed_id, user_id) == False: update.effective_message.reply_text( tld( chat.id, "I can't demote a user which is not even a fed admin! If you wanna bring him to tears, promote him and then demote." )) return res = sql.user_demote_fed(fed_id, user_id) if res == True: update.effective_message.reply_text(tld(chat.id, "Get out of here!")) else: update.effective_message.reply_text( tld(chat.id, "I can not remove him, I am powerless!"))
def user_join_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed owner can do this!")) return msg = update.effective_message # type: Optional[Message] user_id = extract_user(msg, args) if user_id: user = 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(tld(chat.id, "I can't extract a user from this.")) return else: return print(sql.search_user_in_fed(fed_id, user_id)) if not sql.search_user_in_fed(fed_id, user_id) == False: update.effective_message.reply_text( tld( chat.id, "I can't promote a user which is already a fed admin! But I can demote him." )) return res = sql.user_join_fed(fed_id, user_id) update.effective_message.reply_text(tld(chat.id, "Promoted in federation!"))
def user_demote_fed(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id): msg = update.effective_message # type: Optional[Message] user_id = extract_user(msg, args) if user_id: user = 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 cannot extract users from this message") return else: LOGGER.warning('error') if user_id == bot.id: update.effective_message.reply_text("Are you trying to demote me as a federation admin? Do you think I am stupid?") return if sql.search_user_in_fed(fed_id, user_id) == False: update.effective_message.reply_text("I cannot demote people who are not federation admins!") return res = sql.user_demote_fed(fed_id, user_id) if res == True: update.effective_message.reply_text("Get out of here!") else: update.effective_message.reply_text("Demotion failed!") else: update.effective_message.reply_text("Only federation owners can do this!") return
def fed_info(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) if not fed_id: update.effective_message.reply_text("This group is not in any federation!") return if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text("Only a federation admin can do this!") return owner = bot.get_chat(info['owner']) try: owner_name = owner.first_name + " " + owner.last_name except: owner_name = owner.first_name FEDADMIN = sql.all_fed_users(fed_id) FEDADMIN.append(int(owner.id)) TotalAdminFed = len(FEDADMIN) user = update.effective_user # type: Optional[Chat] chat = update.effective_chat # type: Optional[Chat] info = sql.get_fed_info(fed_id) text = "<b>ℹ️ Federation Information:</b>" text += "\nFedID: <code>{}</code>".format(fed_id) text += "\nName: {}".format(info['fname']) text += "\nCreator: {}".format(mention_html(owner.id, owner_name)) text += "\nAll Admins: <code>{}</code>".format(TotalAdminFed) getfban = sql.get_all_fban_users(fed_id) text += "\nTotal banned users: <code>{}</code>".format(len(getfban)) getfchat = sql.all_fed_chats(fed_id) text += "\nNumber of groups in this federation: <code>{}</code>".format(len(getfchat)) update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def set_frules(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text( tld(chat.id, "This chat is not in any federation!")) return if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed admins can do this!")) return if len(args) >= 1: msg = update.effective_message # type: Optional[Message] raw_text = msg.text args = raw_text.split( None, 1) # use python's maxsplit to separate cmd and args if len(args) == 2: txt = args[1] offset = len(txt) - len( raw_text) # set correct offset relative to command markdown_rules = markdown_parser(txt, entities=msg.parse_entities(), offset=offset) x = sql.set_frules(fed_id, markdown_rules) if not x: update.effective_message.reply_text( tld( chat.id, "Big F! There is an error while setting federation rules! If you wondered why please ask it in support group!" )) return rules = sql.get_fed_info(fed_id).fed_name update.effective_message.reply_text( tld(chat.id, f"Rules are set for {rules}!")) else: update.effective_message.reply_text( tld(chat.id, "Please write rules to set it up!"))
def __user_info__(user_id, chat_id): fed_id = sql.get_fed_id(chat_id) if fed_id: fban, fbanreason = sql.get_fban_user(fed_id, user_id) info = sql.get_fed_info(fed_id) infoname = info['fname'] if int(info['owner']) == user_id: text = "This user is the owner of the current Federation: <b>{}</b>.".format(infoname) elif is_user_fed_admin(fed_id, user_id): text = "This user is the admin of the current Federation: <b>{}</b>.".format(infoname) elif fban: text = "Banned in the current Federation: <b>Yes</b>" text += "\n<b>Reason:</b> {}".format(fbanreason) else: text = "Banned in the current Federation: <b>No</b>" else: text = "" return text
def fed_admin(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text("This group is not in any federation!") return if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text("Only federation admins can do this!") return user = update.effective_user # type: Optional[Chat] chat = update.effective_chat # type: Optional[Chat] info = sql.get_fed_info(fed_id) text = "<b>Federation Admin {}:</b>\n\n".format(info['fname']) text += "👑 Owner:\n" owner = bot.get_chat(info['owner']) try: owner_name = owner.first_name + " " + owner.last_name except: owner_name = owner.first_name text += " • {}\n".format(mention_html(owner.id, owner_name)) members = sql.all_fed_members(fed_id) if len(members) == 0: text += "\n🔱 There is no admin in this federation" else: text += "\n🔱 Admin:\n" for x in members: user = bot.get_chat(x) text += " • {}\n".format(mention_html(user.id, user.first_name)) update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def set_frules(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text("Only fed admins can do this!") return if len(args) >= 1: msg = update.effective_message # type: Optional[Message] raw_text = msg.text args = raw_text.split(None, 1) # use python's maxsplit to separate cmd and args if len(args) == 2: txt = args[1] offset = len(txt) - len(raw_text) # set correct offset relative to command markdown_rules = markdown_parser(txt, entities=msg.parse_entities(), offset=offset) sql.set_frules(fed_id, markdown_rules) update.effective_message.reply_text("Rules setuped for this fed!") else: update.effective_message.reply_text("Please write rules!")
def fed_notif(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] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text("This group is not a part of any federation!") return if args: if args[0] in ("yes", "on"): sql.set_feds_setting(user.id, True) msg.reply_text("Reporting turned on! Users fbanned/un-fbanned by you will be notified via PM.") elif args[0] in ("no", "off"): sql.set_feds_setting(user.id, False) msg.reply_text("Reporting turned off! Users fbanned/un-fbanned will not be notified via PM.") else: msg.reply_text("Please enter `on`/`off`", parse_mode="markdown") else: getreport = sql.user_feds_report(user.id) msg.reply_text("Your current Federation report preferences: `{}`".format(getreport), parse_mode="markdown")
def fed_chat(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) user_id = update.effective_message.from_user.id if not is_user_admin(update.effective_chat, user_id): update.effective_message.reply_text("You must be an admin to execute this command.") return if not fed_id: update.effective_message.reply_text("This group is not in any federation!") return user = update.effective_user # type: Optional[Chat] chat = update.effective_chat # type: Optional[Chat] info = sql.get_fed_info(fed_id) text = "This chat is part of the following federation:" text += "\n{} (ID: <code>{}</code>)".format(info['fname'], fed_id) update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def user_demote_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id): msg = update.effective_message # type: Optional[Message] user_id = extract_user(msg, args) if user_id: user = 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 cannot extract users from this message.") return else: LOGGER.warning('error') if user_id == bot.id: update.effective_message.reply_text("Boi, what are you even trying to do?") return if sql.search_user_in_fed(fed_id, user_id) == False: update.effective_message.reply_text("This user isn't even a federation admin!") return res = sql.user_demote_fed(fed_id, user_id) if res == True: update.effective_message.reply_text("Get out of here!") else: update.effective_message.reply_text("Failed to demote!") else: update.effective_message.reply_text("Only federation owners can do this!") return
def join_fed(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message administrators = chat.get_administrators() fed_id = sql.get_fed_id(chat.id) if user.id in SUDO_USERS: pass else: for admin in administrators: status = admin.status if status == "creator": if str(admin.user.id) == str(user.id): pass else: update.effective_message.reply_text("Only group creators can use this command!") return if fed_id: message.reply_text("You cannot join two federations from one chat") return if len(args) >= 1: getfed = sql.search_fed_by_id(args[0]) if getfed == False: message.reply_text("Please enter a valid federation ID") return x = sql.chat_join_fed(args[0], chat.id) if not x: message.reply_text("Failed to join federation! Please contact @onepunchsupport if this problem still persists!") return message.reply_text("This chat has joined the federation: {}!".format(getfed['fname']))
def fed_broadcast(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") msg = update.effective_message # type: Optional[Message] user = update.effective_user # type: Optional[User] if args: chat = update.effective_chat # type: Optional[Chat] fed_id = sql.get_fed_id(chat.id) fedinfo = sql.get_fed_info(fed_id) text = "*New broadcast from the Federation {}*\n".format(fedinfo['fname']) # Parsing md raw_text = msg.text args = raw_text.split(None, 1) # use python's maxsplit to separate cmd and args txt = args[1] offset = len(txt) - len(raw_text) # set correct offset relative to command text_parser = markdown_parser(txt, entities=msg.parse_entities(), offset=offset) text += text_parser try: broadcaster = user.first_name except: broadcaster = user.first_name + " " + user.last_name text += "\n\n- {}".format(mention_markdown(user.id, broadcaster)) chat_list = sql.all_fed_chats(fed_id) failed = 0 for chat in chat_list: try: bot.sendMessage(chat, text, parse_mode="markdown") except TelegramError: failed += 1 LOGGER.warning("Couldn't send broadcast to %s, group name %s", str(chat.chat_id), str(chat.chat_name)) send_text = "The federation broadcast is complete" if failed >= 1: send_text += "{} the group failed to receive the message, probably because it left the Federation.".format(failed) update.effective_message.reply_text(send_text)
def fed_chats(bot: Bot, update: Update, args: List[str]): spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text("Spammer detected! Ignoring user.") chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) if not fed_id: update.effective_message.reply_text("This group is not a part of any federation!") return if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text("Only federation admins can do this!") return getlist = sql.all_fed_chats(fed_id) if len(getlist) == 0: update.effective_message.reply_text("No users are fbanned from the federation {}".format(info['fname']), parse_mode=ParseMode.HTML) return text = "<b>New chat joined the federation {}:</b>\n".format(info['fname']) for chats in getlist: chat_name = dispatcher.bot.getChat(chats).title text += " • {} (<code>{}</code>)\n".format(chat_name, chats) try: update.effective_message.reply_text(text, parse_mode=ParseMode.HTML) except: cleanr = re.compile('<.*?>') cleantext = re.sub(cleanr, '', text) with BytesIO(str.encode(cleantext)) as output: output.name = "fbanlist.txt" update.effective_message.reply_document(document=output, filename="fbanlist.txt", caption="Here is a list of all the chats that joined the federation {}.".format(info['fname']))
def join_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message administrators = chat.get_administrators() fed_id = sql.get_fed_id(chat.id) if user.id in SUDO_USERS: pass else: for admin in administrators: status = admin.status if status == "creator": print(admin) if str(admin.user.id) == str(user.id): pass else: update.effective_message.reply_text("Only the group creator can do it!") return if fed_id: message.reply_text("Uh, you can only join one federation in a chat.") return if len(args) >= 1: fedd = args[0] print(fedd) if sql.search_fed_by_id(fedd) == False: message.reply_text("Please enter a valid federation ID.") return x = sql.chat_join_fed(fedd, chat.id) if not x: message.reply_text("Failed to join federation! Please head to @tohsakas to report this.") return message.reply_text("Chat successfully added to federation!")
def leave_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) administrators = chat.get_administrators() if user.id in SUDO_USERS: pass else: for admin in administrators: status = admin.status if status == "creator": print(admin) if str(admin.user.id) == str(user.id): pass else: update.effective_message.reply_text("Only group creator can do it!") return if sql.chat_leave_fed(chat.id) == True: update.effective_message.reply_text("Leaved from fed!") else: update.effective_message.reply_text("Error!")
def fed_ban_list(bot: Bot, update: Update, args: List[str], chat_data): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) if not fed_id: update.effective_message.reply_text("This group is not a part of any federation!") return if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text("Only federation owners can do this!") return user = update.effective_user # type: Optional[Chat] chat = update.effective_chat # type: Optional[Chat] getfban = sql.get_all_fban_users(fed_id) if len(getfban) == 0: update.effective_message.reply_text("The federation ban list of {} is empty.".format(info['fname']), parse_mode=ParseMode.HTML) return if args: if args[0] == 'json': jam = time.time() new_jam = jam + 1800 cek = get_chat(chat.id, chat_data) if cek.get('status'): if jam <= int(cek.get('value')): waktu = time.strftime("%H:%M:%S %d/%m/%Y", time.localtime(cek.get('value'))) update.effective_message.reply_text("You can back up your data once every 30 minutes!\nYou can back up data again at `{}`".format(waktu), parse_mode=ParseMode.MARKDOWN) return else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) backups = "" for users in getfban: getuserinfo = sql.get_all_fban_users_target(fed_id, users) json_parser = {"user_id": users, "first_name": getuserinfo['first_name'], "last_name": getuserinfo['last_name'], "user_name": getuserinfo['user_name'], "reason": getuserinfo['reason']} backups += json.dumps(json_parser) backups += "\n" with BytesIO(str.encode(backups)) as output: output.name = "rin_fbanned_users.json" update.effective_message.reply_document(document=output, filename="rin_fbanned_users.json", caption="Total {} User are blocked by the Federation {}.".format(len(getfban), info['fname'])) return elif args[0] == 'csv': jam = time.time() new_jam = jam + 1800 cek = get_chat(chat.id, chat_data) if cek.get('status'): if jam <= int(cek.get('value')): waktu = time.strftime("%H:%M:%S %d/%m/%Y", time.localtime(cek.get('value'))) update.effective_message.reply_text("You can back up data once every 30 minutes!\nYou can back up data again at `{}`".format(waktu), parse_mode=ParseMode.MARKDOWN) return else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) backups = "id,firstname,lastname,username,reason\n" for users in getfban: getuserinfo = sql.get_all_fban_users_target(fed_id, users) backups += "{user_id},{first_name},{last_name},{user_name},{reason}".format(user_id=users, first_name=getuserinfo['first_name'], last_name=getuserinfo['last_name'], user_name=getuserinfo['user_name'], reason=getuserinfo['reason']) backups += "\n" with BytesIO(str.encode(backups)) as output: output.name = "rin_fbanned_users.csv" update.effective_message.reply_document(document=output, filename="rin_fbanned_users.csv", caption="Total {} User are blocked by Federation {}.".format(len(getfban), info['fname'])) return text = "<b>{} users have been banned from the federation {}:</b>\n".format(len(getfban), info['fname']) for users in getfban: getuserinfo = sql.get_all_fban_users_target(fed_id, users) if getuserinfo == False: text = "There are no users banned from the federation {}".format(info['fname']) break user_name = getuserinfo['first_name'] if getuserinfo['last_name']: user_name += " " + getuserinfo['last_name'] text += " • {} (<code>{}</code>)\n".format(mention_html(users, user_name), users) try: update.effective_message.reply_text(text, parse_mode=ParseMode.HTML) except: jam = time.time() new_jam = jam + 1800 cek = get_chat(chat.id, chat_data) if cek.get('status'): if jam <= int(cek.get('value')): waktu = time.strftime("%H:%M:%S %d/%m/%Y", time.localtime(cek.get('value'))) update.effective_message.reply_text("You can back up data once every 30 minutes!\nYou can back up data again at `{}`".format(waktu), parse_mode=ParseMode.MARKDOWN) return else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) cleanr = re.compile('<.*?>') cleantext = re.sub(cleanr, '', text) with BytesIO(str.encode(cleantext)) as output: output.name = "fbanlist.txt" update.effective_message.reply_document(document=output, filename="fbanlist.txt", caption="The following is the list of users who are currently fbanned in the Federation {}.".format(info['fname']))
def fed_import_bans(bot: Bot, update: Update, chat_data): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] msg = update.effective_message # type: Optional[Message] fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) if not fed_id: update.effective_message.reply_text("This group is not a part of any federation!") return if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text("Only federation owners can do this!") return if msg.reply_to_message and msg.reply_to_message.document: jam = time.time() new_jam = jam + 1800 cek = get_chat(chat.id, chat_data) if cek.get('status'): if jam <= int(cek.get('value')): waktu = time.strftime("%H:%M:%S %d/%m/%Y", time.localtime(cek.get('value'))) update.effective_message.reply_text("You can backup your data once every 30 minutes!\nYou can backup data again at `{}`".format(waktu), parse_mode=ParseMode.MARKDOWN) return else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) else: if user.id not in SUDO_USERS: put_chat(chat.id, new_jam, chat_data) if int(int(msg.reply_to_message.document.file_size)/1024) >= 200: msg.reply_text("This file is too big!") return success = 0 failed = 0 try: file_info = bot.get_file(msg.reply_to_message.document.file_id) except BadRequest: msg.reply_text("Try downloading and re-uploading the file, this one seems broken!") return fileformat = msg.reply_to_message.document.file_name.split('.')[-1] if fileformat == 'json': with BytesIO() as file: file_info.download(out=file) file.seek(0) reading = file.read().decode('UTF-8') splitting = reading.split('\n') for x in splitting: if x == '': continue try: data = json.loads(x) except json.decoder.JSONDecodeError as err: failed += 1 continue try: import_userid = int(data['user_id']) # Make sure it int import_firstname = str(data['first_name']) import_lastname = str(data['last_name']) import_username = str(data['user_name']) import_reason = str(data['reason']) except ValueError: failed += 1 continue # Checking user if int(import_userid) == bot.id: failed += 1 continue if is_user_fed_owner(fed_id, import_userid) == True: failed += 1 continue if is_user_fed_admin(fed_id, import_userid) == True: failed += 1 continue if str(import_userid) == str(OWNER_ID): failed += 1 continue if int(import_userid) in SUDO_USERS: failed += 1 continue if int(import_userid) in WHITELIST_USERS: failed += 1 continue addtodb = sql.fban_user(fed_id, str(import_userid), import_firstname, import_lastname, import_username, import_reason) if addtodb: success += 1 text = "Successfully imported! {} people are fbanned.".format(success) if failed >= 1: text += " {} Failed to import.".format(failed) elif fileformat == 'csv': with BytesIO() as file: file_info.download(out=file) file.seek(0) reading = file.read().decode('UTF-8') splitting = reading.split('\n') for x in splitting: if x == '': continue data = x.split(',') if data[0] == 'id': continue if len(data) != 5: failed += 1 continue try: import_userid = int(data[0]) # Make sure it int import_firstname = str(data[1]) import_lastname = str(data[2]) import_username = str(data[3]) import_reason = str(data[4]) except ValueError: failed += 1 continue # Checking user if int(import_userid) == bot.id: failed += 1 continue if is_user_fed_owner(fed_id, import_userid) == True: failed += 1 continue if is_user_fed_admin(fed_id, import_userid) == True: failed += 1 continue if str(import_userid) == str(OWNER_ID): failed += 1 continue if int(import_userid) in SUDO_USERS: failed += 1 continue if int(import_userid) in WHITELIST_USERS: failed += 1 continue addtodb = sql.fban_user(fed_id, str(import_userid), import_firstname, import_lastname, import_username, import_reason) if addtodb: success += 1 text = "Successfully imported. {} people are fbanned.".format(success) if failed >= 1: text += " {} failed to import.".format(failed) else: update.effective_message.reply_text("File not supported.") return update.effective_message.reply_text(text)
def unfban(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message # type: Optional[Message] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text( tld(chat.id, "This group is not in any federation!")) return info = sql.get_fed_info(fed_id) if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed admins can do this!")) return user_id = extract_user(message, args) if not user_id: message.reply_text( tld(chat.id, "You don't seem to be referring to a user.")) return user_chat = bot.get_chat(user_id) if user_chat.type != 'private': message.reply_text(tld(chat.id, "That's not a user!")) return if sql.get_fban_user(fed_id, user_id) == False: message.reply_text(tld(chat.id, "This user is not fbanned!")) return banner = update.effective_user # type: Optional[User] message.reply_text( tld(chat.id, "I'll give {} a second chance in this federation.").format( user_chat.first_name)) h = sql.all_fed_chats(fed_id) for O in h: try: member = bot.get_chat_member(O, user_id) if member.status == 'kicked': bot.unban_chat_member(O, user_id) except BadRequest as excp: if excp.message in UNFBAN_ERRORS: pass else: message.reply_text( tld(chat.id, "Could not un-fban due to: {}").format(excp.message)) return except TelegramError: pass try: x = sql.un_fban_user(fed_id, user_id) if not x: message.reply_text( tld(chat.id, "Failed to fban, This user is probably fbanned!")) return except: pass message.reply_text(tld(chat.id, "Person has been un-fbanned.")) OW = bot.get_chat(info.owner_id) HAHA = OW.id FEDADMIN = sql.all_fed_users(fed_id) FEDADMIN.append(int(HAHA)) send_to_list(bot, FEDADMIN, "<b>Un-FedBan</b>" \ "\n<b>Fed:</b> {}" \ "\n<b>FedAdmin:</b> {}" \ "\n<b>User:</b> {}" \ "\n<b>User ID:</b> <code>{}</code>".format(info.fed_name, mention_html(user.id, user.first_name), mention_html(user_chat.id, user_chat.first_name), user_chat.id), html=True)
def user_join_fed(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if is_user_fed_owner(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed owner can do this!")) return msg = update.effective_message # type: Optional[Message] user_id = extract_user(msg, args) if user_id: user = 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(tld(chat.id, "I can't extract a user from this.")) return else: return print(sql.search_user_in_fed(fed_id, user_id)) #if user_id == user_id: # update.effective_message.reply_text(tld(chat.id, "Are you gonna promote yourself?")) # return fed_id = sql.get_fed_id(chat.id) info = sql.get_fed_info(fed_id) OW = bot.get_chat(info.owner_id) HAHA = OW.id if user_id == HAHA: update.effective_message.reply_text( tld(chat.id, "Why are you trying to promote federation owner!?")) return if not sql.search_user_in_fed(fed_id, user_id) == False: update.effective_message.reply_text( tld( chat.id, "I can't promote user which is already a fed admin! But I can demote them." )) return if user_id == bot.id: update.effective_message.reply_text( tld( chat.id, "I am already the federation admin and the one that manage it!" )) return #else: # return res = sql.user_join_fed(fed_id, user_id) if not res: update.effective_message.reply_text( tld( chat.id, "Failed to promoted! It might be because you are admin in another federation! Our code is still buggy, We are sorry for that!" )) return update.effective_message.reply_text(tld(chat.id, "Promoted Successfully!"))
def fed_ban(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] fed_id = sql.get_fed_id(chat.id) if not fed_id: update.effective_message.reply_text( tld(chat.id, "This group is not in any federation!")) return info = sql.get_fed_info(fed_id) OW = bot.get_chat(info.owner_id) HAHA = OW.id FEDADMIN = sql.all_fed_users(fed_id) FEDADMIN.append(int(HAHA)) if is_user_fed_admin(fed_id, user.id) == False: update.effective_message.reply_text( tld(chat.id, "Only fed admins can do this!")) return message = update.effective_message # type: Optional[Message] user_id, reason = extract_user_and_text(message, args) fban = sql.get_fban_user(fed_id, user_id) if not fban == False: update.effective_message.reply_text( tld(chat.id, "This user already fbanned!")) return if not user_id: message.reply_text( tld(chat.id, "You don't seem to be referring to a user.")) return if user_id == bot.id: message.reply_text( tld( chat.id, "You can't fban me, better hit your head against the wall, it's more fun." )) return if is_user_fed_owner(fed_id, user_id) == True: message.reply_text( tld(chat.id, "Why you are trying to fban the federation owner?")) return if is_user_fed_admin(fed_id, user_id) == True: message.reply_text( tld(chat.id, "Why so serious trying to fban the federation admin?")) return if user_id == OWNER_ID: message.reply_text( tld(chat.id, "I'm not fbanning my master, That's pretty dumb idea!")) return if int(user_id) in SUDO_USERS: message.reply_text(tld(chat.id, "I'm not fbanning the bot sudoers!")) return if int(user_id) in WHITELIST_USERS: message.reply_text( tld(chat.id, "This person is whitelisted from being fbanned!")) return try: user_chat = bot.get_chat(user_id) except BadRequest as excp: message.reply_text(excp.message) return if user_chat.type != 'private': message.reply_text(tld(chat.id, "That's not a user!")) return ok123 = mention_html(user_chat.id, user_chat.first_name) ok1234 = info.fed_name text12 = f"Beginning federation ban of {ok123} in {ok1234}." update.effective_message.reply_text(text12, parse_mode=ParseMode.HTML) if reason == "": reason = "No Reason." x = sql.fban_user(fed_id, user_id, reason) if not x: message.reply_text( "Failed to federation ban! Probably this bug is not fixed yet due to the developer is lazy as f**k." ) return h = sql.all_fed_chats(fed_id) for O in h: try: bot.kick_chat_member(O, user_id) #text = tld(chat.id, "I should fban {}, but it's only test fban, right? So i let him live.").format(O) text = "Fbanning {}".format(user_id) #message.reply_text(text) except BadRequest as excp: if excp.message in FBAN_ERRORS: pass else: message.reply_text( tld(chat.id, "Could not fban due to: {}").format(excp.message)) return except TelegramError: pass message.reply_text(tld(chat.id, "Person has been fbanned.")) send_to_list(bot, FEDADMIN, "<b>New FedBan</b>" \ "\n<b>Fed:</b> {}" \ "\n<b>FedAdmin:</b> {}" \ "\n<b>User:</b> {}" \ "\n<b>User ID:</b> <code>{}</code>" \ "\n<b>Reason:</b> {}".format(info.fed_name, mention_html(user.id, user.first_name), mention_html(user_chat.id, user_chat.first_name), user_chat.id, reason), html=True)