digitalhub 0.13.0b3__py3-none-any.whl → 0.14.9__py3-none-any.whl

digitalhub/__init__.py

@@ -95,16 +95,11 @@ from digitalhub.entities.workflow.crud import (
     new_workflow,
     update_workflow,
 )
+from digitalhub.stores.client.utils import get_credentials_and_config, refresh_token
+from digitalhub.stores.configurator.api import get_current_profile, set_current_profile
+from digitalhub.utils.store_utils import get_s3_client, get_sql_engine
 
 try:
     from digitalhub.entities.model.mlflow.utils import from_mlflow_run, get_mlflow_model_metrics
 except ImportError:
     ...
-
-# Register entities into registry
-from digitalhub.factory.utils import register_entities, register_runtimes_entities
-from digitalhub.stores.client.dhcore.utils import refresh_token, set_dhcore_env
-from digitalhub.stores.credentials.api import get_current_profile, set_current_profile
-
-register_entities()
-register_runtimes_entities()

digitalhub/context/api.py

@@ -7,6 +7,11 @@ from __future__ import annotations
 import typing
 
 from digitalhub.context.builder import context_builder
+from digitalhub.entities._commons.enums import EntityTypes
+from digitalhub.factory.entity import entity_factory
+from digitalhub.stores.client.builder import get_client
+from digitalhub.stores.client.enums import ApiCategories, BackendOperations
+from digitalhub.utils.exceptions import ContextError, EntityNotExistsError
 
 if typing.TYPE_CHECKING:
     from digitalhub.context.context import Context
@@ -24,7 +29,7 @@ def build_context(project: Project, overwrite: bool = False) -> Context:
     ----------
     project : Project
         The project object used to build the context.
-    overwrite : bool, optional
+    overwrite : bool
         If True, overwrites existing context if it exists. Default is False.
 
     Returns
@@ -49,20 +54,52 @@ def get_context(project: str) -> Context:
     Context
         The context for the given project name.
     """
-    return context_builder.get(project)
+    try:
+        return context_builder.get(project)
+    except ContextError:
+        try:
+            return get_context_from_remote(project)
+        except EntityNotExistsError as e:
+            raise ContextError(f"Context '{project}' not found remotely nor locally.") from e
 
 
-def delete_context(project: str) -> None:
+def get_context_from_remote(project: str) -> Context:
     """
-    Delete the context for a given project name.
+    Fetch project context from remote backend and create local context.
+
 
     Parameters
     ----------
     project : str
-        Project name.
+        The name of the project to fetch from remote.
 
     Returns
     -------
-    None
+    Context
+        The context instance created from the remote project data.
+    """
+    try:
+        client = get_client()
+        api = client.build_api(
+            ApiCategories.BASE.value,
+            BackendOperations.READ.value,
+            entity_type=EntityTypes.PROJECT.value,
+            entity_name=project,
+        )
+        obj = client.read_object(api)
+        entity_factory.build_entity_from_dict(obj)
+        return context_builder.get(project)
+    except EntityNotExistsError:
+        raise ContextError(f"Project '{project}' not found.")
+
+
+def delete_context(project: str) -> None:
+    """
+    Delete the context for a given project name.
+
+    Parameters
+    ----------
+    project : str
+        Project name.
     """
     context_builder.remove(project)

digitalhub/context/builder.py

@@ -38,7 +38,7 @@ class ContextBuilder:
         ----------
         project : Project
             The project instance to create a context for.
-        overwrite : bool, optional
+        overwrite : bool
             If True, overwrites existing context if project name already exists. Default is False.
 
         Returns
@@ -82,10 +82,6 @@ class ContextBuilder:
         ----------
         project : str
             The name of the project whose context should be removed.
-
-        Returns
-        -------
-        None
             This method does not return anything.
 
         Notes

digitalhub/context/context.py

@@ -4,12 +4,18 @@
 
 from __future__ import annotations
 
+import os
 import typing
 from pathlib import Path
 
+from digitalhub.entities._commons.enums import EntityTypes
+from digitalhub.runtimes.enums import RuntimeEnvVar
+from digitalhub.stores.client.enums import ApiCategories, BackendOperations
+from digitalhub.utils.exceptions import BackendError
+
 if typing.TYPE_CHECKING:
     from digitalhub.entities.project._base.entity import Project
-    from digitalhub.stores.client._base.client import Client
+    from digitalhub.stores.client.client import Client
 
 
 class Context:
@@ -27,8 +33,6 @@ class Context:
         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
@@ -41,12 +45,31 @@ class 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: Path = Path(project.spec.source)
         self.root.mkdir(parents=True, exist_ok=True)
 
         self.is_running: bool = False
         self._run_ctx: str | None = None
+        self._search_run_ctx()
+
+    def _search_run_ctx(self) -> None:
+        """
+        Search for an existing run id in env.
+        """
+        run_id = os.getenv(RuntimeEnvVar.RUN_ID.value)
+        if run_id is not None:
+            try:
+                api = self.client.build_api(
+                    category=ApiCategories.CONTEXT.value,
+                    operation=BackendOperations.READ.value,
+                    project=self.name,
+                    entity_type=EntityTypes.RUN.value,
+                    entity_id=run_id,
+                )
+                run_key = self.client.read_object(api=api)["key"]
+                self.set_run(run_key)
+            except BackendError:
+                pass
 
     def set_run(self, run_ctx: str) -> None:
         """
