From 21c1b266cf4529f37fc476b0cdf668e47fa52ef9 Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <casper.dcl@physics.org>
Date: Wed, 27 Nov 2019 23:17:20 +0000
Subject: [PATCH] add progress for stages; also may need reverting

---
 dvc/repo/add.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/dvc/repo/add.py b/dvc/repo/add.py
index da5b9a2f4..b3d24eefb 100644
--- a/dvc/repo/add.py
+++ b/dvc/repo/add.py
@@ -72,9 +72,9 @@ def _find_all_targets(repo, target, recursive):
             fname
             for fname in Tqdm(
                 repo.tree.walk_files(target),
-                desc="Finding files",
+                desc="Recursing " + target,
                 bar_format=Tqdm.BAR_FMT_NOTOTAL,
-                leave=False,
+                unit="file",
             )
             if not repo.is_dvc_internal(fname)
             if not Stage.is_stage_file(fname)
@@ -87,7 +87,9 @@ def _find_all_targets(repo, target, recursive):
 def _create_stages(repo, targets, fname, pbar=None):
     stages = []
 
-    for out in targets:
+    for out in Tqdm(
+        targets, desc="Creating stages", disable=len(targets) < LARGE_DIR_SIZE
+    ):
         stage = Stage.create(repo, outs=[out], add=True, fname=fname)
 
         if not stage: