fix!

Author: JunchengXue

flow360/cloud/s3_utils.py

@@ -13,6 +13,7 @@
 import boto3
 from boto3.s3.transfer import TransferConfig
 from botocore.config import Config
+
 # pylint: disable=unused-import
 from botocore.exceptions import ClientError as CloudFileNotFoundError
 from pydantic.v1 import BaseModel, Field
@@ -142,22 +143,24 @@ def get_client(self):
         :return:
         """
         customize_boto3_config = None
-        if (self.user_credential is not None
-                and self.user_credential.storage_provider is not None
-                and self.user_credential.storage_provider == "OSS"):
+        if (
+            self.user_credential is not None
+            and self.user_credential.storage_provider is not None
+            and self.user_credential.storage_provider == "OSS"
+        ):
             # OSS does not support aws integrity check
-            customize_boto3_config = Config(request_checksum_calculation="when_required",
-                                            response_checksum_validation="when_required",
-                                            s3={
-                                                "addressing_style": "virtual"
-                                            })
+            customize_boto3_config = Config(
+                request_checksum_calculation="when_required",
+                response_checksum_validation="when_required",
+                s3={"addressing_style": "virtual"},
+            )
         # pylint: disable=no-member
         kwargs = {
             "region_name": Env.current.aws_region,
             "aws_access_key_id": self.user_credential.access_key_id,
             "aws_secret_access_key": self.user_credential.secret_access_key,
             "aws_session_token": self.user_credential.session_token,
-            "config": customize_boto3_config
+            "config": customize_boto3_config,
         }
 
         if self.endpoint is not None:
@@ -175,8 +178,8 @@ def is_expired(self):
         """
         # pylint: disable=no-member
         return (
-                self.user_credential.expiration
-                - datetime.now(tz=self.user_credential.expiration.tzinfo)
+            self.user_credential.expiration
+            - datetime.now(tz=self.user_credential.expiration.tzinfo)
         ).total_seconds() < 300
 
 
@@ -220,9 +223,9 @@ def get_cloud_path_prefix(self, resource_id, file_name):
         return base_path
 
     def create_multipart_upload(
-            self,
-            resource_id: str,
-            remote_file_name: str,
+        self,
+        resource_id: str,
+        remote_file_name: str,
     ):
         """
         Creates a multipart upload for the specified resource ID and remote file name.
@@ -243,12 +246,12 @@ def create_multipart_upload(
 
     # pylint: disable=too-many-arguments
     def upload_part(
-            self,
-            resource_id: str,
-            remote_file_name: str,
-            upload_id: str,
-            part_number: int,
-            compressed_chunk,
+        self,
+        resource_id: str,
+        remote_file_name: str,
+        upload_id: str,
+        part_number: int,
+        compressed_chunk,
     ):
         """
         Uploads a part of the file as part of a multipart upload.
@@ -277,7 +280,7 @@ def upload_part(
         return {"ETag": response["ETag"], "PartNumber": part_number}
 
     def complete_multipart_upload(
-            self, resource_id: str, remote_file_name: str, upload_id: str, uploaded_parts: dict
+        self, resource_id: str, remote_file_name: str, upload_id: str, uploaded_parts: dict
     ):
         """
         Completes a multipart upload for the specified resource ID, remote file name, upload ID, e_tag, and part number.
@@ -302,7 +305,7 @@ def complete_multipart_upload(
         )
 
     def upload_file(
-            self, resource_id: str, remote_file_name: str, file_name: str, progress_callback=None
+        self, resource_id: str, remote_file_name: str, file_name: str, progress_callback=None
     ):
         """
         Upload a file to s3.
@@ -345,15 +348,15 @@ def _call_back(bytes_in_chunk):
 
     # pylint: disable=too-many-arguments
     def download_file(
-            self,
-            resource_id: str,
-            remote_file_name: str,
-            to_file: str = None,
-            to_folder: str = ".",
-            overwrite: bool = True,
-            progress_callback=None,
-            log_error=True,
-            verbose=True,
+        self,
+        resource_id: str,
+        remote_file_name: str,
+        to_file: str = None,
+        to_folder: str = ".",
+        overwrite: bool = True,
+        progress_callback=None,
+        log_error=True,
+        verbose=True,
     ):
         """
         Download a file from s3.