Set up command line arguments

Author: joyguoguo

fuzz/clean_fuzz_dir.py

@@ -1,42 +1,51 @@
 #!/usr/bin/env python3
 import os
 import shutil
+import argparse
 
-ROOT_DIR = "/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects"
+# Default root directory
+DEFAULT_ROOT_DIR = "/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects"
 
 def clean_project_dirs(root_dir):
     removed_files = 0
     removed_dirs = 0
 
-    # 遍历一级项目目录
+    # Walk through the root directory
     for project in os.listdir(root_dir):
         project_path = os.path.join(root_dir, project)
         if not os.path.isdir(project_path):
             continue
 
-        # 删除 fuzz_inputs 文件夹
+        # Delete fuzz_inputs directories
         fuzz_inputs_path = os.path.join(project_path, "fuzz_inputs")
         if os.path.isdir(fuzz_inputs_path):
             shutil.rmtree(fuzz_inputs_path)
             print(f"🗑️ Removed dir: {fuzz_inputs_path}")
             removed_dirs += 1
 
-        # 删除 tests-gen 文件夹
+        # Delete tests-gen directories
         tests_gen_path = os.path.join(project_path, "tests-gen")
         if os.path.isdir(tests_gen_path):
             shutil.rmtree(tests_gen_path)
             print(f"🗑️ Removed dir: {tests_gen_path}")
             removed_dirs += 1
 
-        # # 删除 .inputs.py 文件
-        # for fname in os.listdir(project_path):
-        #     if fname.endswith(".inputs.py"):
-        #         file_path = os.path.join(project_path, fname)
-        #         os.remove(file_path)
-        #         print(f"🗑️ Removed file: {file_path}")
-        #         removed_files += 1
-
-    print(f"\n✅ Done. Removed {removed_files} files and {removed_dirs} directories.")
+    print(f"\n✅ Done. Removed {removed_dirs} directories in total.")
 
 if __name__ == "__main__":
-    clean_project_dirs(ROOT_DIR)
+    # Set up command line arguments
+    parser = argparse.ArgumentParser(
+        description='Clean project directories by removing fuzz_inputs and tests-gen folders',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter
+    )
+    parser.add_argument('--root_dir', default=DEFAULT_ROOT_DIR,
+                        help='Root directory containing project folders')
+    args = parser.parse_args()
+
+    # Validate the root directory exists
+    if not os.path.isdir(args.root_dir):
+        print(f"❌ Error: Specified root directory does not exist: {args.root_dir}")
+        exit(1)
+        
+    print(f"Cleaning projects in: {args.root_dir}")
+    clean_project_dirs(args.root_dir)