Skip to content

Commit

Permalink
feat(ingest/salesforce): helpful error messages on failure
Browse files Browse the repository at this point in the history
  • Loading branch information
mayurinehate committed Aug 29, 2024
1 parent 223650d commit 28e0ec9
Showing 1 changed file with 45 additions and 19 deletions.
64 changes: 45 additions & 19 deletions metadata-ingestion/src/datahub/ingestion/source/salesforce.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,7 +287,13 @@ def __init__(self, config: SalesforceConfig, ctx: PipelineContext) -> None:

except Exception as e:
logger.error(e)
raise ConfigurationError("Salesforce login failed") from e
if "API_CURRENTLY_DISABLED" in str(e):
# https://help.salesforce.com/s/articleView?id=001473830&type=1
error = "Salesforce login failed. Please make sure user has API Enabled Access."
else:
error = "Salesforce login failed. Please verify your credentials."
"Please set `is_sandbox: True` in recipe if this is sandbox account."
raise ConfigurationError(error) from e
if not self.config.api_version:
# List all REST API versions and use latest one
versions_url = "https://{instance}/services/data/".format(
Expand All @@ -314,10 +320,19 @@ def __init__(self, config: SalesforceConfig, ctx: PipelineContext) -> None:
)

def get_workunits_internal(self) -> Iterable[MetadataWorkUnit]:
sObjects = self.get_salesforce_objects()

for sObject in sObjects:
yield from self.get_salesforce_object_workunits(sObject)
try:
sObjects = self.get_salesforce_objects()
except Exception as e:
if "sObject type 'EntityDefinition' is not supported." in str(e):
# https://developer.salesforce.com/docs/atlas.en-us.api_tooling.meta/api_tooling/tooling_api_objects_entitydefinition.htm
raise ConfigurationError(
"Salesforce EntityDefinition query failed. "
"Please verify if user has 'View Setup and Configuration' permission."
) from e
raise e
else:
for sObject in sObjects:
yield from self.get_salesforce_object_workunits(sObject)

def get_salesforce_object_workunits(
self, sObject: dict
Expand Down Expand Up @@ -596,9 +611,9 @@ def _get_schema_field(

TypeClass = FIELD_TYPE_MAPPING.get(fieldType)
if TypeClass is None:
self.report.report_warning(
sObjectName,
f"Unable to map type {fieldType} to metadata schema",
self.report.warning(
message="Unable to map field type to metadata schema",
context=f"{fieldType} for {fieldName} of {sObjectName}",
)
TypeClass = NullTypeClass

Expand Down Expand Up @@ -696,19 +711,30 @@ def get_schema_metadata_workunit(
)
)

sObject_custom_fields_response = self.sf._call_salesforce(
"GET", sObject_custom_fields_query_url
).json()
customFields: Dict[str, Dict] = {}
try:
sObject_custom_fields_response = self.sf._call_salesforce(
"GET", sObject_custom_fields_query_url
).json()

logger.debug(
"Received Salesforce {sObject} custom fields response".format(
sObject=sObjectName
logger.debug(
"Received Salesforce {sObject} custom fields response".format(
sObject=sObjectName
)
)
)
customFields: Dict[str, Dict] = {
record["DeveloperName"]: record
for record in sObject_custom_fields_response["records"]
}

except Exception as e:
error = "Salesforce CustomField query failed. "
if "sObject type 'CustomField' is not supported." in str(e):
# https://github.com/afawcett/apex-toolingapi/issues/19
error += "Please verify if user has 'View All Data' permission."

self.report.warning(message=error, exc=e)
else:
customFields = {
record["DeveloperName"]: record
for record in sObject_custom_fields_response["records"]
}

fields: List[SchemaFieldClass] = []
primaryKeys: List[str] = []
Expand Down

0 comments on commit 28e0ec9

Please sign in to comment.