monolith/utils/loaders/command_loader.py

from os import listdir

from utils.logging.debug import debug
from utils.logging.log import error

from main import CommandMap


def loadCommands(allowDup=False):
    for filename in listdir("commands"):
        if filename.endswith(".py") and filename != "__init__.py":
            commandName = filename[0:-3]
            className = commandName.capitalize() + "Command"
            # try:
            module = __import__("commands.%s" % commandName)
            if commandName not in CommandMap:
                CommandMap[commandName] = getattr(getattr(module, commandName), className)
                debug("Registered command: %s" % commandName)
            else:
                if allowDup:
                    CommandMap[commandName] = getattr(getattr(module, commandName), className)
                    debug("Registered command: %s" % commandName)

                error("Duplicate command: %s" % (commandName))
            # except Exception as err:
            #    error("Exception while loading command %s:\n%s" % (commandName, err))