AlexMathew/scrapple

"""
scrapple.utils.dynamicdispatch
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Functions related to dynamic dispatch of objects
"""

def get_command_class(command):
    """
    Called from runCLI() to select the command class for the selected command.

    :param command: The command to be implemented
    :return: The command class corresponding to the selected command
    """
    from scrapple.commands import genconfig, generate, run, web
    commandMapping = {
        'genconfig': genconfig,
        'generate': generate,
        'run': run,
        'web': web
    }
    cmdClass = getattr(commandMapping.get(command), command.title() + 'Command')
    return cmdClass