Update project.py

This commit is contained in:
Ines Montani 2020-06-27 15:03:21 +02:00
parent 8979dc254f
commit 165c37ccba
1 changed files with 10 additions and 4 deletions

View File

@ -73,7 +73,9 @@ def project_clone_cli(
# fmt: on # fmt: on
): ):
"""Clone a project template from a repository.""" """Clone a project template from a repository."""
project_clone(name, dest, repo=repo, git=git, no_init=no_init, verbose=verbose) project_clone(
name, dest, repo=repo, git=git, no_init=no_init, verbose=verbose, silent=True
)
def project_clone( def project_clone(
@ -83,6 +85,7 @@ def project_clone(
repo: str = about.__projects__, repo: str = about.__projects__,
git: bool = False, git: bool = False,
no_init: bool = False, no_init: bool = False,
silent: bool = False,
verbose: bool = False, verbose: bool = False,
) -> None: ) -> None:
dest = ensure_path(dest) dest = ensure_path(dest)
@ -90,10 +93,13 @@ def project_clone(
# When cloning a subdirectory with DVC, it will create a folder of that name # When cloning a subdirectory with DVC, it will create a folder of that name
# within the destination dir, so we use a tempdir and then copy it into the # within the destination dir, so we use a tempdir and then copy it into the
# parent directory to create the cloned directory # parent directory to create the cloned directory
dest = dest.resolve()
with make_tempdir() as tmp_dir: with make_tempdir() as tmp_dir:
cmd = ["dvc", "get", repo, name, "-o", str(tmp_dir)] cmd = ["dvc", "get", repo, name, "-o", str(tmp_dir)]
if verbose: if verbose:
cmd.append("-v") cmd.append("--verbose")
if silent:
cmd.append("--quiet")
print(" ".join(cmd)) print(" ".join(cmd))
run_command(cmd) run_command(cmd)
shutil.move(str(tmp_dir / Path(name).name), str(dest)) shutil.move(str(tmp_dir / Path(name).name), str(dest))
@ -103,8 +109,8 @@ def project_clone(
if not dir_path.exists(): if not dir_path.exists():
dir_path.mkdir(parents=True) dir_path.mkdir(parents=True)
if not no_init: if not no_init:
project_init(dest, git=git) project_init(dest, git=git, silent=silent)
msg.good(f"Your project is now ready!", dest.resolve()) msg.good(f"Your project is now ready!", dest)
print(f"To fetch the assets, run:\npython -m {NAME} project assets {dest}") print(f"To fetch the assets, run:\npython -m {NAME} project assets {dest}")