use fire

Author: joyguoguo

fuzz/clean_fuzz_dir.py

@@ -1,51 +1,46 @@
 #!/usr/bin/env python3
 import os
 import shutil
-import argparse
+import fire
 
-# Default root directory
-DEFAULT_ROOT_DIR = "/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects"
+def clean_project_dirs(root_dir="/home/jiayiguo/FuzzAug/fuzz/oss-fuzz/projects"):
+    """
+    清理 oss-fuzz 项目目录下的 fuzz_inputs 和 tests-gen 文件夹
 
-def clean_project_dirs(root_dir):
+    Args:
+        root_dir (str): 项目的根目录路径
+    """
     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
 
-        # Delete fuzz_inputs directories
+        # 删除 fuzz_inputs 文件夹
         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
 
-        # Delete tests-gen directories
+        # 删除 tests-gen 文件夹
         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
 
-    print(f"\n✅ Done. Removed {removed_dirs} directories in total.")
+        # 如果需要删除 .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.")
 
 if __name__ == "__main__":
-    # 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)
+    fire.Fire(clean_project_dirs)