Skip to content

Commit

Permalink
Extend load_modules() to handle invalid modules
Browse files Browse the repository at this point in the history
Fixes: #56
Signed-off-by: Tobias Wolf <[email protected]>
  • Loading branch information
NotTheEvilOne committed Jun 19, 2024
1 parent 9b663c6 commit abea9be
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions src/rookify/modules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import importlib
from typing import Any, Dict
from ..logger import get_logger
from .machine import Machine


Expand Down Expand Up @@ -55,6 +56,20 @@ def load_modules(machine: Machine, config: Dict[str, Any]) -> None:
:return: returns tuple of preflight_modules, modules
"""

migration_modules = config["migration_modules"].copy()

for entry in importlib.resources.files("rookify.modules").iterdir():
if entry.is_dir() and entry.name in config["migration_modules"]:
migration_modules.remove(entry.name)
_load_module(machine, config, entry.name)

if len(migration_modules) > 0 or len(config["migration_modules"]) < 1:
logger = get_logger()

if len(config["migration_modules"]) < 1:
logger.error("No modules configured for migration")
elif len(migration_modules) > 0:
for invalid_module_name in migration_modules:
logger.error(
"Invalid module configured: {0}".format(invalid_module_name)
)

0 comments on commit abea9be

Please sign in to comment.