Source code for whisk.cli.commands.project.dvc

import click
from whisk.project import Project
import os
from subprocess import call
[docs]project = Project()

@click.group()
[docs]def cli():
    pass

@cli.command()
@click.option("--bucket", default="whisk-" + project.name.replace("_","-"), show_default=True)
[docs]def remote_add(bucket):
    """
    Adds a DVC S3 remote as the default remote using the provided S3 bucket.
    """
    call("dvc remote add -d s3 s3://{}/".format(bucket), shell=True)

@cli.command()
@click.option("--name", default="s3", show_default=True)
[docs]def remote_remove(name):
    """
    Removes the DVC S3 remote.
    """
    call("dvc remote remove {}".format(name), shell=True)

@cli.command()
[docs]def setup():
    """
    Initializes dvc, adds a local remote, and installs git post-checkout hooks.

    Run `dvc destroy` to revert.
    """
    if os.system("dvc status > /dev/null 2>&1") != 0:
        # Initializing DVC
        call("dvc init > /dev/null", shell=True)
        # Setting up default local DVC remote
        call("dvc remote add -d local /tmp/dvc-storage", shell=True)
    if not os.path.isfile(".git/hooks/post-checkout"):
        # Installing Git hooks into the DVC repository
        call("dvc install > /dev/null", shell=True)