chore: update jinja templates

Author: UdaySagar-Git

pyatlan/generator/templates/methods/asset/alpha__d_q_rule.jinja2

@@ -1,5 +1,5 @@
 
-    @classmethod
+@classmethod
     @init_guid
     def custom_sql_creator(
         cls,
@@ -137,12 +137,31 @@
             ],
         )
         template_config = client.dq_template_config_cache.get_template_config(rule_type)
-        threshold_compare_operator = alpha_DQRule.Attributes._validate_template_features(
-            rule_type,
-            rule_conditions,
-            row_scope_filtering_enabled,
-            template_config,
-            threshold_compare_operator,
+
+        asset_for_validation = asset
+        if row_scope_filtering_enabled and asset.qualified_name:
+            from pyatlan.model.fluent_search import FluentSearch
+
+            search_request = (
+                FluentSearch()
+                .where(Asset.QUALIFIED_NAME.eq(asset.qualified_name))
+                .include_on_results(
+                    Asset.ALPHAASSET_DQ_ROW_SCOPE_FILTER_COLUMN_QUALIFIED_NAME
+                )
+            ).to_request()
+            results = client.asset.search(search_request)
+            if results.count == 1:
+                asset_for_validation = results.current_page()[0]
+
+        threshold_compare_operator = (
+            alpha_DQRule.Attributes._validate_template_features(
+                rule_type,
+                rule_conditions,
+                row_scope_filtering_enabled,
+                template_config,
+                threshold_compare_operator,
+                asset_for_validation,
+            )
         )
 
         attributes = alpha_DQRule.Attributes.creator(
@@ -222,8 +241,8 @@
         retrieved_column = search_result.alpha_dq_rule_base_column  # type: ignore[attr-defined]
         retrieved_alert_priority = search_result.alpha_dq_rule_alert_priority  # type: ignore[attr-defined]
         retrieved_row_scope_filtering_enabled = (
-            search_result.alpha_dq_rule_row_scope_filtering_enabled
-        )  # type: ignore[attr-defined]
+            search_result.alpha_dq_rule_row_scope_filtering_enabled  # type: ignore[attr-defined]
+        )
         retrieved_description = search_result.user_description
         retrieved_asset = search_result.alpha_dq_rule_base_dataset  # type: ignore[attr-defined]
         retrieved_template_rule_name = search_result.alpha_dq_rule_template_name  # type: ignore[attr-defined]
@@ -254,20 +273,29 @@
         template_config = client.dq_template_config_cache.get_template_config(
             retrieved_rule_type
         )
-        validated_threshold_operator = alpha_DQRule.Attributes._validate_template_features(
-            retrieved_rule_type,
-            rule_conditions,
-            row_scope_filtering_enabled,
-            template_config,
-            threshold_compare_operator or retrieved_threshold_compare_operator,
+        validated_threshold_operator = (
+            alpha_DQRule.Attributes._validate_template_features(
+                retrieved_rule_type,
+                rule_conditions,
+                row_scope_filtering_enabled,
+                template_config,
+                threshold_compare_operator or retrieved_threshold_compare_operator,
+                retrieved_asset,
+            )
+        )
+
+        final_compare_operator = (
+            validated_threshold_operator
+            or threshold_compare_operator
+            or retrieved_threshold_compare_operator
         )
 
         config_arguments_raw = alpha_DQRule.Attributes._generate_config_arguments_raw(
             is_alert_enabled=True,
             custom_sql=custom_sql or retrieved_custom_sql,
             display_name=rule_name or retrieved_rule_name,
             dimension=dimension or retrieved_dimension,
-            compare_operator=validated_threshold_operator,
+            compare_operator=final_compare_operator,
             threshold_value=threshold_value or retrieved_threshold_value,
             threshold_unit=threshold_unit or retrieved_threshold_unit,
             column=retrieved_column,
@@ -281,7 +309,7 @@
             name="",
             alpha_dq_rule_config_arguments=alpha_DQRuleConfigArguments(
                 alpha_dq_rule_threshold_object=alpha_DQRuleThresholdObject(
-                    alpha_dq_rule_threshold_compare_operator=validated_threshold_operator,
+                    alpha_dq_rule_threshold_compare_operator=final_compare_operator,
                     alpha_dq_rule_threshold_value=threshold_value
                     or retrieved_threshold_value,
                     alpha_dq_rule_threshold_unit=threshold_unit

pyatlan/generator/templates/methods/attribute/alpha__d_q_rule.jinja2

@@ -1,6 +1,8 @@
         @staticmethod
         def _get_template_config_value(
-            config_value: str, property_name: str = None, value_key: str = "default"
+            config_value: str,
+            property_name: Optional[str] = None,
+            value_key: str = "default",
         ):
             if not config_value:
                 return None
@@ -26,9 +28,10 @@
             threshold_compare_operator: Optional[
                 alpha_DQRuleThresholdCompareOperator
             ] = None,
-        ) -> alpha_DQRuleThresholdCompareOperator:
+            asset: Optional[Asset] = None,
+        ) -> Optional[alpha_DQRuleThresholdCompareOperator]:
             if not template_config or not template_config.get("config"):
-                return
+                return None
 
             config = template_config["config"]
 
@@ -41,15 +44,30 @@
                 )
 
             if row_scope_filtering_enabled:
-                advanced_settings = config.alpha_dq_rule_template_advanced_settings or ""
+                advanced_settings = (
+                    config.alpha_dq_rule_template_advanced_settings or ""
+                )
                 if "alpha_dqRuleRowScopeFilteringEnabled" not in str(advanced_settings):
                     raise ErrorCode.DQ_RULE_TYPE_NOT_SUPPORTED.exception_with_parameters(
                         rule_type, "row scope filtering"
                     )
 
+                if asset and not getattr(
+                    asset,
+                    "alpha_asset_d_q_row_scope_filter_column_qualified_name",
+                    None,
+                ):
+                    raise ErrorCode.DQ_ROW_SCOPE_FILTER_COLUMN_MISSING.exception_with_parameters(
+                        getattr(asset, "qualified_name", "unknown")
+                    )
+
             if rule_conditions:
-                allowed_rule_conditions = alpha_DQRule.Attributes._get_template_config_value(
-                    config.alpha_dq_rule_template_config_rule_conditions, None, "enum"
+                allowed_rule_conditions = (
+                    alpha_DQRule.Attributes._get_template_config_value(
+                        config.alpha_dq_rule_template_config_rule_conditions or "",
+                        None,
+                        "enum",
+                    )
                 )
                 if allowed_rule_conditions:
                     try:
@@ -71,7 +89,8 @@
                 if threshold_compare_operator is None:
                     return alpha_DQRuleThresholdCompareOperator.EQUAL
                 elif (
-                    threshold_compare_operator != alpha_DQRuleThresholdCompareOperator.EQUAL
+                    threshold_compare_operator
+                    != alpha_DQRuleThresholdCompareOperator.EQUAL
                 ):
                     raise ErrorCode.INVALID_PARAMETER_VALUE.exception_with_parameters(
                         f"threshold_compare_operator={threshold_compare_operator.value}",