Skip to content

Commit caacc77

Browse files
committed
Refactor to support verbose mode
1 parent 05c5b07 commit caacc77

1 file changed

Lines changed: 38 additions & 26 deletions

File tree

sqlmesh/core/console.py

Lines changed: 38 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1135,39 +1135,51 @@ def update_signal_progress(
11351135
duration: float,
11361136
) -> None:
11371137
"""Updates the signal checking progress."""
1138-
# Format checked intervals
1139-
check_display = []
1140-
for interval in check_intervals[:3]: # Show first 3 intervals
1141-
interval_str = _format_signal_interval(snapshot, interval)
1142-
if interval_str:
1143-
check_display.append(interval_str)
1144-
if len(check_intervals) > 3:
1145-
check_display.append(f"... and {len(check_intervals) - 3} more")
1146-
1147-
# Format ready intervals
1148-
ready_display = []
1149-
for interval in ready_intervals[:3]: # Show first 3 intervals
1150-
interval_str = _format_signal_interval(snapshot, interval)
1151-
if interval_str:
1152-
ready_display.append(interval_str)
1153-
if len(ready_intervals) > 3:
1154-
ready_display.append(f"... and {len(ready_intervals) - 3} more")
1155-
1156-
# Display signal name
11571138
tree = Tree(f"[{signal_idx + 1}/{total_signals}] {signal_name} {duration:.2f}s")
11581139

1159-
check_str = ", ".join(check_display) if check_display else "no intervals"
1160-
tree.add(f"check: {check_str}")
1140+
formatted_check_intervals = [_format_signal_interval(snapshot, i) for i in check_intervals]
1141+
formatted_ready_intervals = [_format_signal_interval(snapshot, i) for i in ready_intervals]
11611142

1162-
ready_str = ", ".join(ready_display) if ready_display else "no intervals"
1143+
if not formatted_check_intervals:
1144+
formatted_check_intervals = ["no intervals"]
1145+
if not formatted_ready_intervals:
1146+
formatted_ready_intervals = ["no intervals"]
1147+
1148+
# Color coding to help detect partial interval ranges quickly
11631149
if ready_intervals == check_intervals:
1164-
ready_str = f"[green]ready: {ready_str}[/green]"
1150+
color = "green"
11651151
elif ready_intervals:
1166-
ready_str = f"[yellow]ready: {ready_str}[/yellow]"
1152+
color = "yellow"
11671153
else:
1168-
ready_str = f"[red]ready: {ready_str}[/red]"
1154+
color = "red"
1155+
1156+
if self.verbosity < Verbosity.VERY_VERBOSE:
1157+
num_check_intervals = len(formatted_check_intervals)
1158+
if num_check_intervals > 3:
1159+
formatted_check_intervals = formatted_check_intervals[:3]
1160+
formatted_check_intervals.append(f"... and {num_check_intervals - 3} more")
1161+
1162+
num_ready_intervals = len(formatted_ready_intervals)
1163+
if num_ready_intervals > 3:
1164+
formatted_ready_intervals = formatted_ready_intervals[:3]
1165+
formatted_ready_intervals.append(f"... and {num_ready_intervals - 3} more")
1166+
1167+
check = ", ".join(formatted_check_intervals)
1168+
tree.add(f"check: {check}")
1169+
1170+
ready = ", ".join(formatted_ready_intervals)
1171+
tree.add(f"[{color}]ready: {ready}[/{color}]")
1172+
else:
1173+
check_tree = Tree("check")
1174+
tree.add(check_tree)
1175+
for interval in formatted_check_intervals:
1176+
check_tree.add(interval)
1177+
1178+
ready_tree = Tree(f"[{color}]ready[/{color}]")
1179+
tree.add(ready_tree)
1180+
for interval in formatted_ready_intervals:
1181+
ready_tree.add(interval)
11691182

1170-
tree.add(ready_str)
11711183
if self.signal_status_tree is not None:
11721184
self.signal_status_tree.add(tree)
11731185

0 commit comments

Comments
 (0)