Ask user when the only choice is destruction

Author: zanjonke

partial_rendering.py

@@ -20,6 +20,7 @@ class PartialRenderChoice:
     render_range: list[str] | None = None
     msg: str | None = None
     wipe_later_modules: bool = False
+    is_desctructive: bool = False
 
 
 def spec_change(plain_module: PlainModule) -> PlainModule | None:
@@ -78,13 +79,7 @@ def detect_partial_rendering(plain_module: PlainModule) -> PartialRender | None:
         return None
 
     if last_rendered_module_name == plain_module.module_name:
-        if (
-            last_rendered_frid is None
-            or plain_spec.get_next_frid(plain_module.plain_source, last_rendered_frid) is not None
-        ):
-            module = plain_module
-        else:
-            return None
+        module = plain_module
     else:
         found_module: PlainModule | None = None
         for required_module in all_required_modules:
@@ -135,6 +130,7 @@ def get_choices(
             render_range=None,
             msg=f"Start from module [#5593FF]{partial_render.last_render_module.module_name}[/]",
             wipe_later_modules=True,
+            is_desctructive=False,
         )
         choice_idx += 1
 
@@ -169,6 +165,7 @@ def get_choices(
             module=next_module,
             render_range=None,
             msg=f"Start from module [#5593FF]{next_module.module_name}[/]",
+            is_desctructive=partial_render.last_render_module.module_name == plain_module.module_name,
         )
         choice_idx += 1
 
@@ -187,6 +184,7 @@ def get_choices(
                 render_range=None,
                 msg=f"Re-render all affected modules ([#5593FF]{', '.join(all_affected_modules)}[/])",
                 wipe_later_modules=True,
+                is_desctructive=True,
             )
             choice_idx += 1
 
@@ -200,10 +198,9 @@ def get_choices(
                 render_range=None,
                 msg=f"Re-render from first module ([#5593FF]{first_module.module_name}[/])",
                 wipe_later_modules=True,
+                is_desctructive=True,
             )
             choice_idx += 1
 
     choices[str(choice_idx)] = PartialRenderChoice(module=None, render_range=None, msg="Quit")
-    print(choices)
-
     return choices

plain2code.py

@@ -186,7 +186,14 @@ def render(args, run_state: RunState, event_bus: EventBus):  # noqa: C901
         partial_render = detect_partial_rendering(plain_module)
         if partial_render is not None:
             choices = get_choices(plain_module, partial_render, args.force_render)
-            if len(choices) > 2:
+            ask_user = True
+            partial_render_choice = None
+            if len(choices) <= 2:
+                # Last choice is Quit, first choice is the only other actionable choice
+                partial_render_choice = choices[list(choices.keys())[0]]
+                ask_user = partial_render_choice.is_desctructive
+
+            if ask_user:
                 app = PartialRenderTUI(
                     plain_module,
                     partial_render,
@@ -202,9 +209,6 @@ def render(args, run_state: RunState, event_bus: EventBus):  # noqa: C901
                     and partial_render_choice.msg == "Quit"
                 ):
                     sys.exit(0)
-            else:
-                # Last choice is Quit, first choice is the only other actionable choice
-                partial_render_choice = choices[list(choices.keys())[0]]
 
     if partial_render_choice is not None and render_range is not None:
         raise Exception("Partial rendering and render range cannot be used together")

tui/partial_render_tui.py

@@ -128,7 +128,7 @@ def on_mount(self) -> None:
         lv.focus()
 
     def _register_expandable(self, label: Label, prefix: str, full_text: str) -> None:
-        first_line = full_text[:20]
+        first_line = full_text[:100]
         short = f"{prefix} {first_line} [#888](ctrl+o to expand)[/]"
         full = f"{prefix} {full_text} [#888](ctrl+o to collapse)[/]"
         label.update(short)