规范代码

auxiliary.py

@@ -1,33 +1,46 @@
 import time
 import pathlib
 import toml
+import typing
 
-class ConfigFile():
-    def __init__(self, path):
+class ConfigFile:
+    def __init__(self, path: str):
         self.path = pathlib.Path(path)
-        if self.path.exists() == 0:
+        if not self.path.exists():
             self.path.touch()
         self.data = dict()
+        self._load()
+
+    def _load(self):
+        """从文件加载配置数据"""
         with open(self.path, 'r') as f:
-            self.data = toml.load(f)
-    def modify(self, key, value):
+            try:
+                self.data = toml.load(f)
+            except toml.TomlDecodeError:
+                self.data = {}
+
+    def modify(self, key: str, value: typing.Any):
+        """修改配置值并保存"""
         self.data[key] = value
         self.save()
-    def save(self, path=""):
-        if path == "":
-            path = self.path
-        with open(path, 'w') as f:
+
+    def save(self, path: typing.Union[str, pathlib.Path] = ""):
+        """保存配置到文件"""
+        save_path = pathlib.Path(path) if path else self.path
+        with open(save_path, 'w') as f:
             toml.dump(self.data, f)
-    def get(self, key, default = None):
+
+    def get(self, key: str, default: typing.Any = None) -> typing.Any:
+        """获取配置值，如果不存在返回默认值"""
         return self.data.get(key, default)
 
-def get_daystamp() -> int:
-    config = ConfigFile("config.toml")
 
+def get_daystamp() -> int:
+    """获取当前日戳(以天为单位的整数时间戳)"""
+    config = ConfigFile("config.toml")
     time_override = config.get("time_override", -1)
     
-    if time_override is not None and time_override != -1:
-        #print(f"TIME OVERRIDEED TO {time_override}")
+    if time_override != -1:
         return int(time_override)
     
     return int(time.time() // (24 * 3600))

compositions.py

@@ -1,6 +1,16 @@
 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.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
@@ -12,56 +22,70 @@ import re
 import random
 import copy
 
-class Composition():
-    def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict] = pt.Atom.placeholder()):
+
+class Composition:
+    def __init__(
+        self,
+        screen: Screen,
+        reactor,
+        atom: Tuple[pt.Electron, pt.Nucleon, Dict] = pt.Atom.placeholder(),
+    ):
         self.screen = screen
         self.atom = atom
         from reactor import Reactor
+
         self.reactor: Reactor = reactor
         self.reg = dict()
+
     def regid(self, id_):
         self.reg[id_] = id_ + str(uuid.uuid4())
         return self.reg[id_]
+
     def getid(self, id_):
         if id_ not in self.reg.keys():
             return "None"
         return self.reg[id_]
+
     def recid(self, id_):
         return id_[:-36]
+
     def compose(self):
         yield Label("示例标签", id="testlabel")
         yield Button("示例按钮", id="testbtn")
+
     def handler(self, event, type_):
         return 1
-        #if hasattr(event, "button"):
-            #print(event.button.id)
-        #    self.screen.query_one("#testlabel", Label).update("hi")
+
 
 class Finished(Composition):
     def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict]):
         super().__init__(screen, reactor, atom)
+
     def compose(self):
         yield Label("本次记忆进程结束", id=self.regid("msg"))
-        #yield Button("示例按钮", id="testbtn")
+
 
 class Placeholder(Composition):
     def __init__(self, screen: Screen):
         self.screen = screen
+
     def compose(self):
         yield Label("示例标签", id="testlabel")
         yield Button("示例按钮", id="testbtn", classes="choice")
+
     def handler(self, event, type_):
-        #print(event.button.id)
         self.screen.query_one("#testlabel", Label).update("hi")
 
+
 class Recognition(Composition):
     def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict]):
         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'])
+        s = str(self.atom[1]["content"])
         replace_dict = {
             ", ": ",",
             ". ": ".",
@@ -75,132 +99,153 @@ class Recognition(Composition):
         }
         for old, new in replace_dict.items():
             s = s.replace(old, new)
-        result = re.split(r"(?<=[,;:|])", s.replace('/', ' '))
+        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)))) # 致敬传奇去重串 uuid
-        #with Collapsible(title="附加信息", collapsed=True):
+                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]:
-                #print(type(self.atom[1][i]))
-                #print(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():
+                    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_):
-        ##print(event)
         if type_ == "button":
-            #print(event.button.id)
             if event.button.id == self.getid("ok"):
-                #print(1)
-                """"""#assessment = self.reactor.report(self.reactor.current_atom, -1)
-            return 0
+                return 0
         if type_ == 1:
             pass
         return -1
 
+
 class BasicEvaluation(Composition):
-    # 不再使用, 仅作为测试
     def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict]):
         super().__init__(screen, reactor, atom)
+
     def compose(self):
         yield Label(self.atom[1]["content"], id="sentence")
         with Container(id="button_container"):
             btn = {}
