digitalhub 0.7.0b2__py3-none-any.whl → 0.8.0__py3-none-any.whl

digitalhub/entities/project/_base/spec.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from digitalhub.entities._base.entity.spec import Spec, SpecValidator
+
+
+class ProjectSpec(Spec):
+    """
+    ProjectSpec specifications.
+    """
+
+    def __init__(
+        self,
+        context: str | None = None,
+        functions: list | None = None,
+        artifacts: list | None = None,
+        workflows: list | None = None,
+        dataitems: list | None = None,
+        models: list | None = None,
+        **kwargs,
+    ) -> None:
+        self.context = context if context is not None else "./"
+        self.functions = functions if functions is not None else []
+        self.artifacts = artifacts if artifacts is not None else []
+        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 []
+
+
+class ProjectValidator(SpecValidator):
+    """
+    ProjectValidator validator.
+    """
+
+    context: str = None
+    """The project's context."""
+
+    functions: list = None
+    """List of project's functions."""
+
+    artifacts: list = None
+    """List of project's artifacts."""
+
+    workflows: list = None
+    """List of project's workflows."""
+
+    dataitems: list = None
+    """List of project's dataitems."""
+
+    models: list = None
+    """List of project's models."""

digitalhub/entities/project/_base/status.py

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

digitalhub/entities/project/crud.py

