Skip to content

Commit

Permalink
kwargs
Browse files Browse the repository at this point in the history
  • Loading branch information
girarda committed Aug 5, 2022
1 parent 151a1d8 commit 5c1e7a7
Showing 1 changed file with 32 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,14 @@ def request_params(
stream_slice: Optional[StreamSlice] = None,
next_page_token: Optional[Mapping[str, Any]] = None,
) -> Mapping[str, Any]:
return dict(ChainMap(*[s.request_params() for s in self._stream_slicers]))
return dict(
ChainMap(
*[
s.request_params(stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token)
for s in self._stream_slicers
]
)
)

def request_headers(
self,
Expand All @@ -54,7 +61,14 @@ def request_headers(
stream_slice: Optional[StreamSlice] = None,
next_page_token: Optional[Mapping[str, Any]] = None,
) -> Mapping[str, Any]:
return dict(ChainMap(*[s.request_headers(stream_state, stream_slice, next_page_token) for s in self._stream_slicers]))
return dict(
ChainMap(
*[
s.request_headers(stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token)
for s in self._stream_slicers
]
)
)

def request_body_data(
self,
Expand All @@ -63,7 +77,14 @@ def request_body_data(
stream_slice: Optional[StreamSlice] = None,
next_page_token: Optional[Mapping[str, Any]] = None,
) -> Mapping[str, Any]:
return dict(ChainMap(*[s.request_body_data(stream_state, stream_slice, next_page_token) for s in self._stream_slicers]))
return dict(
ChainMap(
*[
s.request_body_data(stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token)
for s in self._stream_slicers
]
)
)

def request_body_json(
self,
Expand All @@ -72,7 +93,14 @@ def request_body_json(
stream_slice: Optional[StreamSlice] = None,
next_page_token: Optional[Mapping[str, Any]] = None,
) -> Optional[Mapping]:
return dict(ChainMap(*[s.request_body_json(stream_state, stream_slice, next_page_token) for s in self._stream_slicers]))
return dict(
ChainMap(
*[
s.request_body_json(stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token)
for s in self._stream_slicers
]
)
)

def get_stream_state(self) -> Mapping[str, Any]:
return dict(ChainMap(*[slicer.get_stream_state() for slicer in self._stream_slicers]))
Expand Down

0 comments on commit 5c1e7a7

Please sign in to comment.