|
10 | 10 | from sqlmesh.core.plan import EvaluatablePlan
|
11 | 11 | from sqlmesh.core.context_diff import ContextDiff
|
12 | 12 | 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 |
14 | 14 | from sqlmesh.core.environment import EnvironmentNamingInfo
|
15 | 15 | from sqlmesh.core.snapshot import (
|
16 | 16 | Snapshot,
|
@@ -197,7 +197,8 @@ class LogError:
|
197 | 197 |
|
198 | 198 | @dataclass
|
199 | 199 | class LogWarning:
|
200 |
| - message: str |
| 200 | + short_message: str |
| 201 | + long_message: t.Optional[str] = None |
201 | 202 |
|
202 | 203 |
|
203 | 204 | @dataclass
|
@@ -252,6 +253,16 @@ class ConsoleException:
|
252 | 253 | exception: Exception
|
253 | 254 |
|
254 | 255 |
|
| 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 | + |
255 | 266 | ConsoleEvent = Union[
|
256 | 267 | StartPlanEvaluation,
|
257 | 268 | StopPlanEvaluation,
|
@@ -294,6 +305,8 @@ class ConsoleException:
|
294 | 305 | StopMigrationProgress,
|
295 | 306 | StartSnapshotMigrationProgress,
|
296 | 307 | ConsoleException,
|
| 308 | + PrintEnvironments, |
| 309 | + ShowTableDiffSummary, |
297 | 310 | ]
|
298 | 311 |
|
299 | 312 | ConsoleEventHandler = Callable[[ConsoleEvent], None]
|
@@ -474,8 +487,10 @@ def log_status_update(self, message: str) -> None:
|
474 | 487 | def log_error(self, message: str) -> None:
|
475 | 488 | self.publish(LogError(message))
|
476 | 489 |
|
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)) |
479 | 494 |
|
480 | 495 | def log_success(self, message: str) -> None:
|
481 | 496 | self.publish(LogSuccess(message))
|
@@ -538,6 +553,12 @@ def remove_handler(self, handler_id: str):
|
538 | 553 | def exception(self, exc: Exception):
|
539 | 554 | self.publish(ConsoleException(exc))
|
540 | 555 |
|
| 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 | + |
541 | 562 |
|
542 | 563 | class DebugEventConsole(EventConsole):
|
543 | 564 | """A console that wraps an existing console and logs all events to a logger"""
|
|
0 commit comments