From e239779c7274c0cd9f73918b607f923edc6fa5dc Mon Sep 17 00:00:00 2001 From: Rusi Popov Date: Thu, 26 Dec 2024 05:01:17 +0200 Subject: [PATCH] 50395 Formatted the source code --- .../sources/declarative/extractors/record_selector.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/airbyte_cdk/sources/declarative/extractors/record_selector.py b/airbyte_cdk/sources/declarative/extractors/record_selector.py index 9c50609ce..18c9aad0a 100644 --- a/airbyte_cdk/sources/declarative/extractors/record_selector.py +++ b/airbyte_cdk/sources/declarative/extractors/record_selector.py @@ -24,6 +24,7 @@ STREAM_SLICE_RESPONSE_KEY = "response" + @dataclass class RecordSelector(HttpSelector): """ @@ -54,7 +55,7 @@ def __post_init__(self, parameters: Mapping[str, Any]) -> None: else self._name ) self.response_root_extractor = DpathExtractor(field_path=[], config={}, parameters={}) - + @property # type: ignore def name(self) -> str: """ @@ -89,17 +90,16 @@ def select_records( :return: List of Records selected from the response """ all_data: Iterable[Mapping[str, Any]] = self.extractor.extract_records(response) - + response_root_data = self.response_root_extractor.extract_records(response) stream_state.update({STREAM_SLICE_RESPONSE_KEY: response_root_data}) - try: + try: yield from self.filter_and_transform( all_data, stream_state, records_schema, stream_slice, next_page_token ) finally: stream_state.pop(STREAM_SLICE_RESPONSE_KEY) - def filter_and_transform( self, all_data: Iterable[Mapping[str, Any]],