calkit-python 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

calkit/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.0.6"
+__version__ = "0.0.8"
 
 from .core import *
 from . import git

calkit/cli/__init__.py

@@ -0,0 +1,6 @@
+from .core import *
+
+
+def run() -> None:
+    from .main import app
+    app()

calkit/cli/config.py

@@ -0,0 +1,42 @@
+"""Config CLI."""
+
+from __future__ import annotations
+
+import typer
+
+from calkit import config
+from calkit.dvc import configure_remote, set_remote_auth
+
+config_app = typer.Typer(no_args_is_help=True)
+
+
+@config_app.command(name="set")
+def set_config_value(key: str, value: str):
+    try:
+        cfg = config.read()
+        cfg = config.Settings.model_validate(cfg.model_dump() | {key: value})
+        # Kind of a hack for setting the password computed field
+        # Types have been validated above, so this won't hurt to do again
+        setattr(cfg, key, value)
+    except FileNotFoundError:
+        # TODO: This fails if we try to set password before any config has
+        # been written
+        # Username is fine
+        cfg = config.Settings.model_validate({key: value})
+    cfg.write()
+
+
+@config_app.command(name="get")
+def get_config_value(key: str) -> None:
+    cfg = config.read()
+    val = getattr(cfg, key)
+    if val is not None:
+        print(val)
+    else:
+        print()
+
+
+@config_app.command(name="setup-remote")
+def setup_remote():
+    configure_remote()
+    set_remote_auth()

calkit/cli/core.py

@@ -0,0 +1,22 @@
+"""Core CLI functionality."""
+
+import os
+import pty
+import subprocess
+
+import typer
+
+
+def print_sep(name: str):
+    width = 66
+    txt_width = len(name) + 2
+    buffer_width = (width - txt_width) // 2
+    buffer = "-" * buffer_width
+    typer.echo(f"{buffer} {name} {buffer}")
+
+
+def run_cmd(cmd: list[str]):
+    if os.name == "nt":
+        subprocess.call(cmd)
+    else:
+        pty.spawn(cmd, lambda fd: os.read(fd, 1024))

calkit/cli/main.py

