Skip to content

Commit

Permalink
50395 Formatted the source code
Browse files Browse the repository at this point in the history
  • Loading branch information
rpopov committed Dec 26, 2024
1 parent 8bfa7ab commit e239779
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions airbyte_cdk/sources/declarative/extractors/record_selector.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@

STREAM_SLICE_RESPONSE_KEY = "response"


@dataclass
class RecordSelector(HttpSelector):
"""
Expand Down Expand Up @@ -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:
"""
Expand Down Expand Up @@ -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]],
Expand Down

0 comments on commit e239779

Please sign in to comment.