Skip to content

Commit 4055f44

Browse files
committed
refactor completion fcn into two
1 parent e46a4af commit 4055f44

File tree

3 files changed

+18
-8
lines changed

3 files changed

+18
-8
lines changed

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

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,22 +41,34 @@
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+
4458
async def complete_file_upload(
4559
uploaded_parts: list[UploadedPart],
4660
upload_completion_link: AnyUrl,
47-
is_directory: bool,
4861
client_session: ClientSession | None = None,
4962
) -> ETag | None:
5063
async with ClientSessionContextManager(client_session) as session:
5164
e_tag: ETag | None = await _filemanager_utils.complete_upload(
5265
session=session,
5366
upload_completion_link=upload_completion_link,
5467
parts=uploaded_parts,
55-
is_directory=is_directory,
68+
is_directory=False,
5669
)
5770
# should not be None because a file is being uploaded
58-
if not is_directory:
59-
assert e_tag is not None # nosec
71+
assert e_tag is not None # nosec
6072
return e_tag
6173

6274

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

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -426,7 +426,6 @@ async def complete_multipart_upload(
426426
e_tag: ETag | None = await complete_file_upload(
427427
uploaded_parts=uploaded_parts.parts,
428428
upload_completion_link=TypeAdapter(AnyUrl).validate_python(f"{complete_link}"),
429-
is_directory=False,
430429
)
431430
assert e_tag is not None # nosec
432431

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

Lines changed: 2 additions & 3 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_file_upload,
17+
complete_directory_upload,
1818
get_upload_links_from_s3,
1919
)
2020
from simcore_service_api_server._service import create_solver_or_program_job
@@ -148,9 +148,8 @@ async def create_program_job(
148148
is_directory=True,
149149
sha256_checksum=None,
150150
)
151-
await complete_file_upload(
151+
await complete_directory_upload(
152152
uploaded_parts=[],
153153
upload_completion_link=file_upload_schema.links.complete_upload,
154-
is_directory=True,
155154
)
156155
return job

0 commit comments

Comments
 (0)