main.py

import cmd2
import shlex
from src.cli import (
    TimeScaleFinding,
    ScaleInfo,
    PrintFamily,
    ListCycles,
    ListSubcycles,
    FindJustChords,
    DegreeApproximations,
    PlayCycle,
    Metastats,
    FindPianoteqDiapason,
    quiz,
)
from src.cli.utils import resolve, ScaleTheoryError

COMMANDS = {
    "time_scale_finding": TimeScaleFinding,
    "scale_info": ScaleInfo,
    "print_family": PrintFamily,
    "find_scales": ListCycles,
    "find_subscales": ListSubcycles,
    "just_chords": FindJustChords,
    "degree_approximation": DegreeApproximations,
    "play": PlayCycle,
    "metastats": Metastats,
    "diapason": FindPianoteqDiapason,
}

MODULES = {
    "quiz": {
        "intervals": quiz.IntervalQuiz,
        "melody": quiz.MelodyQuiz,
        "modes": quiz.ModeQuiz,
        "chords": quiz.ChordsQuiz,
    },
}


def run_command(command):
    def f(argstring):
        try:
            namespace = command.parser.parse_args(shlex.split(argstring))
        except SystemExit:
            return  # argparse likes to SystemExit after --help is called
        try:
            kwargs = resolve(namespace, getattr(command, 'pass_edo_steps', False))
            command.run(**kwargs)
        except ScaleTheoryError as e:
            print(e)

    return f


def run_module(module_name):
    def f(argstring):
        command_name, *args = shlex.split(argstring)
        if command_name in MODULES[module_name]:
            command = MODULES[module_name][command_name]
        else:
            print(f"No command called {module_name} {command_name}")
            return

        try:
            namespace = command.parser.parse_args(args)
        except SystemExit:
            return

        try:
            kwargs = resolve(namespace, getattr(command, 'pass_edo_steps', False))
            command.run(**kwargs)
        except ScaleTheoryError as e:
            print(e)

    return f


class ExplorerShell(cmd2.Cmd):
    intro = "Welcome to the Scale Explorer.\nEnter 'help' to see a list of commands."
    prompt = "\n$> "

    def __init__(self):
        super().__init__(persistent_history_file="~/.scale_theory_history")
        self.debug = True

        for command_name, command in COMMANDS.items():
            command.parser.prog = command_name
            setattr(self, "do_" + command_name, run_command(command))

        for module_name in MODULES:
            setattr(self, "do_" + module_name, run_module(module_name))

    def do_help(self, arg):
        for name in COMMANDS:
            print(name)

        for module_name, command_d in MODULES.items():
            for command_name in command_d:
                print(module_name, command_name)

        print('quit')


if __name__ == "__main__":
    ExplorerShell().cmdloop()