moved to RPC interface

Author: sanderegg

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/storage/__init__.py

@@ -0,0 +1,32 @@
+from pathlib import Path
+
+from models_library.api_schemas_rpc_async_jobs.async_jobs import (
+    AsyncJobNameData,
+)
+from models_library.api_schemas_storage import STORAGE_RPC_NAMESPACE
+from models_library.api_schemas_webserver.storage import StorageAsyncJobGet
+from models_library.projects_nodes_io import LocationID
+from models_library.rabbitmq_basic_types import RPCMethodName
+from models_library.users import UserID
+
+from ..._client_rpc import RabbitMQRPCClient
+from ..async_jobs.async_jobs import submit_job
+
+
+async def compute_path_size(
+    client: RabbitMQRPCClient,
+    *,
+    user_id: UserID,
+    product_name: str,
+    location_id: LocationID,
+    path: Path,
+) -> StorageAsyncJobGet:
+    async_job_rpc_get = await submit_job(
+        rabbitmq_rpc_client=client,
+        rpc_namespace=STORAGE_RPC_NAMESPACE,
+        method_name=RPCMethodName("compute_path_size"),
+        job_id_data=AsyncJobNameData(user_id=user_id, product_name=product_name),
+        location_id=location_id,
+        path=path,
+    )
+    return StorageAsyncJobGet.from_rpc_schema(async_job_rpc_get)

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/storage/paths.py

@@ -44,6 +44,9 @@
     list_jobs,
     submit_job,
 )
+from servicelib.rabbitmq.rpc_interfaces.storage.paths import (
+    compute_path_size as remote_compute_path_size,
+)
 from servicelib.request_keys import RQT_USERID_KEY
 from servicelib.rest_responses import unwrap_envelope
 from yarl import URL
@@ -181,23 +184,22 @@ async def list_paths(request: web.Request) -> web.Response:
 @login_required
 @permission_required("storage.files.*")
 async def compute_path_size(request: web.Request) -> web.Response:
-    _req_ctx = RequestContext.model_validate(request)
-    _path_params = parse_request_path_parameters_as(
+    req_ctx = RequestContext.model_validate(request)
+    path_params = parse_request_path_parameters_as(
         StoragePathComputeSizeParams, request
     )
+
     rabbitmq_rpc_client = get_rabbitmq_rpc_client(request.app)
-    async_job_rpc_get = await submit_job(
-        rabbitmq_rpc_client=rabbitmq_rpc_client,
-        rpc_namespace=STORAGE_RPC_NAMESPACE,
-        method_name="compute_path_size",
-        job_id_data=AsyncJobNameData(
-            user_id=_req_ctx.user_id, product_name=_req_ctx.product_name
-        ),
-        location_id=_path_params.location_id,
-        path=_path_params.path,
+    async_job = await remote_compute_path_size(
+        rabbitmq_rpc_client,
+        user_id=req_ctx.user_id,
+        product_name=req_ctx.product_name,
+        location_id=path_params.location_id,
+        path=path_params.path,
     )
+
     return create_data_response(
-        StorageAsyncJobGet.from_rpc_schema(async_job_rpc_get),
+        async_job,
         status=status.HTTP_202_ACCEPTED,
     )