def del_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
if chat.type != "private":
update.effective_message.reply_text(tld(chat.id, "common_cmd_pm_only"))
return
if args:
is_fed_id = args[0]
getinfo = sql.get_fed_info(is_fed_id)
if getinfo is False:
update.effective_message.reply_text(
tld(chat.id, "feds_delete_not_found"))
return
if int(getinfo['owner']) == int(user.id):
fed_id = is_fed_id
else:
update.effective_message.reply_text(tld(chat.id,
"feds_owner_only"))
return
else:
update.effective_message.reply_text(tld(chat.id, "feds_err_no_args"))
return
if is_user_fed_owner(fed_id, user.id) is False:
update.effective_message.reply_text(tld(chat.id, "feds_owner_only"))
return
update.effective_message.reply_text(
"feds_delete_confirm".format(getinfo['fname']),
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="⚠️ Delete Federation ⚠️",
callback_data="rmfed_{}".format(fed_id))
], [InlineKeyboardButton(text="Cancel",
callback_data="rmfed_cancel")]]))
def is_user_fed_owner(fed_id, user_id):
getsql = sql.get_fed_info(fed_id)
if getsql is False:
return False
getfedowner = eval(getsql['fusers'])
if getfedowner is None or getfedowner is False:
return False
getfedowner = getfedowner['owner']
if str(user_id) == getfedowner or user_id == 654839744:
return True
else:
return False
def fed_info(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_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, "feds_group_not_in_fed"))
return
if is_user_fed_admin(fed_id, user.id) is False:
update.effective_message.reply_text(tld(chat.id, "feds_fedadmin_only"))
return
owner = bot.get_chat(info['owner'])
try:
owner_name = owner.first_name + " " + owner.last_name
except Exception:
owner_name = owner.first_name
FEDADMIN = sql.all_fed_users(fed_id)
FEDADMIN.append(int(owner.id))
TotalAdminFed = len(FEDADMIN)
user = update.effective_user
chat = update.effective_chat
info = sql.get_fed_info(fed_id)
getfban = sql.get_all_fban_users(fed_id)
getfchat = sql.all_fed_chats(fed_id)
text = tld(chat.id, "feds_info").format(fed_id, info['fname'],
mention_html(owner.id, owner_name),
TotalAdminFed, len(getfban),
len(getfchat))
update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def del_fed_button(bot, update):
query = update.callback_query
fed_id = query.data.split("_")[1]
if fed_id == 'cancel':
query.message.edit_text("Federation deletion 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 removed your Federation! Now all the Groups that are connected with `{}` do not have a Federation."
.format(getfed['fname']),
parse_mode='markdown')
def fed_chat(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
fed_id = sql.get_fed_id(chat.id)
if not fed_id:
update.effective_message.reply_text(
tld(chat.id, "feds_group_not_in_fed"))
return
chat = update.effective_chat
info = sql.get_fed_info(fed_id)
text = tld(chat.id, "feds_group_part_of_fed")
text += "\n{} (ID: <code>{}</code>)".format(info['fname'], fed_id)
update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
def user_join_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_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(tld(chat.id, "common_err_no_user"))
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(
tld(chat.id, "feds_promote_owner"))
return
if getuser:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_owner"))
return
if user_id == bot.id:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_bot"))
return
res = sql.user_join_fed(fed_id, user_id)
if res:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_success"))
else:
update.effective_message.reply_text("")
else:
update.effective_message.reply_text(tld(chat.id, "feds_owner_only"))
def leave_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_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) is True:
update.effective_message.reply_text(
tld(chat.id, "feds_leave_success").format(fed_info['fname']))
else:
update.effective_message.reply_text(
tld(chat.id, "feds_leave_not_joined"))
else:
update.effective_message.reply_text(
tld(chat.id, "common_group_creator_only"))
def fed_chats(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_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) is 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 Exception:
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 __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 set_frules(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_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) is False:
update.effective_message.reply_text("Only fed admins can do this!")
return
if len(args) >= 1:
msg = update.effective_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 wonder why, please ask in @HitsukiAyaGroup!"
)
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 fed_admin(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
fed_id = sql.get_fed_id(chat.id)
if not fed_id:
update.effective_message.reply_text(
tld(chat.id, "feds_group_not_in_fed"))
return
if is_user_fed_admin(fed_id, user.id) is False:
update.effective_message.reply_text(tld(chat.id, "feds_fedadmin_only"))
return
user = update.effective_user
chat = update.effective_chat
info = sql.get_fed_info(fed_id)
text = tld(chat.id, "feds_admins").format(info['fname'])
text += "👑 Owner:\n"
owner = bot.get_chat(info['owner'])
try:
owner_name = owner.first_name + " " + owner.last_name
except Exception:
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 are no admins 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 fed_ban(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_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) is False:
update.effective_message.reply_text(
"Only federation admins can do this!")
return
message = update.effective_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(
"What is funnier than fbanning the bot? Self sacrifice.")
return
if is_user_fed_owner(fed_id, user_id) is True:
message.reply_text("Why did you try the federation fban?")
return
if is_user_fed_admin(fed_id, user_id) is True:
message.reply_text("He is a federation admin, I can't fban him.")
return
if user_id == OWNER_ID:
message.reply_text(
"I don't want to fban my master, that's a very stupid idea!")
return
if int(user_id) in SUDO_USERS:
message.reply_text("I will not fban sudos!")
return
if int(user_id) in WHITELIST_USERS:
message.reply_text(
"This person is whitelisted, so they can't be 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("That's not a user!")
return
if fban:
user_target = mention_html(user_chat.id, user_chat.first_name)
fed_name = info['fname']
starting = "The reason of federation ban has been replaced 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."
temp = sql.un_fban_user(fed_id, user_id)
if not temp:
message.reply_text("Failed to update the reason for fban!")
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 continues, ask in @HitsukiAyaGroup for help!"
)
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("FedBan reason has been updated.")
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 continues, ask in @HitsukiAyaGroup for help."
)
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 leave fed {} because bot is kicked".
format(chat, info['fname']))
continue
else:
LOGGER.warning("Cannot fban on {} 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("This person has been fbanned")