@@ -0,0 +1,358 @@
+"""Main CLI app."""
+
+from __future__ import annotations
+
+import os
+import subprocess
+
+import typer
+from typing_extensions import Annotated, Optional
+
+import calkit
+from calkit.cli import print_sep, run_cmd
+from calkit.cli.config import config_app
+from calkit.cli.new import new_app
+from calkit.cli.notebooks import notebooks_app
+
+app = typer.Typer(
+    invoke_without_command=True,
+    no_args_is_help=True,
+    context_settings=dict(help_option_names=["-h", "--help"]),
+    pretty_exceptions_show_locals=False,
+)
+app.add_typer(config_app, name="config", help="Configure Calkit.")
+app.add_typer(
+    new_app, name="new", help="Add new Calkit object (to calkit.yaml)."
+)
+app.add_typer(notebooks_app, name="nb", help="Work with Jupyter notebooks.")
+
+
+@app.callback()
+def main(
+    version: Annotated[
+        bool,
+        typer.Option("--version", help="Show version and exit."),
+    ] = False,
+):
+    if version:
+        typer.echo(calkit.__version__)
+        raise typer.Exit()
+
+
+@app.command(name="status")
+def get_status():
+    """Get a unified Git and DVC status."""
+    print_sep("Code (Git)")
+    run_cmd(["git", "status"])
+    typer.echo()
+    print_sep("Data (DVC)")
+    run_cmd(["dvc", "data", "status"])
+    typer.echo()
+    print_sep("Pipeline (DVC)")
+    run_cmd(["dvc", "status"])
+
+
+@app.command(name="diff")
+def diff():
+    """Get a unified Git and DVC diff."""
+    print_sep("Code (Git)")
+    run_cmd(["git", "diff"])
+    print_sep("Pipeline (DVC)")
+    run_cmd(["dvc", "diff"])
+
+
+@app.command(name="add")
+def add(
+    paths: list[str],
+    commit_message: Annotated[
+        str,
+        typer.Option(
+            "-m",
+            "--commit-message",
+            help="Automatically commit and use this as a message.",
+        ),
+    ] = None,
+    push_commit: Annotated[
+        bool, typer.Option("--push", help="Push after committing.")
+    ] = False,
+    to: Annotated[
+        str,
+        typer.Option(
+            "--to", "-t", help="System with which to add (git or dvc)."
+        ),
+    ] = None,
+):
+    """Add paths to the repo.
+
+    Code will be added to Git and data will be added to DVC.
+
+    Note: This will enable the 'autostage' feature of DVC, automatically
+    adding any .dvc files to Git when adding to DVC.
+    """
+    if to is not None and to not in ["git", "dvc"]:
+        typer.echo(f"Invalid option for 'to': {to}")
+        raise typer.Exit(1)
+    # Ensure autostage is enabled for DVC
+    subprocess.call(["dvc", "config", "core.autostage", "true"])
+    subprocess.call(["git", "add", ".dvc/config"])
+    if to is not None:
+        subprocess.call([to, "add"] + paths)
+    else:
+        dvc_extensions = [
+            ".png",
+            ".h5",
+            ".parquet",
+            ".pickle",
+            ".mp4",
+            ".avi",
+            ".webm",
+            ".pdf",
+        ]
+        dvc_size_thresh_bytes = 1_000_000
+        if "." in paths and to is None:
+            typer.echo("Cannot add '.' with calkit; use git or dvc")
+            raise typer.Exit(1)
+        if to is None:
+            for path in paths:
+                if os.path.isdir(path):
+                    typer.echo("Cannot auto-add directories; use git or dvc")
+                    raise typer.Exit(1)
+        for path in paths:
+            # Detect if this file should be tracked with Git or DVC
+            if os.path.splitext(path)[-1] in dvc_extensions:
+                typer.echo(f"Adding {path} to DVC per its extension")
+                subprocess.call(["dvc", "add", path])
+                continue
+            if os.path.getsize(path) > dvc_size_thresh_bytes:
+                typer.echo(
+                    f"Adding {path} to DVC since it's greater than 1 MB"
+                )
+                subprocess.call(["dvc", "add", path])
+                continue
+            typer.echo(f"Adding {path} to Git")
+            subprocess.call(["git", "add", path])
+    if commit_message is not None:
+        subprocess.call(["git", "commit", "-m", commit_message])
+    if push_commit:
+        push()
+
+
+@app.command(name="commit")
+def commit(
+    all: Annotated[
+        Optional[bool],
+        typer.Option(
+            "--all", "-a", help="Automatically stage all changed files."
+        ),
+    ] = False,
+    message: Annotated[
+        Optional[str], typer.Option("--message", "-m", help="Commit message.")
+    ] = None,
+    push_commit: Annotated[
+        bool,
+        typer.Option(
+            "--push", help="Push to both Git and DVC after committing."
+        ),
+    ] = False,
+):
+    """Commit a change to the repo."""
+    cmd = ["git", "commit"]
+    if all:
+        cmd.append("-a")
+    if message:
+        cmd += ["-m", message]
+    subprocess.call(cmd)
+    if push_commit:
+        push()
+
+
+@app.command(name="pull", help="Pull with both Git and DVC.")
+def pull():
+    typer.echo("Git pulling")
+    subprocess.call(["git", "pull"])
+    typer.echo("DVC pulling")
+    subprocess.call(["dvc", "pull"])
+
+
+@app.command(name="push", help="Push with both Git and DVC.")
+def push():
+    typer.echo("Pushing to Git remote")
+    subprocess.call(["git", "push"])
+    typer.echo("Pushing to DVC remote")
+    subprocess.call(["dvc", "push"])
+
+
+@app.command(name="server", help="Run the local server.")
+def run_server():
+    import uvicorn
+
+    uvicorn.run(
+        "calkit.server:app",
+        port=8866,
+        host="localhost",
+        reload=True,
+        reload_dirs=[os.path.dirname(__file__)],
+    )
+
+
+@app.command(
+    name="run",
+    add_help_option=False,
+    help="Run DVC pipeline (a wrapper for `dvc repro`).",
+)
+def run_dvc_repro(
+    targets: Optional[list[str]] = typer.Argument(default=None),
+    help: Annotated[bool, typer.Option("-h", "--help")] = False,
+    quiet: Annotated[bool, typer.Option("-q", "--quiet")] = False,
+    verbose: Annotated[bool, typer.Option("-v", "--verbose")] = False,
+    force: Annotated[bool, typer.Option("-f", "--force")] = False,
+    interactive: Annotated[bool, typer.Option("-i", "--interactive")] = False,
+    single_item: Annotated[bool, typer.Option("-s", "--single-item")] = False,
+    pipeline: Annotated[
+        Optional[str], typer.Option("-p", "--pipeline")
+    ] = None,
+    all_pipelines: Annotated[
+        bool, typer.Option("-P", "--all-pipelines")
+    ] = False,
+    recursive: Annotated[bool, typer.Option("-R", "--recursive")] = False,
+    downstream: Annotated[
+        Optional[list[str]], typer.Option("--downstream")
+    ] = None,
+    force_downstream: Annotated[
+        bool, typer.Option("--force-downstream")
+    ] = False,
+    pull: Annotated[bool, typer.Option("--pull")] = False,
+    allow_missing: Annotated[bool, typer.Option("--allow-missing")] = False,
+    dry: Annotated[bool, typer.Option("--dry")] = False,
+    keep_going: Annotated[bool, typer.Option("--keep-going", "-k")] = False,
+    ignore_errors: Annotated[bool, typer.Option("--ignore-errors")] = False,
+    glob: Annotated[bool, typer.Option("--glob")] = False,
+    no_commit: Annotated[bool, typer.Option("--no-commit")] = False,
+    no_run_cache: Annotated[bool, typer.Option("--no-run-cache")] = False,
+):
+    """A simple wrapper for ``dvc repro`` that will automatically create any
+    necessary Calkit objects from stage metadata.
+    """
+    if targets is None:
+        targets = []
+    args = targets
+    # Extract any boolean args
+    for name in [
+        "help",
+        "quiet",
+        "verbose",
+        "force",
+        "interactive",
+        "single-item",
+        "all-pipelines",
+        "recursive",
+        "pull",
+        "allow-missing",
+        "dry",
+        "keep-going",
+        "force-downstream",
+        "glob",
+        "no-commit",
+        "no-run-cache",
+    ]:
+        if locals()[name.replace("-", "_")]:
+            args.append("--" + name)
+    if pipeline is not None:
+        args += ["--pipeline", pipeline]
+    if downstream is not None:
+        args += downstream
+    subprocess.call(["dvc", "repro"] + args)
+    # Now parse stage metadata for calkit objects
+    if not os.path.isfile("dvc.yaml"):
+        typer.echo("No dvc.yaml file found")
+        raise typer.Exit(1)
+    objects = []
+    with open("dvc.yaml") as f:
+        pipeline = calkit.ryaml.load(f)
+        for stage_name, stage_info in pipeline.get("stages", {}).items():
+            ckmeta = stage_info.get("meta", {}).get("calkit")
+            if ckmeta is not None:
+                if not isinstance(ckmeta, dict):
+                    typer.echo(
+                        f"Calkit metadata for {stage_name} is not a dictionary"
+                    )
+                    typer.Exit(1)
+                # Stage must have a single output
+                outs = stage_info.get("outs", [])
+                if len(outs) != 1:
+                    typer.echo(
+                        f"Stage {stage_name} does not have exactly one output"
+                    )
+                    raise typer.Exit(1)
+                cktype = ckmeta.get("type")
+                if cktype not in ["figure", "dataset", "publication"]:
+                    typer.echo(f"Invalid Calkit output type '{cktype}'")
+                    raise typer.Exit(1)
+                objects.append(
+                    dict(path=outs[0]) | ckmeta | dict(stage=stage_name)
+                )
+    # Now that we've extracted Calkit objects from stage metadata, we can put
+    # them into the calkit.yaml file, overwriting objects with the same path
+    if os.path.isfile("calkit.yaml"):
+        with open("calkit.yaml") as f:
+            ck_info = calkit.ryaml.load(f)
+    else:
+        ck_info = {}
+    for obj in objects:
+        cktype = obj.pop("type")
+        cktype_plural = cktype + "s"
+        existing = ck_info.get(cktype_plural, [])
+        new = []
+        added = False
+        for ex_obj in existing:
+            if ex_obj.get("path") == obj["path"]:
+                typer.echo(f"Updating {cktype} {ex_obj['path']}")
+                new.append(obj)
+                added = True
+            else:
+                new.append(ex_obj)
+        if not added:
+            typer.echo(f"Adding new {cktype} {obj['path']}")
+            new.append(obj)
+        ck_info[cktype_plural] = new
+    if not dry:
+        with open("calkit.yaml", "w") as f:
+            calkit.ryaml.dump(ck_info, f)
+        run_cmd(["git", "add", "calkit.yaml"])
+
+
+@app.command(name="manual-step", help="Execute a manual step.")
+def manual_step(
+    message: Annotated[
+        str,
+        typer.Option(
+            "--message",
+            "-m",
+            help="Message to display as a prompt.",
+        ),
+    ],
+    cmd: Annotated[str, typer.Option("--cmd", help="Command to run.")] = None,
+    shell: Annotated[
+        bool,
+        typer.Option(
+            "--shell",
+            help="Whether or not to execute the command in shell mode.",
+        ),
+    ] = False,
+    show_stdout: Annotated[
+        bool, typer.Option("--show-stdout", help="Show stdout.")
+    ] = False,
+    show_stderr: Annotated[
+        bool, typer.Option("--show-stderr", help="Show stderr.")
+    ] = False,
+) -> None:
+    if cmd is not None:
+        typer.echo(f"Running command: {cmd}")
+        subprocess.Popen(
+            cmd.split() if not shell else cmd,
+            stderr=subprocess.PIPE if not show_stderr else None,
+            stdout=subprocess.PIPE if not show_stdout else None,
+            shell=shell,
+        )
+    input(message + " (press enter to confirm): ")
+    typer.echo("Done")

