Skip to content

Commit 6702bfe

Browse files
author
Andrei Neagu
committed
simplfy interface
1 parent c014577 commit 6702bfe

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

services/storage/src/simcore_service_storage/simcore_s3_dsm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1311,7 +1311,8 @@ async def create_s3_export(
13111311
)
13121312

13131313
await create_and_upload_export(
1314-
self.app,
1314+
get_s3_client(self.app),
1315+
ProjectRepository.instance(get_db_engine(self.app)),
13151316
self.simcore_bucket_name,
13161317
source_object_keys=source_object_keys,
13171318
destination_object_keys=destination_object_key,

services/storage/src/simcore_service_storage/utils/simcore_s3_dsm_utils.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
from aws_library.s3 import S3MetaData, SimcoreS3API
88
from aws_library.s3._constants import STREAM_READER_CHUNK_SIZE
99
from aws_library.s3._models import S3ObjectKey
10-
from fastapi import FastAPI
1110
from models_library.api_schemas_storage.storage_schemas import S3BucketName
1211
from models_library.projects import ProjectID, ProjectIDStr
1312
from models_library.projects_nodes_io import (
@@ -27,11 +26,9 @@
2726
from ..constants import EXPORTS_S3_PREFIX
2827
from ..exceptions.errors import FileMetaDataNotFoundError, ProjectAccessRightError
2928
from ..models import FileMetaData, FileMetaDataAtDB, GenericCursor, PathMetaData
30-
from ..modules.db import get_db_engine
3129
from ..modules.db.access_layer import AccessLayerRepository
3230
from ..modules.db.file_meta_data import FileMetaDataRepository, TotalChildren
3331
from ..modules.db.projects import ProjectRepository
34-
from ..modules.s3 import get_s3_client
3532
from .utils import convert_db_to_model
3633

3734

@@ -201,17 +198,15 @@ def _replace_node_id_project_id_in_path(
201198

202199

203200
async def create_and_upload_export(
204-
app: FastAPI,
201+
s3_client: SimcoreS3API,
202+
project_repository: ProjectRepository,
205203
bucket: S3BucketName,
206204
*,
207205
source_object_keys: set[tuple[UserSelectionStr, StorageFileID]],
208206
destination_object_keys: StorageFileID,
209207
progress_bar: ProgressBarData,
210208
) -> None:
211-
s3_client = get_s3_client(app)
212-
ids_names_map = await ProjectRepository.instance(
213-
get_db_engine(app)
214-
).get_project_id_and_node_id_to_names_map(
209+
ids_names_map = await project_repository.get_project_id_and_node_id_to_names_map(
215210
project_uuids=_get_project_ids(user_selecton={x[0] for x in source_object_keys})
216211
)
217212

0 commit comments

Comments
 (0)