Skip to content

Commit b150d04

Browse files
committed
upd _prepare_stream_args after review
1 parent f694c10 commit b150d04

File tree

1 file changed

+12
-6
lines changed
  • airbyte-integrations/connectors/source-tiktok-marketing/source_tiktok_marketing

1 file changed

+12
-6
lines changed

airbyte-integrations/connectors/source-tiktok-marketing/source_tiktok_marketing/source.py

+12-6
Original file line numberDiff line numberDiff line change
@@ -68,12 +68,18 @@ def spec(self, *args, **kwargs) -> ConnectorSpecification:
6868
@staticmethod
6969
def _prepare_stream_args(config: Mapping[str, Any]) -> Mapping[str, Any]:
7070
"""Converts an input configure to stream arguments"""
71-
credentials = config.get("credentials") or config
72-
73-
access_token = credentials["access_token"]
74-
secret = credentials.get("secret") or credentials.get("environment", {}).get("secret")
75-
app_id = int(credentials.get("app_id", 0) or credentials.get("environment", {}).get("app_id", 0))
76-
advertiser_id = int(credentials.get("advertiser_id", 0) or credentials.get("environment", {}).get("advertiser_id", 0))
71+
credentials = config.get("credentials")
72+
if credentials:
73+
# used for new config format
74+
access_token = credentials["access_token"]
75+
secret = credentials.get("secret")
76+
app_id = int(credentials.get("app_id", 0))
77+
advertiser_id = int(credentials.get("advertiser_id", 0))
78+
else:
79+
access_token = config["access_token"]
80+
secret = config.get("environment", {}).get("secret")
81+
app_id = int(config.get("environment", {}).get("app_id", 0))
82+
advertiser_id = int(config.get("environment", {}).get("advertiser_id", 0))
7783

7884
return {
7985
"authenticator": TiktokTokenAuthenticator(access_token),

0 commit comments

Comments
 (0)