python-ort 0.9.1__tar.gz → 0.9.3__tar.gz

{python_ort-0.9.1 → python_ort-0.9.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-ort
-Version: 0.9.1
+Version: 0.9.3
 Summary: A Python Ort model serialization library
 License-Expression: MIT
 License-File: LICENSE
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: click>=8.4.1
 Requires-Dist: license-expression>=30.4.4
 Requires-Dist: packageurl-python>=0.17.6
 Requires-Dist: pydantic>=2.13.4

{python_ort-0.9.1 → python_ort-0.9.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-ort"
-version = "0.9.1"
+version = "0.9.3"
 description = "A Python Ort model serialization library"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -18,6 +18,7 @@ classifiers = [
   "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
+  "click>=8.4.1",
   "license-expression>=30.4.4",
   "packageurl-python>=0.17.6",
   "pydantic>=2.13.4",
@@ -162,5 +163,5 @@ skip-magic-trailing-comma = false
 index-url = "https://pypi.org/simple"
 
 [tool.uv.build-backend]
-module-name = "ort"
+module-name = ["ort", "tools"]
 module-root = "src"

python_ort-0.9.3/src/tools/ort_validate.py

@@ -0,0 +1,61 @@
+# SPDX-FileCopyrightText: 2025 Helio Chissini de Castro <heliocastro@gmail.com>
+# SPDX-License-Identifier: MIT
+#
+import logging
+import sys
+from pathlib import Path
+from typing import TypeVar
+
+import click
+from pydantic import BaseModel, ValidationError
+
+from ort import RepositoryConfiguration, ort_yaml_load
+from ort.models import LicenseClassifications
+
+logger = logging.getLogger()
+
+ModelT = TypeVar("ModelT", bound=BaseModel)
+
+
+def load_and_validate(model: type[ModelT], datafile: str) -> None:
+    """Load ``datafile`` as YAML, validate it into ``model`` and pretty-print the result."""
+    try:
+        with Path(datafile).open() as fd:
+            data = ort_yaml_load(fd)
+        parsed = model.model_validate(data or {})
+        logger.debug(parsed.model_dump_json(indent=2))
+        logger.info(f"Successfully validated {datafile} as {model.__name__}.")
+    except ValidationError:
+        logger.error("Validation error while parsing the ORT result:")
+        sys.exit(1)
+    except OSError as e:
+        logger.error(f"Error while opening the file {datafile}: {e}")
+        sys.exit(1)
+
+
+@click.group()
+@click.option("-d", "--debug", is_flag=True, help="Enable debug logging.")
+def main(debug: bool = False) -> None:
+    logging.basicConfig(level=logging.DEBUG)
+    if debug:
+        logging.basicConfig(level=logging.DEBUG)
+    pass
+
+
+@click.command()
+@click.argument("datafile")
+def license_classifications(datafile):
+    load_and_validate(LicenseClassifications, datafile)
+
+
+@click.command()
+@click.argument("datafile")
+def repository_configuration(datafile):
+    load_and_validate(RepositoryConfiguration, datafile)
+
+
+main.add_command(repository_configuration)
+main.add_command(license_classifications)
+
+if __name__ == "__main__":
+    main()