oarepo-runtime 1.5.129__py3-none-any.whl → 1.5.130__py3-none-any.whl

oarepo_runtime/cli/validate.py

@@ -1,4 +1,5 @@
 import sys
+import traceback
 
 import click
 import yaml
@@ -6,6 +7,7 @@ from flask.cli import with_appcontext
 from invenio_db import db
 from invenio_records import Record
 from invenio_records_resources.proxies import current_service_registry
+from tqdm import tqdm
 
 from .base import oarepo
 
@@ -14,7 +16,6 @@ try:
 except ImportError:
     import json
 
-import sys
 from io import StringIO
 
 
@@ -34,9 +35,18 @@ def dump_data(d):
 )
 @click.argument("service-name")
 @click.argument("record-file", required=False)
+@click.option("--community", help="Community name")
 @click.option("--verbose/--no-verbose", is_flag=True)
+@click.option("--with-stacktrace", is_flag=True)
+@click.option(
+    "--fail-on-error",
+    is_flag=True,
+    help="Fail on the first error (for multiple records)",
+)
 @with_appcontext
-def validate(service_name, record_file, verbose):
+def validate(
+    service_name, record_file, community, verbose, with_stacktrace, fail_on_error
+):
     try:
         service = current_service_registry.get(service_name)
     except KeyError:
@@ -61,7 +71,13 @@ def validate(service_name, record_file, verbose):
 
     if not isinstance(data, list):
         data = [data]
-    for idx, d in enumerate(data):
+
+    errors_count = 0
+    for idx, d in enumerate(tqdm(data)):
+        if community:
+            d.setdefault("parent", {}).setdefault("communities", {})[
+                "default"
+            ] = community
         try:
             loaded = schema().load(d)
         except Exception as e:
@@ -71,25 +87,44 @@ def validate(service_name, record_file, verbose):
             )
             click.secho(dump_data(d))
             click.secho(e)
+            if with_stacktrace:
+                traceback.print_exc()
+            if fail_on_error:
+                sys.exit(1)
+            errors_count += 1
             continue
 
-        click.secho(
-            f"Marshmallow validation of record idx {idx+1} has been successful",
-            fg="green",
-        )
+        if verbose:
+            click.secho(
+                f"Marshmallow validation of record idx {idx+1} has been successful",
+                fg="green",
+            )
 
-        rec: Record = config.record_cls(loaded)
+        if hasattr(config, "draft_cls"):
+            record_cls = config.draft_cls
+        else:
+            record_cls = config.record_cls
 
         # Run pre create extensions to check vocabularies
         try:
             with db.session.begin_nested():
+
+                rec: Record = record_cls(
+                    loaded, model=record_cls.model_cls(id=None, data=data)
+                )
+                if record_cls.parent_record_cls:
+                    parent = record_cls.parent_record_cls(loaded["parent"])
+                    rec.parent = parent
+
                 for e in rec._extensions:
                     e.pre_commit(rec)
                 raise CheckOk()
         except CheckOk:
-            click.secho(
-                f"Pre-commit hook of record idx {idx+1} has been successful", fg="green"
-            )
+            if verbose:
+                click.secho(
+                    f"Pre-commit hook of record idx {idx+1} has been successful",
+                    fg="green",
+                )
         except Exception as e:
             click.secho(
                 f"Pre-commit validation of record idx {idx + 1} failed",
@@ -97,7 +132,19 @@ def validate(service_name, record_file, verbose):
             )
             click.secho(dump_data(d))
             click.secho(e)
+            if with_stacktrace:
+                traceback.print_exc()
+            if fail_on_error:
+                sys.exit(1)
+            errors_count += 1
             continue
 
         if verbose:
             yaml.safe_dump(loaded, sys.stdout, allow_unicode=True)
+
+    if errors_count:
+        click.secho(f"Validation finished with {errors_count} errors", fg="red")
+        sys.exit(1)
+    else:
+        click.secho("Validation finished successfully", fg="green")
+        sys.exit(0)

{oarepo_runtime-1.5.129.dist-info → oarepo_runtime-1.5.130.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: oarepo-runtime
-Version: 1.5.129
+Version: 1.5.130
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.129.dist-info → oarepo_runtime-1.5.130.dist-info}/RECORD

@@ -13,7 +13,7 @@ oarepo_runtime/cli/check.py,sha256=sCe2PeokSHvNOXHFZ8YHF8NMhsu5nYjyuZuvXHJ6X18,5
 oarepo_runtime/cli/configuration.py,sha256=M_19zrS0vo-7uHi047a3i4rQsK6pmp4QS0IEBAxMLZ0,2039
 oarepo_runtime/cli/fixtures.py,sha256=_aMQqPnspG0JuL5ewYTBa3HY1RaoOAI0V2R5q7e5frw,5570
 oarepo_runtime/cli/index.py,sha256=8Q2KZlAWjoMyZ9Ft9WIBJDws0ce15VF12o1tdYUP7AI,8936
-oarepo_runtime/cli/validate.py,sha256=HpSvHQCGHlrdgdpKix9cIlzlBoJEiT1vACZdMnOUGEY,2827
+oarepo_runtime/cli/validate.py,sha256=vcN7uj9tXOeQ4_vRv-ZYLc-aW43_qRYltJR8NiPLZKk,4318
 oarepo_runtime/cli/permissions/__init__.py,sha256=2qufYdUoCb9kG7Zy0gKNW5lpRyqbVQSNsf7shLwrThM,198
 oarepo_runtime/cli/permissions/base.py,sha256=Q5txKaofSzAlTVHn1fu5YurwF16Pkh63YB9U6xWYVAU,804
 oarepo_runtime/cli/permissions/evaluate.py,sha256=gxkHySd1vM7lSjR-xXzcACkCKDYurl5c567Kq5FVzpM,2086
@@ -150,9 +150,9 @@ tests/marshmallow_to_json/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 tests/marshmallow_to_json/test_datacite_ui_schema.py,sha256=82iLj8nW45lZOUewpWbLX3mpSkpa9lxo-vK-Qtv_1bU,48552
 tests/marshmallow_to_json/test_simple_schema.py,sha256=izZN9p0v6kovtSZ6AdxBYmK_c6ZOti2_z_wPT_zXIr0,1500
 tests/pkg_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-oarepo_runtime-1.5.129.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
-oarepo_runtime-1.5.129.dist-info/METADATA,sha256=MEE87EgTSva-yYSr22h56-RtZxqaM9r1VQG0HEtOleg,4721
-oarepo_runtime-1.5.129.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-oarepo_runtime-1.5.129.dist-info/entry_points.txt,sha256=k7O5LZUOGsVeSpB7ulU0txBUNp1CVQG7Q7TJIVTPbzU,491
-oarepo_runtime-1.5.129.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
-oarepo_runtime-1.5.129.dist-info/RECORD,,
+oarepo_runtime-1.5.130.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
+oarepo_runtime-1.5.130.dist-info/METADATA,sha256=z3g1h_tsoffkQAq5whumI2ejrVxXc5uxs_2L5wV0-mo,4721
+oarepo_runtime-1.5.130.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+oarepo_runtime-1.5.130.dist-info/entry_points.txt,sha256=k7O5LZUOGsVeSpB7ulU0txBUNp1CVQG7Q7TJIVTPbzU,491
+oarepo_runtime-1.5.130.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
+oarepo_runtime-1.5.130.dist-info/RECORD,,