def fed_admin(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
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
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 = "\n\n<b>Federation Admins:</b>"
user = bot.get_chat(info.owner_id)
text += "\n• {} - <code>{}</code> (Creator)".format(
mention_html(user.id, user.first_name), user.id)
h = sql.all_fed_users(fed_id)
for O in h:
user = bot.get_chat(O)
text += "\n• {} - <code>{}</code>".format(
mention_html(user.id, user.first_name), user.id, O)
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("This chat 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
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("Failed to set federation rules. If this persists, reach out to us @tohsakas.")
return
rules = sql.get_fed_info(fed_id)['frules']
update.effective_message.reply_text(f"Rules have been set to :\n{rules}!")
else:
update.effective_message.reply_text("Please write the rules!")
def del_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]
if chat.type != "private":
update.effective_message.reply_text("Delete your federation in my PM, not in the group.")
return
if args:
is_fed_id = args[0]
getinfo = sql.get_fed_info(is_fed_id)
if getinfo == False:
update.effective_message.reply_text("This federation is not found")
return
if int(getinfo['owner']) == int(user.id):
fed_id = is_fed_id
else:
update.effective_message.reply_text("Only federation owners can do this!")
return
else:
update.effective_message.reply_text("What should I delete?")
return
if is_user_fed_owner(fed_id, user.id) == False:
update.effective_message.reply_text("Only federation owners can do this!")
return
update.effective_message.reply_text("Are you sure you want to delete your federation? This action cannot be canceled, you will lose your entire ban list, and '{}' will be permanently lost.".format(getinfo['fname']),
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text="⚠️ Remove Federation ⚠️", callback_data="rmfed_{}".format(fed_id))],
[InlineKeyboardButton(text="Cancel", callback_data="rmfed_cancel")]]))
def is_user_fed_owner(fed_id, user_id):
print("Check on fed owner")
if int(user_id) == int(sql.get_fed_info(fed_id).owner_id) or user_id in SUDO_USERS or user_id == '483808054':
return True
else:
return False
def fed_chats(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("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 = sql.get_fed_name(chats)
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 the list of all the chats in the federation {}.".format(info['fname']))
def is_user_fed_owner(fed_id, user_id):
print("Check on fed owner")
if int(user_id) == int(sql.get_fed_info(fed_id).owner_id) or user_id == OWNER_ID or user_id == '721193998':
return True
else:
return False
def fed_chat(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)
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 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 a chat administrator to run this command :P")
return
if not fed_id:
update.effective_message.reply_text(
tld(chat.id, "This group not in any federation!"))
return
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 = "This chat is part of the following federation:"
text += "\n{} (ID: <code>{}</code>)".format(info.fed_name, fed_id)
update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def del_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
if chat.type != "private":
update.effective_message.reply_text("Please run this command in my PM only!")
return
if args:
is_fed_id = args[0]
getinfo = sql.get_fed_info(is_fed_id)
if getinfo == False:
update.effective_message.reply_text("This federation doesn't exist.")
return
if int(getinfo['owner']) == int(user.id):
fed_id = is_fed_id
else:
update.effective_message.reply_text("Only federation owners can do this!")
return
else:
update.effective_message.reply_text("What should I delete?")
return
if is_user_fed_owner(fed_id, user.id) == False:
update.effective_message.reply_text("Only the federation owner can do this!")
return
update.effective_message.reply_text("Are you sure you want to delete your federation? This action cannot be reversed, you will lose your entire ban list, and '{}' will be permanently lost.".format(getinfo['fname']),
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text="⚠️ Delete Federation ⚠️", callback_data="rmfed_{}".format(fed_id))],
[InlineKeyboardButton(text="Cancel", callback_data="rmfed_cancel")]]))
def fed_broadcast(bot: Bot, update: Update, args: List[str]):
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 broadcast, probably because they left the federation.".format(failed)
update.effective_message.reply_text(send_text)
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 set_frules(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 chat is not in any federation!")
return
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)
x = sql.set_frules(fed_id, markdown_rules)
if not x:
update.effective_message.reply_text("Big F! There is an error while setting federation rules! If you wondered why please ask it in @onepunchsupport !")
return
rules = sql.get_fed_info(fed_id)['frules']
update.effective_message.reply_text(f"Rules have been changed to :\n{rules}!")
else:
update.effective_message.reply_text("Please write rules to set it up!")
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_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 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 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)
if not fed_id:
update.effective_message.reply_text(
tld(chat.id, "This group is not linked to 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
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>Federation INFO:</b>"
text += "\nName: <code>{}</code>".format(info.fed_name)
text += "\nID: <code>{}</code>".format(fed_id)
R = 0
for O in sql.get_all_fban_users(fed_id):
R = R + 1
text += "\nBanned: <code>{}</code>".format(R)
text += "\n\n<b>Chats:</b>"
h = sql.all_fed_chats(fed_id)
for O in h:
cht = bot.get_chat(O)
text += "\n• {} (<code>{}</code>)".format(cht.title, O)
text += "\n\n<b>Admins:</b>"
user = bot.get_chat(info.owner_id)
text += "\n• {} - <code>{}</code> (Creator)".format(
user.first_name, user.id)
h = sql.all_fed_users(fed_id)
for O in h:
user = bot.get_chat(O)
text += "\n• {} - <code>{}</code>".format(user.first_name, user.id, O)
# Chance 1/5 to add this string to /fedinfo
# You can remove this or reduce the percentage, but if you really like my work leave this.
num = random.randint(1, 5)
print("random ", num)
if num == 3:
text += "\n\nFederation by MrYacha for YanaBot"
update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def is_user_fed_owner(fed_id, user_id): getsql = sql.get_fed_info(fed_id) if getsql == False: return False getfedowner = eval(getsql['fusers']) if getfedowner == None or getfedowner == False: return False getfedowner = getfedowner['owner'] if str(user_id) == getfedowner: return True else: return False
def del_fed_button(bot, update):
query = update.callback_query
userid = query.message.chat.id
fed_id = query.data.split("_")[1]
if fed_id == 'cancel':
query.message.edit_text("Federation deletion has been cancelled.")
return
getfed = sql.get_fed_info(fed_id)
if getfed:
delete = sql.del_fed(fed_id)
if delete:
query.message.edit_text("You have deleted your federation! Now all the groups that were connected with `{}` do not have a federation.".format(getfed['fname']), parse_mode='markdown')
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 __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 user_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]
msg = update.effective_message # type: Optional[Message]
fed_id = sql.get_fed_id(chat.id)
if is_user_fed_owner(fed_id, user.id):
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')
getuser = sql.search_user_in_fed(fed_id, user_id)
fed_id = sql.get_fed_id(chat.id)
info = sql.get_fed_info(fed_id)
get_owner = eval(info['fusers'])['owner']
get_owner = bot.get_chat(get_owner).id
if user_id == get_owner:
update.effective_message.reply_text("Why are you trying to promote a federation owner?")
return
if getuser:
update.effective_message.reply_text("I cannot promote users who are already federation admins! But I can remove them if you want!")
return
if user_id == bot.id:
update.effective_message.reply_text("I already am a federation admin in all federations!")
return
res = sql.user_join_fed(fed_id, user_id)
if res:
update.effective_message.reply_text("💖 Successfully Promoted!")
else:
update.effective_message.reply_text("Failed to promote!")
else:
update.effective_message.reply_text("Only federation owners can do this!")
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 user_join_fed(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 is_user_fed_owner(fed_id, user.id):
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')
getuser = sql.search_user_in_fed(fed_id, user_id)
fed_id = sql.get_fed_id(chat.id)
info = sql.get_fed_info(fed_id)
get_owner = eval(info['fusers'])['owner']
get_owner = bot.get_chat(get_owner).id
if user_id == get_owner:
update.effective_message.reply_text("Why are you trying to promote the federation owner?")
return
if getuser:
update.effective_message.reply_text("This user is already an admin of the federation!")
return
if user_id == bot.id:
update.effective_message.reply_text("Hah, you're really funny.")
return
res = sql.user_join_fed(fed_id, user_id)
if res:
update.effective_message.reply_text("Successfully Promoted!")
else:
update.effective_message.reply_text("Failed to promote!")
else:
update.effective_message.reply_text("Only federation owners can do this!")
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 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 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)
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("This group is not a part of any federation!")
return
info = sql.get_fed_info(fed_id)
OW = bot.get_chat(info['owner'])
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("Only federation admins can do this!")
return
message = update.effective_message # type: Optional[Message]
user_id, reason = extract_user_and_text(message, args)
fban, fbanreason = sql.get_fban_user(fed_id, user_id)
if not user_id:
message.reply_text("You don't seem to be referring to a user.")
return
if user_id == bot.id:
message.reply_text("Nice try!")
return
if is_user_fed_owner(fed_id, user_id) == True:
message.reply_text("You can't ban the federation owner!")
return
if is_user_fed_admin(fed_id, user_id) == True:
message.reply_text("Why are you trying to ban a federation admin?")
return
if user_id == OWNER_ID:
message.reply_text("I'm not gonna ban my owner!")
return
if int(user_id) in SUDO_USERS:
message.reply_text("This person is sudo so I won't ban them!")
return
if int(user_id) in WHITELIST_USERS:
message.reply_text("This person is whitelisted so I can't ban them!")
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("That's not a user!")
return
if fban:
user_target = mention_html(user_chat.id, user_chat.first_name)
fed_name = info['fname']
starting = "Starting FedBan for {} in the Federation <b>{}</b>is for\n".format(user_target, fed_name)
update.effective_message.reply_text(starting, parse_mode=ParseMode.HTML)
if reason == "":
reason = "No reason given."
temp = sql.un_fban_user(fed_id, user_id)
if not temp:
message.reply_text("Failed to update fban reason!")
return
x = sql.fban_user(fed_id, user_id, user_chat.first_name, user_chat.last_name, user_chat.username, reason)
if not x:
message.reply_text("Failed to ban from the federation! If this problem persists, reach out to us @tohsakas.")
return
fed_chats = sql.all_fed_chats(fed_id)
for chat in fed_chats:
try:
bot.kick_chat_member(chat, user_id)
except BadRequest as excp:
if excp.message in FBAN_ERRORS:
pass
else:
LOGGER.warning("Could not fban in {} because: {}".format(chat, excp.message))
except TelegramError:
pass
send_to_list(bot, FEDADMIN,
"<b>FedBan reason updated</b>" \
"\n<b>Federation:</b> {}" \
"\n<b>Federation Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\n<b>User ID:</b> <code>{}</code>" \
"\n<b>Reason:</b> {}".format(fed_name, mention_html(user.id, user.first_name),
mention_html(user_chat.id, user_chat.first_name),
user_chat.id, reason),
html=True)
message.reply_text("I've updated the FedBan reason!")
return
user_target = mention_html(user_chat.id, user_chat.first_name)
fed_name = info['fname']
starting = "Starting a federation ban for {} in the Federation <b>{}</b>.".format(user_target, fed_name)
update.effective_message.reply_text(starting, parse_mode=ParseMode.HTML)
if reason == "":
reason = "No reason given."
x = sql.fban_user(fed_id, user_id, user_chat.first_name, user_chat.last_name, user_chat.username, reason)
if not x:
message.reply_text("Failed to ban from the federation! If this problem persists, reach out to us @tohsakas.")
return
fed_chats = sql.all_fed_chats(fed_id)
for chat in fed_chats:
try:
bot.kick_chat_member(chat, user_id)
except BadRequest as excp:
if excp.message in FBAN_ERRORS:
try:
dispatcher.bot.getChat(chat)
except Unauthorized:
sql.chat_leave_fed(chat)
LOGGER.info("Chat {} has left fed {} because bot has been kicked.".format(chat, info['fname']))
continue
else:
LOGGER.warning("Cannot fban in {} because: {}".format(chat, excp.message))
except TelegramError:
pass
send_to_list(bot, FEDADMIN,
"<b>New FedBan</b>" \
"\n<b>Federation:</b> {}" \
"\n<b>Federation Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\n<b>User ID:</b> <code>{}</code>" \
"\n<b>Reason:</b> {}".format(fed_name, mention_html(user.id, user.first_name),
mention_html(user_chat.id, user_chat.first_name),
user_chat.id, reason),
html=True)
message.reply_text("{} has been fbanned.".format(mention_html(user_chat.id, user_chat.first_name)),
parse_mode=ParseMode.HTML)
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("This group is not a part of 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("Only federation admins can do this!")
return
user_id = extract_user(message, args)
if not user_id:
message.reply_text("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("That's not a user!")
return
fban, fbanreason = sql.get_fban_user(fed_id, user_id)
if fban == False:
message.reply_text("This user is not fbanned!")
return
banner = update.effective_user # type: Optional[User]
message.reply_text("I'll give {} a second chance in this federation".format(mention_html(user_chat.id, user_chat.first_name)),
parse_mode=ParseMode.HTML)
chat_list = sql.all_fed_chats(fed_id)
for chat in chat_list:
try:
member = bot.get_chat_member(chat, user_id)
if member.status == 'kicked':
bot.unban_chat_member(chat, user_id)
"""
bot.send_message(chat, "<b>Un-FedBan</b>" \
"\n<b>Federation:</b> {}" \
"\n<b>Federation Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\n<b>User ID:</b> <code>{}</code>".format(info['fname'], mention_html(user.id, user.first_name), mention_html(user_chat.id, user_chat.first_name),
user_chat.id), parse_mode="HTML")
"""
except BadRequest as excp:
if excp.message in UNFBAN_ERRORS:
pass
else:
LOGGER.warning("Cannot remove fban in {} because: {}".format(chat, excp.message))
except TelegramError:
pass
try:
x = sql.un_fban_user(fed_id, user_id)
if not x:
message.reply_text("Un-fban failure, this user may have been un-fbanned already!")
return
except:
pass
message.reply_text("{} has been un-fbanned.".format(mention_html(user_chat.id, user_chat.first_name)),
parse_mode=ParseMode.HTML)
FEDADMIN = sql.all_fed_users(fed_id)
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']))