试图移植

src/heurams/interface/screens/preparation.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+from textual.app import ComposeResult
+from textual.widgets import (
+    Header,
+    Footer,
+    Label,
+    Static,
+    Button,
+    Markdown,
+)
+from textual.containers import Container
+from textual.screen import Screen
+
+import heurams.kernel.particles as pt
+import heurams.services.hasher as hasher
+from heurams.context import *
+
+class PreparationScreen(Screen):
+    BINDINGS = [
+        ("q", "go_back", "返回"),
+        ("escape", "quit_app", "退出"),
+        ("p", "precache", "预缓存音频")
+    ]
+
+    def __init__(
+        self, nucleon_file: pathlib.Path, electron_file: pathlib.Path
+    ) -> None:
+        super().__init__(name=None, id=None, classes=None)
+        self.nucleon_file = nucleon_file
+        self.electron_file = electron_file
+        self.nucleon_union = pt.load_nucleon(self.nucleon_file)
+
+    def compose(self) -> ComposeResult:
+        yield Header(show_clock=True)
+        with Container(id="vice_container"):
+            yield Label(f"准备就绪: [b]{self.nucleon_file.stem}[/b]\n")
+            yield Label(f"内容源文件对象: ./nucleon/[b]{self.nucleon_file.name}[/b].toml")
+            yield Label(f"元数据文件对象: ./electron/[b]{self.electron_file.name}[/b].toml")
+            yield Label(f"\n单元数量:{len(self.nucleon_union)}\n")
+
+            yield Button(
+                    "开始记忆",
+                    id="start_memorizing_button",
+                    variant="primary",
+                    classes="start-button",
+            )
+            yield Button(
+                "预缓存音频",
+                id="precache_button",
+                variant="success",
+                classes="precache-button",
+            )
+
+            yield Static(f"\n单元预览:\n")
+            yield Markdown(self._get_full_content().replace("/", ""), classes="full")
+        yield Footer()
+
+    def _get_full_content(self):
+        content = ""
+        for i in self.nucleon_union:
+            content += " - " + i["content"] + "  \n"
+        return content
+
+    def action_go_back(self):
+        self.app.pop_screen()
+
+    def action_precache(self):
+        from ..screens.precache import PrecachingScreen
+        precache_screen = PrecachingScreen(self.nucleon_file)
+        self.app.push_screen(precache_screen)
+
+    def action_quit_app(self):
+        self.app.exit()
+
+    def on_button_pressed(self, event) -> None:
+        if event.button.id == "start_memorizing_button":
+            from .memorizor import MemScreen
+            newscr = MemScreen(
+                self.nucleon_file, self.electron_file, 6
+            )
+            self.app.push_screen(newscr)
+        elif event.button.id == "precache_button":
+            self.action_precache()