@@ -0,0 +1,357 @@
+from __future__ import annotations
+
+import importlib.util as imputil
+import typing
+from pathlib import Path
+
+from digitalhub.client.api import build_client, get_client
+from digitalhub.context.api import delete_context
+from digitalhub.entities._base.api_utils import delete_entity_api_base, read_entity_api_base, update_entity_api_base
+from digitalhub.entities.utils.entity_types import EntityTypes
+from digitalhub.factory.api import build_entity_from_dict, build_entity_from_params
+from digitalhub.utils.exceptions import BackendError, EntityAlreadyExistsError, EntityError
+from digitalhub.utils.io_utils import read_yaml
+
+if typing.TYPE_CHECKING:
+    from digitalhub.entities.project._base.entity import Project
+
+
+ENTITY_TYPE = EntityTypes.PROJECT.value
+
+
+def new_project(
+    name: str,
+    description: str | None = None,
+    labels: list[str] | None = None,
+    local: bool = False,
+    config: dict | None = None,
+    context: str | None = None,
+    setup_kwargs: dict | None = None,
+    **kwargs,
+) -> Project:
+    """
+    Create a new object.
+
+    Parameters
+    ----------
+    name : str
+        Object name.
+    description : str
+        Description of the object (human readable).
+    labels : list[str]
+        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.
+    setup_kwargs : dict
+        Setup keyword arguments passed to setup_project() function.
+    **kwargs : dict
+        Keyword arguments.
+
+    Returns
+    -------
+    Project
+        Object instance.
+
+    Examples
+    --------
+    >>> obj = new_project("my-project")
+    """
+    build_client(local, config)
+    if context is None:
+        context = name
+    obj = build_entity_from_params(
+        name=name,
+        kind="project",
+        description=description,
+        labels=labels,
+        local=local,
+        context=context,
+        **kwargs,
+    )
+    obj.save()
+    return _setup_project(obj, setup_kwargs)
+
+
+def get_project(
+    name: str,
+    local: bool = False,
+    config: dict | None = None,
+    setup_kwargs: dict | None = None,
+    **kwargs,
+) -> Project:
+    """
+    Retrieves project details from backend.
+
+    Parameters
+    ----------
+    name : str
+        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
+        Parameters to pass to the API call.
+
+    Returns
+    -------
+    Project
+        Object instance.
+
+    Examples
+    --------
+    >>> obj = get_project("my-project")
+    """
+    build_client(local, config)
+    client = get_client(local)
+    obj = read_entity_api_base(client, ENTITY_TYPE, name, **kwargs)
+    obj["local"] = local
+    project = build_entity_from_dict(obj)
+    return _setup_project(project, setup_kwargs)
+
+
+def import_project(
+    file: str,
+    local: bool = False,
+    config: dict | None = None,
+    setup_kwargs: dict | None = None,
+) -> Project:
+    """
+    Import object from a YAML file.
+
+    Parameters
+    ----------
+    file : str
+        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.
+
+    Returns
+    -------
+    Project
+        Object instance.
+
+    Examples
+    --------
+    >>> obj = import_project("my-project.yaml")
+    """
+    build_client(local, config)
+    dict_obj: dict = read_yaml(file)
+    dict_obj["local"] = local
+    obj = build_entity_from_dict(dict_obj)
+    obj = _setup_project(obj, setup_kwargs)
+
+    try:
+        obj.save()
+    except EntityAlreadyExistsError:
+        pass
+
+    # Import related entities
+    obj._import_entities(dict_obj)
+
+    obj.refresh()
+
+    return obj
+
+
+def load_project(
+    name: str | None = None,
+    filename: str | None = None,
+    local: bool = False,
+    config: dict | None = None,
+    setup_kwargs: dict | None = None,
+    **kwargs,
+) -> Project:
+    """
+    Load project and context from backend or file. Name or
+    filename must be provided. Name takes precedence over filename.
+
+    Parameters
+    ----------
+    name : str
+        Project name.
+    filename : str
+        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.
+    **kwargs : dict
+        Keyword arguments.
+
+    Returns
+    -------
+    Project
+        Object instance.
+
+    Examples
+    --------
+    If name is provided, load project from backend.
+    >>> obj = load_project(name="my-project")
+
+    If filename is provided, load project from file.
+    >>> obj = load_project(filename="my-project.yaml")
+    """
+    if name is not None:
+        return get_project(name=name, local=local, config=config, setup_kwargs=setup_kwargs, **kwargs)
+    if filename is not None:
+        return import_project(filename, local=local, config=config, setup_kwargs=setup_kwargs)
+    raise EntityError("Either name or filename must be provided.")
+
+
+def get_or_create_project(
+    name: str,
+    local: bool = False,
+    config: dict | None = None,
+    context: str | None = None,
+    setup_kwargs: dict | None = None,
+    **kwargs,
+) -> Project:
+    """
+    Try to get project. If not exists, create it.
+
+    Parameters
+    ----------
+    name : str
+        Project name.
+    local : bool
+        Flag to determine if backend is local.
+    config : dict
+        DHCore environment configuration.
+    context : str
+        Folder where the project will saves its context locally.
+    setup_kwargs : dict
+        Setup keyword arguments passed to setup_project() function.
+    **kwargs : dict
+        Keyword arguments.
+
+    Returns
+    -------
+    Project
+        Object instance.
+    """
+    try:
+        return get_project(
+            name,
+            local=local,
+            config=config,
+            setup_kwargs=setup_kwargs,
+            **kwargs,
+        )
+    except BackendError:
+        return new_project(
+            name,
+            local=local,
+            config=config,
+            setup_kwargs=setup_kwargs,
+            context=context,
+            **kwargs,
+        )
+
+
+def update_project(entity: Project, local: bool = False, **kwargs) -> Project:
+    """
+    Update object. Note that object spec are immutable.
+
+    Parameters
+    ----------
+    entity : Project
+        Object to update.
+    local : bool
+        Flag to determine if backend is local.
+    **kwargs : dict
+        Parameters to pass to the API call.
+
+    Returns
+    -------
+    Project
+        The updated object.
+
+    Examples
+    --------
+    >>> obj = update_project(obj)
+    """
+    client = get_client(local)
+    obj = update_entity_api_base(client, ENTITY_TYPE, entity.name, entity.to_dict(), **kwargs)
+    return build_entity_from_dict(obj)
+
+
+def delete_project(
+    name: str,
+    cascade: bool = True,
+    clean_context: bool = True,
+    local: bool = False,
+    **kwargs,
+) -> list[dict]:
+    """
+    Delete a project.
+
+    Parameters
+    ----------
+    name : str
+        Project name.
+    cascade : bool
+        Flag to determine if delete is cascading.
+    clean_context : bool
+        Flag to determine if context will be deleted. If a context is deleted,
+        all its objects are unreacheable.
+    local : bool
+        Flag to determine if backend is local.
+    **kwargs : dict
+        Parameters to pass to the API call.
+
+    Returns
+    -------
+    dict
+        Response from backend.
+
+    Examples
+    --------
+    >>> delete_project("my-project")
+    """
+    client = get_client(local)
+    obj = delete_entity_api_base(client, ENTITY_TYPE, name, cascade=cascade, **kwargs)
+    if clean_context:
+        delete_context(name)
+    return obj
+
+
+def _setup_project(project: Project, setup_kwargs: dict | None = None) -> Project:
+    """
+    Search for setup_project.py file and launch setup hanlder as project hook.
+
+    Parameters
+    ----------
+    project : Project
+        The project to scafold.
+    setup_kwargs : dict
+        Arguments to pass to setup handler.
+
+    Returns
+    -------
+    Project
+        Set up project.
+    """
+    setup_kwargs = setup_kwargs if setup_kwargs is not None else {}
+    check_pth = Path(project.spec.context, ".CHECK")
+    setup_pth = Path(project.spec.context, "setup_project.py")
+    if setup_pth.exists() and not check_pth.exists():
+        spec = imputil.spec_from_file_location("setup_project", setup_pth)
+        mod = imputil.module_from_spec(spec)
+        spec.loader.exec_module(mod)
+        handler = getattr(mod, "setup")
+        project = handler(project, **setup_kwargs)
+        check_pth.touch()
+    return project

