2
0
mirror of https://github.com/Nick80835/microbot synced 2025-08-22 18:19:16 +00:00
microbot/ubot/loader.py

228 lines
7.8 KiB
Python
Raw Normal View History

2020-03-16 21:02:05 -04:00
# SPDX-License-Identifier: GPL-2.0-or-later
2023-10-16 09:46:31 -04:00
import glob
2020-05-16 22:50:25 -04:00
from concurrent.futures import ThreadPoolExecutor
2023-10-16 09:46:31 -04:00
from importlib import import_module, reload
from os.path import basename, dirname, isfile
2020-05-09 13:15:08 -04:00
from aiohttp import ClientSession
from telethon.tl.types import DocumentAttributeFilename
2023-10-16 09:46:31 -04:00
from .command_handler import CommandHandler
class Loader():
2020-07-18 15:11:09 -04:00
aioclient = ClientSession()
thread_pool = ThreadPoolExecutor()
help_dict = {}
help_hidden_dict = {}
loaded_modules = []
all_modules = []
2023-10-16 09:46:31 -04:00
def __init__(self, client, logger, settings):
self.client = client
self.logger = logger
self.settings = settings
2020-07-18 15:11:09 -04:00
self.command_handler = CommandHandler(client, settings, self)
2023-10-16 09:46:31 -04:00
def load_all_modules(self):
self._find_all_modules()
for module_name in self.all_modules:
try:
self.loaded_modules.append(import_module("ubot.modules." + module_name))
except Exception as exception:
self.logger.error(f"Error while loading {module_name}: {exception}")
2023-10-16 09:46:31 -04:00
def reload_all_modules(self):
2020-06-16 17:00:09 -04:00
self.command_handler.incoming_commands = []
self.command_handler.inline_photo_commands = []
self.command_handler.inline_article_commands = []
self.command_handler.callback_queries = []
self.help_dict = {}
self.help_hidden_dict = {}
2023-10-16 09:46:31 -04:00
errors = ""
for module in self.loaded_modules:
try:
reload(module)
2020-04-17 11:29:58 -04:00
except ModuleNotFoundError:
pass
2023-10-16 09:46:31 -04:00
except Exception as exception:
errors += f"`Error while reloading {module.__name__} -> {exception}\n\n`"
raise exception
return errors or None
def add(self, pattern=None, **args):
pattern = args.get("pattern", pattern)
2020-07-16 13:41:42 -04:00
pattern_extra = args.get("pattern_extra", "")
2023-10-16 09:46:31 -04:00
def decorator(func):
if args.get("hide_help", False):
if func.__module__.split(".")[-1] in self.help_hidden_dict:
self.help_hidden_dict[func.__module__.split(".")[-1]] += [pattern]
else:
self.help_hidden_dict[func.__module__.split(".")[-1]] = [pattern]
else:
if func.__module__.split(".")[-1] in self.help_dict:
self.help_dict[func.__module__.split(".")[-1]] += [pattern]
else:
self.help_dict[func.__module__.split(".")[-1]] = [pattern]
2020-06-16 17:00:09 -04:00
self.command_handler.incoming_commands.append({
"pattern": pattern,
2020-07-16 13:41:42 -04:00
"pattern_extra": pattern_extra,
"function": func,
2020-06-26 19:17:50 -04:00
"simple_pattern": args.get('simple_pattern', False),
2020-06-26 19:43:36 -04:00
"raw_pattern": args.get('raw_pattern', False),
"sudo": args.get('sudo', False),
"extras": args.get('extras', None),
"nsfw": args.get('nsfw', False),
2020-06-26 15:02:38 -04:00
"pass_nsfw": args.get('pass_nsfw', False),
"admin": args.get('admin', False),
2020-06-16 15:16:44 -04:00
"owner": args.get('owner', False),
"locking": args.get('locking', False),
2020-06-16 15:51:26 -04:00
"lockreason": None,
"userlocking": args.get('userlocking', False),
2020-06-29 15:29:20 -04:00
"lockedusers": [],
"chance": args.get('chance', None)
2020-06-16 17:00:09 -04:00
})
2023-10-16 09:46:31 -04:00
2020-04-17 16:34:12 -04:00
return func
2023-10-16 09:46:31 -04:00
return decorator
2020-06-16 17:18:11 -04:00
def add_list(self, pattern=None, **args):
pattern_list = args.get("pattern", pattern)
pattern_extra = args.get("pattern_extra", "")
def decorator(func):
for pattern in pattern_list:
if args.get("hide_help", False):
if func.__module__.split(".")[-1] in self.help_hidden_dict:
self.help_hidden_dict[func.__module__.split(".")[-1]] += [pattern]
else:
self.help_hidden_dict[func.__module__.split(".")[-1]] = [pattern]
2020-06-16 17:18:11 -04:00
else:
if func.__module__.split(".")[-1] in self.help_dict:
self.help_dict[func.__module__.split(".")[-1]] += [pattern]
else:
self.help_dict[func.__module__.split(".")[-1]] = [pattern]
2020-06-16 17:18:11 -04:00
self.command_handler.incoming_commands.append({
2020-07-16 13:41:42 -04:00
"pattern": pattern,
"pattern_extra": pattern_extra,
2020-06-16 17:18:11 -04:00
"function": func,
2020-06-26 19:17:50 -04:00
"simple_pattern": args.get('simple_pattern', False),
2020-06-26 19:43:36 -04:00
"raw_pattern": args.get('raw_pattern', False),
2020-06-16 17:18:11 -04:00
"sudo": args.get('sudo', False),
"extras": args.get('extras', pattern),
"nsfw": args.get('nsfw', False),
2020-06-26 23:24:13 -04:00
"pass_nsfw": args.get('pass_nsfw', False),
2020-06-16 17:18:11 -04:00
"admin": args.get('admin', False),
"owner": args.get('owner', False),
"locking": args.get('locking', False),
"lockreason": None,
"userlocking": args.get('userlocking', False),
2020-07-03 23:00:56 -04:00
"lockedusers": [],
"chance": args.get('chance', None)
2020-06-16 17:18:11 -04:00
})
return func
return decorator
2020-05-21 13:53:11 -04:00
def add_inline_photo(self, pattern=None, **args):
pattern = args.get("pattern", pattern)
2020-05-16 17:52:29 -04:00
def decorator(func):
2020-06-16 17:00:09 -04:00
self.command_handler.inline_photo_commands.append({
"pattern": pattern,
2020-05-21 13:53:11 -04:00
"function": func,
"default": args.get("default", None)
2020-06-16 17:00:09 -04:00
})
2020-05-16 17:52:29 -04:00
return func
return decorator
2020-05-25 14:04:56 -04:00
def add_inline_article(self, pattern=None, **args):
pattern = args.get("pattern", pattern)
def decorator(func):
2020-06-16 17:00:09 -04:00
self.command_handler.inline_article_commands.append({
"pattern": pattern,
2020-05-25 14:04:56 -04:00
"function": func,
"default": args.get("default", None)
2020-06-16 17:00:09 -04:00
})
2020-05-25 14:04:56 -04:00
return func
return decorator
def add_callback_query(self, data_id=None, **args):
data_id = args.get("data_id", data_id)
def decorator(func):
2020-06-16 17:00:09 -04:00
self.command_handler.callback_queries.append({
"data_id": data_id,
"function": func,
"extras": args.get('extras', None)
2020-06-16 17:00:09 -04:00
})
return func
return decorator
async def get_text(self, event, with_reply=True, return_msg=False, default=None):
if event.args:
if return_msg:
2020-05-04 19:08:13 -04:00
if event.is_reply:
return event.args, await event.get_reply_message()
return event.args, event
return event.args
elif event.is_reply and with_reply:
reply = await event.get_reply_message()
if return_msg:
return reply.text, reply
return reply.text
else:
if return_msg:
return default, event
return default
async def get_image(self, event):
if event and event.media:
if event.photo:
2020-05-09 12:22:02 -04:00
return event.photo
elif event.document:
if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in event.media.document.attributes:
2020-05-09 12:22:02 -04:00
return None
if event.gif or event.video or event.audio or event.voice:
2020-05-09 12:22:02 -04:00
return None
2020-05-09 12:22:02 -04:00
return event.media.document
else:
return None
else:
2020-05-09 12:22:02 -04:00
return None
2020-06-23 15:28:09 -04:00
def prefix(self):
return (self.settings.get_list('cmd_prefix') or ['.'])[0]
2023-10-16 09:46:31 -04:00
def _find_all_modules(self):
2020-06-24 14:13:40 -04:00
module_paths = glob.glob(dirname(__file__) + "/modules/*.py")
2023-10-16 09:46:31 -04:00
2020-06-24 14:13:40 -04:00
self.all_modules = sorted([
basename(f)[:-3] for f in module_paths
2023-10-16 09:46:31 -04:00
if isfile(f) and f.endswith(".py")
2020-06-24 14:13:40 -04:00
])