Format

Author: nickles-lee

databricks/sdk/mixins/files.py

@@ -11,8 +11,8 @@
 from enum import Enum
 from io import BytesIO
 from types import TracebackType
-from typing import (TYPE_CHECKING, AnyStr, BinaryIO, Generator, Iterable, Iterator, Type, Union)
-from typing import Dict, List, Optional
+from typing import (TYPE_CHECKING, AnyStr, BinaryIO, Dict, Generator, Iterable,
+                    Iterator, List, Optional, Type, Union)
 from urllib import parse
 
 from .._property import _cached_property
@@ -724,7 +724,8 @@ def _files_multipart_upload_complete(self, file_path: str, session_token: str, e
             body={'etags': etags})
         return
 
-    def multipart_upload_create_part_urls(self, session_token: str, *, page_token: str, page_size: int) -> MultipartUploadCreatePartUrlsResponse:
+    def multipart_upload_create_part_urls(self, session_token: str, *, page_token: str,
+                                          page_size: int) -> MultipartUploadCreatePartUrlsResponse:
         """Request a set of presigned URLs for uploading parts of a file in a multipart upload session."""
 
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json'}
@@ -763,7 +764,11 @@ def execute_presigned_url_request(self,
 
         request_headers = {**presigned_url.headers, **headers}
         resp_headers = {}
-        resp = self._api.do(presigned_url.method, presigned_url.url, headers=request_headers, data=data, response_headers = resp_headers)
+        resp = self._api.do(presigned_url.method,
+                            presigned_url.url,
+                            headers=request_headers,
+                            data=data,
+                            response_headers=resp_headers)
         return (resp, resp_headers)
 
 

tests/integration/test_files.py

@@ -8,7 +8,8 @@
 import pytest
 
 from databricks.sdk.core import DatabricksError
-from databricks.sdk.mixins.files import MultipartUploadCreate, MultipartUploadCreatePartUrlsResponse
+from databricks.sdk.mixins.files import (MultipartUploadCreate,
+                                         MultipartUploadCreatePartUrlsResponse)
 from databricks.sdk.service.catalog import VolumeType
 
 
@@ -229,6 +230,7 @@ def test_files_api_upload_download(ucws, random):
             with w.files.download(target_file).contents as f:
                 assert f.read() == b"some text data"
 
+
 def test_files_multipart_upload_download_baremetal(ucws, random):
     w = ucws
     schema = 'filesit-' + random()
@@ -241,24 +243,27 @@ def test_files_multipart_upload_download_baremetal(ucws, random):
             ## Create a session, upload the file in a single part, and then finalize the session
             session: MultipartUploadCreate = w.files.multipart_upload_create(target_file)
 
-            url_page_one: MultipartUploadCreatePartUrlsResponse = w.files.multipart_upload_create_part_urls(session.session_token, page_size=1)
+            url_page_one: MultipartUploadCreatePartUrlsResponse = w.files.multipart_upload_create_part_urls(
+                session.session_token, page_size=1)
 
             # Verify that a second page of URLs may be requested
-            w.files.multipart_upload_create_part_urls(session.session_token, page_token=url_page_one.next_page_token, page_size=1)
-
+            w.files.multipart_upload_create_part_urls(session.session_token,
+                                                      page_token=url_page_one.next_page_token,
+                                                      page_size=1)
 
-            (part_upload_resp, part_upload_resp_headers) = w.files.execute_presigned_url_request(url_page_one.upload_part_urls[0], data=f)
+            (part_upload_resp, part_upload_resp_headers) = w.files.execute_presigned_url_request(
+                url_page_one.upload_part_urls[0], data=f)
             print(part_upload_resp)
             print(part_upload_resp_headers)
 
-            w.files.multipart_upload_complete(target_file, session.session_token, [part_upload_resp_headers["etag"]])
+            w.files.multipart_upload_complete(target_file, session.session_token,
+                                              [part_upload_resp_headers["etag"]])
 
             ## Download the file & assert that it's what we expect
             with w.files.download(target_file).contents as f:
                 assert f.read() == b"some text data"
 
 
-
 def test_files_api_read_twice_from_one_download(ucws, random):
     w = ucws
     schema = 'filesit-' + random()