ID

7114174

date

2023-08-05 08:51:48+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

5e6645c

message

allow specifying a dir to empty

changed files

1 file, 7 additions, 1 deletion

• fs/emptydir

Changed files

fs/emptydir (1386) → fs/emptydir (1477)

diff --git a/fs/emptydir b/fs/emptydir
index 8fc767a..d9d173c 100755
--- a/fs/emptydir
+++ b/fs/emptydir
@@ -8,6 +8,7 @@ It prints the name of every directory it removes.
 
 import os
 import shutil
+import sys
 
 import humanize
 import termcolor
@@ -41,7 +42,12 @@ def delete_directory(d):
 if __name__ == "__main__":
     total_deleted = 0
 
-    for d, _, _ in os.walk("."):
+    try:
+        root = sys.argv[1]
+    except IndexError:
+        root = "."
+
+    for d, _, _ in os.walk(root):
         if can_be_deleted(d):
             total_deleted += delete_directory(d)