monolith/utils/loaders/command_loader.py

from os import listdir

import commands
from main import CommandMap
from utils.logging.debug import debug
from utils.logging.log import *


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 not commandName 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))