v0.2.4 基本可用型

main.py

@@ -2,17 +2,19 @@ from textual.app import App, ComposeResult
 from textual.widgets import Header, Footer, ListView, ListItem, Label, Static, Button
 from textual.containers import Container, Horizontal
 from textual.screen import Screen
-import particles as pt
-from reactor import Reactor
 import pathlib
 import threading
 import edge_tts as tts
 from playsound import playsound
 from textual.logging import TextualHandler
 
-ver = '0.2.1'
+import particles as pt
+from reactor import Reactor
+import auxiliary as aux
 
-debug = TextualHandler(stderr=True)
+ver = '0.2.4'
+
+config = aux.ConfigFile("config.toml")
 
 class MemScreen(Screen):
     BINDINGS = [
@@ -37,7 +39,7 @@ class MemScreen(Screen):
         self,
         nucleon_file: pt.AtomicFile,
         electron_file: pt.AtomicFile,
-        tasked_num: int
+        tasked_num
     ):
         super().__init__(name=None, id=None, classes=None)
         self.reactor = Reactor(nucleon_file, electron_file, tasked_num)
@@ -104,13 +106,16 @@ class MemScreen(Screen):
             ret = self.reactor.forward(1)
             if ret == -1:
                 if self.reactor.round_set == 0:
-                    if self.stage == 3:
+                    if self.stage == 4:
                         # NOTE #
-                        #self.reactor.save()
+                        if config.get("save"):
+                            self.reactor.save()
                         self._show_finished_screen("今日目标已完成")
                     else:
-                        self.stage += 1
                         self.reactor.set_round_templated(self.stage)
+                        self.reactor.forward(1)
+                        self._update_ui()
+                        self.stage += 1
                 return
         #feedback_label.update("") # 清除反馈消息
             self._update_ui()
@@ -168,7 +173,7 @@ class PreparationScreen(Screen):
         pass
         if event.button.id == "start_memorizing_button":
             #init_file(Path(self.atom_file).name)
-            newscr = MemScreen(self.nucleon_file, self.electron_file, 8)
+            newscr = MemScreen(self.nucleon_file, self.electron_file, config.get("tasked_number", 8))
             self.app.push_screen(
                 newscr
             )
@@ -236,6 +241,5 @@ class AppLauncher(App):
         self.push_screen("file_selection_screen")
 
 if __name__ == "__main__":
-    css_path = pathlib.Path("styles_dashboard.tcss")
     app = AppLauncher()
     app.run()