Skip to content

chore: upgrade sqlmesh to v0.163.0 #17

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Mar 7, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 25 additions & 4 deletions dagster_sqlmesh/console.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from sqlmesh.core.plan import EvaluatablePlan
from sqlmesh.core.context_diff import ContextDiff
from sqlmesh.core.plan import PlanBuilder
from sqlmesh.core.table_diff import RowDiff, SchemaDiff
from sqlmesh.core.table_diff import RowDiff, SchemaDiff, TableDiff
from sqlmesh.core.environment import EnvironmentNamingInfo
from sqlmesh.core.snapshot import (
Snapshot,
Expand Down Expand Up @@ -197,7 +197,8 @@ class LogError:

@dataclass
class LogWarning:
message: str
short_message: str
long_message: t.Optional[str] = None


@dataclass
Expand Down Expand Up @@ -252,6 +253,16 @@ class ConsoleException:
exception: Exception


@dataclass
class PrintEnvironments:
environments_summary: t.Dict[str, int]


@dataclass
class ShowTableDiffSummary:
table_diff: TableDiff


ConsoleEvent = Union[
StartPlanEvaluation,
StopPlanEvaluation,
Expand Down Expand Up @@ -294,6 +305,8 @@ class ConsoleException:
StopMigrationProgress,
StartSnapshotMigrationProgress,
ConsoleException,
PrintEnvironments,
ShowTableDiffSummary,
]

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

def log_warning(self, message):
self.publish(LogWarning(message))
def log_warning(
self, short_message: str, long_message: t.Optional[str] = None
) -> None:
self.publish(LogWarning(short_message, long_message))

def log_success(self, message: str) -> None:
self.publish(LogSuccess(message))
Expand Down Expand Up @@ -538,6 +553,12 @@ def remove_handler(self, handler_id: str):
def exception(self, exc: Exception):
self.publish(ConsoleException(exc))

def print_environments(self, environments_summary: t.Dict[str, int]) -> None:
self.publish(PrintEnvironments(environments_summary))

def show_table_diff_summary(self, table_diff: TableDiff) -> None:
self.publish(ShowTableDiffSummary(table_diff))


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