yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-07 05:53:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added check for policy tags (#2655)

Browse Source
This commit is contained in:
Ayush Shah 2022-02-08 00:19:32 +05:30 committed by GitHub
parent b4ebf5ce87
commit 0e62db641d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ingestion/src/metadata/ingestion/source/sql_source.py
View File

@ -564,7 +564,11 @@ class SQLSource(Source[OMetaDatabaseAndTable]):
] ]
col_dict = Column(**parsed_string) col_dict = Column(**parsed_string)
try: try:
if "policy_tags" in column and column["policy_tags"]: if (
self.config.enable_policy_tags
and "policy_tags" in column
and column["policy_tags"]
):
self.metadata.create_primary_tag_category( self.metadata.create_primary_tag_category(
category=self.config.tag_category_name, category=self.config.tag_category_name,
data=Tag( data=Tag(
@ -572,7 +576,7 @@ class SQLSource(Source[OMetaDatabaseAndTable]):
), ),
) )
except APIError: except APIError:
if column["policy_tags"]: if column["policy_tags"] and self.config.enable_policy_tags:
col_dict.tags = [ col_dict.tags = [
TagLabel( TagLabel(
tagFQN=f"{self.config.tag_category_name}.{column['policy_tags']}", tagFQN=f"{self.config.tag_category_name}.{column['policy_tags']}",
@ -581,9 +585,8 @@ class SQLSource(Source[OMetaDatabaseAndTable]):
) )
] ]
except Exception as err: except Exception as err:
logger.error(traceback.print_exc()) logger.debug(traceback.print_exc())
logger.error(err) logger.debug(err)
om_column = col_dict om_column = col_dict
except Exception as err: except Exception as err:
logger.debug(traceback.print_exc()) logger.debug(traceback.print_exc())