Skip to content

Commit 6241445

Browse files
authored
feat: log sqlmesh errors to dagster (#10)
1 parent d74cf01 commit 6241445

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

dagster_sqlmesh/resource.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def warning(self, message: str, obj: t.Optional[t.Dict[str, t.Any]] = None):
8787
self.log("warning", message, obj)
8888

8989
def error(self, message: str, obj: t.Optional[t.Dict[str, t.Any]] = None):
90-
self.log("warning", message, obj)
90+
self.log("error", message, obj)
9191

9292
def log(self, level: str | int, message: str, obj: t.Optional[t.Dict[str, t.Any]]):
9393
self._handler.log(level, message, self.ensure_standard_obj(obj))
@@ -180,15 +180,17 @@ def report_event(self, event: console.ConsoleEvent):
180180
"duration_ms": duration_ms,
181181
},
182182
)
183-
184183
case console.LogSuccess(success):
185184
self.update_stage("done")
186185
if success:
187186
log_context.info("sqlmesh ran successfully")
188187
else:
189188
log_context.error("sqlmesh failed")
190189
raise Exception("sqlmesh failed during run")
191-
190+
case console.LogError(message):
191+
log_context.error(
192+
message,
193+
)
192194
case _:
193195
log_context.debug("Received event")
194196

@@ -201,6 +203,10 @@ def log(
201203
message: str,
202204
obj: t.Optional[t.Dict[str, t.Any]] = None,
203205
):
206+
if level == "error":
207+
self._logger.error(message)
208+
return
209+
204210
obj = obj or {}
205211
final_obj = obj.copy()
206212
final_obj["message"] = message

0 commit comments

Comments
 (0)