|
7 | 7 | from aws_library.s3 import S3MetaData, SimcoreS3API |
8 | 8 | from aws_library.s3._constants import STREAM_READER_CHUNK_SIZE |
9 | 9 | from aws_library.s3._models import S3ObjectKey |
10 | | -from fastapi import FastAPI |
11 | 10 | from models_library.api_schemas_storage.storage_schemas import S3BucketName |
12 | 11 | from models_library.projects import ProjectID, ProjectIDStr |
13 | 12 | from models_library.projects_nodes_io import ( |
|
27 | 26 | from ..constants import EXPORTS_S3_PREFIX |
28 | 27 | from ..exceptions.errors import FileMetaDataNotFoundError, ProjectAccessRightError |
29 | 28 | from ..models import FileMetaData, FileMetaDataAtDB, GenericCursor, PathMetaData |
30 | | -from ..modules.db import get_db_engine |
31 | 29 | from ..modules.db.access_layer import AccessLayerRepository |
32 | 30 | from ..modules.db.file_meta_data import FileMetaDataRepository, TotalChildren |
33 | 31 | from ..modules.db.projects import ProjectRepository |
34 | | -from ..modules.s3 import get_s3_client |
35 | 32 | from .utils import convert_db_to_model |
36 | 33 |
|
37 | 34 |
|
@@ -201,17 +198,15 @@ def _replace_node_id_project_id_in_path( |
201 | 198 |
|
202 | 199 |
|
203 | 200 | async def create_and_upload_export( |
204 | | - app: FastAPI, |
| 201 | + s3_client: SimcoreS3API, |
| 202 | + project_repository: ProjectRepository, |
205 | 203 | bucket: S3BucketName, |
206 | 204 | *, |
207 | 205 | source_object_keys: set[tuple[UserSelectionStr, StorageFileID]], |
208 | 206 | destination_object_keys: StorageFileID, |
209 | 207 | progress_bar: ProgressBarData, |
210 | 208 | ) -> 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( |
215 | 210 | project_uuids=_get_project_ids(user_selecton={x[0] for x in source_object_keys}) |
216 | 211 | ) |
217 | 212 |
|
|
0 commit comments