add recursive file find progress

for profiling; may need to revert
iterative · Jan 26, 2020 · 80e8270 · 80e8270
1 parent 0293f71
commit 80e8270
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dvc/repo/add.py b/dvc/repo/add.py
@@ -70,7 +70,12 @@ def _find_all_targets(repo, target, recursive):
     if os.path.isdir(target) and recursive:
         return [
             fname
-            for fname in repo.tree.walk_files(target)
+            for fname in Tqdm(
+                repo.tree.walk_files(target),
+                desc="Finding files",
+                bar_format=Tqdm.BAR_FMT_NOTOTAL,
+                leave=False,
+            )
             if not repo.is_dvc_internal(fname)
             if not Stage.is_stage_file(fname)
             if not repo.scm.belongs_to_scm(fname)