yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-31 02:29:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

salesforce table description from label if not through query (#18286)

Browse Source
This commit is contained in:
harshsoni2024 2024-10-16 12:56:44 +05:30 committed by harshsoni2024
parent 7bbc92b960
commit 4ac6111893
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ingestion/src/metadata/ingestion/source/database/salesforce/metadata.py
View File

@ -203,15 +203,18 @@ class SalesforceSource(DatabaseServiceSource):
)
)
def get_table_description(self, table_name: str) -> Optional[str]:
def get_table_description(
self, table_name: str, object_label: Optional[str]
) -> Optional[str]:
"""
Method to get the table description for salesforce with Tooling API
"""
table_description = None
try:
result = self.client.toolingexecute(
f"query/?q=SELECT+Description+FROM+EntityDefinition+WHERE+QualifiedApiName='{table_name}'"
)
return result["records"][0]["Description"]
table_description = result["records"][0]["Description"]
except KeyError as err:
logger.warning(
f"Unable to get required key from Tooling API response for table [{table_name}]: {err}"
@ -225,7 +228,7 @@ class SalesforceSource(DatabaseServiceSource):
logger.warning(
f"Unable to get description with Tooling API for table [{table_name}]: {exc}"
)
return None
return table_description if table_description else object_label
def yield_table(
self, table_name_and_type: Tuple[str, TableType]
@ -241,11 +244,13 @@ class SalesforceSource(DatabaseServiceSource):
f"sobjects/{table_name}/describe/",
params=None,
)
columns = self.get_columns(salesforce_objects["fields"])
columns = self.get_columns(salesforce_objects.get("fields", []))
table_request = CreateTableRequest(
name=EntityName(table_name),
tableType=table_type,
description=self.get_table_description(table_name),
description=self.get_table_description(
table_name, salesforce_objects.get("label")
),
columns=columns,
tableConstraints=table_constraints,
databaseSchema=FullyQualifiedEntityName(