Fix extraneous spacing between annotation and run time

Author: treysp

sqlmesh/core/console.py

@@ -891,7 +891,7 @@ def update_snapshot_evaluation_progress(
                     self.evaluation_column_widths["duration"]
                 )
 
-                msg = f"{batch} {display_name}  {annotation} {duration}".replace(
+                msg = f"{batch} {display_name}   {annotation}   {duration}".replace(
                     CHECK_MARK, GREEN_CHECK_MARK
                 )
 
@@ -3359,54 +3359,56 @@ def _create_evaluation_model_annotation(snapshot: Snapshot, interval_info: t.Opt
     return interval_info if interval_info else ""
 
 
-def _calculate_interval_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
+def _calculate_interval_str_len(snapshot: Snapshot, intervals: t.List[Interval]) -> int:
     interval_str_len = 0
-    for snapshot, intervals in batched_intervals.items():
-        for interval in intervals:
-            interval_str_len = max(
-                interval_str_len,
-                len(
-                    _create_evaluation_model_annotation(
-                        snapshot, _format_evaluation_model_interval(snapshot, interval)
-                    )
-                ),
-            )
+    for interval in intervals:
+        interval_str_len = max(
+            interval_str_len,
+            len(
+                _create_evaluation_model_annotation(
+                    snapshot, _format_evaluation_model_interval(snapshot, interval)
+                )
+            ),
+        )
     return interval_str_len
 
 
-def _calculate_audit_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
+def _calculate_audit_str_len(snapshot: Snapshot) -> int:
     # The annotation includes audit results. We cannot build the audits result string
     # until after evaluation occurs, but we must determine the annotation column width here.
     # Therefore, we add enough padding for the longest possible audits result string.
     audit_str_len = 0
     audit_base_str_len = len(f", audits ") + 1  # +1 for check/X
-    for snapshot in batched_intervals:
-        if snapshot.is_audit:
+    if snapshot.is_audit:
+        # +1 for "1" audit count, +1 for red X
+        audit_str_len = max(
+            audit_str_len, audit_base_str_len + (2 if not snapshot.audit.blocking else 1)
+        )
+    if snapshot.is_model and snapshot.model.audits:
+        num_audits = len(snapshot.model.audits_with_args)
+        num_nonblocking_audits = sum(
+            1
+            for audit in snapshot.model.audits_with_args
+            if not audit[0].blocking
+            or ("blocking" in audit[1] and audit[1]["blocking"] == exp.false())
+        )
+        if num_audits == 1:
             # +1 for "1" audit count, +1 for red X
-            audit_str_len = max(
-                audit_str_len, audit_base_str_len + (2 if not snapshot.audit.blocking else 1)
-            )
-        if snapshot.is_model and snapshot.model.audits:
-            num_audits = len(snapshot.model.audits_with_args)
-            num_nonblocking_audits = sum(
-                1
-                for audit in snapshot.model.audits_with_args
-                if not audit[0].blocking
-                or ("blocking" in audit[1] and audit[1]["blocking"] == exp.false())
-            )
-            if num_audits == 1:
-                # +1 for "1" audit count, +1 for red X
-                audit_len = audit_base_str_len + (2 if num_nonblocking_audits else 1)
-            else:
-                audit_len = audit_base_str_len + len(str(num_audits))
-                if num_nonblocking_audits:
-                    # +1 for space, +1 for red X
-                    audit_len += len(str(num_nonblocking_audits)) + 2
-            audit_str_len = max(audit_str_len, audit_len)
+            audit_len = audit_base_str_len + (2 if num_nonblocking_audits else 1)
+        else:
+            audit_len = audit_base_str_len + len(str(num_audits))
+            if num_nonblocking_audits:
+                # +1 for space, +1 for red X
+                audit_len += len(str(num_nonblocking_audits)) + 2
+        audit_str_len = max(audit_str_len, audit_len)
     return audit_str_len
 
 
 def _calculate_annotation_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
-    return _calculate_interval_str_len(batched_intervals) + _calculate_audit_str_len(
-        batched_intervals
-    )
+    annotation_str_len = 0
+    for snapshot, intervals in batched_intervals.items():
+        annotation_str_len = max(
+            annotation_str_len,
+            _calculate_interval_str_len(snapshot, intervals) + _calculate_audit_str_len(snapshot),
+        )
+    return annotation_str_len

tests/cli/test_cli.py

@@ -182,7 +182,7 @@ def test_plan(runner, tmp_path):
     assert_plan_success(result)
     # 'Models needing backfill' section and eval progress bar should display the same inclusive intervals
     assert "sqlmesh_example.incremental_model: [2020-01-01 - 2022-12-31]" in result.output
-    assert "sqlmesh_example.incremental_model  [insert 2020-01-01 - 2022-12-31]" in result.output
+    assert "sqlmesh_example.incremental_model   [insert 2020-01-01 - 2022-12-31]" in result.output
 
 
 def test_plan_skip_tests(runner, tmp_path):
@@ -243,7 +243,7 @@ def test_plan_restate_model(runner, tmp_path):
     assert result.exit_code == 0
     assert_duckdb_test(result)
     assert "No changes to plan: project files match the `prod` environment" in result.output
-    assert "sqlmesh_example.full_model  [full refresh" in result.output
+    assert "sqlmesh_example.full_model   [full refresh" in result.output
     assert_model_batches_executed(result)
     assert_virtual_layer_updated(result)
 
@@ -568,8 +568,8 @@ def test_plan_nonbreaking(runner, tmp_path):
     assert "+  'a' AS new_col" in result.output
     assert "Directly Modified: sqlmesh_example.incremental_model (Non-breaking)" in result.output
     assert "sqlmesh_example.full_model (Indirect Non-breaking)" in result.output
-    assert "sqlmesh_example.incremental_model  [insert" in result.output
-    assert "sqlmesh_example.full_model  [full refresh" not in result.output
+    assert "sqlmesh_example.incremental_model   [insert" in result.output
+    assert "sqlmesh_example.full_model   [full refresh" not in result.output
     assert_backfill_success(result)
 
 
@@ -626,8 +626,8 @@ def test_plan_breaking(runner, tmp_path):
     assert result.exit_code == 0
     assert "+  item_id + 1 AS item_id," in result.output
     assert "Directly Modified: sqlmesh_example.full_model (Breaking)" in result.output
-    assert "sqlmesh_example.full_model  [full refresh" in result.output
-    assert "sqlmesh_example.incremental_model  [insert" not in result.output
+    assert "sqlmesh_example.full_model   [full refresh" in result.output
+    assert "sqlmesh_example.incremental_model   [insert" not in result.output
     assert_backfill_success(result)
 
 
@@ -665,8 +665,8 @@ def test_plan_dev_select(runner, tmp_path):
     assert "+  item_id + 1 AS item_id," not in result.output
     assert "Directly Modified: sqlmesh_example__dev.full_model (Breaking)" not in result.output
     # only incremental_model backfilled
-    assert "sqlmesh_example__dev.incremental_model  [insert" in result.output
-    assert "sqlmesh_example__dev.full_model  [full refresh" not in result.output
+    assert "sqlmesh_example__dev.incremental_model   [insert" in result.output
+    assert "sqlmesh_example__dev.full_model   [full refresh" not in result.output
     assert_backfill_success(result)
 
 
@@ -704,8 +704,8 @@ def test_plan_dev_backfill(runner, tmp_path):
         "Directly Modified: sqlmesh_example__dev.incremental_model (Non-breaking)" in result.output
     )
     # only incremental_model backfilled
-    assert "sqlmesh_example__dev.incremental_model  [insert" in result.output
-    assert "sqlmesh_example__dev.full_model  [full refresh" not in result.output
+    assert "sqlmesh_example__dev.incremental_model   [insert" in result.output
+    assert "sqlmesh_example__dev.full_model   [full refresh" not in result.output
     assert_backfill_success(result)