address comments

Signed-off-by: Godot Bian <[email protected]>

src/deadline_worker_agent/sessions/actions/run_attachment_download.py

@@ -9,7 +9,7 @@
 import sys
 import json
 from shlex import quote
-from logging import getLogger, LoggerAdapter
+from logging import LoggerAdapter
 import sysconfig
 from typing import Any, TYPE_CHECKING, Optional
 from dataclasses import asdict
@@ -32,6 +32,7 @@
 
 from openjd.sessions import (
     LOG as OPENJD_LOG,
+    LogContent,
     PathMappingRule as OpenjdPathMapping,
     PosixSessionUser,
     WindowsSessionUser,
@@ -52,30 +53,19 @@
 from .openjd_action import OpenjdAction
 
 if TYPE_CHECKING:
-    from concurrent.futures import Future
     from ..session import Session
     from ..job_entities import JobAttachmentDetails, StepDetails
 
 
-logger = getLogger(__name__)
-
-
-class SyncCanceled(Exception):
-    """Exception indicating the synchronization was canceled"""
-
-    pass
-
-
 class AttachmentDownloadAction(OpenjdAction):
-    """Action to synchronize input job attachments for a AWS Deadline Cloud job
+    """Action to synchronize input job attachments for a AWS Deadline Cloud Session
 
     Parameters
     ----------
     id : str
         The unique action identifier
     """
 
-    _future: Future[None]
     _job_attachment_details: Optional[JobAttachmentDetails]
     _step_details: Optional[StepDetails]
     _step_script: Optional[StepScript_2023_09]
@@ -102,6 +92,7 @@ def __init__(
         self._logger = LoggerAdapter(OPENJD_LOG, extra={"session_id": session_id})
 
     def set_step_script(self, manifests, path_mapping, s3_settings) -> None:
+        # TODO - update to run python as embedded file
         profile = os.environ.get("AWS_PROFILE")
         deadline_path = os.path.join(Path(sysconfig.get_path("scripts")), "deadline")
 
@@ -151,17 +142,24 @@ def start(
             An executor for running futures
         """
 
-        self._logger.info(f"Syncing inputs using session {session}")
-
         if self._step_details:
             section_title = "Job Attachments Download for Step"
         else:
             section_title = "Job Attachments Download for Job"
 
         # Banner mimicing the one printed by the openjd-sessions runtime
-        self._logger.info("==============================================")
-        self._logger.info(f"--------- AttachmentDownloadAction  {section_title}")
-        self._logger.info("==============================================")
+        self._logger.info(
+            "==============================================",
+            extra={"openjd_log_content": LogContent.BANNER},
+        )
+        self._logger.info(
+            f"--------- AttachmentDownloadAction  {section_title}",
+            extra={"openjd_log_content": LogContent.BANNER},
+        )
+        self._logger.info(
+            "==============================================",
+            extra={"openjd_log_content": LogContent.BANNER},
+        )
 
         if not (job_attachment_settings := session._job_details.job_attachment_settings):
             raise RuntimeError("Job attachment settings were not contained in JOB_DETAILS entity")
@@ -232,7 +230,7 @@ def start(
             )
         )
 
-        vfs_handled = self._vfs_handling(
+        vfs_handled = self._start_vfs(
             session=session,
             attachments=attachments,
             merged_manifests_by_root=merged_manifests_by_root,
@@ -286,7 +284,7 @@ def start(
                 task_parameter_values=dict[str, ParameterValue](),
             )
 
-    def _vfs_handling(
+    def _start_vfs(
         self,
         session: Session,
         attachments: Attachments,