yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-09-01 13:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ingest): salesforce - add sandbox support (#6105)

Browse Source 
Co-authored-by: rgudic <roko.gudic@infobip.com>
This commit is contained in:
Roko Gudić 2022-10-04 18:05:53 +02:00 committed by GitHub
parent e2c5d2a4ce
commit d52fde126f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
metadata-ingestion/src/datahub/ingestion/source/salesforce.py
View File

@ -90,6 +90,10 @@ class SalesforceConfig(DatasetSourceConfigBase):
instance_url: Optional[str] = Field( instance_url: Optional[str] = Field(
description="Salesforce instance url. e.g. https://MyDomainName.my.salesforce.com" description="Salesforce instance url. e.g. https://MyDomainName.my.salesforce.com"
) )
# Flag to indicate whether the instance is production or sandbox
is_sandbox: bool = Field(
default=False, description="Connect to Sandbox instance of your Salesforce"
)
access_token: Optional[str] = Field(description="Access token for instance url") access_token: Optional[str] = Field(description="Access token for instance url")
ingest_tags: Optional[bool] = Field( ingest_tags: Optional[bool] = Field(
@ -206,6 +210,7 @@ class SalesforceSource(Source):
instance_url=self.config.instance_url, instance_url=self.config.instance_url,
session_id=self.config.access_token, session_id=self.config.access_token,
session=self.session, session=self.session,
domain="test" if self.config.is_sandbox else None,
) )
elif self.config.auth is SalesforceAuthType.USERNAME_PASSWORD: elif self.config.auth is SalesforceAuthType.USERNAME_PASSWORD:
logger.debug("Username/Password Provided in Config") logger.debug("Username/Password Provided in Config")
@ -224,6 +229,7 @@ class SalesforceSource(Source):
password=self.config.password, password=self.config.password,
security_token=self.config.security_token, security_token=self.config.security_token,
session=self.session, session=self.session,
domain="test" if self.config.is_sandbox else None,
) )
except Exception as e: except Exception as e: