snewell/pennyworth

#!/usr/bin/python3

"""
This module is the driver modules for the execution of various tools.
"""

import sys

import pennyworth.plugin


def _do_help():
    print("usage: pennyworth tool [tool args]")
    print("Arguments:")
    print("\t--help\tDisplay this help")
    print("\t--list\tDisplay available tools")


_TOOLS = pennyworth.plugin.query_plugins("pennyworth.commands")


def _do_list():
    for tool in sorted(_TOOLS):
        print("{} - {}".format(tool, _TOOLS[tool][1]))


_EX_TOOLS = {
    "--help": _do_help,
    "--list": _do_list
}


def main():
    # pylint: disable=missing-docstring
    if len(sys.argv) > 1:
        tool = _TOOLS.get(sys.argv[1])
        if tool:
            tool[0](sys.argv[2:])
        else:
            tool = _EX_TOOLS.get(sys.argv[1])
            if tool:
                tool()
            else:
                print("{} isn't an available tool".format(sys.argv[1]))
                sys.exit(1)
    else:
        _do_help()


if __name__ == "__main__":
    main()