ssrjson-benchmark 0.0.3__cp310-cp310-win_amd64.whl → 0.0.5__cp310-cp310-win_amd64.whl

ssrjson_benchmark/__init__.py

@@ -1,17 +1,19 @@
-from .benchmark_main import (
-    run_benchmark,
+from .benchmark_impl import (
+    generate_report_pdf,
     generate_report_markdown,
-    generate_report,
-    run_benchmark_default,
+    run_benchmark,
 )
-from importlib.metadata import version
 
-__version__ = version("ssrjson-benchmark")
+try:
+    from importlib.metadata import version
+
+    __version__ = version("ssrjson-benchmark")
+except Exception:
+    __version__ = "0.0.0"
 
 __all__ = [
     "run_benchmark",
     "generate_report_markdown",
-    "generate_report",
-    "run_benchmark_default",
+    "generate_report_pdf",
     "__version__",
 ]

ssrjson_benchmark/__main__.py

@@ -1,31 +1,53 @@
-import argparse
-import json
-import os
-from ssrjson_benchmark import (
-    run_benchmark,
-    generate_report_markdown,
-    generate_report,
-)
+def main():
+    import argparse
+    import json
+    import os
+    import pathlib
+    import sys
 
+    from .benchmark_impl import (
+        generate_report_markdown,
+        generate_report_pdf,
+        parse_file_result,
+        run_benchmark,
+    )
 
-def main():
     parser = argparse.ArgumentParser()
 
     parser.add_argument(
         "-f", "--file", help="record JSON file", required=False, default=None
     )
+    parser.add_argument(
+        "-d",
+        "--in-dir",
+        help="Benchmark JSON files directory",
+        required=False,
+    )
     parser.add_argument(
         "-m",
         "--markdown",
-        help="Generate markdown report",
+        help="Generate Markdown report",
         required=False,
         action="store_true",
     )
     parser.add_argument(
-        "--process-bytes",
-        help="Total process bytes per test, default 1e8",
+        "--no-pdf",
+        help="Don't generate PDF report",
         required=False,
-        default=1e8,
+        action="store_true",
+    )
+    parser.add_argument(
+        "--process-gigabytes",
+        help="Total process gigabytes per test, default 0.1 (float)",
+        required=False,
+        default=0.1,
+        type=float,
+    )
+    parser.add_argument(
+        "--bin-process-megabytes",
+        help="Maximum bytes to process per read for binary formats, default 32 (int)",
+        required=False,
+        default=32,
         type=int,
     )
     parser.add_argument(
@@ -35,19 +57,40 @@ def main():
         default=os.getcwd(),
     )
     args = parser.parse_args()
+    if args.file and args.no_pdf and not args.markdown:
+        print("Nothing to do.")
+        sys.exit(0)
+
+    _benchmark_files_dir = args.in_dir
+    if not _benchmark_files_dir:
+        _benchmark_files_dir = os.path.join(
+            os.path.dirname(os.path.abspath(__file__)), "_files"
+        )
+    benchmark_files_dir = sorted(pathlib.Path(_benchmark_files_dir).glob("*.json"))
+    if not benchmark_files_dir:
+        print(f"No benchmark file found using given path: {_benchmark_files_dir}")
+        sys.exit(0)
 
     if args.file:
-        with open(args.file, "r") as f:
-            j = json.load(f)
+        with open(args.file, "rb") as f:
+            result_ = json.load(f)
+        result = parse_file_result(result_)
         file = args.file.split("/")[-1]
     else:
-        j, file = run_benchmark(args.process_bytes)
+        process_bytes = int(args.process_gigabytes * 1024 * 1024 * 1024)
+        bin_process_bytes = args.bin_process_megabytes * 1024 * 1024
+        if process_bytes <= 0 or bin_process_bytes <= 0:
+            print("process-gigabytes and bin-process-megabytes must be positive.")
+            sys.exit(1)
+        result, file = run_benchmark(
+            benchmark_files_dir, process_bytes, bin_process_bytes
+        )
         file = file.split("/")[-1]
 
     if args.markdown:
-        generate_report_markdown(j, file, args.out_dir)
-    else:
-        generate_report(j, file, args.out_dir)
+        generate_report_markdown(result, file, args.out_dir)
+    if not args.no_pdf:
+        generate_report_pdf(result, file, args.out_dir)
 
 
 if __name__ == "__main__":