Added check for policy tags (#2655)

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

View File

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