Update benchmark_requests.py

scripts/benchmark/benchmark_requests.py

@@ -2,10 +2,15 @@
 
 import argparse
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from datetime import datetime
+from pathlib import Path
 
 from tqdm.auto import tqdm
 from util import make_request
 
+RESULTS_DIR = Path(__file__).parent / "results"
+RESULTS_DIR.mkdir(exist_ok=True)
+
 if __name__ == "__main__":
     """Main function to benchmark the PESUAuth API.
 
@@ -123,14 +128,30 @@
             else:
                 success.append(0)
 
-    outfile = (
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    if output:
+        outfile = Path(output)
+    else:
+        filename = (
+            f"benchmark_[t={timestamp}]"
+            f"_[n={num_requests}]"
+            f"_[w={max_workers}]"
+            f"_[r={route}]"
+            f"_[execution_mode={'par' if parallel else 'seq'}]"
+            ".csv"
+        )
+        outfile = RESULTS_DIR / filename
+
+    outfile.parent.mkdir(parents=True, exist_ok=True)
+
+    """outfile = (
         output
         if output
         else (
             f"benchmark_[num_requests={num_requests}]_[max_workers={max_workers}]_"
             f"[parallel={parallel}]_[route={route}]_[timeout={timeout}].csv"
         )
-    )
+    )"""
 
     with open(
         outfile,
@@ -139,6 +160,7 @@
         f.write("status,time\n")
         f.writelines(f"{s},{t}\n" for s, t in zip(success, times, strict=False))
 
+    print(f"Results saved to the following directory: {outfile}")
     print(f"Benchmark completed. Successful requests: {sum(success)} out of {len(success)}")
     print(f"Average time per request: {sum(times) / len(times):.2f} seconds")
     print(f"Total time taken: {sum(times):.2f} seconds")