oarepo-runtime 2.0.0.dev14__tar.gz → 2.0.0.dev16__tar.gz

{oarepo_runtime-2.0.0.dev14 → oarepo_runtime-2.0.0.dev16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oarepo-runtime
-Version: 2.0.0.dev14
+Version: 2.0.0.dev16
 Summary: A set of runtime extensions of Invenio repository
 Project-URL: Homepage, https://github.com/oarepo/oarepo-runtime
 License-Expression: MIT

{oarepo_runtime-2.0.0.dev14 → oarepo_runtime-2.0.0.dev16}/oarepo_runtime/__init__.py

@@ -19,6 +19,6 @@ from .api import Model
 from .ext import OARepoRuntime
 from .proxies import current_runtime
 
-__version__ = "2.0.0dev14"
+__version__ = "2.0.0dev16"
 
 __all__ = ("Model", "OARepoRuntime", "__version__", "current_runtime")

{oarepo_runtime-2.0.0.dev14 → oarepo_runtime-2.0.0.dev16}/oarepo_runtime/api.py

@@ -41,6 +41,21 @@ if TYPE_CHECKING:
     )
 
 
+@dataclasses.dataclass
+class ModelMetadata:
+    """Model metadata configuration.
+
+    ModeMetadata is used in oarepo-model to add metadata to the model's oarepo_model_arguments dict.
+    """
+
+    types: dict[str, Any]
+    """Dictionary of types passed from InvenioModelBuilder.type_registry"""
+    metadata_type: str | None = None
+    """Metadata type"""
+    record_type: str | None = None
+    """Record type"""
+
+
 @dataclasses.dataclass
 class Export:
     """Configuration of an export format.
@@ -110,6 +125,7 @@ class Model[
         media_draft_file_service: FileService | None = None,
         exports: list[Export] | None = None,
         records_alias_enabled: bool = True,
+        model_metadata: ModelMetadata | None = None,
     ):
         """Initialize the model configuration.
 
@@ -132,6 +148,7 @@ class Model[
             If not provided, no exports are available.
         :param records_alias_enabled: Whether the records alias is enabled for this model.
             Such models will be searchable via the `/api/records` endpoint.
+        :param model_metadata: Metadata of the model.
         """
         self._code = code
         self._name = name
@@ -153,6 +170,7 @@ class Model[
         self._resource = resource
         self._resource_config = resource_config
         self._exports = exports or []
+        self._model_metadata = model_metadata
 
     @property
     def code(self) -> str:
@@ -183,6 +201,11 @@ class Model[
         """
         return self._records_alias_enabled
 
+    @property
+    def model_metadata(self) -> ModelMetadata | None:
+        """Get the model metadata."""
+        return self._model_metadata
+
     @property
     def ui_model(self) -> Mapping[str, Any]:
         """Get the UI model."""

{oarepo_runtime-2.0.0.dev14 → oarepo_runtime-2.0.0.dev16}/oarepo_runtime/ext.py

@@ -29,6 +29,7 @@ if TYPE_CHECKING:  # pragma: no cover
     from flask import Flask
     from invenio_drafts_resources.records.api import Draft
     from invenio_records_resources.services.base.service import Service
+    from invenio_records_resources.services.files.service import FileService
     from invenio_records_resources.services.records import RecordService
 
     from .api import Model
@@ -170,6 +171,16 @@ class OARepoRuntime:
             raise ValueError("Need to pass a record instance, got None")
         return self.get_record_service_for_record_class(type(record))
 
+    def get_file_service_for_record(self, record: Any) -> FileService | None:
+        """Return the file service for the given record (draft or published)."""
+        model = self.models_by_record_class.get(type(record))
+        if not model:
+            raise KeyError(f"No model found for record class '{type(record).__name__}'.")
+        is_draft = getattr(record, "is_draft", False)
+        if is_draft:
+            return model.draft_file_service
+        return model.file_service
+
     def get_record_service_for_record_class(self, record_cls: type[RecordBase]) -> RecordService:
         """Retrieve the service associated with a given record class."""
         for t in record_cls.mro():