entrypoint

Author: sanderegg

packages/models-library/src/models_library/api_schemas_storage/storage_schemas.py

@@ -455,3 +455,31 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
     model_config = ConfigDict(
         extra="forbid", json_schema_extra=_update_json_schema_extra
     )
+
+
+class PathTotalSizeCreate(BaseModel):
+    path: Path
+    size: ByteSize
+
+    @staticmethod
+    def _update_json_schema_extra(schema: JsonDict) -> None:
+        schema.update(
+            {
+                "examples": [
+                    # a folder
+                    {
+                        "path": "f8da77a9-24b9-4eab-aee7-1f0608da1e3e",
+                        "size": 15728640,
+                    },
+                    # 1 file
+                    {
+                        "path": f"f8da77a9-24b9-4eab-aee7-1f0608da1e3e/2f94f80f-633e-4dfa-a983-226b7babe3d7/outputs/output5/{FileMetaDataGet.model_json_schema()['examples'][0]['file_name']}",
+                        "size": 1024,
+                    },
+                ]
+            }
+        )
+
+    model_config = ConfigDict(
+        extra="forbid", json_schema_extra=_update_json_schema_extra
+    )

services/storage/src/simcore_service_storage/api/rest/_paths.py

@@ -5,8 +5,13 @@
 from fastapi import APIRouter, Depends
 from fastapi_pagination import create_page
 from fastapi_pagination.cursor import CursorPage, CursorParams
-from models_library.api_schemas_storage.storage_schemas import PathMetaDataGet
+from models_library.api_schemas_storage.storage_schemas import (
+    PathMetaDataGet,
+    PathTotalSizeCreate,
+)
+from models_library.generics import Envelope
 from models_library.users import UserID
+from pydantic import ByteSize
 
 from ...dsm_factory import BaseDataManager
 from .dependencies.dsm_prodiver import get_data_manager
@@ -43,3 +48,17 @@ async def list_paths(
         params=page_params,
         next_=next_cursor,
     )
+
+
+@router.post(
+    "/locations/{location_id}/paths/{path:path}:size",
+    response_model=Envelope[PathTotalSizeCreate],
+)
+async def compute_path_total_size(
+    dsm: Annotated[BaseDataManager, Depends(get_data_manager)],
+    user_id: UserID,
+    path: Path,
+):
+    return Envelope[PathTotalSizeCreate](
+        data=PathTotalSizeCreate(path=path, size=ByteSize(238723))
+    )

services/storage/tests/unit/test_handlers_paths.py

@@ -18,7 +18,10 @@
 import sqlalchemy as sa
 from fastapi import FastAPI, status
 from fastapi_pagination.cursor import CursorPage
-from models_library.api_schemas_storage.storage_schemas import PathMetaDataGet
+from models_library.api_schemas_storage.storage_schemas import (
+    PathMetaDataGet,
+    PathTotalSizeCreate,
+)
 from models_library.projects_nodes_io import LocationID, NodeID, SimcoreS3FileID
 from models_library.users import UserID
 from pydantic import ByteSize, TypeAdapter
@@ -482,10 +485,11 @@ async def test_path_compute_size(
     ).with_query(user_id=user_id)
     response = await client.post(f"{url}")
 
-    received_size, _ = assert_status(
+    received, _ = assert_status(
         response,
         status.HTTP_200_OK,
-        ByteSize,
+        PathTotalSizeCreate,
     )
-    assert received_size
-    assert received_size == 169
+    assert received
+    assert received.path == Path(project["uuid"])
+    assert received.size == 169