utils/src/cli.py

import code
import argparse
import importlib
import sys

import src

BANNER = """
                .    o8o  oooo   o8o      .    o8o                     
              .o8    `"'  `888   `"'    .o8    `"'                     
oooo  oooo  .o888oo oooo   888  oooo  .o888oo oooo   .ooooo.   .oooo.o 
`888  `888    888   `888   888  `888    888   `888  d88' `88b d88(  "8 
 888   888    888    888   888   888    888    888  888ooo888 `"Y88b.  
 888   888    888 .  888   888   888    888 .  888  888    .o o.  )88b 
 `V88V"V8P'   "888" o888o o888o o888o   "888" o888o `Y8bod8P' 8""888P' 
"""


def main():
    parser = argparse.ArgumentParser(
        description="Run a specific script by name. If no name is provided, start a REPL."
    )
    parser.add_argument(
        "user_input",
        type=str,
        nargs="*",
        help="The name of the script that you want to run plus any arguments for that script.",
    )
    parser.add_argument(
        "--version", action="store_true", help="Print out the version string."
    )

    args = parser.parse_args()
    user_input = args.user_input
    if args.version is True:
        print(src.__version__)
        sys.exit()
    if len(user_input) == 0:
        code.interact(banner=BANNER, local=locals())
    else:
        command_name = user_input.pop(0)
        module = importlib.import_module(f"src.commands.{command_name}")
        sys.exit(module.main(user_input))