wordscape/main.py

import argparse
from string import ascii_lowercase
import cmd2

from dictionary import DEFAULT, ALTERNATE

LOWERCASE = set(ascii_lowercase)
DICTS = {"default": DEFAULT, "alt": ALTERNATE}
# argparsers
dictionary_manage_parser = argparse.ArgumentParser()
commands = dictionary_manage_parser.add_mutually_exclusive_group()
commands.add_argument(
    "-l", "--list", action="store_true", help="List available dictionaries"
)
commands.add_argument("-s", "--switch", help="Dictionary to swap to")

letter_set_manage = argparse.ArgumentParser()
letter_set_manage.add_argument("new_letters")


class MainLoop(cmd2.Cmd):
    """Loop for wordscape commands

    """

    prompt = "?$ "

    def __init__(self):
        self.dict = "default"
        self.letters = [l for l in input("Enter letters: ").lower() if l in LOWERCASE]
        self.prompt = ", ".join(self.letters) + " " + MainLoop.prompt
        super().__init__()

    @cmd2.with_argparser(dictionary_manage_parser)
    def do_dict(self, args):
        """list/switch dict"""
        if args.list:
            print(
                "\n".join(
                    (
                        "{} {}".format("*" if key == self.dict else " ", key)
                        for key in DICTS.keys()
                    )
                )
            )
        else:
            DICTS[args.switch]
            self.dict = args.switch

    @cmd2.with_argparser(letter_set_manage)
    def do_change_letters(self, args):
        pass


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