execute

Author: pcrespov

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

@@ -1,7 +1,6 @@
 import logging
 
 import sqlalchemy as sa
-
 from aiohttp import web
 from models_library.projects import ProjectID
 from models_library.projects_nodes import Node, PartialNode
@@ -10,8 +9,8 @@
 from simcore_postgres_database.webserver_models import projects_nodes
 from sqlalchemy.ext.asyncio import AsyncConnection
 
-from .exceptions import NodeNotFoundError
 from ..db.plugin import get_asyncpg_engine
+from .exceptions import NodeNotFoundError
 
 _logger = logging.getLogger(__name__)
 
@@ -44,21 +43,18 @@ async def get(
     node_id: NodeID,
 ) -> Node:
     async with transaction_context(get_asyncpg_engine(app), connection) as conn:
-        get_stmt = sa.select(
-            *_SELECTION_PROJECTS_NODES_DB_ARGS
-        ).where(
+        get_stmt = sa.select(*_SELECTION_PROJECTS_NODES_DB_ARGS).where(
             (projects_nodes.c.project_uuid == f"{project_id}")
             & (projects_nodes.c.node_id == f"{node_id}")
         )
 
-        result = await conn.stream(get_stmt)
+        result = await conn.execute(get_stmt)
         assert result  # nosec
 
-        row = await result.first()
+        row = result.one_or_none()
         if row is None:
             raise NodeNotFoundError(
-                project_uuid=f"{project_id}",
-                node_uuid=f"{node_id}"
+                project_uuid=f"{project_id}", node_uuid=f"{node_id}"
             )
         assert row  # nosec
         return Node.model_validate(row, from_attributes=True)
@@ -75,7 +71,7 @@ async def update(
     values = partial_node.model_dump(mode="json", exclude_unset=True)
 
     async with transaction_context(get_asyncpg_engine(app), connection) as conn:
-        await conn.stream(
+        await conn.execute(
             projects_nodes.update()
             .values(**values)
             .where(