async def emailPhase(message):
user = message.from_user
LOGGER.info(f"Email from : {user.id} | {user.first_name}")
if not await isUserMember(user.id):
return await message.reply("press /start first.")
if await isUserInDb(user.id):
return await message.reply(
f"You already have access to the drives, for any queries join {config.get('group_link')} and seek admins assistance"
)
email_id = message.text
out = ""
out += f"<b>FirstName:</b> <code>{user.first_name}</code>\n"
if user.last_name:
out += f"<b>LastName:</b> <code>{user.last_name}</code>\n"
if user.username:
out += f"<b>UserName:</b> @{user.username}\n"
out += f"<b>UserId:</b> <code>{user.id}</code>\n"
out += f"<b>EmailId:</b> <code>{email_id}</code>\n"
out += f"<b>Link:</b> <a href='tg://user?id={user.id}'>here</a>"
await bot.send_message(config.get('admin_chat'), out)
await addAccessUser(user.id, email_id)
await message.reply(
"Aight, we received your application now be patient and wait for approval (approval process take upto a week)."
)
def __list_all_modules():
from os.path import dirname, basename, isfile
import glob
# This generates a list of modules in this folder for the * in __main__ to work.
mod_paths = glob.glob(dirname(__file__) + "/*.py")
all_modules = [
basename(f)[:-3] for f in mod_paths
if isfile(f) and f.endswith(".py") and not f.endswith('__init__.py')
]
if LOAD or NO_LOAD:
to_load = LOAD
if to_load:
if not all(
any(mod == module_name for module_name in all_modules)
for mod in to_load):
LOGGER.error("Invalid load order names. Quitting.")
quit(1)
else:
to_load = all_modules
if NO_LOAD:
LOGGER.info("Not loading: {}".format(NO_LOAD))
return [item for item in to_load if item not in NO_LOAD]
return to_load
return all_modules
def new_fed(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
if chat.type != "private":
update.effective_message.reply_text(tld(chat.id, "common_cmd_pm_only"))
return
fednam = message.text.split(None, 1)[1]
if not fednam == '':
fed_id = str(uuid.uuid4())
fed_name = fednam
LOGGER.info(fed_id)
if user.id == int(OWNER_ID):
fed_id = fed_name
x = sql.new_fed(user.id, fed_name, fed_id)
if not x:
update.effective_message.reply_text(
tld(chat.id, "feds_create_fail"))
return
update.effective_message.reply_text(tld(chat.id,
"feds_create_success").format(
fed_name, fed_id, fed_id),
parse_mode=ParseMode.MARKDOWN)
try:
bot.send_message(MESSAGE_DUMP,
tld(chat.id, "feds_create_success_logger").format(
fed_name, fed_id),
parse_mode=ParseMode.HTML)
except Exception:
LOGGER.warning("Cannot send a message to MESSAGE_DUMP")
else:
update.effective_message.reply_text(tld(chat.id, "feds_err_no_args"))
def bannedUserFilter(message):
banned = isUserBanned(message.from_user.id)
if banned:
LOGGER.info(
f"Message from banned user: {message.from_user.id} : {message.from_user.first_name}"
)
return not banned
async def getAllBannedUsers():
users = []
try:
async for user in bannedUserDb.find({}):
LOGGER.info(f"Adding {user.userId} in banned.")
users.append(user.userId)
except Exception as e:
LOGGER.error(e)
return users
async def removeBanUser(userId):
LOGGER.info(f"UnBanning User: {userId}")
try:
result = await bannedUserDb.delete_one({"userId": userId})
LOGGER.info(result)
except Exception as e:
LOGGER.error(e)
return False
BANNED_USERS.remove(userId)
return True
async def addBanUser(userId):
LOGGER.info(f"Banning User: {userId}")
try:
result = await bannedUserDb.insert_one({"userId": userId})
LOGGER.info(result)
except Exception as e:
LOGGER.error(e)
return False
BANNED_USERS.append(userId)
return True
async def isUserMember(userId):
LOGGER.info(f"Checking if {userId} is in our kholi!")
try:
member = await bot.get_chat_member(config.get("base_chat"),userId)
if member.status == "left" or member.status == "kicked":
return False
except Exception as e:
LOGGER.error(e)
return False
return True
async def isUserInDb(userId):
LOGGER.info(f"Checking {userId} in db")
try:
result = await accessUserDb.find_one({"userId": userId})
LOGGER.info(result)
if result is None:
return False
except Exception as e:
LOGGER.error(e)
return False
return True
async def startPhase(message):
user = message.from_user
LOGGER.info(f"Start from : {user.id} | {user.first_name}")
if not await isUserMember(user.id):
return await message.reply(
f"Join my group first : {config.get('group_link')}")
if await isUserInDb(user.id):
return await message.reply(
f"You already have access to the drives, for any queries join {config.get('group_link')} and seek admins assistance"
)
await message.reply("Send your EMAIL-ID.")
async def addAccessUser(userId, email):
LOGGER.info(f"Adding {userId} : {email}")
try:
result = await accessUserDb.insert_one({
"userId": userId,
"email": email
})
LOGGER.info(result)
except Exception as e:
LOGGER.error(e)
return False
return True
def main():
# test_handler = CommandHandler("test", test) #Unused variable
start_handler = CommandHandler("start", start, pass_args=True)
help_handler = CommandHandler("help", get_help)
help_callback_handler = CallbackQueryHandler(help_button, pattern=r"help_")
start_callback_handler = CallbackQueryHandler(send_start,
pattern=r"bot_start")
migrate_handler = MessageHandler(Filters.status_update.migrate,
migrate_chats)
# dispatcher.add_handler(test_handler)
dispatcher.add_handler(start_handler)
dispatcher.add_handler(start_callback_handler)
dispatcher.add_handler(help_handler)
dispatcher.add_handler(help_callback_handler)
dispatcher.add_handler(migrate_handler)
# dispatcher.add_error_handler(error_callback)
# add antiflood processor
Dispatcher.process_update = process_update
LOGGER.info("Using long polling.")
# updater.start_polling(timeout=15, read_latency=4, clean=True)
updater.start_polling(poll_interval=0.0,
timeout=10,
clean=True,
bootstrap_retries=-1,
read_latency=3.0)
LOGGER.info("Successfully loaded")
if len(argv) not in (1, 3, 4):
tbot.disconnect()
else:
tbot.run_until_disconnected()
updater.idle()
# Stop processing with any other handler.
except DispatcherHandlerStop:
self.logger.debug(
'Stopping further handlers due to DispatcherHandlerStop')
break
# Dispatch any error.
except TelegramError as te:
self.logger.warning(
'A TelegramError was raised while processing the Update')
try:
self.dispatch_error(update, te)
except DispatcherHandlerStop:
self.logger.debug('Error handler stopped further handlers')
break
except Exception:
self.logger.exception(
'An uncaught error was raised while handling the error')
# Errors should not stop the thread.
except Exception:
self.logger.exception(
'An uncaught error was raised while processing the update')
if __name__ == '__main__':
LOGGER.info("Successfully loaded modules: " + str(ALL_MODULES))
tbot.start(bot_token=TOKEN)
main()
mod_paths = glob.glob(dirname(__file__) + "/*.py")
all_modules = [
basename(f)[:-3] for f in mod_paths
if isfile(f) and f.endswith(".py") and not f.endswith('__init__.py')
]
if LOAD or NO_LOAD:
to_load = LOAD
if to_load:
if not all(
any(mod == module_name for module_name in all_modules)
for mod in to_load):
LOGGER.error("Invalid load order names. Quitting.")
quit(1)
else:
to_load = all_modules
if NO_LOAD:
LOGGER.info("Not loading: {}".format(NO_LOAD))
return [item for item in to_load if item not in NO_LOAD]
return to_load
return all_modules
ALL_MODULES = sorted(__list_all_modules())
LOGGER.info("Modules to load: %s", str(ALL_MODULES))
__all__ = ALL_MODULES + ["ALL_MODULES"]
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) == 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) == True:
message.reply_text("Why did you try the federation fban?")
return
if is_user_fed_admin(fed_id, user_id) == 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 @tgbotAyaGroup 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 @tgbotAyaGroup 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")
import tgbot.modules.sql.feds_sql as sql
from tgbot.modules.tr_engine.strings import tld
# Greeting all bot owners that is using this module,
# The following people
# - MrYacha [Module Maker] - 10 Hours
# - RealAkito (used to be peaktogoo) [Module Reworker 01] - 22 Hours
# - AyraHikari [Module Reworker 03] - 26 Hours
# have spent so much time of their life into making this module better, stable, and well feature-rich.
# Please don't remove these comment, if you're still respecting us, the module maker and reworkers.
#
# Total times spend for this module is approx. 58+ hours
LOGGER.info(
"feds: Original: MrYacha, Reworked 01: RealAkito, Reworked 02: AyraHikari."
)
FBAN_ERRORS = {
"User is an administrator of the chat", "Chat not found",
"Not enough rights to restrict/unrestrict chat member",
"User_not_participant", "Peer_id_invalid", "Group chat was deactivated",
"Need to be inviter of a user to kick it from a basic group",
"Chat_admin_required",
"Only the creator of a basic group can kick group administrators",
"Channel_private", "Not in the chat", "Have no rights to send a message"
}
UNFBAN_ERRORS = {
"User is an administrator of the chat", "Chat not found",
"Not enough rights to restrict/unrestrict chat member",
from tgbot.events import register
from tgbot import LOGGER
from tgbot.modules.tr_engine.strings import tld
from requests import get
import rapidjson as json
# Greeting all bot owners that is using this module,
# - RealAkito (used to be peaktogoo) [Module Maker]
# have spent so much time of their life into making this module better, stable, and well more supports.
# Please don't remove these comment, if you're still respecting me, the module maker.
#
# This module was inspired by Android Helper Bot by Vachounet.
# None of the code is taken from the bot itself, to avoid confusion.
LOGGER.info("android: Original Android Modules by @RealAkito on Telegram")
@register(pattern=r"^/los(?: |$)(\S*)")
async def los(event):
if event.from_id == None:
return
chat_id = event.chat_id
try:
device_ = event.pattern_match.group(1)
device = urllib.parse.quote_plus(device_)
except Exception:
device = ''
if device == '':
async def initDatabase():
LOGGER.info("Initialing Database.")
users = await getAllBannedUsers()
BANNED_USERS.extend(users)