Sami
28 lines
962 B
Python
28 lines
962 B
Python
"""Test helper for importing plugin modules from a hyphenated directory."""
|
|
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def load_module(filename: str, module_name: str, package: bool = False):
|
|
path = ROOT / filename
|
|
if package:
|
|
pkg_name = "digital_pet_pkg"
|
|
if pkg_name not in sys.modules:
|
|
pkg_spec = importlib.util.spec_from_file_location(pkg_name, ROOT / "__init__.py", submodule_search_locations=[str(ROOT)])
|
|
pkg = importlib.util.module_from_spec(pkg_spec)
|
|
sys.modules[pkg_name] = pkg
|
|
full_name = f"{pkg_name}.{path.stem}"
|
|
else:
|
|
full_name = module_name
|
|
spec = importlib.util.spec_from_file_location(full_name, path)
|
|
mod = importlib.util.module_from_spec(spec)
|
|
sys.modules[full_name] = mod
|
|
assert spec.loader is not None
|
|
spec.loader.exec_module(mod)
|
|
return mod
|