disable super progress bar for single-files

TODO: actually disable sub-bars instead
This commit is contained in:
Casper da Costa-Luis
2020-01-26 23:33:38 +00:00
parent dee51991c3
commit 426972e55f
+11 -1
View File
@@ -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