-            btn['5'] = Button("完美回想", variant="success", id=self.regid("feedback5"), classes="choice")
-            btn['4'] = Button("犹豫后正确", variant="success", id=self.regid("feedback4"), classes="choice")
-            btn['3'] = Button("困难地正确", variant="warning", id=self.regid("feedback3"), classes="choice")
-            btn['2'] = Button("错误但熟悉", variant="warning", id=self.regid("feedback2"), classes="choice")
-            btn['1'] = Button("错误且不熟", variant="error", id=self.regid("feedback1"), classes="choice")
-            btn['0'] = Button("完全空白", variant="error", id=self.regid("feedback0"), classes="choice")
-            yield Horizontal(btn['5'], btn['4'])
-            yield Horizontal(btn['3'], btn['2'])
-            yield Horizontal(btn['1'], btn['0'])
+            btn["5"] = Button(
+                "完美回想", variant="success", id=self.regid("feedback5"), classes="choice"
+            )
+            btn["4"] = Button(
+                "犹豫后正确", variant="success", id=self.regid("feedback4"), classes="choice"
+            )
+            btn["3"] = Button(
+                "困难地正确", variant="warning", id=self.regid("feedback3"), classes="choice"
+            )
+            btn["2"] = Button(
+                "错误但熟悉", variant="warning", id=self.regid("feedback2"), classes="choice"
+            )
+            btn["1"] = Button(
+                "错误且不熟", variant="error", id=self.regid("feedback1"), classes="choice"
+            )
+            btn["0"] = Button(
+                "完全空白", variant="error", id=self.regid("feedback0"), classes="choice"
+            )
+            yield Horizontal(btn["5"], btn["4"])
+            yield Horizontal(btn["3"], btn["2"])
+            yield Horizontal(btn["1"], btn["0"])
+
     def handler(self, event, type_):
         if "feedback" in event.button.id:
-            #print(self.recid(event.button.id)[8:9])
             assess = int(self.recid(event.button.id)[8:9])
             ret = self.reactor.report(self.atom, assess)
             return ret
 
+
 class FillBlank(Composition):
     def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict]):
         super().__init__(screen, reactor, atom)
         self.inputlist = []
         self.hashtable = {}
         self._work()
+
     def _work(self):
         self.puzzle = pz.BlankPuzzle(self.atom[1]["content"], 4)
         self.puzzle.refresh()
         self.ans = copy.copy(self.puzzle.answer)
         random.shuffle(self.ans)
+
     def compose(self):
         yield Label(self.puzzle.wording, id=self.regid("sentence"))
         yield Label(f"当前输入: {self.inputlist}", id=self.regid("inputpreview"))
-        #yield Label(renderable=f"答案: {self.puzzle.answer}", id=self.regid("ans"))
         for i in self.ans:
             self.hashtable[str(hash(i))] = i
             yield Button(i, id=self.regid(f"select{hash(i)}"))
         yield Button("退格", id=self.regid(f"delete"))
+
     def handler(self, event, type_):
         if type_ == "button":
-            if self.recid(event.button.id) == "delete" and len(self.inputlist) > 0:
-                self.inputlist.pop()
+            if self.recid(event.button.id) == "delete":
+                if len(self.inputlist) > 0:
+                    self.inputlist.pop()
+                else:
+                    return 1
             else:
                 self.inputlist.append(self.hashtable[self.recid(event.button.id)[6:]])
             if len(self.inputlist) < len(self.puzzle.answer):
                 return 1
             else:
                 if self.inputlist == self.puzzle.answer:
-                    print("ok")
                     return 0
                 else:
                     self.inputlist = []
                     return 1
 
+
 class DrawCard(Composition):
     def __init__(self, screen: Screen, reactor, atom: Tuple[pt.Electron, pt.Nucleon, Dict]):
         super().__init__(screen, reactor, atom)
         self.inputlist = []
         self.hashtable = {}
         self._work()
+
     def _work(self):
-        self.puzzle = pz.SelectionPuzzle(self.atom[1]["keyword_note"], [], 2, "选择正确词义: ")
+        self.puzzle = pz.SelectionPuzzle(self.atom[1]["keyword_note"], [], 2, "选择正确词义: ")  # type: ignore
         self.puzzle.refresh()
+
     def compose(self):
-        print(len(self.inputlist))
-        yield Label(self.atom[1].content, id=self.regid("sentence"))
+        yield Label(self.atom[1].content.replace("/",""), id=self.regid("sentence"))
         yield Label(self.puzzle.wording[len(self.inputlist)], id=self.regid("puzzle"))
         yield Label(f"当前输入: {self.inputlist}", id=self.regid("inputpreview"))
-        yield Label(renderable=f"答案: {self.puzzle.answer}", id=self.regid("ans"))
         for i in self.puzzle.options[len(self.inputlist)]:
             self.hashtable[str(hash(i))] = i
             yield Button(i, id=self.regid(f"select{hash(i)}"))
         yield Button("退格", id=self.regid(f"delete"))
+
     def handler(self, event, type_):
         if type_ == "button":
