ssrjson-benchmark 0.0.9__cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

ssrjson_benchmark/__init__.py

@@ -0,0 +1,21 @@
+from . import _ssrjson_benchmark as internal
+from .benchmark_impl import (
+    generate_report_markdown,
+    generate_report_pdf,
+    run_benchmark,
+)
+
+try:
+    from importlib.metadata import version
+
+    __version__ = version("ssrjson-benchmark")
+except Exception:
+    __version__ = "0.0.0"
+
+__all__ = [
+    "generate_report_markdown",
+    "generate_report_pdf",
+    "internal",
+    "run_benchmark",
+    "__version__",
+]

ssrjson_benchmark/__main__.py

@@ -0,0 +1,113 @@
+import sys
+
+
+def main():
+    try:
+        import msgspec
+        import orjson
+        import ssrjson
+        import ujson
+    except ImportError:
+        print("Please install required packages: msgspec, orjson, ssrjson, ujson")
+        return 1
+    import argparse
+    import json
+    import os
+    import pathlib
+
+    from .benchmark_impl import (
+        generate_report_markdown,
+        generate_report_pdf,
+        parse_file_result,
+        run_benchmark,
+    )
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        "-f",
+        "--file",
+        help="Use a result JSON file generated in previous benchmark to print report. Will skip all tests.",
+        required=False,
+        default=None,
+    )
+    parser.add_argument(
+        "-d",
+        "--in-dir",
+        help="Benchmark JSON files directory. If not provided, use the files bundled in this package.",
+        required=False,
+    )
+    parser.add_argument(
+        "-m",
+        "--markdown",
+        help="Generate Markdown report",
+        required=False,
+        action="store_true",
+    )
+    parser.add_argument(
+        "--no-pdf",
+        help="Don't generate PDF report",
+        required=False,
+        action="store_true",
+    )
+    parser.add_argument(
+        "--process-gigabytes",
+        help="Total gigabytes to process per test case, default 0.1 (float)",
+        required=False,
+        default=0.1,
+        type=float,
+    )
+    parser.add_argument(
+        "--bin-process-megabytes",
+        help="Maximum bytes to process per bin, default 32 (int)",
+        required=False,
+        default=32,
+        type=int,
+    )
+    parser.add_argument(
+        "--out-dir",
+        help="Output directory for reports",
+        required=False,
+        default=os.getcwd(),
+    )
+    args = parser.parse_args()
+    skip_tests = bool(args.file)
+    if skip_tests and args.no_pdf and not args.markdown:
+        print("Nothing to do.")
+        return 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}")
+        return 1
+
+    if skip_tests:
+        with open(args.file, "rb") as f:
+            result_ = json.load(f)
+        result = parse_file_result(result_)
+        file = args.file.split("/")[-1]
+    else:
+        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.")
+            return 1
+        result, file = run_benchmark(
+            benchmark_files_dir, process_bytes, bin_process_bytes
+        )
+        file = file.split("/")[-1]
+
+    if args.markdown:
+        generate_report_markdown(result, file, args.out_dir)
+    if not args.no_pdf:
+        generate_report_pdf(result, file, args.out_dir)
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())