__init__.py.template

# This is a default __init__ file for the Duckietown Shell commands
#
# Maintainer: Andrea F. Daniele
from os.path import (
    exists as _exists,
    dirname as _dirname,
    basename as _basename,
    isdir as _isdir,
    isfile as _isfile,
    relpath as _relpath,
    join as _join,
)
import glob as _glob

from dt_shell import dtslogger

import utils
from utils.exceptions import ShellNeedsUpdate

# constants
_this_dir = _dirname(__file__)
_command_file = "command.py"

# import current command
if _exists(_join(_this_dir, _command_file)):
    try:
        # noinspection PyUnresolvedReferences
        from .command import *
    except ShellNeedsUpdate as e:
        _root: str = _join(_dirname(utils.__file__), "..")
        _command: str = _relpath(_this_dir, _root).strip("/")
        dtslogger.warning(
            f"Command '{_command}' was not loaded because the shell needs to be "
            f"updated. Current version is {e.current_version}, required version is "
            f"{e.version_needed}"
        )
        from utils.command_utils import failed_to_load_command as command

# find all modules
_modules = [m for m in _glob.glob(_join(_this_dir, "*")) if _isdir(m)]
# this is important to avoid name clashing with commands at lower levels
_modules.sort(key=lambda p: int(_isfile(_join(p, _command_file))))

# load submodules
for _mod in _modules:
    try:
        exec("from .%s import *" % _basename(_mod))
    except ImportError as e:
        if _exists(_join(_mod, _command_file)):
            raise EnvironmentError(e)
    except EnvironmentError as e:
        raise e