Factor out severity

PicnicSupermarket · Aug 8, 2024 · c887978 · c887978
1 parent 81a5a86
commit c887978
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/dbt_score/formatters/json_formatter.py b/src/dbt_score/formatters/json_formatter.py
@@ -68,22 +68,23 @@ def model_evaluated(
             "results": {},
         }
         for rule, result in results.items():
+            severity = rule.severity.name.lower()
             if result is None:
                 self._model_results[model.name]["results"][rule.source()] = {
                     "result": "OK",
-                    "severity": rule.severity.name.lower(),
+                    "severity": severity,
                     "message": None,
                 }
             elif isinstance(result, RuleViolation):
                 self._model_results[model.name]["results"][rule.source()] = {
                     "result": "WARN",
-                    "severity": rule.severity.name.lower(),
+                    "severity": severity,
                     "message": result.message,
                 }
             else:
                 self._model_results[model.name]["results"][rule.source()] = {
                     "result": "ERR",
-                    "severity": rule.severity.name.lower(),
+                    "severity": severity,
                     "message": str(result),
                 }