Archived
1
0
Fork 0
This repository has been archived on 2024-04-26. You can view files and clone it, but cannot push or open issues or pull requests.
akari-bot/core/loader.py

79 lines
2.4 KiB
Python
Raw Normal View History

2021-02-01 15:13:11 +00:00
import importlib
import os
import re
import traceback
2021-07-07 16:00:24 +00:00
from .logger import Logger
2021-07-06 18:00:26 +00:00
from .elements import Plugin
2021-02-01 15:13:11 +00:00
2021-04-28 12:21:33 +00:00
err_prompt = []
2021-02-03 14:43:24 +00:00
2021-07-07 16:00:24 +00:00
def load_modules():
fun_file = None
load_dir_path = os.path.abspath('./modules/')
dir_list = os.listdir(load_dir_path)
for file_name in dir_list:
try:
2021-04-07 15:18:41 +00:00
file_path = f'{load_dir_path}/{file_name}'
if os.path.isdir(file_path):
if file_name != '__pycache__':
2021-04-07 15:18:41 +00:00
fun_file = file_name
else:
2021-07-06 18:00:26 +00:00
continue
2021-04-07 15:18:41 +00:00
if os.path.isfile(file_path):
b = re.match(r'(.*)(.py)', file_path)
if b:
fun_file = b.group(1)
2021-07-06 18:00:26 +00:00
else:
continue
if fun_file is not None:
2021-07-07 16:00:24 +00:00
Logger.info(f'Loading modules.{fun_file}...')
2021-07-06 18:00:26 +00:00
modules = 'modules.' + fun_file
importlib.import_module(modules)
2021-07-07 16:00:24 +00:00
Logger.info(f'Succeeded loaded modules.{fun_file}!')
except:
tb = traceback.format_exc()
Logger.info(f'Failed to load modules.{fun_file}: {tb}')
err_prompt.append(str(tb))
class PluginManager:
_plugin_list = set()
@classmethod
def add_plugin(cls, plugin: Plugin):
PluginManager._plugin_list.add(plugin)
@classmethod
def return_plugin_list(cls):
return PluginManager._plugin_list
2021-02-01 15:13:11 +00:00
2021-07-08 15:51:58 +00:00
@classmethod
def return_plugin_as_dict(cls):
return {x.bind_prefix: x for x in PluginManager._plugin_list}
@classmethod
def return_plugin_alias_map(cls):
alias_map = {}
for x in PluginManager._plugin_list:
if isinstance(x.alias, str):
alias_map.update({x.alias: x.bind_prefix})
if isinstance(x.alias, tuple):
for y in x.alias:
alias_map.update({y: x.bind_prefix})
return alias_map
load_modules()
Modules = PluginManager.return_plugin_as_dict()
ModulesAliases = PluginManager.return_plugin_alias_map()
2021-02-01 15:13:11 +00:00
2021-04-28 12:21:33 +00:00
loadercache = os.path.abspath('.cache_loader')
openloadercache = open(loadercache, 'w')
if err_prompt:
2021-05-27 15:13:40 +00:00
err_prompt = re.sub(' File "<frozen importlib.*?>", .*?\n', '', '\n'.join(err_prompt))
openloadercache.write('加载模块中发生了以下错误,对应模块未加载:\n' + err_prompt)
2021-04-28 12:21:33 +00:00
else:
openloadercache.write('所有模块已正确加载。')
openloadercache.close()