Added the raw config argument generator

Author: vaibhavatlan

pyatlan/model/assets/core/alpha__d_q_rule.py

@@ -15,6 +15,8 @@
     alpha_DQRuleAlertPriority,
     alpha_DQRuleStatus,
     alpha_DQSourceSyncStatus,
+    alpha_DQRuleThresholdCompareOperator,
+    alpha_DQRuleThresholdUnit
 )
 from pyatlan.model.fields.atlan_fields import (
     KeywordField,
@@ -24,12 +26,47 @@
     TextField,
 )
 from pyatlan.model.structs import alpha_DQRuleConfigArguments
+import json
 
 from .data_quality import DataQuality
 
 
 class alpha_DQRule(DataQuality):
     """Description"""
+    @classmethod
+    def _generate_config_arguments_raw(
+        *,
+        is_alert_enabled: bool = True,
+        custom_sql: Optional[str] = None,
+        display_name: Optional[str] = None,
+        dimension: Optional[alpha_DQDimension] = None,
+        compare_operator: alpha_DQRuleThresholdCompareOperator,
+        threshold_value: int,
+        threshold_unit: Optional[alpha_DQRuleThresholdUnit] = None,
+        dq_priority: alpha_DQRuleAlertPriority,
+    ) -> str:
+
+        config = {
+            "isAlertEnabled": is_alert_enabled,
+            "alpha_dqRuleTemplateConfigThresholdObject": {
+                "alpha_dqRuleTemplateConfigThresholdCompareOperator": compare_operator,
+                "alpha_dqRuleTemplateConfigThresholdValue": threshold_value,
+                "alpha_dqRuleTemplateConfigThresholdUnit": threshold_unit,
+            },
+            "alpha_dqRuleTemplateAdvancedSettings.dqPriority": dq_priority,
+        }
+        
+        if custom_sql is not None:
+            config["alpha_dqRuleTemplateConfigCustomSQL"] = custom_sql
+            
+        if display_name is not None:
+            config["alpha_dqRuleTemplateConfigDisplayName"] = display_name
+                    
+        if dimension is not None:
+            config["alpha_dqRuleTemplateConfigDimension"] = dimension
+
+
+        return json.dumps(config)
 
     type_name: str = Field(default="alpha_DQRule", allow_mutation=False)
 

pyatlan/model/enums.py

@@ -2970,3 +2970,8 @@ class alpha_DQSourceSyncStatus(str, Enum):
     FAILURE = "FAILURE"
     IN_PROGRESS = "IN_PROGRESS"
     WAITING_FOR_SCHEDULE = "WAITING_FOR_SCHEDULE"
+
+class alpha_DQRuleThresholdCompareOperator(str, Enum):
+    GREATER_THAN_EQUAL = "GTE" 
+    LESS_THAN_EQUAL = "LTE"
+    EQUAL = "EQ"