Sanity check. Just list the directories

dandi · Nov 8, 2024 · a443081 · a443081
1 parent 04b4193
commit a443081
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/.github/scripts/du.py b/.github/scripts/du.py
@@ -41,6 +41,7 @@ def prepare_report(directory):
     for user, data in report.items():
         data["disk_usage_human_readable"] = bytes_to_human_readable(data["disk_usage_bytes"])
 
+
     with open(OUTPUT_FILE, 'w') as f:
         json.dump(report, f, indent=4)
 
@@ -51,5 +52,9 @@ def prepare_report(directory):
     if len(sys.argv) != 2:
         print("Usage: du.py <directory_to_check>")
     else:
-        directory = sys.argv[1]
-        prepare_report(directory)
+        path = sys.argv[1]
+        directories = [d for d in os.listdir(path) if os.path.isdir(os.path.join(path, d))]
+
+        with open(OUTPUT_FILE, 'w') as f:
+            f.write("\n".join(directories))
+        # prepare_report(directory)