chore: suppress and warn during state interaction

src/cdf/core/state.py

@@ -16,6 +16,7 @@
 )
 from sqlmesh.core.engine_adapter import EngineAdapter
 
+import cdf.core.logger as logger
 from cdf.core.context import active_project, execution_id
 from cdf.types import M, P
 
@@ -130,7 +131,7 @@ def _execute(self, sql: str) -> None:
 
     def store_json(self, key: str, value: JSON) -> None:
         """Store a JSON value"""
-        with self.adapter.transaction(value is not None):
+        with self.adapter.transaction(value is not None), logger.suppress_and_warn():
             self.adapter.delete_from(self.kv_table, f"key = '{key}'")
             if value is not None:
                 self.adapter.insert_append(
@@ -208,7 +209,7 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
                 audit_event["success"] = not isinstance(rv, M.Err)
                 audit_event["properties"].update(output_props(rv))
                 audit_event["properties"] = json.dumps(audit_event["properties"])
-                with self.adapter.transaction():
+                with self.adapter.transaction(), logger.suppress_and_warn():
                     self.adapter.insert_append(
                         self.audit_table,
                         pd.DataFrame([audit_event]),
@@ -231,7 +232,7 @@ def audit(
             "properties": json.dumps(properties),
             "execution_id": execution_id.get(),
         }
-        with self.adapter.transaction():
+        with self.adapter.transaction(), logger.suppress_and_warn():
             self.adapter.insert_append(
                 self.audit_table,
                 pd.DataFrame([payload]),
@@ -267,23 +268,23 @@ def capture_extract_info(self, info: ExtractInfo) -> None:
         d = self._info_to_payload(info)
         if not d:
             return
-        with self.adapter.transaction():
+        with self.adapter.transaction(), logger.suppress_and_warn():
             self.adapter.insert_append(self.extract_table, pd.DataFrame(d))
 
     def capture_normalize_info(self, info: NormalizeInfo) -> None:
         """Capture normalize information"""
         d = self._info_to_payload(info)
         if not d:
             return
-        with self.adapter.transaction():
+        with self.adapter.transaction(), logger.suppress_and_warn():
             self.adapter.insert_append(self.normalize_table, pd.DataFrame(d))
 
     def capture_load_info(self, info: LoadInfo) -> None:
         """Capture load information"""
         d = self._info_to_payload(info)
         if not d:
             return
-        with self.adapter.transaction():
+        with self.adapter.transaction(), logger.suppress_and_warn():
             self.adapter.insert_append(self.load_table, pd.DataFrame(d))
 
     @staticmethod