试图移植

2025-11-02 23:42:53 +08:00
parent f86187868b
commit 0e08fb3a41
18 changed files with 689 additions and 156 deletions
--- a/src/heurams/interface/widgets/recognition.py
+++ b/src/heurams/interface/widgets/recognition.py
@@ -0,0 +1,74 @@
+from textual.app import App, ComposeResult
+from textual.events import Event
+from textual.widgets import (
+    Collapsible,
+    Header,
+    Footer,
+    Markdown,
+    ListView,
+    ListItem,
+    Label,
+    Static,
+    Button,
+)
+from textual.containers import Container, Horizontal, Center
+from textual.screen import Screen
+from textual.widget import Widget
+import uuid
+from typing import Tuple, Dict
+import heurams.kernel.particles as pt
+import heurams.kernel.puzzles as pz
+import re
+import random
+import copy
+
+class Recognition(Composition):
+    def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict], extra = {}):
+        super().__init__(screen, reactor, atom)
+
+    def compose(self):
+        with Center():
+            yield Static(f"[dim]{self.atom[1]['translation']}[/]")
+            yield Label(f"")
+        s = str(self.atom[1]["content"])
+        replace_dict = {
+            ", ": ",",
+            ". ": ".",
+            "; ": ";",
+            ": ": ":",
+            "/,": ",",
+            "./": ".",
+            "/;": ";",
+            ";/": ";",
+            ":/": ":",
+        }
+        for old, new in replace_dict.items():
+            s = s.replace(old, new)
+        result = re.split(r"(?<=[,;:|])", s.replace("/", " "))
+        for i in result:
+            with Center():
+                yield Label(
+                    f"[b][b]{i.replace('/', ' ')}[/][/]",
+                    id=self.regid("sentence" + str(hash(i))),
+                )
+        for i in self.atom[2]["testdata"]["additional_inf"]:
+            if self.atom[1][i]:
+                if isinstance(self.atom[1][i], list):
+                    for j in self.atom[1][i]:
+                        yield Markdown(f"### {self.atom[2]['keydata'][i]}: {j}")
+                        continue
+                if isinstance(self.atom[1][i], Dict):
+                    t = ""
+                    for j, k in self.atom[1][i].items(): # type: ignore
+                        # 弱智的 Pylance 类型推导 
+                        t += f"> **{j}**: {k}  \n"
+                    yield Markdown(t, id=self.regid("tran"))
+        with Center():
+            yield Button("我已知晓", id=self.regid("ok"))
+
+    def handler(self, event, type_):
+        if type_ == "button":
+            if event.button.id == self.getid("ok"):
+                self.reactor.report(self.atom, 5)
+                return 0
+        return -1