diff --git a/airbyte-integrations/connectors/source-sendgrid/source_sendgrid/source.py b/airbyte-integrations/connectors/source-sendgrid/source_sendgrid/source.py index 4f864a9f92950..6834137e74fd8 100644 --- a/airbyte-integrations/connectors/source-sendgrid/source_sendgrid/source.py +++ b/airbyte-integrations/connectors/source-sendgrid/source_sendgrid/source.py @@ -25,7 +25,15 @@ from datetime import datetime from typing import Generator -from airbyte_protocol import AirbyteCatalog, AirbyteConnectionStatus, AirbyteMessage, AirbyteRecordMessage, Status, Type +from airbyte_protocol import ( + AirbyteCatalog, + AirbyteConnectionStatus, + AirbyteMessage, + AirbyteRecordMessage, + ConfiguredAirbyteCatalog, + Status, + Type, +) from base_python import AirbyteLogger, ConfigContainer, Source from python_http_client import ForbiddenError @@ -58,10 +66,13 @@ def read( ) -> Generator[AirbyteMessage, None, None]: client = self._client(config_container) - catalog = AirbyteCatalog.parse_obj(self.read_config(catalog_path)) + config = self.read_config(catalog_path) + catalog = ConfiguredAirbyteCatalog.parse_obj(config) logger.info("Starting syncing sendgrid") - for stream in catalog.streams: + for configured_stream in catalog.streams: + # TODO handle incremental syncs + stream = configured_stream.stream if stream.name not in client.ENTITY_MAP.keys(): logger.warn(f"Stream '{stream}' not found in the recognized entities") continue