From 102ea93e8344ea4cb50ecf7d98c83dd6cf0f8e22 Mon Sep 17 00:00:00 2001 From: Pere Miquel Brull Date: Tue, 23 Aug 2022 07:39:23 +0200 Subject: [PATCH] Check if table exists in datamodel rest (#6859) --- .../src/metadata/ingestion/sink/metadata_rest.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/ingestion/src/metadata/ingestion/sink/metadata_rest.py b/ingestion/src/metadata/ingestion/sink/metadata_rest.py index 0024cc21745..e01550686bb 100644 --- a/ingestion/src/metadata/ingestion/sink/metadata_rest.py +++ b/ingestion/src/metadata/ingestion/sink/metadata_rest.py @@ -189,11 +189,16 @@ class MetadataRestSink(Sink[Entity]): :param datamodel_link: Table ID + Data Model """ - table = self.metadata.get_by_name(entity=Table, fqn=datamodel_link.fqn) + table: Table = self.metadata.get_by_name(entity=Table, fqn=datamodel_link.fqn) - self.metadata.ingest_table_data_model( - table=table, data_model=datamodel_link.datamodel - ) + if table: + self.metadata.ingest_table_data_model( + table=table, data_model=datamodel_link.datamodel + ) + else: + logger.warning( + f"Could not find any entity by Table FQN [{datamodel_link.fqn}] when adding DBT models." + ) def write_table_location_link(self, table_location_link: TableLocationLink) -> None: """