From 426972e55f155d817e1db975afa2f25dbf860445 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Wed, 27 Nov 2019 21:40:19 +0000 Subject: [PATCH] disable super progress bar for single-files TODO: actually disable sub-bars instead --- dvc/repo/add.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/dvc/repo/add.py b/dvc/repo/add.py index 5e8da0b96..0b7c4e4c6 100644 --- a/dvc/repo/add.py +++ b/dvc/repo/add.py @@ -23,7 +23,14 @@ def add(repo, targets, recursive=False, no_commit=False, fname=None): targets = [targets] stages_list = [] - with Tqdm(total=len(targets), desc="Add", unit="file", leave=True) as pbar: + num_targets = len(targets) + with Tqdm( + total=num_targets, + desc="Add", + unit="file", + leave=True, + disable=True if num_targets < 2 else None, + ) as pbar: for target in targets: sub_targets = _find_all_targets(repo, target, recursive) pbar.total += len(sub_targets) - 1 @@ -55,6 +62,9 @@ def add(repo, targets, recursive=False, no_commit=False, fname=None): stages_list += stages + if pbar.disable: + pbar.write("100% Add") + return stages_list