From 0768d3cab25f56ceafe1dfaa7c064cae49e2d7f9 Mon Sep 17 00:00:00 2001 From: sajarin Date: Wed, 25 Oct 2023 14:26:06 -0400 Subject: [PATCH] fix: format files --- .../connectors/source-asana/source_asana/source.py | 4 ++-- .../connectors/source-asana/source_asana/streams.py | 6 +++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/airbyte-integrations/connectors/source-asana/source_asana/source.py b/airbyte-integrations/connectors/source-asana/source_asana/source.py index e9faf578de538e..7f797bd06ff8b4 100644 --- a/airbyte-integrations/connectors/source-asana/source_asana/source.py +++ b/airbyte-integrations/connectors/source-asana/source_asana/source.py @@ -16,9 +16,9 @@ Attachments, AttachmentsCompact, CustomFields, - Portfolio, - PortfolioMemberships, OrganizationExports, + Portfolio, + PortfolioMemberships, Projects, Sections, SectionsCompact, diff --git a/airbyte-integrations/connectors/source-asana/source_asana/streams.py b/airbyte-integrations/connectors/source-asana/source_asana/streams.py index 967425d0d97a10..0b84e5cfc032dd 100644 --- a/airbyte-integrations/connectors/source-asana/source_asana/streams.py +++ b/airbyte-integrations/connectors/source-asana/source_asana/streams.py @@ -261,10 +261,12 @@ class Tags(WorkspaceRequestParamsRelatedStream): def path(self, **kwargs) -> str: return "tags" + class Portfolio(WorkspaceRequestParamsRelatedStream): def path(self, **kwargs) -> str: return "portfolios" + class PortfolioRelatedStream(AsanaStream, ABC): """ Stream Portfolio memberships depends on `portfolio`: argument in request. @@ -273,15 +275,17 @@ class PortfolioRelatedStream(AsanaStream, ABC): def stream_slices(self, **kwargs) -> Iterable[Optional[Mapping[str, Any]]]: yield from self.read_slices_from_records(stream_class=Portfolio, slice_field="gid") + class PortfolioMemberships(PortfolioRelatedStream): def path(self, **kwargs) -> str: return "portfolio_memberships" - + def request_params(self, stream_slice: Mapping[str, Any] = None, **kwargs) -> MutableMapping[str, Any]: params = super().request_params(stream_slice=stream_slice, **kwargs) params["portfolio"] = stream_slice["gid"] return params + class Tasks(ProjectRelatedStream): use_cache = True