digitalhub 0.10.2__py3-none-any.whl → 0.11.0b0__py3-none-any.whl

digitalhub/entities/function/_base/entity.py

@@ -5,14 +5,16 @@ from concurrent.futures import ThreadPoolExecutor
 
 from digitalhub.entities._base.executable.entity import ExecutableEntity
 from digitalhub.entities._commons.enums import EntityTypes, Relationship
-from digitalhub.factory.api import get_run_kind, get_task_kind_from_action
+from digitalhub.factory.factory import factory
 from digitalhub.utils.exceptions import BackendError
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
+    from digitalhub.entities._base.entity.spec import SpecValidator
     from digitalhub.entities.function._base.spec import FunctionSpec
     from digitalhub.entities.function._base.status import FunctionStatus
     from digitalhub.entities.run._base.entity import Run
+    from digitalhub.entities.trigger._base.entity import Trigger
 
 
 class Function(ExecutableEntity):
@@ -72,8 +74,8 @@ class Function(ExecutableEntity):
             Run instance.
         """
         # Get task and run kind
-        task_kind = get_task_kind_from_action(self.kind, action)
-        run_kind = get_run_kind(self.kind)
+        task_kind = factory.get_task_kind_from_action(self.kind, action)
+        run_kind = factory.get_run_kind(self.kind)
 
         # Create or update new task
         task = self._get_or_create_task(task_kind)
@@ -100,3 +102,47 @@ class Function(ExecutableEntity):
             result = executor.submit(run.run)
             r = result.result()
         return r
+
+    def trigger(self, action: str, trigger_kind: str, trigger_name: str, **kwargs) -> Trigger:
+        """
+        Trigger function.
+
+        Parameters
+        ----------
+        action : str
+            Action to execute.
+        trigger_kind : str
+            Trigger kind.
+        **kwargs : dict
+            Keyword arguments passed to Run builder.
+
+        Returns
+        -------
+        Run
+            Run instance.
+        """
+        # Get task
+        task_kind = factory.get_task_kind_from_action(self.kind, action)
+        task = self._get_or_create_task(task_kind)
+        task_string = task._get_task_string()
+
+        # Get run validator for building trigger template
+        run_kind = factory.get_run_kind(self.kind)
+        run_validator: SpecValidator = factory.get_spec_validator(run_kind)
+        if kwargs is None:
+            kwargs = {}
+        kwargs[self.ENTITY_TYPE] = self._get_executable_string()
+        kwargs["task"] = task_string
+
+        template = run_validator(**kwargs).to_dict()
+
+        # Override kwargs
+        kwargs["project"] = self.project
+        kwargs["kind"] = trigger_kind
+        kwargs["name"] = trigger_name
+        kwargs["template"] = template
+
+        # Create object instance
+        trigger: Trigger = factory.build_entity_from_params(**kwargs)
+        trigger.save()
+        return trigger

digitalhub/entities/project/_base/builder.py

@@ -26,6 +26,7 @@ class ProjectProjectBuilder(EntityBuilder):
         description: str | None = None,
         labels: list[str] | None = None,
         local: bool = False,
+        config: dict | None = None,
         context: str | None = None,
         **kwargs,
     ) -> Project:
@@ -44,6 +45,8 @@ class ProjectProjectBuilder(EntityBuilder):
             List of labels.
         local : bool
             If True, use local backend, if False use DHCore backend. Default to False.
+        config : dict
+            DHCore environment configuration.
         context : str
             The context local folder of the project.
         **kwargs : dict
@@ -62,6 +65,7 @@ class ProjectProjectBuilder(EntityBuilder):
             labels=labels,
         )
         spec = self.build_spec(
+            config=config,
             context=context,
             **kwargs,
         )

digitalhub/entities/project/_base/entity.py

@@ -67,7 +67,7 @@ from digitalhub.entities.workflow.crud import (
     new_workflow,
     update_workflow,
 )
-from digitalhub.factory.api import build_entity_from_dict
+from digitalhub.factory.factory import factory
 from digitalhub.stores.client.api import get_client
 from digitalhub.utils.exceptions import BackendError, EntityAlreadyExistsError, EntityError
 from digitalhub.utils.generic_utils import get_timestamp
@@ -106,6 +106,9 @@ class Project(Entity):
         local: bool = False,
     ) -> None:
         super().__init__(kind, metadata, spec, status, user)
+        self.spec: ProjectSpec
+        self.status: ProjectStatus
+
         self.id = name
         self.name = name
         self.key = base_processor.build_project_key(self.name, local=local)
@@ -329,7 +332,7 @@ class Project(Entity):
                         entity["metadata"]["embedded"] = True
 
                     try:
-                        build_entity_from_dict(entity).save()
+                        factory.build_entity_from_dict(entity).save()
                     except EntityAlreadyExistsError:
                         pass
 

digitalhub/entities/project/_base/models.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+_DEFAULT_FILE_STORE = "s3://datalake"
+
+
+class ProfileConfig(BaseModel):
+    """
+    Configuration profiles.
+    """
+
+    default_files_store: Optional[str] = _DEFAULT_FILE_STORE
+
+    def to_dict(self) -> dict:
+        return self.model_dump(by_alias=True, exclude_none=True)

digitalhub/entities/project/_base/spec.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import Optional
 
 from digitalhub.entities._base.entity.spec import Spec, SpecValidator
+from digitalhub.entities.project._base.models import ProfileConfig
 
 
 class ProjectSpec(Spec):
@@ -18,6 +19,7 @@ class ProjectSpec(Spec):
         workflows: list | None = None,
         dataitems: list | None = None,
         models: list | None = None,
+        config: dict | None = None,
         **kwargs,
     ) -> None:
         self.context = context if context is not None else "./"
@@ -26,6 +28,7 @@ class ProjectSpec(Spec):
         self.workflows = workflows if workflows is not None else []
         self.dataitems = dataitems if dataitems is not None else []
         self.models = models if models is not None else []
+        self.config = config if config is not None else {}
 
 
 class ProjectValidator(SpecValidator):
@@ -50,3 +53,6 @@ class ProjectValidator(SpecValidator):
 
     models: Optional[list] = None
     """List of project's models."""
