Skip to content

Commit fe77b13

Browse files
committed
file-mode-api: fix lint, andd ruff
1 parent cb5884e commit fe77b13

File tree

2 files changed

+11
-7
lines changed

2 files changed

+11
-7
lines changed

airbyte_cdk/sources/file_based/file_based_stream_reader.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from enum import Enum
99
from io import IOBase
1010
from os import makedirs, path
11-
from typing import Any, Callable, Iterable, List, Optional, Set, Tuple, MutableMapping
11+
from typing import Any, Callable, Iterable, List, MutableMapping, Optional, Set, Tuple
1212

1313
from wcmatch.glob import GLOBSTAR, globmatch
1414

@@ -176,7 +176,12 @@ def upload(
176176
"""
177177
...
178178

179-
def _get_file_transfer_paths(self, file: RemoteFile, local_directory: str, parse_file_path_from_uri: Optional[Callable] = None) -> MutableMapping[str, Any]:
179+
def _get_file_transfer_paths(
180+
self,
181+
file: RemoteFile,
182+
local_directory: str,
183+
parse_file_path_from_uri: Optional[Callable[[str], str]] = None,
184+
) -> MutableMapping[str, Any]:
180185
preserve_directory_structure = self.preserve_directory_structure()
181186
if not parse_file_path_from_uri:
182187
file_path = file.uri
@@ -200,6 +205,5 @@ def _get_file_transfer_paths(self, file: RemoteFile, local_directory: str, parse
200205
self.FILE_NAME: file_name,
201206
self.FILE_FOLDER: file_folder,
202207
self.SOURCE_FILE_URI: file.uri,
203-
204208
}
205209
return file_paths

unit_tests/sources/file_based/test_file_based_stream_reader.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -441,11 +441,11 @@ def test_preserve_sub_directories_scenarios(
441441
)
442442
reader = TestStreamReader()
443443
reader.config = TestSpec(**config)
444-
file_paths = reader._get_file_transfer_paths(
445-
remote_file, "/tmp/transfer-files/"
446-
)
444+
file_paths = reader._get_file_transfer_paths(remote_file, "/tmp/transfer-files/")
447445

448-
assert file_paths[AbstractFileBasedStreamReader.FILE_RELATIVE_PATH] == expected_file_relative_path
446+
assert (
447+
file_paths[AbstractFileBasedStreamReader.FILE_RELATIVE_PATH] == expected_file_relative_path
448+
)
449449
assert file_paths[AbstractFileBasedStreamReader.LOCAL_FILE_PATH] == expected_local_file_path
450450
assert file_paths[AbstractFileBasedStreamReader.SOURCE_FILE_URI] == source_file
451451
assert file_paths[AbstractFileBasedStreamReader.FILE_NAME] == path.basename(source_file)

0 commit comments

Comments
 (0)