calkit/cli/new.py

@@ -0,0 +1,107 @@
+"""CLI for creating new objects."""
+
+from __future__ import annotations
+
+import os
+
+import git
+import typer
+from typing_extensions import Annotated
+
+from calkit.core import ryaml
+
+new_app = typer.Typer(no_args_is_help=True)
+
+
+@new_app.command(name="figure")
+def new_figure(
+    path: str,
+    title: Annotated[str, typer.Option("--title")],
+    description: Annotated[str, typer.Option("--desc")] = None,
+    commit: Annotated[bool, typer.Option("--commit")] = False,
+):
+    """Add a new figure."""
+    if os.path.isfile("calkit.yaml"):
+        with open("calkit.yaml") as f:
+            ck_info = ryaml.load(f)
+    else:
+        ck_info = {}
+    figures = ck_info.get("figures", [])
+    paths = [f.get("path") for f in figures]
+    if path in paths:
+        raise ValueError(f"Figure at path {path} already exists")
+    obj = dict(path=path, title=title)
+    if description is not None:
+        obj["description"] = description
+    figures.append(obj)
+    ck_info["figures"] = figures
+    with open("calkit.yaml", "w") as f:
+        ryaml.dump(ck_info, f)
+    if commit:
+        repo = git.Repo()
+        repo.git.add("calkit.yaml")
+        repo.git.commit(["-m", f"Add figure {path}"])
+
+
+@new_app.command("question")
+def new_question(
+    question: str,
+    commit: Annotated[bool, typer.Option("--commit")] = False,
+):
+    """Add a new question."""
+    if os.path.isfile("calkit.yaml"):
+        with open("calkit.yaml") as f:
+            ck_info = ryaml.load(f)
+    else:
+        ck_info = {}
+    questions = ck_info.get("questions", [])
+    if question in questions:
+        raise ValueError("Question already exists")
+    if not question.endswith("?"):
+        raise ValueError("Questions must end with a question mark")
+    questions.append(question)
+    ck_info["questions"] = questions
+    with open("calkit.yaml", "w") as f:
+        ryaml.dump(ck_info, f)
+    if commit:
+        repo = git.Repo()
+        repo.git.add("calkit.yaml")
+        repo.git.commit(["-m", "Add question"])
+
+
+@new_app.command("notebook")
+def new_notebook(
+    path: Annotated[str, typer.Argument(help="Notebook path (relative)")],
+    title: Annotated[str, typer.Option("--title")],
+    description: Annotated[str, typer.Option("--desc")] = None,
+    commit: Annotated[bool, typer.Option("--commit")] = False,
+):
+    """Add a new notebook."""
+    if os.path.isabs(path):
+        raise ValueError("Path must be relative")
+    if not os.path.isfile(path):
+        raise ValueError("Path is not a file")
+    if not path.endswith(".ipynb"):
+        raise ValueError("Path does not have .ipynb extension")
+    # TODO: Add option to create stages that run `calkit nb clean` and
+    # `calkit nb execute`
+    if os.path.isfile("calkit.yaml"):
+        with open("calkit.yaml") as f:
+            ck_info = ryaml.load(f)
+    else:
+        ck_info = {}
+    notebooks = ck_info.get("notebooks", [])
+    paths = [f.get("path") for f in notebooks]
+    if path in paths:
+        raise ValueError(f"Notebook at path {path} already exists")
+    obj = dict(path=path, title=title)
+    if description is not None:
+        obj["description"] = description
+    notebooks.append(obj)
+    ck_info["notebooks"] = notebooks
+    with open("calkit.yaml", "w") as f:
+        ryaml.dump(ck_info, f)
+    if commit:
+        repo = git.Repo()
+        repo.git.add("calkit.yaml")
+        repo.git.commit(["-m", f"Add notebook {path}"])

