f6bd91018b
* bump connector version * Source: MailChimp, New Stream Automations * fix: Schema error fix * Fixes formatting. * fix: Fixed acceptance test schema errors * fix: Update schema, update dockerlabel * refactor: Update docs * refactor: Update DockerVersionTag * fix: Update EOF and configured catalog * fix: Schema, Update docs * fix: Schema on catalog * fix: update DocketVersionTag * fix: schema, update docs, bump docker version * fix: spec, update doc in streams.py * fix: schema compatability for null * update connector version to 0.4.0 * Update Dockerfile * recreate source spec * restore source spec form main * rollback source spec from main * source spec * run gradle config seed * fix schema and remove unused code * fix docs * bump connector seed file versino * auto-bump connector version --------- Co-authored-by: marcosmarxm <marcosmarxm@gmail.com> Co-authored-by: nataly <nataly@airbyte.io> Co-authored-by: Octavia Squidington III <octavia-squidington-iii@users.noreply.github.com> Co-authored-by: Marcos Marx <marcosmarxm@users.noreply.github.com>
74 lines
2.9 KiB
Python
74 lines
2.9 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
|
|
import base64
|
|
from typing import Any, List, Mapping, Tuple
|
|
|
|
import requests
|
|
from airbyte_cdk import AirbyteLogger
|
|
from airbyte_cdk.sources import AbstractSource
|
|
from airbyte_cdk.sources.streams import Stream
|
|
from airbyte_cdk.sources.streams.http.auth import TokenAuthenticator
|
|
from requests.auth import AuthBase
|
|
|
|
from .streams import Automations, Campaigns, EmailActivity, Lists, Reports
|
|
|
|
|
|
class MailChimpAuthenticator:
|
|
@staticmethod
|
|
def get_server_prefix(access_token: str) -> str:
|
|
try:
|
|
response = requests.get(
|
|
"https://login.mailchimp.com/oauth2/metadata", headers={"Authorization": "OAuth {}".format(access_token)}
|
|
)
|
|
return response.json()["dc"]
|
|
except Exception as e:
|
|
raise Exception(f"Cannot retrieve server_prefix for you account. \n {repr(e)}")
|
|
|
|
def get_auth(self, config: Mapping[str, Any]) -> AuthBase:
|
|
authorization = config.get("credentials", {})
|
|
auth_type = authorization.get("auth_type")
|
|
if auth_type == "apikey" or not authorization:
|
|
# API keys have the format <key>-<data_center>.
|
|
# See https://mailchimp.com/developer/marketing/docs/fundamentals/#api-structure
|
|
apikey = authorization.get("apikey") or config.get("apikey")
|
|
if not apikey:
|
|
raise Exception("No apikey in creds")
|
|
auth_string = f"anystring:{apikey}".encode("utf8")
|
|
b64_encoded = base64.b64encode(auth_string).decode("utf8")
|
|
auth = TokenAuthenticator(token=b64_encoded, auth_method="Basic")
|
|
auth.data_center = apikey.split("-").pop()
|
|
|
|
elif auth_type == "oauth2.0":
|
|
access_token = authorization["access_token"]
|
|
auth = TokenAuthenticator(token=access_token, auth_method="Bearer")
|
|
auth.data_center = self.get_server_prefix(access_token)
|
|
|
|
else:
|
|
raise Exception(f"Invalid auth type: {auth_type}")
|
|
|
|
return auth
|
|
|
|
|
|
class SourceMailchimp(AbstractSource):
|
|
def check_connection(self, logger: AirbyteLogger, config: Mapping[str, Any]) -> Tuple[bool, Any]:
|
|
try:
|
|
authenticator = MailChimpAuthenticator().get_auth(config)
|
|
requests.get(f"https://{authenticator.data_center}.api.mailchimp.com/3.0/ping", headers=authenticator.get_auth_header())
|
|
return True, None
|
|
except Exception as e:
|
|
return False, repr(e)
|
|
|
|
def streams(self, config: Mapping[str, Any]) -> List[Stream]:
|
|
authenticator = MailChimpAuthenticator().get_auth(config)
|
|
campaign_id = config.get("campaign_id")
|
|
return [
|
|
Lists(authenticator=authenticator),
|
|
Campaigns(authenticator=authenticator),
|
|
Automations(authenticator=authenticator),
|
|
EmailActivity(authenticator=authenticator, campaign_id=campaign_id),
|
|
Reports(authenticator=authenticator),
|
|
]
|