digitalhub/entities/run/_base/builder.py

@@ -0,0 +1,94 @@
+from __future__ import annotations
+
+import typing
+
+from digitalhub.entities._base.runtime_entity.builder import EntityError, RuntimeEntityBuilder
+from digitalhub.entities._base.unversioned.builder import UnversionedBuilder
+from digitalhub.entities.utils.entity_types import EntityTypes
+
+if typing.TYPE_CHECKING:
+    from digitalhub.entities.run._base.entity import Run
+
+
+class RunBuilder(UnversionedBuilder, RuntimeEntityBuilder):
+    """
+    Run builder.
+    """
+
+    ENTITY_TYPE = EntityTypes.RUN.value
+
+    def build(
+        self,
+        project: str,
+        kind: str,
+        uuid: str | None = None,
+        labels: list[str] | None = None,
+        task: str | None = None,
+        local_execution: bool = False,
+        **kwargs,
+    ) -> Run:
+        """
+        Create a new object.
+
+        Parameters
+        ----------
+        project : str
+            Project name.
+        kind : str
+            Kind the object.
+        uuid : str
+            ID of the object.
+        labels : list[str]
+            List of labels.
+        task : str
+            Name of the task associated with the run.
+        local_execution : bool
+            Flag to determine if object has local execution.
+        **kwargs : dict
+            Spec keyword arguments.
+
+        Returns
+        -------
+        Run
+            Object instance.
+        """
+        if task is None:
+            raise EntityError("Missing task in run spec")
+        self._check_kind_validity(task)
+        uuid = self.build_uuid(uuid)
+        metadata = self.build_metadata(
+            project=project,
+            name=uuid,
+            labels=labels,
+        )
+        spec = self.build_spec(
+            task=task,
+            local_execution=local_execution,
+            **kwargs,
+        )
+        status = self.build_status()
+        return self.build_entity(
+            project=project,
+            uuid=uuid,
+            kind=kind,
+            metadata=metadata,
+            spec=spec,
+            status=status,
+        )
+
+    def _check_kind_validity(self, task: str) -> None:
+        """
+        Check kind validity.
+
+        Parameters
+        ----------
+        task : str
+            Task string.
+
+        Returns
+        -------
+        None
+        """
+        task_kind = task.split("://")[0]
+        if task_kind not in self.get_all_kinds():
+            raise EntityError(f"Invalid run '{self.ENTITY_KIND}' for task kind '{task_kind}'")