Skip to content

Commit 562324f

Browse files
committed
@sanderegg consolidate complete upload methdos
1 parent f866265 commit 562324f

File tree

2 files changed

+9
-20
lines changed
  • packages/simcore-sdk/src/simcore_sdk/node_ports_common
  • services/api-server/src/simcore_service_api_server/api/routes

2 files changed

+9
-20
lines changed

packages/simcore-sdk/src/simcore_sdk/node_ports_common/filemanager.py

Lines changed: 6 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -41,35 +41,23 @@
4141
_logger = logging.getLogger(__name__)
4242

4343

44-
async def complete_directory_upload(
45-
uploaded_parts: list[UploadedPart],
46-
upload_completion_link: AnyUrl,
47-
client_session: ClientSession | None = None,
48-
) -> None:
49-
async with ClientSessionContextManager(client_session) as session:
50-
await _filemanager_utils.complete_upload(
51-
session=session,
52-
upload_completion_link=upload_completion_link,
53-
parts=uploaded_parts,
54-
is_directory=True,
55-
)
56-
57-
5844
async def complete_file_upload(
5945
uploaded_parts: list[UploadedPart],
6046
upload_completion_link: AnyUrl,
6147
client_session: ClientSession | None = None,
62-
) -> ETag:
48+
is_directory: bool = False,
49+
) -> ETag | None:
6350
async with ClientSessionContextManager(client_session) as session:
6451
e_tag: ETag | None = await _filemanager_utils.complete_upload(
6552
session=session,
6653
upload_completion_link=upload_completion_link,
6754
parts=uploaded_parts,
68-
is_directory=False,
55+
is_directory=is_directory,
6956
)
7057
# should not be None because a file is being uploaded
71-
assert e_tag is not None # nosec
72-
return e_tag
58+
if not is_directory:
59+
assert e_tag is not None # nosec
60+
return e_tag
7361

7462

7563
async def get_download_link_from_s3(

services/api-server/src/simcore_service_api_server/api/routes/programs.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from servicelib.fastapi.dependencies import get_reverse_url_mapper
1515
from simcore_sdk.node_ports_common.constants import SIMCORE_LOCATION
1616
from simcore_sdk.node_ports_common.filemanager import (
17-
complete_directory_upload,
17+
complete_file_upload,
1818
get_upload_links_from_s3,
1919
)
2020
from simcore_service_api_server._service import create_solver_or_program_job
@@ -154,8 +154,9 @@ async def create_program_job(
154154
is_directory=True,
155155
sha256_checksum=None,
156156
)
157-
await complete_directory_upload(
157+
await complete_file_upload(
158158
uploaded_parts=[],
159159
upload_completion_link=file_upload_schema.links.complete_upload,
160+
is_directory=True,
160161
)
161162
return job

0 commit comments

Comments
 (0)