Skip to main content

Print progress in the cdir script

ID
263b64b
date
2024-08-01 12:59:20+00:00
author
Alex Chan <alex@alexwlchan.net>
parent
c839328
message
Print progress in the `cdir` script
changed files
2 files, 11 additions, 5 deletions

Changed files

config.fish (7240) → config.fish (7280)

diff --git a/config.fish b/config.fish
index 3b49112..a190c0b 100644
--- a/config.fish
+++ b/config.fish
@@ -195,6 +195,7 @@ end
 
 __create_bash_script_alias text/pp_xml.sh
 
+__create_python_script_alias fs/cdir.py
 __create_python_script_alias git/find_big_commits.py
 __create_python_script_alias git/git-cloc.py
 __create_python_script_alias images/chunky_pixels.py

fs/cdir (1675) → fs/cdir.py (1799)

diff --git a/fs/cdir b/fs/cdir.py
similarity index 92%
rename from fs/cdir
rename to fs/cdir.py
index af4def4..1cf01c6 100755
--- a/fs/cdir
+++ b/fs/cdir.py
@@ -29,15 +29,20 @@ import collections
 import os
 import sys
 
+import tqdm
+
 
 def count_entries_under(d):
-    total = 1
+    with tqdm.tqdm(desc=d, leave=False) as pbar:
+        total = 1
+
+        for _, dirnames, filenames in os.walk(d):
+            update = len(dirnames) + len(filenames)
 
-    for _, dirnames, filenames in os.walk(d):
-        total += len(dirnames)
-        total += len(filenames)
+            total += update
+            pbar.update(update)
 
-    return total
+        return total
 
 
 if __name__ == "__main__":