@@ -56,10 +79,6 @@ class Context:
         ----------
         run_ctx : str
             The run key to set.
-
-        Returns
-        -------
-        None
         """
         self.is_running = True
         self._run_ctx = run_ctx
@@ -67,10 +86,6 @@ class Context:
     def unset_run(self) -> None:
         """
         Clear the current run key and reset running state.
-
-        Returns
-        -------
-        None
         """
         self.is_running = False
         self._run_ctx = None

digitalhub/entities/_base/_base/entity.py

@@ -54,26 +54,11 @@ class Base:
         ----------
         **kwargs : dict
             Keyword arguments to be set as attributes.
-
-        Returns
-        -------
-        None
         """
         for k, v in kwargs.items():
             if k not in self.__dict__:
                 setattr(self, k, v)
 
-    def _get_private_attrs(self) -> dict:
-        """
-        Return all private attributes of the object.
-
-        Returns
-        -------
-        dict
-            A dictionary containing the private attributes of the entity instance.
-        """
-        return {k: v for k, v in self.__dict__.items() if k.startswith("_")}
-
     def __repr__(self) -> str:
         """
         Return string representation of the entity object.

digitalhub/entities/_base/context/entity.py

@@ -8,8 +8,7 @@ import typing
 
 from digitalhub.context.api import get_context
 from digitalhub.entities._base.entity.entity import Entity
-from digitalhub.entities._processors.context import context_processor
-from digitalhub.utils.generic_utils import get_timestamp
+from digitalhub.entities._processors.processors import context_processor
 from digitalhub.utils.io_utils import write_yaml
 
 if typing.TYPE_CHECKING:
@@ -81,8 +80,6 @@ class ContextEntity(Entity):
         ContextEntity
             Entity updated.
         """
-        if self._context().local:
-            self.metadata.updated = self.metadata.updated = get_timestamp()
         new_obj = context_processor.update_context_entity(self.project, self.ENTITY_TYPE, self.id, self.to_dict())
         self._update_attributes(new_obj)
         return self

digitalhub/entities/_base/entity/builder.py

@@ -7,11 +7,11 @@ from __future__ import annotations
 import typing
 from abc import abstractmethod
 
-from digitalhub.entities._base.entity._constructors.metadata import build_metadata
-from digitalhub.entities._base.entity._constructors.name import build_name
-from digitalhub.entities._base.entity._constructors.spec import build_spec
-from digitalhub.entities._base.entity._constructors.status import build_status
-from digitalhub.entities._base.entity._constructors.uuid import build_uuid
+from digitalhub.entities._constructors.metadata import build_metadata
+from digitalhub.entities._constructors.name import build_name
+from digitalhub.entities._constructors.spec import build_spec
+from digitalhub.entities._constructors.status import build_status
+from digitalhub.entities._constructors.uuid import build_uuid
 from digitalhub.utils.exceptions import BuilderError
 
 if typing.TYPE_CHECKING:

digitalhub/entities/_base/entity/entity.py

@@ -67,10 +67,6 @@ class Entity(Base):
         ----------
         obj : dict
             Mapping representation of object.
-
-        Returns
-        -------
-        None
         """
         self.metadata = obj.metadata
         self.spec = obj.spec
@@ -95,10 +91,6 @@ class Entity(Base):
             The target entity.
         source : str
             The source entity.
-
-        Returns
-        -------
-        None
         """
         if self.metadata.relationships is None:
             self.metadata.relationships = []