90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
# pragma pylint: disable=attribute-defined-outside-init
|
|
|
|
"""
|
|
This module load custom objects
|
|
"""
|
|
import importlib.util
|
|
import inspect
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Any, List, Optional, Tuple, Type, Union
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class IResolver(object):
|
|
"""
|
|
This class contains all the logic to load custom classes
|
|
"""
|
|
|
|
@staticmethod
|
|
def _get_valid_object(object_type, module_path: Path,
|
|
object_name: str) -> Optional[Type[Any]]:
|
|
"""
|
|
Returns the first object with matching object_type and object_name in the path given.
|
|
:param object_type: object_type (class)
|
|
:param module_path: absolute path to the module
|
|
:param object_name: Class name of the object
|
|
:return: class or None
|
|
"""
|
|
|
|
# Generate spec based on absolute path
|
|
spec = importlib.util.spec_from_file_location('unknown', str(module_path))
|
|
module = importlib.util.module_from_spec(spec)
|
|
try:
|
|
spec.loader.exec_module(module) # type: ignore # importlib does not use typehints
|
|
except (ModuleNotFoundError, SyntaxError) as err:
|
|
# Catch errors in case a specific module is not installed
|
|
logger.warning(f"Could not import {module_path} due to '{err}'")
|
|
|
|
valid_objects_gen = (
|
|
obj for name, obj in inspect.getmembers(module, inspect.isclass)
|
|
if object_name == name and object_type in obj.__bases__
|
|
)
|
|
return next(valid_objects_gen, None)
|
|
|
|
@staticmethod
|
|
def _search_object(directory: Path, object_type, object_name: str,
|
|
kwargs: dict = {}) -> Union[Tuple[Any, Path], Tuple[None, None]]:
|
|
"""
|
|
Search for the objectname in the given directory
|
|
:param directory: relative or absolute directory path
|
|
:return: object instance
|
|
"""
|
|
logger.debug("Searching for %s %s in '%s'", object_type.__name__, object_name, directory)
|
|
for entry in directory.iterdir():
|
|
# Only consider python files
|
|
if not str(entry).endswith('.py'):
|
|
logger.debug('Ignoring %s', entry)
|
|
continue
|
|
module_path = Path.resolve(directory.joinpath(entry))
|
|
obj = IResolver._get_valid_object(
|
|
object_type, module_path, object_name
|
|
)
|
|
if obj:
|
|
return (obj(**kwargs), module_path)
|
|
return (None, None)
|
|
|
|
@staticmethod
|
|
def _load_object(paths: List[Path], object_type, object_name: str,
|
|
kwargs: dict = {}) -> Union[Any, None]:
|
|
"""
|
|
Try to load object from path list.
|
|
"""
|
|
|
|
for _path in paths:
|
|
try:
|
|
(module, module_path) = IResolver._search_object(directory=_path,
|
|
object_type=object_type,
|
|
object_name=object_name,
|
|
kwargs=kwargs)
|
|
if module:
|
|
logger.info(
|
|
f"Using resolved {object_type.__name__.lower()[1:]} {object_name} "
|
|
f"from '{module_path}'...")
|
|
return module
|
|
except FileNotFoundError:
|
|
logger.warning('Path "%s" does not exist.', _path.relative_to(Path.cwd()))
|
|
|
|
return None
|