update nodes

Author: giancarloromeo

services/web/server/src/simcore_service_webserver/projects/_nodes_repository.py

@@ -1,11 +1,16 @@
+from typing import Any
+
 from aiohttp import web
 from models_library.projects import ProjectID
-from models_library.projects_nodes import Node
+from models_library.projects_nodes import Node, PartialNode
 from models_library.projects_nodes_io import NodeID
 from models_library.services_types import ServiceKey, ServiceVersion
 from pydantic import TypeAdapter
 from simcore_postgres_database.utils_projects_nodes import ProjectNodesRepo
-from simcore_postgres_database.utils_repos import pass_or_acquire_connection
+from simcore_postgres_database.utils_repos import (
+    pass_or_acquire_connection,
+    transaction_context,
+)
 from simcore_service_webserver.db.plugin import get_asyncpg_engine
 
 
@@ -35,3 +40,24 @@ async def get_project_nodes_map(
         for project_node in project_nodes
     }
     return TypeAdapter(dict[NodeID, Node]).validate_python(workbench)
+
+
+async def update_project_nodes_map(
+    app: web.Application,
+    *,
+    project_id: ProjectID,
+    partial_nodes_map: dict[NodeID, PartialNode],
+) -> dict[NodeID, Node]:
+    repo = ProjectNodesRepo(project_uuid=project_id)
+
+    workbench: dict[NodeID, dict[str, Any]] = {}
+    async with transaction_context(get_asyncpg_engine(app)) as conn:
+        for node_id, node in partial_nodes_map.items():
+            project_node = await repo.update(
+                conn,
+                node_id=node_id,
+                node=node,
+            )
+            workbench[node_id] = project_node.model_dump_as_node()
+
+    return TypeAdapter(dict[NodeID, Node]).validate_python(workbench)

services/web/server/src/simcore_service_webserver/projects/_nodes_service.py

@@ -10,7 +10,7 @@
 from models_library.api_schemas_storage.storage_schemas import FileMetaDataGet
 from models_library.basic_types import KeyIDStr
 from models_library.projects import ProjectID
-from models_library.projects_nodes import Node
+from models_library.projects_nodes import Node, PartialNode
 from models_library.projects_nodes_io import NodeID, SimCoreFileLink
 from models_library.services_types import ServiceKey, ServiceVersion
 from models_library.users import UserID
@@ -90,6 +90,17 @@ async def get_project_nodes_map(
     return await _nodes_repository.get_project_nodes_map(app, project_id=project_id)
 
 
+async def update_project_nodes_map(
+    app: web.Application,
+    *,
+    project_id: ProjectID,
+    partial_nodes_map: dict[NodeID, PartialNode],
+) -> dict[NodeID, Node]:
+    return await _nodes_repository.update_project_nodes_map(
+        app, project_id=project_id, partial_nodes_map=partial_nodes_map
+    )
+
+
 #
 # PREVIEWS
 #