112 lines
4.4 KiB
Python
112 lines
4.4 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
from typing import Any, List, Mapping
|
|
|
|
from airbyte_cdk.sources.declarative.yaml_declarative_source import YamlDeclarativeSource
|
|
from airbyte_cdk.sources.streams import Stream
|
|
from airbyte_cdk.sources.streams.http.requests_native_auth import TokenAuthenticator
|
|
|
|
from .streams import (
|
|
CampaignsMetrics,
|
|
CustomEvent,
|
|
EmailBounce,
|
|
EmailClick,
|
|
EmailComplaint,
|
|
EmailOpen,
|
|
EmailSend,
|
|
EmailSendSkip,
|
|
EmailSubscribe,
|
|
EmailUnsubscribe,
|
|
HostedUnsubscribeClick,
|
|
InAppClick,
|
|
InAppClose,
|
|
InAppDelete,
|
|
InAppDelivery,
|
|
InAppOpen,
|
|
InAppSend,
|
|
InAppSendSkip,
|
|
InboxMessageImpression,
|
|
InboxSession,
|
|
Purchase,
|
|
PushBounce,
|
|
PushOpen,
|
|
PushSend,
|
|
PushSendSkip,
|
|
PushUninstall,
|
|
SmsBounce,
|
|
SmsClick,
|
|
SmsReceived,
|
|
SmsSend,
|
|
SmsSendSkip,
|
|
SmsUsageInfo,
|
|
Templates,
|
|
WebPushClick,
|
|
WebPushSend,
|
|
WebPushSendSkip,
|
|
)
|
|
|
|
"""
|
|
This file provides the necessary constructs to interpret a provided declarative YAML configuration file into
|
|
source connector.
|
|
|
|
WARNING: Do not modify this file.
|
|
"""
|
|
|
|
# Declarative Source
|
|
class SourceIterable(YamlDeclarativeSource):
|
|
def __init__(self):
|
|
super().__init__(**{"path_to_yaml": "manifest.yaml"})
|
|
|
|
def streams(self, config: Mapping[str, Any]) -> List[Stream]:
|
|
streams = super().streams(config=config)
|
|
|
|
authenticator = TokenAuthenticator(token=config["api_key"], auth_header="Api-Key", auth_method="")
|
|
# end date is provided for integration tests only
|
|
start_date, end_date = config["start_date"], config.get("end_date")
|
|
date_range = {"start_date": start_date, "end_date": end_date}
|
|
|
|
# TODO: migrate streams below to low code as slicer logic will be migrated to generator based
|
|
streams.extend(
|
|
[
|
|
CampaignsMetrics(authenticator=authenticator, **date_range),
|
|
Templates(authenticator=authenticator, **date_range),
|
|
EmailBounce(authenticator=authenticator, **date_range),
|
|
EmailClick(authenticator=authenticator, **date_range),
|
|
EmailComplaint(authenticator=authenticator, **date_range),
|
|
EmailOpen(authenticator=authenticator, **date_range),
|
|
EmailSend(authenticator=authenticator, **date_range),
|
|
EmailSendSkip(authenticator=authenticator, **date_range),
|
|
EmailSubscribe(authenticator=authenticator, **date_range),
|
|
EmailUnsubscribe(authenticator=authenticator, **date_range),
|
|
PushSend(authenticator=authenticator, **date_range),
|
|
PushSendSkip(authenticator=authenticator, **date_range),
|
|
PushOpen(authenticator=authenticator, **date_range),
|
|
PushUninstall(authenticator=authenticator, **date_range),
|
|
PushBounce(authenticator=authenticator, **date_range),
|
|
WebPushSend(authenticator=authenticator, **date_range),
|
|
WebPushClick(authenticator=authenticator, **date_range),
|
|
WebPushSendSkip(authenticator=authenticator, **date_range),
|
|
InAppSend(authenticator=authenticator, **date_range),
|
|
InAppOpen(authenticator=authenticator, **date_range),
|
|
InAppClick(authenticator=authenticator, **date_range),
|
|
InAppClose(authenticator=authenticator, **date_range),
|
|
InAppDelete(authenticator=authenticator, **date_range),
|
|
InAppDelivery(authenticator=authenticator, **date_range),
|
|
InAppSendSkip(authenticator=authenticator, **date_range),
|
|
InboxSession(authenticator=authenticator, **date_range),
|
|
InboxMessageImpression(authenticator=authenticator, **date_range),
|
|
SmsSend(authenticator=authenticator, **date_range),
|
|
SmsBounce(authenticator=authenticator, **date_range),
|
|
SmsClick(authenticator=authenticator, **date_range),
|
|
SmsReceived(authenticator=authenticator, **date_range),
|
|
SmsSendSkip(authenticator=authenticator, **date_range),
|
|
SmsUsageInfo(authenticator=authenticator, **date_range),
|
|
Purchase(authenticator=authenticator, **date_range),
|
|
CustomEvent(authenticator=authenticator, **date_range),
|
|
HostedUnsubscribeClick(authenticator=authenticator, **date_range),
|
|
]
|
|
)
|
|
return streams
|