digitalhub 0.10.3__py3-none-any.whl → 0.11.0__py3-none-any.whl

digitalhub/__init__.py

@@ -72,6 +72,16 @@ from digitalhub.entities.task.crud import (
     new_task,
     update_task,
 )
+from digitalhub.entities.trigger.crud import (
+    delete_trigger,
+    get_trigger,
+    get_trigger_versions,
+    import_trigger,
+    list_triggers,
+    load_trigger,
+    new_trigger,
+    update_trigger,
+)
 from digitalhub.entities.workflow.crud import (
     delete_workflow,
     get_workflow,

digitalhub/context/api.py

@@ -9,20 +9,26 @@ if typing.TYPE_CHECKING:
     from digitalhub.entities.project._base.entity import Project
 
 
-def build_context(project: Project, overwrite: bool = False) -> None:
+def build_context(project: Project, overwrite: bool = False) -> Context:
     """
-    Wrapper for ContextBuilder.build().
+    Build a new context for a project.
+
+    Creates or updates a context instance for the given project in the global
+    context registry.
 
     Parameters
     ----------
     project : Project
         The project object used to build the context.
+    overwrite : bool, optional
+        If True, overwrites existing context if it exists, by default False.
 
     Returns
     -------
-    None
+    Context
+        The newly created or existing context instance.
     """
-    context_builder.build(project, overwrite)
+    return context_builder.build(project, overwrite)
 
 
 def get_context(project: str) -> Context:

digitalhub/context/builder.py

@@ -11,51 +11,60 @@ if typing.TYPE_CHECKING:
 
 class ContextBuilder:
     """
-    ContextBuilder class.
-    It implements the builder pattern to create a context instance.
-    It allows to use multiple projects as context at the same time
-    by adding them to the _instances registry with their name.
+    A builder class for managing project contexts.
+
+    This class implements the builder pattern to create and manage Context instances.
+    It maintains a registry of project contexts, allowing multiple projects to be
+    used simultaneously by storing them with their respective names.
+
+    Attributes
+    ----------
+    _instances : dict[str, Context]
+        Internal registry mapping project names to their Context instances.
     """
 
     def __init__(self) -> None:
         self._instances: dict[str, Context] = {}
 
-    def build(self, project_object: Project, overwrite: bool = False) -> None:
+    def build(self, project: Project, overwrite: bool = False) -> Context:
         """
-        Add a project as context.
+        Add a project as context and return the created Context instance.
 
         Parameters
         ----------
-        project_object : Project
-            The project to add.
-        overwrite : bool
-            If True, the project will be overwritten if it already exists.
+        project : Project
+            The project instance to create a context for.
+        overwrite : bool, optional
+            If True, overwrites existing context if project name already exists,
+            by default False.
 
         Returns
         -------
-        None
+        Context
+            The newly created or existing Context instance.
         """
-        if (project_object.name not in self._instances) or overwrite:
-            self._instances[project_object.name] = Context(project_object)
+        if (project.name not in self._instances) or overwrite:
+            self._instances[project.name] = Context(project)
+        return self._instances[project.name]
 
     def get(self, project: str) -> Context:
         """
-        Get a context from project name if it exists.
+        Retrieve a context instance by project name.
 
         Parameters
         ----------
         project : str
-            The project name.
+            The name of the project whose context to retrieve.
 
         Returns
         -------
         Context
-            The project context.
+            The context instance associated with the project.
 
         Raises
         ------
-        ValueError
-            If the project is not in the context.
+        ContextError
+            If no context exists for the specified project name.
         """
         try:
             return self._instances[project]
@@ -64,16 +73,22 @@ class ContextBuilder:
 
     def remove(self, project: str) -> None:
         """
-        Remove a project from the context.
+        Remove a project's context from the registry.
 
         Parameters
         ----------
         project : str
-            The project name.
+            The name of the project whose context should be removed.
 
         Returns
         -------
         None
+            This method doesn't return anything.
+
+        Notes
+        -----
+        If the project doesn't exist in the registry, this method
+        silently does nothing.
         """
         self._instances.pop(project, None)
 

digitalhub/context/context.py

@@ -5,60 +5,71 @@ from pathlib import Path
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities.project._base.entity import Project
+    from digitalhub.stores.client._base.client import Client
 
 
 class Context:
     """
-    Context class built forom a `Project` instance. It contains
-    some information about the project, such as the project name,
-    a client instance (local or non-local), the local context
-    project path and information about client locality.
+    Context class built from a Project instance.
+
+    Contains project-specific information and state, including project name,
+    client instance, local context paths, and run-time information.
+
+    Attributes
+    ----------
+    name : str
+        The name of the project.
+    client : BaseClient
+        The client instance (local or remote) associated with the project.
+    config : dict
+        Project configuration profile.
+    local : bool
+        Whether the client is local or remote.
+    root : Path
+        The local context project path.
+    is_running : bool
+        Flag indicating if the context has an active run.
+    _run_ctx : str | None
+        Current run key, if any.
     """
 
     def __init__(self, project: Project) -> None:
-        self.name = project.name
-        self.client = project._client
-        self.local = project._client.is_local()
-        self.root = Path(project.spec.context)
+        self.name: str = project.name
+        self.client: Client = project._client
+        self.config: dict = project.spec.config
+        self.local: bool = project._client.is_local()
+        self.root: Path = Path(project.spec.context)
         self.root.mkdir(parents=True, exist_ok=True)
 
         self.is_running: bool = False
-        self._run_ctx: str = None
+        self._run_ctx: str | None = None
 
     def set_run(self, run_ctx: str) -> None:
         """
-        Set run identifier.
+        Set the current run key.
 
         Parameters
         ----------
         run_ctx : str
-            Run key.
-
-        Returns
-        -------
-        None
+            The run key to set.
         """
         self.is_running = True
         self._run_ctx = run_ctx
 
     def unset_run(self) -> None:
         """
-        Unset run identifier.
-
-        Returns
-        -------
-        None
+        Clear the current run key and reset running state.
         """
         self.is_running = False
         self._run_ctx = None
 
-    def get_run_ctx(self) -> str:
+    def get_run_ctx(self) -> str | None:
         """
-        Get run identifier.
+        Get the current run key.
 
         Returns
         -------
-        str
-            Run key.
+        str | None
+            The current run key if set, None otherwise.
         """
         return self._run_ctx

digitalhub/entities/_base/entity/builder.py

@@ -178,3 +178,14 @@ class EntityBuilder:
             Entity kind.
         """
         return self.ENTITY_KIND
+
+    def get_spec_validator(self) -> type[SpecValidator]:
+        """
+        Get entity spec validator.
+
+        Returns
+        -------
+        type[SpecValidator]
+            Entity spec validator.
+        """
+        return self.ENTITY_SPEC_VALIDATOR

digitalhub/entities/_base/executable/entity.py

@@ -7,15 +7,16 @@ from digitalhub.entities._commons.enums import EntityTypes
 from digitalhub.entities._processors.context import context_processor
 from digitalhub.entities.run.crud import delete_run, get_run, list_runs
 from digitalhub.entities.task.crud import delete_task
-from digitalhub.factory.api import build_entity_from_dict, build_entity_from_params
+from digitalhub.factory.factory import factory
 from digitalhub.utils.exceptions import EntityAlreadyExistsError, EntityError
 
 if typing.TYPE_CHECKING:
     from digitalhub.entities._base.entity.metadata import Metadata
-    from digitalhub.entities._base.entity.spec import Spec
+    from digitalhub.entities._base.entity.spec import Spec, SpecValidator
     from digitalhub.entities._base.entity.status import Status
     from digitalhub.entities.run._base.entity import Run
     from digitalhub.entities.task._base.entity import Task
+    from digitalhub.entities.trigger._base.entity import Trigger
 
 
 class ExecutableEntity(VersionedEntity):
@@ -102,7 +103,7 @@ class ExecutableEntity(VersionedEntity):
             # Create a new object from dictionary.
             # the form in which tasks are stored in function
             # status
-            task_obj = build_entity_from_dict(task)
+            task_obj: Task = factory.build_entity_from_dict(task)
 
             # Try to save it in backend to been able to use
             # it for launching runs. In fact, tasks must be
@@ -145,7 +146,7 @@ class ExecutableEntity(VersionedEntity):
         kwargs["kind"] = task_kind
 
         # Create object instance
-        task = build_entity_from_params(**kwargs)
+        task: Task = factory.build_entity_from_params(**kwargs)
         task.save()
 
         self._tasks[task_kind] = task
@@ -207,7 +208,7 @@ class ExecutableEntity(VersionedEntity):
         kwargs["uuid"] = self._tasks[kind].id
 
         # Update task
-        task = build_entity_from_params(**kwargs)
+        task: Task = factory.build_entity_from_params(**kwargs)
         task.save(update=True)
         self._tasks[kind] = task
         return task
@@ -403,3 +404,49 @@ class ExecutableEntity(VersionedEntity):
             project=self.project,
             **kwargs,
         )
+
+    ##############################
+    #  Trigger
+    ##############################
+
+    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 = {}
+
+        # Override kwargs
+        kwargs["project"] = self.project
+        kwargs["kind"] = trigger_kind
+        kwargs["name"] = trigger_name
+        kwargs[self.ENTITY_TYPE] = self._get_executable_string()
+        kwargs["task"] = task_string
+        kwargs["template"] = run_validator(**kwargs).to_dict()
+
+        # Create object instance
+        trigger: Trigger = factory.build_entity_from_params(**kwargs)
+        trigger.save()
+        return trigger

digitalhub/entities/_base/material/utils.py

@@ -2,9 +2,9 @@ from __future__ import annotations
 
 from pathlib import Path
 
-from digitalhub.stores.data.s3.utils import get_s3_bucket_from_env
+from digitalhub.stores.data.utils import get_default_store
 from digitalhub.utils.file_utils import eval_zip_type
-from digitalhub.utils.uri_utils import S3Schemes, has_local_scheme
+from digitalhub.utils.uri_utils import has_local_scheme
 
 
 def eval_local_source(source: str | list[str]) -> None:
@@ -34,7 +34,7 @@ def eval_local_source(source: str | list[str]) -> None:
         raise ValueError("Invalid source path. Source must be a local path.")
 
 
-def eval_zip_sources(source: str | list[str]) -> str:
+def eval_zip_sources(source: str | list[str]) -> bool:
     """
     Evaluate zip sources.
 
@@ -45,21 +45,21 @@ def eval_zip_sources(source: str | list[str]) -> str:
 
     Returns
     -------
-    str
-        S3Schemes.
+    bool
+        True if source is zip.
     """
     if isinstance(source, list):
         if len(source) > 1:
-            return S3Schemes.S3.value
+            return False
         path = source[0]
     else:
         if Path(source).is_dir():
-            return S3Schemes.S3.value
+            return False
         path = source
 
     if not eval_zip_type(path):
-        return S3Schemes.S3.value
-    return S3Schemes.ZIP_S3.value
+        return False
+    return True
 
 
 def build_log_path_from_source(
@@ -90,8 +90,8 @@ def build_log_path_from_source(
     str
         Log path.
     """
-    scheme = eval_zip_sources(source)
-    path = f"{scheme}://{get_s3_bucket_from_env()}/{project}/{entity_type}/{name}/{uuid}"
+    prefix = "zip+" if eval_zip_sources(source) else ""
+    path = f"{prefix}{get_default_store(project)}/{project}/{entity_type}/{name}/{uuid}"
 
     if isinstance(source, list) and len(source) >= 1:
         if len(source) > 1:

digitalhub/entities/_commons/enums.py

@@ -17,6 +17,7 @@ class EntityTypes(Enum):
     WORKFLOW = "workflow"
     TASK = "task"
     RUN = "run"
+    TRIGGER = "trigger"
 
 
 class Relationship(Enum):
@@ -55,6 +56,7 @@ class State(Enum):
     STOPPED = "STOPPED"
     SUCCESS = "SUCCESS"
     UNKNOWN = "UNKNOWN"
+    UPLOADING = "UPLOADING"
 
 
 class ApiCategories(Enum):
@@ -103,3 +105,5 @@ class EntityKinds(Enum):
     MODEL_HUGGINGFACE = "huggingface"
     MODEL_SKLEARN = "sklearn"
     SECRET_SECRET = "secret"
+    TRIGGER_SCHEDULER = "scheduler"
+    TRIGGER_LIFECYCLE = "lifecycle"

digitalhub/entities/_processors/base.py

@@ -4,7 +4,7 @@ import typing
 
 from digitalhub.context.api import delete_context
 from digitalhub.entities._commons.enums import ApiCategories, BackendOperations
-from digitalhub.factory.api import build_entity_from_dict, build_entity_from_params
+from digitalhub.factory.factory import factory
 from digitalhub.stores.client.api import get_client
 from digitalhub.utils.exceptions import EntityAlreadyExistsError, EntityError, EntityNotExistsError
 from digitalhub.utils.io_utils import read_yaml
@@ -84,11 +84,11 @@ class BaseEntityOperationsProcessor:
             client = _entity._client
             obj = _entity
         else:
-            client = get_client(kwargs.get("local"), kwargs.pop("config", None))
-            obj = build_entity_from_params(**kwargs)
+            client = get_client(kwargs.get("local"))
+            obj = factory.build_entity_from_params(**kwargs)
         ent = self._create_base_entity(client, obj.ENTITY_TYPE, obj.to_dict())
         ent["local"] = client.is_local()
-        return build_entity_from_dict(ent)
+        return factory.build_entity_from_dict(ent)
 
     def _read_base_entity(
         self,
@@ -147,10 +147,10 @@ class BaseEntityOperationsProcessor:
         Project
             Object instance.
         """
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         obj = self._read_base_entity(client, entity_type, entity_name, **kwargs)
         obj["local"] = client.is_local()
-        return build_entity_from_dict(obj)
+        return factory.build_entity_from_dict(obj)
 
     def import_project_entity(
         self,
@@ -172,11 +172,11 @@ class BaseEntityOperationsProcessor:
         Project
             Object instance.
         """
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         obj: dict = read_yaml(file)
         obj["status"] = {}
         obj["local"] = client.is_local()
-        ent: Project = build_entity_from_dict(obj)
+        ent: Project = factory.build_entity_from_dict(obj)
 
         try:
             self._create_base_entity(ent._client, ent.ENTITY_TYPE, ent.to_dict())
@@ -208,10 +208,10 @@ class BaseEntityOperationsProcessor:
         Project
             Object instance.
         """
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         obj: dict = read_yaml(file)
         obj["local"] = client.is_local()
-        ent: Project = build_entity_from_dict(obj)
+        ent: Project = factory.build_entity_from_dict(obj)
 
         try:
             self._update_base_entity(ent._client, ent.ENTITY_TYPE, ent.name, ent.to_dict())
@@ -278,7 +278,7 @@ class BaseEntityOperationsProcessor:
         entities = []
         for obj in objs:
             obj["local"] = client.is_local()
-            ent = build_entity_from_dict(obj)
+            ent = factory.build_entity_from_dict(obj)
             entities.append(ent)
         return entities
 
@@ -345,10 +345,10 @@ class BaseEntityOperationsProcessor:
         Project
             Object instance.
         """
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         obj = self._update_base_entity(client, entity_type, entity_name, entity_dict, **kwargs)
         obj["local"] = client.is_local()
-        return build_entity_from_dict(obj)
+        return factory.build_entity_from_dict(obj)
 
     def _delete_base_entity(
         self,
@@ -414,7 +414,7 @@ class BaseEntityOperationsProcessor:
         """
         if kwargs.pop("clean_context", True):
             delete_context(entity_name)
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         return self._delete_base_entity(
             client,
             entity_type,
@@ -493,7 +493,7 @@ class BaseEntityOperationsProcessor:
         -------
         None
         """
-        client = get_client(kwargs.pop("local", False), kwargs.pop("config", None))
+        client = get_client(kwargs.pop("local", False))
         api = client.build_api(
             ApiCategories.BASE.value,
             BackendOperations.SHARE.value,