New Cleo CLI

CHANGELOG.md

@@ -7,6 +7,7 @@
 - Added scooze_id to CardModel and Card. Changed the MongoDB _id to scooze_id. ([#193](https://github.com/arcavios/scooze/pull/193))
 - Added AsyncScoozeApi as a way to use API endpoints in an async context (fixes Jupyter compatability) ([#199](https://github.com/arcavios/scooze/pull/199))
 - Add Docker support for starting MongoDB via the CLI ([#200](https://github.com/arcavios/scooze/pull/200))
+- CLI rework to be more robust ([#203](https://github.com/arcavios/scooze/pull/203))
 
 ### Changed
 

pyproject.toml

@@ -34,6 +34,7 @@ requests = "^2.31.0"
 frozendict = "^2.3.8"
 ijson = "^3.2.3"
 docker = "^6.1.3"
+cleo = "^2.0.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.0"
@@ -43,7 +44,7 @@ isort = "^5.12.0"
 mongomock = "^4.1.2"
 
 [tool.poetry.scripts]
-scooze = "scooze.cli:run_cli"
+scooze = "scooze.console.cli:run_cli"
 
 [tool.pytest.ini_options]
 # addopts = "-m 'not slow'"

src/scooze/api/bulkdata.py

@@ -49,7 +49,7 @@ async def load_batch(batch) -> int:
 
     except FileNotFoundError:
         print(file_path)
-        download_now = input(f"{file_type} file not found; would you like to download it now? [y/n] ") in "yY"
+        download_now = input(f"{file_type} file not found; would you like to download it now? [y/N] ") in "yY"
         if not download_now:
             print("No cards loaded into database.")
             return

src/scooze/console/cli.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+from importlib import import_module
+from typing import TYPE_CHECKING
+
+import pkg_resources
+from cleo.application import Application
+from cleo.loaders.factory_command_loader import FactoryCommandLoader
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+    from cleo.commands.command import Command
+
+# Accepted scooze CLI commands
+COMMANDS = [
+    "delete",
+    "run",
+    # Load commands
+    "load cards",
+    "load decks",
+    # Setup commands
+    "setup docker",
+    "setup local",
+    # Teardown commands
+    "teardown docker",
+    "teardown local",
+]
+
+
+def load_scooze_command(name: str) -> Callable[[], Command]:
+    def _scooze_command_factory() -> Command:
+        cli_words = name.split(" ")
+        module = import_module("scooze.console.commands." + ".".join(cli_words))
+        scooze_command_class = getattr(module, "".join(c.title() for c in cli_words) + "Command")
+        command: Command = scooze_command_class()
+        return command
+
+    return _scooze_command_factory
+
+
+class ScoozeApplication(Application):
+    def __init__(self):
+        pkg_name = "scooze"
+        super().__init__(pkg_name, f"{pkg_resources.get_distribution(pkg_name).version}")
+
+        command_loader = FactoryCommandLoader({name: load_scooze_command(name) for name in COMMANDS})
+        self.set_command_loader(command_loader)
+
+
+def run_cli() -> int:
+    exit_code: int = ScoozeApplication().run()
+    return exit_code
+
+
+if __name__ == "__main__":
+    run_cli()

src/scooze/console/commands/delete.py

@@ -0,0 +1,49 @@
+import scooze.database.deck as deck_db
+from cleo.commands.command import Command
+from cleo.helpers import argument
+from scooze.api import ScoozeApi
+from scooze.catalogs import DbCollection
+
+
+class DeleteCommand(Command):
+    name = "delete"
+    description = "Delete collections from the database."
+
+    arguments = [
+        argument(
+            "collections",
+            f"Which collections to remove from the database. Can be any of: <fg=cyan>all, cards, decks</>",
+            multiple=True,
+        )
+    ]
+
+    def handle(self):
+        to_delete: list[DbCollection] = []
+        delete_args = self.argument("collections")
+
+        if "all" in delete_args:
+            to_delete.extend(DbCollection.list())
+        else:
+            if "cards" in delete_args:
+                to_delete.append(DbCollection.CARDS)
+            if "decks" in delete_args:
+                to_delete.append(DbCollection.DECKS)
+
+        if len(to_delete) == 0:
+            print("No valid collections were given to delete.")
+
+        for collection in to_delete:
+            delete_collection(collection)
+
+
+def delete_collection(coll: DbCollection):
+    clean = input(f"Delete existing {coll}? [y/N] ") in "yY"
+    if clean:
+        print(f"Deleting all {coll} from your local database...")
+        match coll:
+            case DbCollection.CARDS:
+                with ScoozeApi() as s:
+                    s.delete_cards_all()
+            case DbCollection.DECKS:
+                # TODO(#145): Use the ScoozeApi for this
+                deck_db.delete_decks_all()