yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-25 18:30:00 +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 GitHub
parent 51448452d0
commit 4f89dc582b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 Method to get the table description for salesforce with Tooling API
""" """
table_description = None
try: try:
result = self.client.toolingexecute( result = self.client.toolingexecute(
f"query/?q=SELECT+Description+FROM+EntityDefinition+WHERE+QualifiedApiName='{table_name}'" 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: except KeyError as err:
logger.warning( logger.warning(
f"Unable to get required key from Tooling API response for table [{table_name}]: {err}" f"Unable to get required key from Tooling API response for table [{table_name}]: {err}"
@ -225,7 +228,7 @@ class SalesforceSource(DatabaseServiceSource):
logger.warning( logger.warning(
f"Unable to get description with Tooling API for table [{table_name}]: {exc}" 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( def yield_table(
self, table_name_and_type: Tuple[str, TableType] self, table_name_and_type: Tuple[str, TableType]
@ -241,11 +244,13 @@ class SalesforceSource(DatabaseServiceSource):
f"sobjects/{table_name}/describe/", f"sobjects/{table_name}/describe/",
params=None, params=None,
) )
columns = self.get_columns(salesforce_objects["fields"]) columns = self.get_columns(salesforce_objects.get("fields", []))
table_request = CreateTableRequest( table_request = CreateTableRequest(
name=EntityName(table_name), name=EntityName(table_name),
tableType=table_type, tableType=table_type,
description=self.get_table_description(table_name), description=self.get_table_description(
table_name, salesforce_objects.get("label")
),
columns=columns, columns=columns,
tableConstraints=table_constraints, tableConstraints=table_constraints,
databaseSchema=FullyQualifiedEntityName( databaseSchema=FullyQualifiedEntityName(