Skip to content

Commit 35c5afe

Browse files
authored
chore: upgrade sqlmesh to v0.163.0 (#17)
* chore: bump sqlmesh version * chore: bump package version
1 parent c105fbb commit 35c5afe

File tree

3 files changed

+864
-845
lines changed

3 files changed

+864
-845
lines changed

dagster_sqlmesh/console.py

Lines changed: 25 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from sqlmesh.core.plan import EvaluatablePlan
1111
from sqlmesh.core.context_diff import ContextDiff
1212
from sqlmesh.core.plan import PlanBuilder
13-
from sqlmesh.core.table_diff import RowDiff, SchemaDiff
13+
from sqlmesh.core.table_diff import RowDiff, SchemaDiff, TableDiff
1414
from sqlmesh.core.environment import EnvironmentNamingInfo
1515
from sqlmesh.core.snapshot import (
1616
Snapshot,
@@ -197,7 +197,8 @@ class LogError:
197197

198198
@dataclass
199199
class LogWarning:
200-
message: str
200+
short_message: str
201+
long_message: t.Optional[str] = None
201202

202203

203204
@dataclass
@@ -252,6 +253,16 @@ class ConsoleException:
252253
exception: Exception
253254

254255

256+
@dataclass
257+
class PrintEnvironments:
258+
environments_summary: t.Dict[str, int]
259+
260+
261+
@dataclass
262+
class ShowTableDiffSummary:
263+
table_diff: TableDiff
264+
265+
255266
ConsoleEvent = Union[
256267
StartPlanEvaluation,
257268
StopPlanEvaluation,
@@ -294,6 +305,8 @@ class ConsoleException:
294305
StopMigrationProgress,
295306
StartSnapshotMigrationProgress,
296307
ConsoleException,
308+
PrintEnvironments,
309+
ShowTableDiffSummary,
297310
]
298311

299312
ConsoleEventHandler = Callable[[ConsoleEvent], None]
@@ -474,8 +487,10 @@ def log_status_update(self, message: str) -> None:
474487
def log_error(self, message: str) -> None:
475488
self.publish(LogError(message))
476489

477-
def log_warning(self, message):
478-
self.publish(LogWarning(message))
490+
def log_warning(
491+
self, short_message: str, long_message: t.Optional[str] = None
492+
) -> None:
493+
self.publish(LogWarning(short_message, long_message))
479494

480495
def log_success(self, message: str) -> None:
481496
self.publish(LogSuccess(message))
@@ -538,6 +553,12 @@ def remove_handler(self, handler_id: str):
538553
def exception(self, exc: Exception):
539554
self.publish(ConsoleException(exc))
540555

556+
def print_environments(self, environments_summary: t.Dict[str, int]) -> None:
557+
self.publish(PrintEnvironments(environments_summary))
558+
559+
def show_table_diff_summary(self, table_diff: TableDiff) -> None:
560+
self.publish(ShowTableDiffSummary(table_diff))
561+
541562

542563
class DebugEventConsole(EventConsole):
543564
"""A console that wraps an existing console and logs all events to a logger"""

0 commit comments

Comments
 (0)