|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import contextlib |
|
|
import glob |
|
|
import os |
|
|
from importlib import import_module |
|
|
from logging import Logger |
|
|
|
|
|
from . import LOGS |
|
|
from .fns.tools import get_all_files |
|
|
|
|
|
|
|
|
class Loader: |
|
|
def __init__(self, path="plugins", key="Official", logger: Logger = LOGS): |
|
|
self.path = path |
|
|
self.key = key |
|
|
self._logger = logger |
|
|
|
|
|
def load( |
|
|
self, |
|
|
log=True, |
|
|
func=import_module, |
|
|
include=None, |
|
|
exclude=None, |
|
|
after_load=None, |
|
|
load_all=False, |
|
|
): |
|
|
_single = os.path.isfile(self.path) |
|
|
if include: |
|
|
if log: |
|
|
self._logger.info("Including: {}".format("• ".join(include))) |
|
|
files = glob.glob(f"{self.path}/_*.py") |
|
|
for file in include: |
|
|
path = f"{self.path}/{file}.py" |
|
|
if os.path.exists(path): |
|
|
files.append(path) |
|
|
elif _single: |
|
|
files = [self.path] |
|
|
else: |
|
|
if load_all: |
|
|
files = get_all_files(self.path, ".py") |
|
|
else: |
|
|
files = glob.glob(f"{self.path}/*.py") |
|
|
if exclude: |
|
|
for path in exclude: |
|
|
if not path.startswith("_"): |
|
|
with contextlib.suppress(ValueError): |
|
|
files.remove(f"{self.path}/{path}.py") |
|
|
if log and not _single: |
|
|
self._logger.info( |
|
|
f"• Installing {self.key} Plugins || Count : {len(files)} •" |
|
|
) |
|
|
for plugin in sorted(files): |
|
|
if func == import_module: |
|
|
plugin = plugin.replace(".py", "").replace("/", ".").replace("\\", ".") |
|
|
try: |
|
|
modl = func(plugin) |
|
|
except ModuleNotFoundError as er: |
|
|
modl = None |
|
|
self._logger.error(f"{plugin}: '{er.name}' not installed!") |
|
|
continue |
|
|
except Exception as exc: |
|
|
modl = None |
|
|
self._logger.error(f"pyUltroid - {self.key} - ERROR - {plugin}") |
|
|
self._logger.exception(exc) |
|
|
continue |
|
|
if _single and log: |
|
|
self._logger.info(f"Successfully Loaded {plugin}!") |
|
|
if callable(after_load): |
|
|
if func == import_module: |
|
|
plugin = plugin.split(".")[-1] |
|
|
after_load(self, modl, plugin_name=plugin) |
|
|
|
