Move filtering into console method

Author: treysp

sqlmesh/core/console.py

@@ -968,10 +968,13 @@ def start_promotion_progress(
                 "Updating virtual layer ", self.console, justify="left"
             )
 
+            snapshots_with_virtual_views = [
+                s for s in snapshots if s.is_model and not s.is_symbolic
+            ]
             self.promotion_progress.start()
             self.promotion_task = self.promotion_progress.add_task(
                 f"Virtually updating {environment_naming_info.name}...",
-                total=len(snapshots),
+                total=len(snapshots_with_virtual_views),
             )
 
             # determine name column widths if we're printing names
@@ -984,15 +987,20 @@ def start_promotion_progress(
                             dialect=self.dialect,
                         )
                     )
-                    for snapshot in snapshots
+                    for snapshot in snapshots_with_virtual_views
                 )
 
             self.environment_naming_info = environment_naming_info
             self.default_catalog = default_catalog
 
     def update_promotion_progress(self, snapshot: SnapshotInfoLike, promoted: bool) -> None:
         """Update the snapshot promotion progress."""
-        if self.promotion_progress is not None and self.promotion_task is not None:
+        if (
+            self.promotion_progress is not None
+            and self.promotion_task is not None
+            and snapshot.is_model
+            and not snapshot.is_symbolic
+        ):
             if self.verbosity >= Verbosity.VERBOSE:
                 display_name = snapshot.display_name(
                     self.environment_naming_info,
@@ -1007,9 +1015,7 @@ def update_promotion_progress(self, snapshot: SnapshotInfoLike, promoted: bool)
                         else "[green]created[/green]"
                     )
                 action_str = action_str or "[red]dropped[/red]"
-                self.promotion_progress.live.console.print(
-                    f"{snapshot.display_name(self.environment_naming_info, self.default_catalog if self.verbosity < Verbosity.VERY_VERBOSE else None, dialect=self.dialect).ljust(self.PROGRESS_BAR_COLUMN_WIDTHS['name'])} {action_str.ljust(7)}"
-                )
+                self.promotion_progress.live.console.print(f"{display_name} {action_str}")
             self.promotion_progress.update(self.promotion_task, refresh=True, advance=1)
 
     def stop_promotion_progress(self, success: bool = True) -> None:

sqlmesh/core/plan/evaluator.py

@@ -376,7 +376,7 @@ def _update_views(
         environment = plan.environment
 
         self.console.start_promotion_progress(
-            len(promotion_result.added) + len(promotion_result.removed),
+            promotion_result.added + promotion_result.removed,
             environment.naming_info,
             self.default_catalog,
         )

sqlmesh/core/scheduler.py

@@ -427,7 +427,7 @@ def run_merged_intervals(
         batched_intervals = self.batch_intervals(merged_intervals, deployability_index)
 
         self.console.start_evaluation_progress(
-            {snapshot: len(intervals) for snapshot, intervals in batched_intervals.items()},
+            batched_intervals,
             environment_naming_info,
             self.default_catalog,
         )

tests/cli/test_cli.py

@@ -294,8 +294,8 @@ def test_plan_verbose(runner, tmp_path):
         cli, ["--log-file-dir", tmp_path, "--paths", tmp_path, "plan", "--verbose"], input="y\n"
     )
     assert_plan_success(result)
-    assert "sqlmesh_example.seed_model                           created" in result.output
-    assert "sqlmesh_example.full_model                           created" in result.output
+    assert "sqlmesh_example.seed_model        created" in result.output
+    assert "sqlmesh_example.full_model        created" in result.output
 
 
 def test_plan_very_verbose(runner, tmp_path, copy_to_temp_path):

web/server/console.py

@@ -172,11 +172,7 @@ def start_promotion_progress(
 
         self.log_event_plan_apply()
 
-    def update_promotion_progress(
-        self,
-        snapshot: SnapshotInfoLike,
-        promoted: bool,
-    ) -> None:
+    def update_promotion_progress(self, snapshot: SnapshotInfoLike, promoted: bool) -> None:
         if self.plan_apply_stage_tracker and self.plan_apply_stage_tracker.promote:
             self.plan_apply_stage_tracker.promote.update(
                 {"num_tasks": self.plan_apply_stage_tracker.promote.num_tasks + 1}