Adding key selection, handling headless mode and migrating to iterative

function

Author: zanjonke

partial_rendering.py

@@ -60,7 +60,6 @@ def module_comes_before_or_equal(
     module1: PlainModule,
     module2: PlainModule,
 ) -> bool:
-
     for module in all_required_modules:
         if module.module_name == module1.module_name:
             return True
@@ -120,7 +119,7 @@ def get_all_affected_modules_from_change(
     plain_module: PlainModule,
     plain_module_render_state: PlainModuleRenderState,
 ) -> list[PlainModule]:
-    all_affected_modules = list[PlainModule]()
+    all_affected_modules = dict[str, PlainModule]()
 
     if plain_module_render_state.change_type == "spec_change":
         start_module = plain_module_render_state.change
@@ -138,10 +137,10 @@ def get_all_affected_modules_from_change(
         if module.module_name == start_module.module_name:
             affected_module = True
 
-        if affected_module:
-            all_affected_modules.append(module)
+        if affected_module and module.module_name not in all_affected_modules:
+            all_affected_modules[module.module_name] = module
 
-    return all_affected_modules
+    return list(all_affected_modules.values())
 
 
 def get_render_choices(

plain2code.py

@@ -193,7 +193,7 @@ def render(args, run_state: RunState, event_bus: EventBus):  # noqa: C901
                 render_choice = render_choices[list(render_choices.keys())[0]]
                 ask_user = render_choice.is_destructive
 
-            if ask_user:
+            if ask_user and not args.headless:
                 app = PlainModuleRenderChoiceTUI(
                     plain_module,
                     plain_module_render_state,
@@ -209,6 +209,11 @@ def render(args, run_state: RunState, event_bus: EventBus):  # noqa: C901
                     and render_choice.choice_type == "quit"
                 ):
                     sys.exit(0)
+            elif ask_user and args.headless:
+                # ignore the default choice if it requires user input due to headless mode
+                # fallback to --render-from
+                # default choice is only used only when the action is not destructive
+                render_choice = None
 
     if render_choice is not None and render_range is not None:
         raise Exception("Partial rendering and render range cannot be used together")

plain_modules.py

@@ -71,15 +71,14 @@ def get_codeplain_folder(self):
         return os.path.join(self.module_build_folder, CODEPLAIN_METADATA_FOLDER)
 
     def get_module_render_status(self) -> tuple[str | None, str | None]:
-        if len(self.required_modules) == 0:
-            return git_utils.get_last_rendered_functionality(self.module_build_folder)
-
         module_name, frid = git_utils.get_last_rendered_functionality(self.module_build_folder)
         if module_name is not None and module_name == self.module_name:
             return module_name, frid
 
-        for module in reversed(self.required_modules):
-            last_rendered_module_name, last_rendered_frid = module.get_module_render_status()
+        for module in reversed(self.all_required_modules):
+            last_rendered_module_name, last_rendered_frid = git_utils.get_last_rendered_functionality(
+                module.module_build_folder
+            )
             if last_rendered_module_name is not None:
                 return last_rendered_module_name, last_rendered_frid
 

tui/plain_module_render_choice_tui.py

@@ -36,7 +36,14 @@ def __init__(
 
     def get_msg_from_choice(self, render_choice: RenderChoice) -> str:
         if render_choice.choice_type == "module_start":
-            return f"Start from module [#5593FF]{render_choice.module.module_name}[/]"
+            msg = ""
+            if render_choice.module.module_name == self.plain_module.module_name:
+                if render_choice.module.is_module_fully_rendered():
+                    return f"Re-render the current module ([#5593FF]{render_choice.module.module_name}[/])"
+                else:
+                    return f"Start the current module ([#5593FF]{render_choice.module.module_name}[/])"
+            else:
+                return f"Start from module [#5593FF]{render_choice.module.module_name}[/]"
         elif render_choice.choice_type == "rerender_affected" and self.plain_module_render_state.change is not None:
             all_affected_modules = get_all_affected_modules_from_change(
                 self.plain_module, self.plain_module_render_state
@@ -109,8 +116,7 @@ def on_mount(self) -> None:
             )
 
         elif pr.last_render_module.is_module_fully_rendered():
-            change_box.mount(Label("--- Rendering finished ---", classes="rendering-info-row"))
-            change_box.mount(Label("The current module was fully rendered.", classes="rendering-info-title"))
+            change_box.mount(Label("The current module is fully rendered.", classes="rendering-info-title"))
         else:
             interrupted_frid = "1"
             interrupted_module = pr.last_render_module
@@ -147,7 +153,7 @@ def on_mount(self) -> None:
 
         # Populate the ListView
         lv = self.query_one("#choice-list", ListView)
-        self.mount(Label("How would you like to continue?", classes="partial-render-question"), before=lv)
+        self.mount(Label("How would you like to proceed?", classes="partial-render-question"), before=lv)
         for key, choice in self.render_choices.items():
             lv.append(ListItem(Label(f"[bold]{key}.[/bold] {self.get_msg_from_choice(choice)}"), id=f"choice-{key}"))
         lv.focus()
@@ -170,6 +176,16 @@ def on_choice_selected(self, event: ListView.Selected) -> None:
         if not item_id or not item_id.startswith("choice-"):
             raise ValueError(f"Invalid item ID: {item_id}")
         key = item_id.split("-", 1)[1]
+        self._select_choice(key)
+
+    def on_key(self, event) -> None:
+        """Allow selecting a choice by typing its number directly."""
+        if event.key in self.render_choices:
+            event.stop()
+            event.prevent_default()
+            self._select_choice(event.key)
+
+    def _select_choice(self, key: str) -> None:
         self.selected_choice = self.render_choices[key]
         self.exit(self.selected_choice)