+
+    config: Optional[ProfileConfig] = None
+    """Project's config."""

digitalhub/entities/project/crud.py

@@ -75,7 +75,6 @@ def new_project(
 def get_project(
     name: str,
     local: bool = False,
-    config: dict | None = None,
     setup_kwargs: dict | None = None,
     **kwargs,
 ) -> Project:
@@ -88,8 +87,6 @@ def get_project(
         The Project name.
     local : bool
         Flag to determine if backend is local.
-    config : dict
-        DHCore environment configuration.
     setup_kwargs : dict
         Setup keyword arguments passed to setup_project() function.
     **kwargs : dict
@@ -108,7 +105,6 @@ def get_project(
         entity_type=ENTITY_TYPE,
         entity_name=name,
         local=local,
-        config=config,
         **kwargs,
     )
     return setup_project(obj, setup_kwargs)
@@ -117,7 +113,6 @@ def get_project(
 def import_project(
     file: str,
     local: bool = False,
-    config: dict | None = None,
     setup_kwargs: dict | None = None,
 ) -> Project:
     """
@@ -129,8 +124,6 @@ def import_project(
         Path to YAML file.
     local : bool
         Flag to determine if backend is local.
-    config : dict
-        DHCore environment configuration.
     setup_kwargs : dict
         Setup keyword arguments passed to setup_project() function.
 
@@ -143,14 +136,13 @@ def import_project(
     --------
     >>> obj = import_project("my-project.yaml")
     """
-    obj = base_processor.import_project_entity(file=file, local=local, config=config)
+    obj = base_processor.import_project_entity(file=file, local=local)
     return setup_project(obj, setup_kwargs)
 
 
 def load_project(
     file: str,
     local: bool = False,
-    config: dict | None = None,
     setup_kwargs: dict | None = None,
 ) -> Project:
     """
@@ -162,8 +154,6 @@ def load_project(
         Path to YAML file.
     local : bool
         Flag to determine if backend is local.
-    config : dict
-        DHCore environment configuration.
     setup_kwargs : dict
         Setup keyword arguments passed to setup_project() function.
 
@@ -176,7 +166,7 @@ def load_project(
     --------
     >>> obj = load_project("my-project.yaml")
     """
-    obj = base_processor.load_project_entity(file=file, local=local, config=config)
+    obj = base_processor.load_project_entity(file=file, local=local)
     return setup_project(obj, setup_kwargs)
 
 
@@ -234,7 +224,6 @@ def get_or_create_project(
         return get_project(
             name,
             local=local,
-            config=config,
             setup_kwargs=setup_kwargs,
             **kwargs,
         )

digitalhub/entities/run/_base/entity.py

@@ -7,13 +7,7 @@ from digitalhub.entities._base.unversioned.entity import UnversionedEntity
 from digitalhub.entities._commons.enums import EntityTypes, State
 from digitalhub.entities._commons.metrics import MetricType, set_metrics, validate_metric_value
 from digitalhub.entities._processors.context import context_processor
-from digitalhub.factory.api import (
-    build_runtime,
-    build_spec,
-    build_status,
-    get_entity_type_from_kind,
-    get_executable_kind,
-)
+from digitalhub.factory.factory import factory
 from digitalhub.utils.exceptions import EntityError
 from digitalhub.utils.logger import LOGGER
 
@@ -61,7 +55,7 @@ class Run(UnversionedEntity):
         executable = self._get_executable()
         task = self._get_task()
         new_spec = self._get_runtime().build(executable, task, self.to_dict())
-        self.spec = build_spec(self.kind, **new_spec)
+        self.spec = factory.build_spec(self.kind, **new_spec)
         self._set_state(State.BUILT.value)
         self.save(update=True)
 
@@ -302,7 +296,7 @@ class Run(UnversionedEntity):
         -------
         None
         """
-        self.status: RunStatus = build_status(self.kind, **status)
+        self.status: RunStatus = factory.build_status(self.kind, **status)
 
     def _set_state(self, state: str) -> None:
         """
@@ -343,7 +337,7 @@ class Run(UnversionedEntity):
         Runtime
             Runtime object.
         """
-        return build_runtime(self.kind, self.project)
+        return factory.build_runtime(self.kind, self.project)
 
     def _get_executable(self) -> dict:
         """
@@ -355,8 +349,8 @@ class Run(UnversionedEntity):
         dict
             Executable (function or workflow) from backend.
         """
-        exec_kind = get_executable_kind(self.kind)
-        exec_type = get_entity_type_from_kind(exec_kind)
+        exec_kind = factory.get_executable_kind(self.kind)
+        exec_type = factory.get_entity_type_from_kind(exec_kind)
         string_to_split = getattr(self.spec, exec_type)
         exec_name, exec_id = string_to_split.split("://")[-1].split("/")[-1].split(":")
         return context_processor.read_context_entity(

digitalhub/entities/task/_base/entity.py

@@ -5,7 +5,7 @@ import typing
 from digitalhub.entities._base.unversioned.entity import UnversionedEntity
 from digitalhub.entities._commons.enums import EntityTypes
 from digitalhub.entities._processors.context import context_processor
-from digitalhub.factory.api import build_entity_from_params, get_entity_type_from_kind, get_executable_kind
+from digitalhub.factory.factory import factory
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
@@ -63,8 +63,8 @@ class Task(UnversionedEntity):
         Run
             Run object.
         """
-        exec_kind = get_executable_kind(self.kind)
-        exec_type = get_entity_type_from_kind(exec_kind)
+        exec_kind = factory.get_executable_kind(self.kind)
+        exec_type = factory.get_entity_type_from_kind(exec_kind)
         kwargs[exec_type] = getattr(self.spec, exec_type)
         return self.new_run(
             save=save,
@@ -108,7 +108,7 @@ class Task(UnversionedEntity):
         """
         if save:
             return context_processor.create_context_entity(**kwargs)
-        return build_entity_from_params(**kwargs)
+        return factory.build_entity_from_params(**kwargs)
 
     def get_run(self, entity_key: str) -> Run:
         """

digitalhub/entities/task/_base/models.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Optional
+from typing import Optional, Union
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -15,6 +15,26 @@ class VolumeType(Enum):
     EMPTY_DIR = "empty_dir"
 
 
+class SpecEmptyDir(BaseModel):
+    """
+    Spec empty dir model.
+    """
+
+    medium: Optional[str] = None
+
+    size_limit: Optional[str] = "128Mi"
+
+
+class SpecPVC(BaseModel):
+    """
+    Spec PVC model.
+    """
+
+    size: str
+
+    storage_class: str
+
+
 class Volume(BaseModel):
     """
     Volume model.
@@ -31,7 +51,7 @@ class Volume(BaseModel):
     mount_path: str
     """Volume mount path inside the container."""
 
-    spec: Optional[dict[str, str]] = None
+    spec: Optional[Union[SpecEmptyDir, SpecPVC]] = None
     """Volume spec."""
 
 

digitalhub/entities/trigger/_base/builder.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from digitalhub.entities._base.versioned.builder import VersionedBuilder
+from digitalhub.entities._commons.enums import EntityTypes
+from digitalhub.entities.trigger._base.entity import Trigger
+
+
+class TriggerBuilder(VersionedBuilder):
+    """
+    TriggerTriggerBuilder builder.
+    """
+
+    ENTITY_TYPE = EntityTypes.TRIGGER.value
+
+    def build(
+        self,
+        kind: str,
+        project: str,
+        name: str,
+        uuid: str | None = None,
+        description: str | None = None,
+        labels: list[str] | None = None,
+        **kwargs,
+    ) -> Trigger:
+        """
+        Create a new object.
+
+        Parameters
+        ----------
+        project : str
+            Project name.
+        name : str
+            Object name.
+        kind : str
+            Kind the object.
+        uuid : str
+            ID of the object.
+        description : str
+            Description of the object (human readable).
+        labels : list[str]
+            List of labels.
+        **kwargs : dict
+            Spec keyword arguments.
+
+        Returns
+        -------
+        Trigger
+            Object instance.
+        """
+        name = self.build_name(name)
+        uuid = self.build_uuid(uuid)
+        metadata = self.build_metadata(
+            project=project,
+            name=name,
+            description=description,
+            labels=labels,
+        )
+        spec = self.build_spec(
+            **kwargs,
+        )
+        status = self.build_status()
+        return self.build_entity(
+            project=project,
+            name=name,
+            uuid=uuid,
+            kind=kind,
+            metadata=metadata,
+            spec=spec,
+            status=status,
+        )

digitalhub/entities/trigger/_base/entity.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+import typing
+
+from digitalhub.entities._base.versioned.entity import VersionedEntity
+from digitalhub.entities._commons.enums import EntityTypes
+
+if typing.TYPE_CHECKING:
+    from digitalhub.entities._base.entity.metadata import Metadata
+    from digitalhub.entities.trigger._base.spec import TriggerSpec
+    from digitalhub.entities.trigger._base.status import TriggerStatus
+
+
+class Trigger(VersionedEntity):
+    """
+    A class representing a trigger.
+    """
+
+    ENTITY_TYPE = EntityTypes.TRIGGER.value
+
+    def __init__(
+        self,
+        project: str,
+        name: str,
+        uuid: str,
+        kind: str,
+        metadata: Metadata,
+        spec: TriggerSpec,
+        status: TriggerStatus,
+        user: str | None = None,
+    ) -> None:
+        super().__init__(project, name, uuid, kind, metadata, spec, status, user)
+        self.spec: TriggerSpec
+        self.status: TriggerStatus

digitalhub/entities/trigger/_base/spec.py

@@ -0,0 +1,30 @@
+from __future__ import annotations
+
+from digitalhub.entities._base.entity.spec import Spec, SpecValidator
+
+
+class TriggerSpec(Spec):
+    """
+    TriggerSpec specifications.
+    """
+
+    def __init__(self, task: str, function: str, template: dict) -> None:
+        super().__init__()
+        self.task = task
+        self.function = function
+        self.template = template
+
+
+class TriggerValidator(SpecValidator):
+    """
+    TriggerValidator validator.
+    """
+
+    task: str
+    """Task string."""
+
+    function: str
+    """Function string."""
+
+    template: dict
+    """Template string."""

digitalhub/entities/trigger/_base/status.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+from digitalhub.entities._base.entity.status import Status
+
+
+class TriggerStatus(Status):
+    """
+    TriggerStatus status.
+    """