calkit/cli/notebooks.py

@@ -0,0 +1,76 @@
+"""Notebooks CLI."""
+
+from __future__ import annotations
+
+import os
+import subprocess
+
+import typer
+from typing_extensions import Annotated
+
+notebooks_app = typer.Typer(no_args_is_help=True)
+
+
+@notebooks_app.command("clean")
+def clean_notebook_outputs(path: str):
+    """Clean notebook and place a copy in the cleaned notebooks directory.
+
+    This can be useful to use as a preprocessing DVC stage to use a clean
+    notebook as a dependency for a stage that caches and executed notebook.
+    """
+    if os.path.isabs(path):
+        raise ValueError("Path must be relative")
+    fpath_out = os.path.join(".calkit", "notebooks", "cleaned", path)
+    folder = os.path.dirname(fpath_out)
+    os.makedirs(folder, exist_ok=True)
+    subprocess.call(
+        [
+            "jupyter",
+            "nbconvert",
+            path,
+            "--clear-output",
+            "--to",
+            "notebook",
+            "--output",
+            fpath_out,
+        ]
+    )
+
+
+@notebooks_app.command("execute")
+def execute_notebook(
+    path: str,
+    to: Annotated[
+        str, typer.Option("--to", help="Output format ('html' or 'notebook').")
+    ] = "notebook",
+):
+    """Execute notebook and place a copy in the relevant directory.
+
+    This can be useful to use as a preprocessing DVC stage to use a clean
+    notebook as a dependency for a stage that caches and executed notebook.
+    """
+    if os.path.isabs(path):
+        raise ValueError("Path must be relative")
+    if to == "html":
+        subdir = "html"
+        fname_out = path.removesuffix(".ipynb") + ".html"
+    elif to == "notebook":
+        subdir = "executed"
+        fname_out = path
+    else:
+        raise ValueError(f"Invalid output format: '{to}'")
+    fpath_out = os.path.join(".calkit", "notebooks", subdir, fname_out)
+    folder = os.path.dirname(fpath_out)
+    os.makedirs(folder, exist_ok=True)
+    subprocess.call(
+        [
+            "jupyter",
+            "nbconvert",
+            path,
+            "--execute",
+            "--to",
+            to,
+            "--output",
+            fpath_out,
+        ]
+    )