-            if self.recid(event.button.id) == "delete" and len(self.inputlist) > 0:
-                self.inputlist.pop()
+            if self.recid(event.button.id) == "delete":
+                if len(self.inputlist) > 0:
+                    self.inputlist.pop()
+                else:
+                    return 1
             else:
                 self.inputlist.append(self.hashtable[self.recid(event.button.id)[6:]])
             if len(self.inputlist) < len(self.puzzle.answer):
                 return 1
             else:
                 if self.inputlist == self.puzzle.answer:
-                    print("ok")
                     return 0
                 else:
                     self.inputlist = []
                     return 1
 
+
 registry = {
     "sample": Composition,
     "recognition": Recognition,
@@ -210,16 +255,16 @@ registry = {
 }
 
 
-# TEST
-
 class TestScreen(Screen):
     def __init__(self):
         super().__init__(name=None, id=None, classes=None)
         self.comp = Recognition(self, None, pt.Atom.advanced_placeholder())
+
     def compose(self) -> ComposeResult:
         yield Header(show_clock=True)
         yield from self.comp.compose()
         yield Footer()
+
     def on_mount(self) -> None:
         pass
 
@@ -229,9 +274,10 @@ class TestScreen(Screen):
     def action_quit_app(self) -> None:
         self.app.exit()
 
+
 class AppLauncher(App):
     CSS_PATH = "styles.tcss"
-    TITLE = '测试布局'
+    TITLE = "测试布局"
     BINDINGS = [("escape", "quit", "退出"), ("d", "toggle_dark", "改变色调")]
     SCREENS = {
         "testscreen": TestScreen,
@@ -241,6 +287,7 @@ class AppLauncher(App):
         self.action_toggle_dark()
         self.push_screen("testscreen")
 
+
 if __name__ == "__main__":
     app = AppLauncher()
     app.run()

main.py

@@ -1,35 +1,43 @@
 from textual.app import App, ComposeResult
-from textual.widgets import Header, Footer, ListView, ProgressBar, DirectoryTree, ListItem, Label, Static, Button
+from textual.widgets import (
+    Header,
+    Footer,
+    ListView,
+    ProgressBar,
+    DirectoryTree,
+    ListItem,
+    Label,
+    Static,
+    Button,
+)
 from textual.containers import Container, Horizontal, Center
 from textual.screen import Screen
 import pathlib
 import threading
 import edge_tts as tts
 from playsound import playsound
-
 import particles as pt
 from reactor import Reactor, Apparatus
 import auxiliary as aux
 import compositions as compo
-
 import builtins
 
-# 保存原始的 open 函数
 _original_open = builtins.open
 
-# 定义新的 open 函数，默认使用 UTF-8
+
 def _open(*args, **kwargs):
-    if 'encoding' not in kwargs:
-        kwargs['encoding'] = 'utf-8'
+    if "encoding" not in kwargs:
+        kwargs["encoding"] = "utf-8"
     return _original_open(*args, **kwargs)
 
-# 替换全局的 open
+
 builtins.open = _open
 
-ver = '0.3.0b'
+ver = "0.3.0b"
 
 config = aux.ConfigFile("config.toml")
 
+
 class MemScreen(Screen):
     BINDINGS = [
         ("d", "toggle_dark", "改变色调"),
@@ -49,38 +57,30 @@ class MemScreen(Screen):
         ("/", "press('q0')", None),
     ]
     btn = dict()
+
     def __init__(
         self,
         nucleon_file: pt.NucleonUnion,
         electron_file: pt.ElectronUnion,
-        tasked_num
+        tasked_num,
     ):
         super().__init__(name=None, id=None, classes=None)
         self.reactor = Reactor(nucleon_file, electron_file, self, tasked_num)
         self.stage = 1
         self.stage += self.reactor.set_round_templated(self.stage)
-        ##print(self.reactor.procession)
         self.reactor.forward()
-        #self.compo:compo.Composition = compo.Placeholder(self)
         self.compo = next(self.reactor.current_appar)
 
-
     def compose(self) -> ComposeResult:
-        #print(self.compo)
         yield Header(show_clock=True)
         with Center():
-            yield Static(f"{len(self.reactor.procession) - self.reactor.index}/{len(self.reactor.procession)}")
-            #yield ProgressBar(total=len(self.reactor.procession) - 1, show_percentage=False, show_eta=False, id="progress")
+            yield Static(
+                f"{len(self.reactor.procession) - self.reactor.index}/{len(self.reactor.procession)}"
+            )
         yield from self.compo.compose()
         yield Footer()
 
-    """    
-    def _get_progress_text(self):
-        return f"{len(self.reactor.procession) - self.reactor.index}/{len(self.reactor.procession)}"
-    """
-
     def on_mount(self):
-        # 首次挂载时调用
         pass
 
     def on_button_pressed(self, event):
@@ -90,13 +90,12 @@ class MemScreen(Screen):
     def _forward_judge(self, ret):
         if ret == -1:
             return
-        if ret == 0: # 成功
+        if ret == 0:
             try:
                 self.compo = next(self.reactor.current_appar)
                 self.refresh_ui()
             except StopIteration:
                 nxt = self.reactor.forward(1)
-                #print(2)
                 try:
                     self.compo = next(self.reactor.current_appar)
                 except:
@@ -106,13 +105,13 @@ class MemScreen(Screen):
                         if self.stage == 4:
                             if config.get("save"):
                                 self.reactor.save()
-                            self.compo = compo.Finished(self, None, pt.Atom.placeholder())
+                            self.compo = compo.Finished(
+                                self, None, pt.Atom.placeholder()
+                            )
                             self.refresh_ui()
-                            #self._show_finished_screen("今日目标已完成")
                         else:
                             self.reactor.set_round_templated(self.stage)
                             self.reactor.forward(1)
-                            #self._update_ui()
                             self.stage += 1
                             self.compo = next(self.reactor.current_appar)
                             self.refresh_ui()
@@ -121,71 +120,47 @@ class MemScreen(Screen):
                 else:
                     self.refresh_ui()
                     return
-        if ret == 1: # 不允许前进
+        if ret == 1:
             self.refresh_ui()
             return
 
     def refresh_ui(self):
         self.call_later(self.recompose)
-        #self.call_later(lambda: self.query_one("#progress", expect_type=ProgressBar).advance(self.reactor.index))
-        ##print(area.children)
-        #for child in list(area.children):
-        #    child.remove() # 致敬传奇组件树 DOM
-        ##print(1,list(self.compo.compose()))
-        #area.mount(*list(self.compo.compose()))
 
     def report(self, quality):
         assessment = self.reactor.report(self.reactor.current_atom, quality)
         return assessment
-        """if assessment == 1:
-            # 需要复习
-            feedback_label.update(f"评分为 {quality}, 已经加入至复习, 请重复记忆")
-        else:
-            ret = self.reactor.forward(1)
-            if ret == -1:
-                if self.reactor.round_set == 0:
-                    if self.stage == 4:
-                        # NOTE #
-                        if config.get("save"):
-                            self.reactor.save()
-                        self._show_finished_screen("今日目标已完成")
-                    else:
-                        self.reactor.set_round_templated(self.stage)
-                        self.reactor.forward(1)
-                        self._update_ui()
-                        self.stage += 1
-                return
-        #feedback_label.update("") # 清除反馈消息
-            self._update_ui()"""
-
-    #def action_press(self, btnid):
-    #    self.on_button_pressed(btnid)
 
     def action_play_voice(self):
         def play():
             cache_dir = pathlib.Path(f"./cache/voice/")
-            cache_dir.mkdir(parents = True, exist_ok = True)
-            cache = cache_dir / f"{self.reactor.current_atom[1].content.replace("/","")}.wav"
+            cache_dir.mkdir(parents=True, exist_ok=True)
+            cache = cache_dir / f"{self.reactor.current_atom[1].content.replace('/','')}.wav"
             if not cache.exists():
-                communicate = tts.Communicate(self.reactor.current_atom[1].content.replace("/",""), "zh-CN-YunjianNeural")
-                communicate.save_sync(f"./cache/voice/{self.reactor.current_atom[1].content.replace("/","")}.wav")
+                communicate = tts.Communicate(
+                    self.reactor.current_atom[1].content.replace("/", ""),
+                    "zh-CN-YunjianNeural",
+                )
+                communicate.save_sync(
+                    f"./cache/voice/{self.reactor.current_atom[1].content.replace('/','')}.wav"
+                )
             playsound(str(cache))
+
         threading.Thread(target=play).start()
 
     def action_toggle_dark(self):
         self.app.action_toggle_dark()
 
     def action_pop_screen(self):
-        """返回到上一个屏幕"""
         self.app.pop_screen()
 
-class PreparationScreen(Screen):
-    BINDINGS = [
-        ("q", "go_back", "返回"),
-        ("escape", "quit_app", "退出")
-    ]
 
-    def __init__(self, nucleon_file: pt.NucleonUnion, electron_file: pt.ElectronUnion) -> None:
+class PreparationScreen(Screen):
+    BINDINGS = [("q", "go_back", "返回"), ("escape", "quit_app", "退出")]
+
+    def __init__(
+        self, nucleon_file: pt.NucleonUnion, electron_file: pt.ElectronUnion
+    ) -> None:
         super().__init__(name=None, id=None, classes=None)
         self.nucleon_file = nucleon_file
         self.electron_file = electron_file
@@ -194,10 +169,19 @@ class PreparationScreen(Screen):
         yield Header(show_clock=True)
         with Container(id="learning_screen_container"):
             yield Label(f"记忆项目: [b]{self.nucleon_file.name}[/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"核子文件对象: ./nucleon/[b]{self.nucleon_file.name}[/b].toml"
+            )
+            yield Label(
+                f"电子文件对象: ./electron/[b]{self.electron_file.name}[/b].toml"
+            )
             yield Label(f"核子数量:{len(self.nucleon_file)}")
-            yield Button("开始记忆", id="start_memorizing_button", variant="primary", classes="start-button")
+            yield Button(
+                "开始记忆",
+                id="start_memorizing_button",
+                variant="primary",
+                classes="start-button",
+            )
             yield Static(f"\n全文如下:\n")
             yield Static(self._get_full_content(), classes="full")
         yield Footer()
@@ -205,7 +189,7 @@ class PreparationScreen(Screen):
     def _get_full_content(self):
         content = ""
         for i in self.nucleon_file.nucleons:
-            content += i['content']
+            content += i["content"]
         return content
 
     def action_go_back(self):
@@ -215,76 +199,79 @@ class PreparationScreen(Screen):
         self.app.exit()
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
-        pass
         if event.button.id == "start_memorizing_button":
-            #init_file(Path(self.atom_file).name)
-            newscr = MemScreen(self.nucleon_file, self.electron_file, config.get("tasked_number", 8))
-            self.app.push_screen(
-                newscr
+            newscr = MemScreen(
+                self.nucleon_file, self.electron_file, config.get("tasked_number", 8)
             )
-        #if event.button.id == "edit_metadata_button":
-        #    init_file(Path(self.atom_file).name)
-        #    os.system("reset;nano ./data/" + str(Path(self.atom_file).name.replace(".txt", "_atoms.json")))
+            self.app.push_screen(newscr)
+
 
 class FileSelectorScreen(Screen):
     global ver
+
     def compose(self) -> ComposeResult:
         yield Header(show_clock=True)
         yield Container(
             Label(f'欢迎使用 "潜进" 辅助记忆软件, 版本 {ver}', classes="title-label"),
             Label("选择要学习的文件:", classes="title-label"),
-            ListView(id="file-list", classes="file-list-view")
+            ListView(id="file-list", classes="file-list-view"),
         )
         yield Footer()
 
     def on_mount(self) -> None:
         file_list_widget = self.query_one("#file-list", ListView)
         nucleon_path = pathlib.Path("./nucleon")
-        nucleon_files = sorted([f.name for f in nucleon_path.iterdir() if f.suffix == ".toml"])
+        nucleon_files = sorted(
+            [f.name for f in nucleon_path.iterdir() if f.suffix == ".toml"]
+        )
 
         if nucleon_files:
             for filename in nucleon_files:
                 file_list_widget.append(ListItem(Label(filename)))
         else:
-            file_list_widget.append(ListItem(Static("在 ./nucleon/ 中未找到任何核子文件. 请放置文件后重启应用.")))
+            file_list_widget.append(
+                ListItem(Static("在 ./nucleon/ 中未找到任何核子文件. 请放置文件后重启应用."))
+            )
             file_list_widget.disabled = True
 
     def on_list_view_selected(self, event: ListView.Selected) -> None:
         if not isinstance(event.item, ListItem):
-            self.notify("无法选择此项。", severity="error")
             return
 
         selected_label = event.item.query_one(Label)
         if "未找到任何 .toml 文件" in str(selected_label.renderable):
-            self.notify("请先在 `./atoms/` 目录中放置 .toml 文件。", severity="warning")
             return
 
         selected_filename = str(selected_label.renderable)
-        nucleon_file = pt.NucleonUnion(pathlib.Path("./nucleon") / selected_filename)
+        nucleon_file = pt.NucleonUnion(
+            pathlib.Path("./nucleon") / selected_filename
+        )
         electron_file_path = pathlib.Path("./electron") / selected_filename
         if electron_file_path.exists():
             pass
         else:
             electron_file_path.touch()
-        electron_file = pt.ElectronUnion(pathlib.Path("./electron") / selected_filename)
-        # self.notify(f"已选择: {selected_filename}", timeout=2)
+        electron_file = pt.ElectronUnion(
+            pathlib.Path("./electron") / selected_filename
+        )
         self.app.push_screen(PreparationScreen(nucleon_file, electron_file))
 
     def action_quit_app(self) -> None:
         self.app.exit()
 
+
 class AppLauncher(App):
     CSS_PATH = "styles.tcss"
-    TITLE = '潜进 - 辅助记忆程序'
+    TITLE = "潜进 - 辅助记忆程序"
     BINDINGS = [("escape", "quit", "退出"), ("d", "toggle_dark", "改变色调")]
     SCREENS = {
         "file_selection_screen": FileSelectorScreen,
     }
 
     def on_mount(self) -> None:
-        #self.action_toggle_dark()
         self.push_screen("file_selection_screen")
 
+
 if __name__ == "__main__":
     app = AppLauncher()
     app.run()

particles.py

@@ -3,30 +3,29 @@ import toml
 import time
 import auxiliary as aux
 
-import time
 
-class Electron():
+class Electron:
     """电子: 记忆分析元数据及算法"""
-    algorithm = "SM-2" # 暂时使用 SM-2 算法进行记忆拟合, 考虑 SM-15 替代
+    algorithm = "SM-2"  # 暂时使用 SM-2 算法进行记忆拟合, 考虑 SM-15 替代
 
     def __init__(self, content: str, metadata: dict):
         self.content = content
         self.metadata = metadata
         if metadata == {}:
-            #print("NULL")
+            # print("NULL")
             self._default_init()
 
     def _default_init(self):
         defaults = {
-            'efactor': 2.5, # 易度系数, 越大越简单, 最大为5 
-            'real_rept': 0, # (实际)重复次数
-            'rept': 0, # (有效)重复次数
-            'interval': 0, # 最佳间隔
-            'last_date': 0, # 上一次复习的时间戳
-            'next_date': 0, # 将要复习的时间戳
-            'is_activated': 0, # 激活状态
+            'efactor': 2.5,  # 易度系数, 越大越简单, 最大为5
+            'real_rept': 0,  # (实际)重复次数
+            'rept': 0,  # (有效)重复次数
+            'interval': 0,  # 最佳间隔
+            'last_date': 0,  # 上一次复习的时间戳
+            'next_date': 0,  # 将要复习的时间戳
+            'is_activated': 0,  # 激活状态
             # *NOTE: 此处"时间戳"是以天为单位的整数, 即 UNIX 时间戳除以一天的秒数取整
-            'last_modify': time.time() # 最后修改时间戳(此处是UNIX时间戳)
+            'last_modify': time.time()  # 最后修改时间戳(此处是UNIX时间戳)
         }
         self.metadata = defaults
 
@@ -52,40 +51,46 @@ class Electron():
         if quality == -1:
             return -1
 
-        self.metadata['efactor'] = self.metadata['efactor'] + (0.1 - (5 - quality) * (0.08 + (5 - quality) * 0.02))
+        self.metadata['efactor'] = self.metadata['efactor'] + (
+            0.1 - (5 - quality) * (0.08 + (5 - quality) * 0.02)
+        )
         self.metadata['efactor'] = max(1.3, self.metadata['efactor'])
 
         if quality < 3:
             # 若保留率低于 3，重置重复次数
             self.metadata['rept'] = 0
-            self.metadata['interval'] = 0 # 设为0，以便下面重新计算 I(1)
+            self.metadata['interval'] = 0  # 设为0，以便下面重新计算 I(1)
         else:
             self.metadata['rept'] += 1
 
         self.metadata['real_rept'] += 1
 
-        if is_new_activation: # 初次激活
+        if is_new_activation:  # 初次激活
             self.metadata['rept'] = 0
             self.metadata['efactor'] = 2.5
 
-        if self.metadata['rept'] == 0: # 刚被重置或初次激活后复习
-            self.metadata['interval'] = 1 # I(1) 
+        if self.metadata['rept'] == 0:  # 刚被重置或初次激活后复习
+            self.metadata['interval'] = 1  # I(1)
         elif self.metadata['rept'] == 1:
-            self.metadata['interval'] = 6 # I(2) 经验公式
+            self.metadata['interval'] = 6  # I(2) 经验公式
         else:
-            self.metadata['interval'] = round(self.metadata['interval'] * self.metadata['efactor'])
+            self.metadata['interval'] = round(
+                self.metadata['interval'] * self.metadata['efactor']
+            )
 
         self.metadata['last_date'] = aux.get_daystamp()
         self.metadata['next_date'] = aux.get_daystamp() + self.metadata['interval']
         self.metadata['last_modify'] = time.time()
 
     def __str__(self):
-        return (f"记忆单元预览 \n"
-                f"内容: '{self.content}' \n"
-                f"易度系数: {self.metadata['efactor']:.2f} \n"
-                f"已经重复的次数: {self.metadata['rept']} \n"
-                f"下次间隔: {self.metadata['interval']} 天 \n"
-                f"下次复习日期时间戳: {self.metadata['next_date']}")
+        return (
+            f"记忆单元预览 \n"
+            f"内容: '{self.content}' \n"
+            f"易度系数: {self.metadata['efactor']:.2f} \n"
+            f"已经重复的次数: {self.metadata['rept']} \n"
+            f"下次间隔: {self.metadata['interval']} 天 \n"
+            f"下次复习日期时间戳: {self.metadata['next_date']}"
+        )
 
     def __eq__(self, other):
         if self.content == other.content:
@@ -106,10 +111,6 @@ class Electron():
     def __setitem__(self, key, value):
         if key == "content":
             raise AttributeError("content 应为只读")
-        
-        # 可以在此处添加更复杂的验证逻辑，例如只允许修改预定义的 metadata 键
-        # 或者根据键进行类型检查等。
-        
         self.metadata[key] = value
         self.metadata['last_modify'] = time.time()
 
@@ -123,6 +124,7 @@ class Electron():
     def placeholder():
         return Electron("电子对象样例内容", {})
 
+
 class Nucleon:
     """核子: 材料元数据"""
 
@@ -151,7 +153,8 @@ class Nucleon:
     def placeholder():
         return Nucleon("核子对象样例内容", {})
 
-class NucleonUnion():
+
+class NucleonUnion:
     """
     替代原有 NucleonFile 类, 支持复杂逻辑
 
@@ -166,6 +169,7 @@ class NucleonUnion():
     Parameters:
         path (Path): 包含核子数据的文件路径。
     """
+
     def __init__(self, path):
         self.path = path
         self.name = path.name.replace(path.suffix, "")
@@ -189,8 +193,10 @@ class NucleonUnion():
             tmp = {i.content: i.metadata for i in self.nucleons}
             toml.dump(tmp, f)
 
-class ElectronUnion():
-    "取代原有 ElectronFile 类, 以支持复杂逻辑"
+
+class ElectronUnion:
+    """取代原有 ElectronFile 类, 以支持复杂逻辑"""
+
     def __init__(self, path):
         self.path = path
         self.name = path.name.replace(path.suffix, "")
@@ -207,13 +213,15 @@ class ElectronUnion():
         self.electrons = self.electrons_dict.values()
 
     def save(self):
-        #print(1)
+        # print(1)
         with open(self.path, 'w') as f:
             tmp = {i.content: i.metadata for i in self.electrons}
-            #print(tmp)
+            # print(tmp)
             toml.dump(tmp, f)
 
-"""class AtomicFile():
+
+"""
+class AtomicFile():
     def __init__(self, path, type_="unknown"):
         self.path = path
         self.type_ = type_
@@ -237,10 +245,10 @@ class ElectronUnion():
         return ""
     def get_len(self):
         return len(self.datalist)
-
 """
 
-class Atom():
+
+class Atom:
     @staticmethod
     def placeholder():
         return (Electron.placeholder(), Nucleon.placeholder(), {})
@@ -249,18 +257,30 @@ class Atom():
     def advanced_placeholder():
         return (
             Electron("两只黄鹤鸣翠柳", {}),
-            Nucleon("两只黄鹤鸣翠柳", {"note": [],
-            "translation": "臣子李密陈言：我因命运不好，小时候遭遇到了不幸",
-            "keyword_note": {"险衅":"凶险祸患（这里指命运不好）", "夙":"早时，这里指年幼的时候", "闵":"通'悯'，指可忧患的事", "凶":"不幸，指丧父"}}),
+            Nucleon(
+                "两只黄鹤鸣翠柳",
+                {
+                    "note": [],
+                    "translation": "臣子李密陈言：我因命运不好，小时候遭遇到了不幸",
+                    "keyword_note": {
+                        "险衅": "凶险祸患（这里指命运不好）",
+                        "夙": "早时，这里指年幼的时候",
+                        "闵": "通'悯'，指可忧患的事",
+                        "凶": "不幸，指丧父"
+                    }
+                }
+            ),
             {
-                "keydata":{
+                "keydata": {
                     "note": "笔记",
                     "keyword_note": "关键词翻译",
-                    "translation": "语句翻译"},
-                "testdata":{
+                    "translation": "语句翻译"
+                },
+                "testdata": {
                     "additional_inf": ["translation", "note", "keyword_note"],
                     "fill_blank_test": ["translation"],
                     "draw_card_test": ["keyword_note"]
                 },
                 "is_new_activation": 0
-            })
+            }
+        )

precaching.py

@@ -4,40 +4,49 @@ import edge_tts as tts
 from pathlib import Path
 import shutil
 
-def precache(text):
-    cache_dir = Path(f"./cache/voice/")
-    cache_dir.mkdir(parents = True, exist_ok = True)
+
+def precache(text: str):
+    """预缓存单个文本的音频"""
+    cache_dir = Path("./cache/voice/")
+    cache_dir.mkdir(parents=True, exist_ok=True)
     cache = cache_dir / f"{text}.wav"
     if not cache.exists():
         communicate = tts.Communicate(text, "zh-CN-YunjianNeural")
         communicate.save_sync(f"./cache/voice/{text}.wav")
 
-def proc_file(path):
+
+def proc_file(path: Path):
+    """处理单个文件"""
     nu = pt.NucleonUnion(path)
     c = 0
     for i in nu.nucleons:
         c += 1
-        print(f"预缓存 [{nu.name}] ({c}/{len(nu)}): {i["content"]}")
-        precache(f"{i["content"]}")
+        print(f"预缓存 [{nu.name}] ({c}/{len(nu)}): {i['content']}")
+        precache(i['content'])
 
-def walk(path_str):
+
+def walk(path_str: str):
+    """遍历目录处理所有文件"""
     path = Path(path_str)
-
     print(f"正在遍历目录: {path}")
+    
     for item in path.iterdir():
-        if item.is_file():
-            if item.suffix == ".toml":
-                print(f"正预缓存文件: {item.name}")
-                proc_file(item)
+        if item.is_file() and item.suffix == ".toml":
+            print(f"正预缓存文件: {item.name}")
+            proc_file(item)
         elif item.is_dir():
             print(f"进入目录: {item.name}")
 
-print("音频预缓存实用程序")
-print("需要?")
-print("全部缓存: A")
-print("清空缓存: C")
-choice = input("输入选项 $ ")
-if choice == "a" or choice == "A":
-    walk("./nucleon")
-if choice == "c" or choice == "C":
-    shutil.rmtree("./cache/voice")
+
+if __name__ == "__main__":
+    print("音频预缓存实用程序")
+    print("A: 全部缓存")
+    print("C: 清空缓存")
+    
+    choice = input("输入选项 $ ").upper()
+    
+    if choice == "A":
+        walk("./nucleon")
+    elif choice == "C":
+        shutil.rmtree("./cache/voice", ignore_errors=True)
+        print("缓存已清空")

puzzles.py

@@ -1,8 +1,10 @@
 import random
 
-class Puzzle():
+
+class Puzzle:
     pass
 
+
 class BlankPuzzle(Puzzle):
     """填空题谜题生成器
     
@@ -10,18 +12,19 @@ class BlankPuzzle(Puzzle):
         text: 原始字符串(需要 "/" 分割句子, 末尾应有 "/")
         min_denominator: 最小概率倒数(如占所有可生成填空数的 1/7 中的 7, 若期望值小于 1, 则取 1)
     """
-    def __init__(self, text, min_denominator):
+
+    def __init__(self, text: str, min_denominator: int):
         self.text = text
         self.min_denominator = min_denominator
         self.wording = "填空题 - 尚未刷新谜题"
         self.answer = ["填空题 - 尚未刷新谜题"]
 
-    def refresh(self): # 刷新谜题
+    def refresh(self):  # 刷新谜题
         placeholder = "___SLASH___"
         tmp_text = self.text.replace("/", placeholder)
         words = tmp_text.split(placeholder)
         if not words:
-            return ""
+            return
         words = [word for word in words if word]
         num_blanks = min(max(1, len(words) // self.min_denominator), len(words))
         indices_to_blank = random.sample(range(len(words)), num_blanks)
@@ -31,15 +34,13 @@ class BlankPuzzle(Puzzle):
         for index in indices_to_blank:
             blanked_words[index] = "__" * len(words[index])
             answer.append(words[index])
-        result = []
-        for word in blanked_words:
-            result.append(word)
         self.answer = answer
-        self.wording = "".join(result)
+        self.wording = "".join(blanked_words)
 
     def __str__(self):
         return f"{self.wording}\n{str(self.answer)}"
 
+
 class SelectionPuzzle(Puzzle):
     """选择题谜题生成器
     
@@ -49,12 +50,20 @@ class SelectionPuzzle(Puzzle):
         max_riddles_num: 最大生成谜题数 (默认2个)
         prefix: 问题前缀
     """
-    def __init__(self, mapping, jammer: list, max_riddles_num: int = 2, prefix: str = ""):
-        jammer += ["1","2","3","4"]
+
+    def __init__(
+        self,
+        mapping: dict,
+        jammer: list,
+        max_riddles_num: int = 2,
+        prefix: str = ""
+    ):
         self.prefix = prefix
         self.mapping = mapping
-        self.jammer = list(set(jammer + list(mapping.values())))  # 合并干扰项和正确答案并去重
-        self.max_riddles_num = max(1, min(max_riddles_num, 5))  # 限制1-5个谜题
+        self.jammer = list(set(jammer + list(mapping.values())))
+        while len(self.jammer) < 4:
+            self.jammer.append(" ")
+        self.max_riddles_num = max(1, min(max_riddles_num, 5))
         self.wording = "选择题 - 尚未刷新谜题"
         self.answer = ["选择题 - 尚未刷新谜题"]
         self.options = []
@@ -67,35 +76,29 @@ class SelectionPuzzle(Puzzle):
             self.options = []
             return
         
-        # 确定实际生成的谜题数量
         num_questions = min(self.max_riddles_num, len(self.mapping))
         questions = random.sample(list(self.mapping.items()), num_questions)
-        
-        # 生成谜题
         puzzles = []
         answers = []
         all_options = []
         
         for question, correct_answer in questions:
-            # 生成选项 (正确答案 + 3个干扰项)
             options = [correct_answer]
-            available_jammers = [j for j in self.jammer if j != correct_answer]
-            
+            available_jammers = [
+                j for j in self.jammer if j != correct_answer
+            ]
             if len(available_jammers) >= 3:
                 selected_jammers = random.sample(available_jammers, 3)
             else:
                 selected_jammers = random.choices(available_jammers, k=3)
-            
             options.extend(selected_jammers)
             random.shuffle(options)
-            
             puzzles.append(question)
             answers.append(correct_answer)
             all_options.append(options)
         
         question_texts = []
-        for i, (puzzle, options) in enumerate(zip(puzzles, all_options)):
-            #options_text = "\n".join([f"   {chr(97+j)}. {opt}" for j, opt in enumerate(options)])
+        for i, puzzle in enumerate(puzzles):
             question_texts.append(f"{self.prefix}:\n  {i+1}. {puzzle}")
         
         self.wording = question_texts
@@ -106,8 +109,12 @@ class SelectionPuzzle(Puzzle):
         return f"{self.wording}\n正确答案: {', '.join(self.answer)}"
 
 
-
-puz = SelectionPuzzle({"1+1":"2", "1+2":"3", "1+3": "4"}, ["2","5","0"], 3, '求值: ')
+puz = SelectionPuzzle(
+    {"1+1": "2", "1+2": "3", "1+3": "4"},
+    ["2", "5", "0"],
+    3,
+    '求值: '
+)
 puz.refresh()
 print(puz.wording)
 print(puz.answer)