digitalhub 0.10.0b4__py3-none-any.whl → 0.10.0b6__py3-none-any.whl

digitalhub/entities/_base/unversioned/entity.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import typing
 
 from digitalhub.entities._base.context.entity import ContextEntity
-from digitalhub.entities._operations.processor import processor
+from digitalhub.entities._processors.context import context_processor
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
@@ -25,4 +25,4 @@ class UnversionedEntity(ContextEntity):
         super().__init__(project, kind, metadata, spec, status, user)
         self.id = uuid
         self.name = uuid
-        self.key = processor.build_context_entity_key(project, self.ENTITY_TYPE, kind, uuid)
+        self.key = context_processor.build_context_entity_key(project, self.ENTITY_TYPE, kind, uuid)

digitalhub/entities/_base/versioned/entity.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import typing
 
 from digitalhub.entities._base.context.entity import ContextEntity
-from digitalhub.entities._operations.processor import processor
+from digitalhub.entities._processors.context import context_processor
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
@@ -26,4 +26,4 @@ class VersionedEntity(ContextEntity):
         super().__init__(project, kind, metadata, spec, status, user)
         self.name = name
         self.id = uuid
-        self.key = processor.build_context_entity_key(project, self.ENTITY_TYPE, kind, name, uuid)
+        self.key = context_processor.build_context_entity_key(project, self.ENTITY_TYPE, kind, name, uuid)

digitalhub/entities/_commons/enums.py

@@ -73,6 +73,7 @@ class BackendOperations(Enum):
 
     CREATE = "create"
     READ = "read"
+    READ_ALL_VERSIONS = "read_all_versions"
     UPDATE = "update"
     DELETE = "delete"
     LIST = "list"

digitalhub/entities/_commons/metrics.py

@@ -0,0 +1,164 @@
+from __future__ import annotations
+
+from typing import Any, Union
+
+from pydantic import BaseModel, ValidationError
+
+MetricType = Union[float, int, list[Union[float, int]]]
+
+
+class Metric(BaseModel):
+    """
+    Metric.
+    """
+
+    value: MetricType
+
+
+def validate_metric_value(value: Any) -> MetricType:
+    """
+    Validate metric value.
+
+    Parameters
+    ----------
+    value : Any
+        The value to validate.
+
+    Returns
+    -------
+    MetricType
+        The validated value.
+    """
+    try:
+        return Metric(value=value).value
+    except ValidationError as e:
+        raise ValueError("Invalid metric value. Must be a list of floats or ints or a float or an int.") from e
+
+
+def set_metrics(
+    metrics: dict[str, MetricType],
+    key: str,
+    value: Any,
+    overwrite: bool,
+    single_value: bool,
+) -> dict[str, MetricType]:
+    """
+    Set metric value.
+
+    Parameters
+    ----------
+    metrics : dict[str, MetricType]
+        The metrics dictionary.
+    key : str
+        The key of the entity.
+    value : Any
+        The value to set.
+    overwrite : bool
+        Whether to overwrite the metric.
+
+    Returns
+    -------
+    dict[str, MetricType]
+        The metrics dictionary.
+    """
+    if isinstance(value, list):
+        return handle_metric_list(metrics, key, value, overwrite)
+    elif single_value:
+        return handle_metric_single(metrics, key, value, overwrite)
+    return handle_metric_list_append(metrics, key, value, overwrite)
+
+
+def handle_metric_single(
+    metrics: dict[str, MetricType],
+    key: str,
+    value: float | int,
+    overwrite: bool,
+) -> dict:
+    """
+    Handle metric single value.
+
+    Parameters
+    ----------
+    metrics : dict[str, MetricType]
+        Metrics dictionary.
+    key : str
+        Key of the metric.
+    value : float
+        Value of the metric.
+    overwrite : bool
+        If True, overwrite existing metric.
+
+    Returns
+    -------
+    dict
+        Metrics dictionary.
+    """
+    if key not in metrics or overwrite:
+        metrics[key] = value
+    return metrics
+
+
+def handle_metric_list_append(
+    metrics: dict[str, MetricType],
+    key: str,
+    value: float | int,
+    overwrite: bool,
+) -> dict:
+    """
+    Handle metric list append.
+
+    Parameters
+    ----------
+    metrics : dict[str, MetricType]
+        Metrics dictionary.
+    key : str
+        Key of the metric.
+    value : float
+        Value of the metric.
+    overwrite : bool
+        If True, overwrite existing metric.
+
+    Returns
+    -------
+    dict
+        Metrics dictionary.
+    """
+    if key not in metrics or overwrite:
+        metrics[key] = [value]
+    elif isinstance(metrics[key], list):
+        metrics[key].append(value)
+    else:
+        metrics[key] = [metrics[key], value]
+    return metrics
+
+
+def handle_metric_list(
+    metrics: dict[str, MetricType],
+    key: str,
+    value: list[int | float],
+    overwrite: bool,
+) -> dict:
+    """
+    Handle metric list.
+
+    Parameters
+    ----------
+    metrics : dict[str, MetricType]
+        Metrics dictionary.
+    key : str
+        Key of the metric.
+    value : list[int | float]
+        Value of the metric.
+    overwrite : bool
+        If True, overwrite existing metric.
+
+    Returns
+    -------
+    dict
+        Metrics dictionary.
+    """
+    if key not in metrics or overwrite:
+        metrics[key] = value
+    else:
+        metrics[key].extend(value)
+    return metrics

digitalhub/entities/_commons/types.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from typing import Union
+
+MetricType = Union[float, int, list[Union[float, int]]]

digitalhub/entities/_commons/utils.py

@@ -1,12 +1,6 @@
 from __future__ import annotations
 
-from typing import Any
-
-from pydantic import ValidationError
-
 from digitalhub.entities._commons.enums import EntityTypes
-from digitalhub.entities._commons.models import Metric
-from digitalhub.utils.types import MetricType
 
 
 def parse_entity_key(key: str) -> tuple[str, str, str, str | None, str]:
@@ -87,23 +81,3 @@ def get_project_from_key(key: str) -> str:
     """
     project, _, _, _, _ = parse_entity_key(key)
     return project
-
-
-def validate_metric_value(value: Any) -> MetricType:
-    """
-    Validate metric value.
-
-    Parameters
-    ----------
-    value : Any
-        The value to validate.
-
-    Returns
-    -------
-    MetricType
-        The validated value.
-    """
-    try:
-        return Metric(value=value).value
-    except ValidationError as e:
-        raise ValueError("Invalid metric value. Must be a list of floats or ints or a float or an int.") from e