def load_command_executable(self, root: str, command_data: dict, plugin_data: dict):
command_data.update({'path': os.path.join(root)})
command_function = self.load_function(root, command_data)
cmd = SigmaCommand(self.bot, command_function, plugin_data, command_data)
if cmd.alts:
for alt in cmd.alts:
self.alts.update({alt: cmd.name})
self.commands.update({command_data.get('name'): cmd})
def load_command_executable(self, root, command_data, module_data):
"""
Loads the command's executable and defining function call.
:type root: str
:type command_data: dict
:type module_data: dict
"""
command_data.update({'path': os.path.join(root)})
command_function = self.load_function(root, command_data)
cmd = SigmaCommand(self.bot, command_function, module_data, command_data)
if cmd.alts:
for alt in cmd.alts:
self.alts.update({alt: cmd.name})
self.commands.update({command_data.get('name'): cmd})
def load_command_executable(self, root, command_data, module_data):
"""
Loads the command's executable and defining function call.
:param root: The command origin path.
:type root: str
:param command_data: Document containing the command's details.
:type command_data: dict
:param module_data: Document containing the module's details.
:type module_data: dict
:return:
:rtype:
"""
command_data.update({'path': os.path.join(root)})
command_function = self.load_function(root, command_data)
cmd = SigmaCommand(self.bot, command_function, module_data,
command_data)
if cmd.alts:
for alt in cmd.alts:
self.alts.update({alt: cmd.name})
self.commands.update({command_data.get('name'): cmd})
def load_all_modules(self):
self.alts = {}
self.commands = {}
self.events = {}
directory = 'sigma/modules'
for root, dirs, files in os.walk(directory):
for file in files:
if file == 'module.yml':
file_path = (os.path.join(root, file))
with open(file_path, encoding='utf-8') as plugin_file:
plugin_data = yaml.safe_load(plugin_file)
if plugin_data['enabled']:
if self.init:
self.log.info(
f'Loading the {plugin_data["name"]} Module'
)
if 'commands' in plugin_data:
if plugin_data[
'category'] not in self.categories:
self.categories.append(
plugin_data['category'])
for command_data in plugin_data['commands']:
if self.bot.cfg.pref.music_only:
if plugin_data['category'] == 'music':
add_cmd = True
else:
add_cmd = False
elif self.bot.cfg.pref.text_only:
if plugin_data['category'] != 'music':
add_cmd = True
else:
add_cmd = False
else:
add_cmd = True
if add_cmd:
if command_data['enabled']:
module_root_location = os.path.join(
root)
command_module_location = os.path.join(
root, command_data["name"])
command_module_location = command_module_location.replace(
'/', '.')
command_module_location = command_module_location.replace(
'\\', '.')
command_function = importlib.import_module(
command_module_location)
importlib.reload(command_function)
command_data.update(
{'path': module_root_location})
cmd = SigmaCommand(
self.bot, command_function,
plugin_data, command_data)
if cmd.alts:
for alt in cmd.alts:
self.alts.update(
{alt: cmd.name})
self.commands.update(
{command_data['name']: cmd})
if self.bot.cfg.dsc.bot:
if not self.bot.cfg.pref.music_only:
if 'events' in plugin_data:
for event_data in plugin_data[
'events']:
if event_data['enabled']:
command_module_location = os.path.join(
root, event_data["name"])
command_module_location = command_module_location.replace(
'/', '.')
command_module_location = command_module_location.replace(
'\\', '.')
event_function = importlib.import_module(
command_module_location)
importlib.reload(
event_function)
event = SigmaEvent(
self.bot, event_function,
plugin_data, event_data)
if event.event_type in self.events:
event_list = self.events[
event.event_type]
else:
event_list = []
event_list.append(event)
self.events.update({
event.event_type:
event_list
})