{calkit_python-0.0.6.dist-info → calkit_python-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: calkit-python
-Version: 0.0.6
+Version: 0.0.8
 Summary: Reproducibility simplified.
 Project-URL: Homepage, https://github.com/calkit/calkit
 Project-URL: Issues, https://github.com/calkit/calkit/issues

{calkit_python-0.0.6.dist-info → calkit_python-0.0.8.dist-info}/RECORD

@@ -1,5 +1,4 @@
-calkit/__init__.py,sha256=MLdE9hjYmVEWkHdbusBCZDUzUC3oiY_AL-IfCGKX8eI,142
-calkit/cli.py,sha256=Ko3ntXUTqYeiLv00PQz68SWRp_vY1hruQgjmgwXnuzc,11788
+calkit/__init__.py,sha256=IQMt4pG5Q-NtyYJJWLzR88yFOvC8riMn-DQ52wqYi8w,142
 calkit/cloud.py,sha256=CnQKms4mBQo4X8jVLIa-70-3jPDl38zRNFyK5TjiaHM,2209
 calkit/config.py,sha256=NPFRk5inn66Wy44pyx8WppyqvIzkW1sO6VjXwIcCacE,2044
 calkit/core.py,sha256=qLNjBwlECzMitL85pMGIHzne4j7FpQAR9-ukYtdIMig,1225
@@ -10,11 +9,17 @@ calkit/gui.py,sha256=2UCrMyosc5v4CLlDcHO7oDwW9eLn5_WbNaLaG5bjMTc,23
 calkit/jupyter.py,sha256=S6HUOVafaLQJ7_ZfC1ddBwF3fT5q3GcbouLHdlH-oSo,1034
 calkit/models.py,sha256=ejl032XIrbaCFkO-4VbCaKVQyHqNl7N4WWU20RjeAfc,1447
 calkit/server.py,sha256=gq3643ABKTQu1eVR_OqJ-SdQh2dPLHSFsipmmAK5ttA,5959
+calkit/cli/__init__.py,sha256=2Id-_5dsHbL29qqScLkRUhWXzTgxsZ_A_lMPs9rf5uY,77
+calkit/cli/config.py,sha256=y-TWv150qWvUtlMfySgCyMyk8n5SNL-8hNRm9NycJ98,1098
+calkit/cli/core.py,sha256=N-I2Yw1STgEMN5xLmfOuVq1od1EJx2oMCyVQtNa8Nwg,424
+calkit/cli/main.py,sha256=_zrl3nYszxDwpvis5rd0Qa5jP78XjLvril1pvn9BnhA,11378
+calkit/cli/new.py,sha256=Kn80X_z47uJ27Nkl8WJVWYqdwXyuaKpIy19_C1A26Q4,3386
+calkit/cli/notebooks.py,sha256=GsNmm8pdFkbs_KyZxClfIUThD78GCdu0R5udCXhXc9E,2083
 calkit/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 calkit/tests/test_core.py,sha256=CdRqvYnq3MidyTknvfIbkmmhMfydpyKtTW9hakAvzsc,167
 calkit/tests/test_jupyter.py,sha256=YTL6zI740UM2KUjskSipzvSKxsyQ8rVPFQIX5cb2tMQ,114
-calkit_python-0.0.6.dist-info/METADATA,sha256=kL3dH09fqVyHjPjkKnclC21IAHwmwP8NNdvWgOEEthg,4004
-calkit_python-0.0.6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-calkit_python-0.0.6.dist-info/entry_points.txt,sha256=tiPfbDDZNB-YStw5YVO_4Gww0W3Dcdy8Mh6muX_GWnY,42
-calkit_python-0.0.6.dist-info/licenses/LICENSE,sha256=9ZamCaSUTZk9rcrnf-sWFKLOHr3ws-S_dgKMegW4nw8,1056
-calkit_python-0.0.6.dist-info/RECORD,,
+calkit_python-0.0.8.dist-info/METADATA,sha256=I1R0dOcHch17Q1K3vbqQ83CWU95Eu0ygri_rWfBsZbc,4004
+calkit_python-0.0.8.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+calkit_python-0.0.8.dist-info/entry_points.txt,sha256=tiPfbDDZNB-YStw5YVO_4Gww0W3Dcdy8Mh6muX_GWnY,42
+calkit_python-0.0.8.dist-info/licenses/LICENSE,sha256=9ZamCaSUTZk9rcrnf-sWFKLOHr3ws-S_dgKMegW4nw8,1056
+calkit_python-0.0.8.dist-info/RECORD,,

calkit/cli.py

@@ -1,392 +0,0 @@
-"""The command line interface."""
-
-from __future__ import annotations
-
-import os
-import pty
-import subprocess
-import sys
-
-import git
-import typer
-from typing_extensions import Annotated, Optional
-
-import calkit
-from calkit.core import ryaml
-
-from . import config
-from .dvc import configure_remote, set_remote_auth
-
-app = typer.Typer(
-    invoke_without_command=True,
-    no_args_is_help=True,
-    context_settings=dict(help_option_names=["-h", "--help"]),
-)
-config_app = typer.Typer(no_args_is_help=True)
-new_app = typer.Typer(no_args_is_help=True)
-notebooks_app = typer.Typer(no_args_is_help=True)
-app.add_typer(config_app, name="config", help="Configure Calkit.")
-app.add_typer(
-    new_app, name="new", help="Add new Calkit object (to calkit.yaml)."
-)
-app.add_typer(notebooks_app, name="nb", help="Work with Jupyter notebooks.")
-
-
-@app.callback()
-def main(
-    version: Annotated[
-        bool,
-        typer.Option("--version", help="Show version and exit."),
-    ] = False,
-):
-    if version:
-        typer.echo(calkit.__version__)
-        raise typer.Exit()
-
-
-@config_app.command(name="set")
-def set_config_value(key: str, value: str):
-    try:
-        cfg = config.read()
-        cfg = config.Settings.model_validate(cfg.model_dump() | {key: value})
-        # Kind of a hack for setting the password computed field
-        # Types have been validated above, so this won't hurt to do again
-        setattr(cfg, key, value)
-    except FileNotFoundError:
-        # TODO: This fails if we try to set password before any config has
-        # been written
-        # Username is fine
-        cfg = config.Settings.model_validate({key: value})
-    cfg.write()
-
-
-@config_app.command(name="get")
-def get_config_value(key: str) -> None:
-    cfg = config.read()
-    val = getattr(cfg, key)
-    if val is not None:
-        print(val)
-    else:
-        print()
-
-
-@config_app.command(name="setup-remote")
-def setup_remote():
-    configure_remote()
-    set_remote_auth()
-
-
-@app.command(name="status")
-def get_status():
-    """Get a unified Git and DVC status."""
-
-    def print_sep(name: str):
-        width = 66
-        txt_width = len(name) + 2
-        buffer_width = (width - txt_width) // 2
-        buffer = "-" * buffer_width
-        typer.echo(f"{buffer} {name} {buffer}")
-
-    def run_cmd(cmd: list[str]):
-        if os.name == "nt":
-            subprocess.call(cmd)
-        else:
-            pty.spawn(cmd, lambda fd: os.read(fd, 1024))
-
-    print_sep("Code (Git)")
-    run_cmd(["git", "status"])
-    typer.echo()
-    print_sep("Data (DVC)")
-    run_cmd(["dvc", "data", "status"])
-    typer.echo()
-    print_sep("Pipeline (DVC)")
-    run_cmd(["dvc", "status"])
-
-
-@app.command(name="add")
-def add(paths: list[str]):
-    """Add paths to the repo.
-
-    Code will be added to Git and data will be added to DVC.
-    """
-    dvc_extensions = [".png", ".h5", ".parquet", ".pickle"]
-    dvc_size_thresh_bytes = 1_000_000
-    dvc_folders = ["data", "figures"]
-    if "." in paths:
-        print("ERROR: Cannot add '.' with calkit; use git or dvc")
-        sys.exit(1)
-    for path in paths:
-        if os.path.isdir(path):
-            print("ERROR: Cannot add directories with calkit; use git or dvc")
-            sys.exit(1)
-        # Detect if this file should be tracked with Git or DVC
-        # TODO: Add to whatever
-
-
-@app.command(name="commit")
-def commit(
-    all: Annotated[Optional[bool], typer.Option("--all", "-a")] = False,
-    message: Annotated[Optional[str], typer.Option("--message", "-m")] = None,
-):
-    """Commit a change to the repo."""
-    print(all, message)
-    raise NotImplementedError
-
-
-@app.command(name="pull", help="Pull with both Git and DVC.")
-def pull():
-    typer.echo("Git pulling")
-    subprocess.call(["git", "pull"])
-    typer.echo("DVC pulling")
-    subprocess.call(["dvc", "pull"])
-
-
-@app.command(name="push", help="Push with both Git and DVC.")
-def push():
-    typer.echo("Pushing to Git remote")
-    subprocess.call(["git", "push"])
-    typer.echo("Pushing to DVC remote")
-    subprocess.call(["dvc", "push"])
-
-
-@new_app.command(name="figure")
-def new_figure(
-    path: str,
-    title: Annotated[str, typer.Option("--title")],
-    description: Annotated[str, typer.Option("--desc")] = None,
-    commit: Annotated[bool, typer.Option("--commit")] = False,
-):
-    """Add a new figure."""
-    if os.path.isfile("calkit.yaml"):
-        with open("calkit.yaml") as f:
-            ck_info = ryaml.load(f)
-    else:
-        ck_info = {}
-    figures = ck_info.get("figures", [])
-    paths = [f.get("path") for f in figures]
-    if path in paths:
-        raise ValueError(f"Figure at path {path} already exists")
-    obj = dict(path=path, title=title)
-    if description is not None:
-        obj["description"] = description
-    figures.append(obj)
-    ck_info["figures"] = figures
-    with open("calkit.yaml", "w") as f:
-        ryaml.dump(ck_info, f)
-    if commit:
-        repo = git.Repo()
-        repo.git.add("calkit.yaml")
-        repo.git.commit(["-m", f"Add figure {path}"])
-
-
-@new_app.command("question")
-def new_question(
-    question: str,
-    commit: Annotated[bool, typer.Option("--commit")] = False,
-):
-    """Add a new question."""
-    if os.path.isfile("calkit.yaml"):
-        with open("calkit.yaml") as f:
-            ck_info = ryaml.load(f)
-    else:
-        ck_info = {}
-    questions = ck_info.get("questions", [])
-    if question in questions:
-        raise ValueError("Question already exists")
-    if not question.endswith("?"):
-        raise ValueError("Questions must end with a question mark")
-    questions.append(question)
-    ck_info["questions"] = questions
-    with open("calkit.yaml", "w") as f:
-        ryaml.dump(ck_info, f)
-    if commit:
-        repo = git.Repo()
-        repo.git.add("calkit.yaml")
-        repo.git.commit(["-m", "Add question"])
-
-
-@new_app.command("notebook")
-def new_notebook(
-    path: Annotated[str, typer.Argument(help="Notebook path (relative)")],
-    title: Annotated[str, typer.Option("--title")],
-    description: Annotated[str, typer.Option("--desc")] = None,
-    commit: Annotated[bool, typer.Option("--commit")] = False,
-):
-    """Add a new notebook."""
-    if os.path.isabs(path):
-        raise ValueError("Path must be relative")
-    if not os.path.isfile(path):
-        raise ValueError("Path is not a file")
-    if not path.endswith(".ipynb"):
-        raise ValueError("Path does not have .ipynb extension")
-    # TODO: Add option to create stages that run `calkit nb clean` and
-    # `calkit nb execute`
-    if os.path.isfile("calkit.yaml"):
-        with open("calkit.yaml") as f:
-            ck_info = ryaml.load(f)
-    else:
-        ck_info = {}
-    notebooks = ck_info.get("notebooks", [])
-    paths = [f.get("path") for f in notebooks]
-    if path in paths:
-        raise ValueError(f"Notebook at path {path} already exists")
-    obj = dict(path=path, title=title)
-    if description is not None:
-        obj["description"] = description
-    notebooks.append(obj)
-    ck_info["notebooks"] = notebooks
-    with open("calkit.yaml", "w") as f:
-        ryaml.dump(ck_info, f)
-    if commit:
-        repo = git.Repo()
-        repo.git.add("calkit.yaml")
-        repo.git.commit(["-m", f"Add notebook {path}"])
-
-
-@notebooks_app.command("clean")
-def clean_notebook_outputs(path: str):
-    """Clean notebook and place a copy in the cleaned notebooks directory.
-
-    This can be useful to use as a preprocessing DVC stage to use a clean
-    notebook as a dependency for a stage that caches and executed notebook.
-    """
-    if os.path.isabs(path):
-        raise ValueError("Path must be relative")
-    fpath_out = os.path.join(".calkit", "notebooks", "cleaned", path)
-    folder = os.path.dirname(fpath_out)
-    os.makedirs(folder, exist_ok=True)
-    subprocess.call(
-        [
-            "jupyter",
-            "nbconvert",
-            path,
-            "--clear-output",
-            "--to",
-            "notebook",
-            "--output",
-            fpath_out,
-        ]
-    )
-
-
-@notebooks_app.command("execute")
-def execute_notebook(
-    path: str,
-    to: Annotated[
-        str, typer.Option("--to", help="Output format ('html' or 'notebook').")
-    ] = "notebook",
-):
-    """Execute notebook and place a copy in the relevant directory.
-
-    This can be useful to use as a preprocessing DVC stage to use a clean
-    notebook as a dependency for a stage that caches and executed notebook.
-    """
-    if os.path.isabs(path):
-        raise ValueError("Path must be relative")
-    if to == "html":
-        subdir = "html"
-        fname_out = path.removesuffix(".ipynb") + ".html"
-    elif to == "notebook":
-        subdir = "executed"
-        fname_out = path
-    else:
-        raise ValueError(f"Invalid output format: '{to}'")
-    fpath_out = os.path.join(".calkit", "notebooks", subdir, fname_out)
-    folder = os.path.dirname(fpath_out)
-    os.makedirs(folder, exist_ok=True)
-    subprocess.call(
-        [
-            "jupyter",
-            "nbconvert",
-            path,
-            "--execute",
-            "--to",
-            to,
-            "--output",
-            fpath_out,
-        ]
-    )
-
-
-@app.command(name="server", help="Run the local server.")
-def run_server():
-    import uvicorn
-
-    uvicorn.run(
-        "calkit.server:app",
-        port=8866,
-        host="localhost",
-        reload=True,
-        reload_dirs=[os.path.dirname(__file__)],
-    )
-
-
-@app.command(
-    name="run",
-    add_help_option=False,
-    help="Run DVC pipeline (a wrapper for `dvc repro`).",
-)
-def run_dvc_repro(
-    targets: Optional[list[str]] = typer.Argument(default=None),
-    help: Annotated[bool, typer.Option("-h", "--help")] = False,
-    quiet: Annotated[bool, typer.Option("-q", "--quiet")] = False,
-    verbose: Annotated[bool, typer.Option("-v", "--verbose")] = False,
-    force: Annotated[bool, typer.Option("-f", "--force")] = False,
-    interactive: Annotated[bool, typer.Option("-i", "--interactive")] = False,
-    single_item: Annotated[bool, typer.Option("-s", "--single-item")] = False,
-    pipeline: Annotated[
-        Optional[str], typer.Option("-p", "--pipeline")
-    ] = None,
-    all_pipelines: Annotated[
-        bool, typer.Option("-P", "--all-pipelines")
-    ] = False,
-    recursive: Annotated[bool, typer.Option("-R", "--recursive")] = False,
-    downstream: Annotated[
-        Optional[list[str]], typer.Option("--downstream")
-    ] = None,
-    force_downstream: Annotated[
-        bool, typer.Option("--force-downstream")
-    ] = False,
-    pull: Annotated[bool, typer.Option("--pull")] = False,
-    allow_missing: Annotated[bool, typer.Option("--allow-missing")] = False,
-    dry: Annotated[bool, typer.Option("--dry")] = False,
-    keep_going: Annotated[bool, typer.Option("--keep-going", "-k")] = False,
-    ignore_errors: Annotated[bool, typer.Option("--ignore-errors")] = False,
-    glob: Annotated[bool, typer.Option("--glob")] = False,
-    no_commit: Annotated[bool, typer.Option("--no-commit")] = False,
-    no_run_cache: Annotated[bool, typer.Option("--no-run-cache")] = False,
-):
-    """A simple wrapper for ``dvc repro``."""
-    if targets is None:
-        targets = []
-    args = targets
-    # Extract any boolean args
-    for name in [
-        "help",
-        "quiet",
-        "verbose",
-        "force",
-        "interactive",
-        "single-item",
-        "all-pipelines",
-        "recursive",
-        "pull",
-        "allow-missing",
-        "dry",
-        "keep-going",
-        "force-downstream",
-        "glob",
-        "no-commit",
-        "no-run-cache",
-    ]:
-        if locals()[name.replace("-", "_")]:
-            args.append("--" + name)
-    if pipeline is not None:
-        args += ["--pipeline", pipeline]
-    if downstream is not None:
-        args += downstream
-    subprocess.call(["dvc", "repro"] + args)
-
-
-def run() -> None:
-    app()