Compare commits
13 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| d347471cc0 | |||
| 0a365b568a | |||
| e303d4dc1e | |||
| cb78290f05 | |||
| e0417981b1 | |||
| a0660d3348 | |||
| f5e0417292 | |||
| 142d30347b | |||
| e57cea7219 | |||
| 98ec6504a4 | |||
| 243eea864b | |||
| cfb1385f4d | |||
| a1462206a2 |
6
.gitignore
vendored
6
.gitignore
vendored
@@ -11,14 +11,14 @@ electron/test.toml
|
|||||||
build/
|
build/
|
||||||
dist/
|
dist/
|
||||||
old/
|
old/
|
||||||
|
|
||||||
# Project specific directories
|
|
||||||
# config/
|
# config/
|
||||||
data/cache/
|
data/cache/
|
||||||
data/electron/
|
data/electron/
|
||||||
data/nucleon/
|
data/nucleon/
|
||||||
!data/nucleon/test*
|
data/global/
|
||||||
|
!data/nucleon/TEST*
|
||||||
data/orbital/
|
data/orbital/
|
||||||
|
config/config_dev.toml
|
||||||
AGENTS.md
|
AGENTS.md
|
||||||
|
|
||||||
# Byte-compiled / optimized / DLL files
|
# Byte-compiled / optimized / DLL files
|
||||||
|
|||||||
@@ -10,13 +10,15 @@
|
|||||||
- `dev` 分支: 开发版本
|
- `dev` 分支: 开发版本
|
||||||
- 功能分支: 从 `dev` 分支创建, 命名格式为 `feature/描述` 或 `fix/描述` 或 `refactor/描述`
|
- 功能分支: 从 `dev` 分支创建, 命名格式为 `feature/描述` 或 `fix/描述` 或 `refactor/描述`
|
||||||
2. **代码风格**:
|
2. **代码风格**:
|
||||||
- 请使用 Black 格式化代码
|
- 请使用 Black 与 isort 格式化代码
|
||||||
- 遵循 PEP 8 规范
|
- 遵循 PEP 8 规范
|
||||||
- 添加适当的文档字符串
|
- 添加适当的文档字符串
|
||||||
3. **提交消息**:
|
3. **提交消息**:
|
||||||
- 使用简体中文或英文撰写清晰的提交消息
|
- 使用简体中文或英文撰写清晰的提交消息
|
||||||
- 格式: 遵循 Conventional Commits 规范
|
- 格式: 遵循 Conventional Commits 规范
|
||||||
|
4. **合并方式**:
|
||||||
|
- 不使用 Fast-forward 合并
|
||||||
|
- 可以设置 `git config merge.ff false`
|
||||||
## 设置开发环境
|
## 设置开发环境
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
|
|||||||
@@ -82,7 +82,7 @@ python -m heurams.interface
|
|||||||
|
|
||||||
## 配置
|
## 配置
|
||||||
|
|
||||||
配置文件位于 `config/config.toml`(相对于工作目录). 如果不存在, 会使用内置的默认配置.
|
配置文件位于 `config/config.toml`(相对于工作目录). 如果不存在, 会使用内置的默认配置.
|
||||||
|
|
||||||
## 项目结构
|
## 项目结构
|
||||||
|
|
||||||
|
|||||||
@@ -14,6 +14,14 @@ scheduled_num = 8
|
|||||||
# UTC 时间戳修正 仅用于 UNIX 日时间戳的生成修正, 单位为秒
|
# UTC 时间戳修正 仅用于 UNIX 日时间戳的生成修正, 单位为秒
|
||||||
timezone_offset = +28800 # 中国标准时间 (UTC+8)
|
timezone_offset = +28800 # 中国标准时间 (UTC+8)
|
||||||
|
|
||||||
|
[interface]
|
||||||
|
|
||||||
|
[interface.memorizor]
|
||||||
|
autovoice = true # 自动语音播放, 仅限于 recognition 组件
|
||||||
|
|
||||||
|
[algorithm]
|
||||||
|
default = "SM-2" # 主要算法; 可选项: SM-2, SM-15M, FSRS
|
||||||
|
|
||||||
[puzzles] # 谜题默认配置
|
[puzzles] # 谜题默认配置
|
||||||
|
|
||||||
[puzzles.mcq]
|
[puzzles.mcq]
|
||||||
@@ -25,6 +33,7 @@ min_denominator = 3
|
|||||||
[paths] # 相对于配置文件的 ".." (即工作目录) 而言 或绝对路径
|
[paths] # 相对于配置文件的 ".." (即工作目录) 而言 或绝对路径
|
||||||
nucleon_dir = "./data/nucleon"
|
nucleon_dir = "./data/nucleon"
|
||||||
electron_dir = "./data/electron"
|
electron_dir = "./data/electron"
|
||||||
|
global_dir = "./data/global" # 全局数据路径, SM-15 等算法需要
|
||||||
orbital_dir = "./data/orbital"
|
orbital_dir = "./data/orbital"
|
||||||
cache_dir = "./data/cache"
|
cache_dir = "./data/cache"
|
||||||
template_dir = "./data/template"
|
template_dir = "./data/template"
|
||||||
@@ -34,6 +43,9 @@ audio = "playsound" # 可选项: playsound(通用), termux(仅用于支持 Andro
|
|||||||
tts = "edgetts" # 可选项: edgetts
|
tts = "edgetts" # 可选项: edgetts
|
||||||
llm = "openai" # 可选项: openai
|
llm = "openai" # 可选项: openai
|
||||||
|
|
||||||
|
[providers.tts.edgetts] # EdgeTTS 设置
|
||||||
|
voice = "zh-CN-XiaoxiaoNeural" # 可选项: zh-CN-YunjianNeural (男声), zh-CN-XiaoxiaoNeural (女声)
|
||||||
|
|
||||||
[providers.llm.openai] # 与 OpenAI 相容的语言模型接口服务设置
|
[providers.llm.openai] # 与 OpenAI 相容的语言模型接口服务设置
|
||||||
url = ""
|
url = ""
|
||||||
key = ""
|
key = ""
|
||||||
|
|||||||
@@ -25,3 +25,4 @@ readme = "README.md"
|
|||||||
|
|
||||||
[tool.setuptools.packages.find]
|
[tool.setuptools.packages.find]
|
||||||
where = ["src"]
|
where = ["src"]
|
||||||
|
|
||||||
|
|||||||
@@ -3,8 +3,9 @@
|
|||||||
以及基准路径
|
以及基准路径
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from contextvars import ContextVar
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
from contextvars import ContextVar
|
||||||
|
|
||||||
from heurams.services.config import ConfigFile
|
from heurams.services.config import ConfigFile
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
@@ -31,7 +32,11 @@ try:
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
print("未能加载自定义用户配置")
|
print("未能加载自定义用户配置")
|
||||||
logger.warning("未能加载自定义用户配置, 错误: %s", e)
|
logger.warning("未能加载自定义用户配置, 错误: %s", e)
|
||||||
|
if pathlib.Path(rootdir / "default" / "config" / "config_dev.toml").exists():
|
||||||
|
logger.debug("使用开发设置")
|
||||||
|
config_var: ContextVar[ConfigFile] = ContextVar(
|
||||||
|
"config_var", default=ConfigFile(workdir / "config" / "config_dev.toml")
|
||||||
|
)
|
||||||
# runtime_var: ContextVar = ContextVar('runtime_var', default=dict()) # 运行时共享数据
|
# runtime_var: ContextVar = ContextVar('runtime_var', default=dict()) # 运行时共享数据
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -6,10 +6,10 @@ daystamp_override = -1
|
|||||||
timestamp_override = -1
|
timestamp_override = -1
|
||||||
|
|
||||||
# [调试] 一键通过
|
# [调试] 一键通过
|
||||||
quick_pass = 0
|
quick_pass = 1
|
||||||
|
|
||||||
# 对于每个项目的默认新记忆原子数量
|
# 对于每个项目的默认新记忆原子数量
|
||||||
tasked_number = 8
|
scheduled_num = 8
|
||||||
|
|
||||||
# UTC 时间戳修正 仅用于 UNIX 日时间戳的生成修正, 单位为秒
|
# UTC 时间戳修正 仅用于 UNIX 日时间戳的生成修正, 单位为秒
|
||||||
timezone_offset = +28800 # 中国标准时间 (UTC+8)
|
timezone_offset = +28800 # 中国标准时间 (UTC+8)
|
||||||
@@ -22,8 +22,18 @@ max_riddles_num = 2
|
|||||||
[puzzles.cloze]
|
[puzzles.cloze]
|
||||||
min_denominator = 3
|
min_denominator = 3
|
||||||
|
|
||||||
[paths] # 相对于工作目录而言 或绝对路径
|
[paths] # 相对于配置文件的 ".." (即工作目录) 而言 或绝对路径
|
||||||
nucleon_dir = "./data/nucleon"
|
nucleon_dir = "./data/nucleon"
|
||||||
electron_dir = "./data/electron"
|
electron_dir = "./data/electron"
|
||||||
orbital_dir = "./data/orbital"
|
orbital_dir = "./data/orbital"
|
||||||
cache_dir = "./data/cache"
|
cache_dir = "./data/cache"
|
||||||
|
template_dir = "./data/template"
|
||||||
|
|
||||||
|
[services] # 定义服务到提供者的映射
|
||||||
|
audio = "playsound" # 可选项: playsound(通用), termux(仅用于支持 Android Termux), mpg123(TODO)
|
||||||
|
tts = "edgetts" # 可选项: edgetts
|
||||||
|
llm = "openai" # 可选项: openai
|
||||||
|
|
||||||
|
[providers.llm.openai] # 与 OpenAI 相容的语言模型接口服务设置
|
||||||
|
url = ""
|
||||||
|
key = ""
|
||||||
|
|||||||
0
src/heurams/default/config/config_dev.toml
Normal file
0
src/heurams/default/config/config_dev.toml
Normal file
60
src/heurams/interface/__init__.py
Normal file
60
src/heurams/interface/__init__.py
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
from textual.app import App
|
||||||
|
from textual.widgets import Button
|
||||||
|
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .screens.about import AboutScreen
|
||||||
|
from .screens.dashboard import DashboardScreen
|
||||||
|
from .screens.nucreator import NucleonCreatorScreen
|
||||||
|
from .screens.precache import PrecachingScreen
|
||||||
|
|
||||||
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def environment_check():
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
logger.debug("检查环境路径")
|
||||||
|
|
||||||
|
for i in config_var.get()["paths"].values():
|
||||||
|
i = Path(i)
|
||||||
|
if not i.exists():
|
||||||
|
logger.info("创建目录: %s", i)
|
||||||
|
print(f"创建 {i}")
|
||||||
|
i.mkdir(exist_ok=True, parents=True)
|
||||||
|
else:
|
||||||
|
logger.debug("目录已存在: %s", i)
|
||||||
|
print(f"找到 {i}")
|
||||||
|
logger.debug("环境检查完成")
|
||||||
|
|
||||||
|
|
||||||
|
class HeurAMSApp(App):
|
||||||
|
TITLE = "潜进"
|
||||||
|
CSS_PATH = "css/main.tcss"
|
||||||
|
SUB_TITLE = "启发式辅助记忆调度器"
|
||||||
|
BINDINGS = [
|
||||||
|
("q", "quit", "退出"),
|
||||||
|
("d", "toggle_dark", "切换色调"),
|
||||||
|
("1", "app.push_screen('dashboard')", "仪表盘"),
|
||||||
|
("2", "app.push_screen('precache_all')", "缓存管理器"),
|
||||||
|
("3", "app.push_screen('nucleon_creator')", "创建新单元"),
|
||||||
|
("0", "app.push_screen('about')", "版本信息"),
|
||||||
|
]
|
||||||
|
SCREENS = {
|
||||||
|
"dashboard": DashboardScreen,
|
||||||
|
"nucleon_creator": NucleonCreatorScreen,
|
||||||
|
"precache_all": PrecachingScreen,
|
||||||
|
"about": AboutScreen,
|
||||||
|
}
|
||||||
|
|
||||||
|
def on_mount(self) -> None:
|
||||||
|
environment_check()
|
||||||
|
self.push_screen("dashboard")
|
||||||
|
|
||||||
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||||||
|
self.exit(event.button.id)
|
||||||
|
|
||||||
|
def action_do_nothing(self):
|
||||||
|
print("DO NOTHING")
|
||||||
|
self.refresh()
|
||||||
@@ -1,87 +1,18 @@
|
|||||||
from textual.app import App
|
from textual.app import App
|
||||||
from textual.widgets import Button
|
from textual.widgets import Button
|
||||||
|
|
||||||
|
from heurams.services.logger import get_logger
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.interface import HeurAMSApp
|
||||||
|
|
||||||
|
from .screens.about import AboutScreen
|
||||||
from .screens.dashboard import DashboardScreen
|
from .screens.dashboard import DashboardScreen
|
||||||
from .screens.nucreator import NucleonCreatorScreen
|
from .screens.nucreator import NucleonCreatorScreen
|
||||||
from .screens.precache import PrecachingScreen
|
from .screens.precache import PrecachingScreen
|
||||||
from .screens.about import AboutScreen
|
|
||||||
from heurams.services.logger import get_logger
|
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class HeurAMSApp(App):
|
|
||||||
TITLE = "潜进"
|
|
||||||
CSS_PATH = "css/main.tcss"
|
|
||||||
SUB_TITLE = "启发式辅助记忆调度器"
|
|
||||||
BINDINGS = [
|
|
||||||
("q", "quit", "退出"),
|
|
||||||
("d", "toggle_dark", "切换色调"),
|
|
||||||
("1", "app.push_screen('dashboard')", "仪表盘"),
|
|
||||||
("2", "app.push_screen('precache_all')", "缓存管理器"),
|
|
||||||
("3", "app.push_screen('nucleon_creator')", "创建新单元"),
|
|
||||||
("0", "app.push_screen('about')", "版本信息"),
|
|
||||||
]
|
|
||||||
SCREENS = {
|
|
||||||
"dashboard": DashboardScreen,
|
|
||||||
"nucleon_creator": NucleonCreatorScreen,
|
|
||||||
"precache_all": PrecachingScreen,
|
|
||||||
"about": AboutScreen,
|
|
||||||
}
|
|
||||||
|
|
||||||
def on_mount(self) -> None:
|
|
||||||
self.push_screen("dashboard")
|
|
||||||
|
|
||||||
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
||||||
self.exit(event.button.id)
|
|
||||||
|
|
||||||
def action_do_nothing(self):
|
|
||||||
print("DO NOTHING")
|
|
||||||
self.refresh()
|
|
||||||
|
|
||||||
|
|
||||||
def environment_check():
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
logger.debug("检查环境路径")
|
|
||||||
|
|
||||||
for i in config_var.get()["paths"].values():
|
|
||||||
i = Path(i)
|
|
||||||
if not i.exists():
|
|
||||||
logger.info("创建目录: %s", i)
|
|
||||||
print(f"创建 {i}")
|
|
||||||
i.mkdir(exist_ok=True, parents=True)
|
|
||||||
else:
|
|
||||||
logger.debug("目录已存在: %s", i)
|
|
||||||
print(f"找到 {i}")
|
|
||||||
logger.debug("环境检查完成")
|
|
||||||
|
|
||||||
|
|
||||||
def is_subdir(parent, child):
|
|
||||||
try:
|
|
||||||
child.relative_to(parent)
|
|
||||||
logger.debug("is_subdir: %s 是 %s 的子目录", child, parent)
|
|
||||||
return 1
|
|
||||||
except:
|
|
||||||
logger.debug("is_subdir: %s 不是 %s 的子目录", child, parent)
|
|
||||||
return 0
|
|
||||||
|
|
||||||
|
|
||||||
# 开发模式
|
|
||||||
from heurams.context import rootdir, workdir, config_var
|
|
||||||
from pathlib import Path
|
|
||||||
from heurams.context import rootdir
|
|
||||||
import os
|
|
||||||
|
|
||||||
if is_subdir(Path(rootdir), Path(os.getcwd())):
|
|
||||||
os.chdir(Path(rootdir) / ".." / "..")
|
|
||||||
print(f'转入开发数据目录: {Path(rootdir)/".."/".."}')
|
|
||||||
|
|
||||||
environment_check()
|
|
||||||
|
|
||||||
app = HeurAMSApp()
|
app = HeurAMSApp()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
app.run()
|
app.run()
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
app.run()
|
|
||||||
|
|||||||
@@ -1,15 +1,8 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from textual.app import ComposeResult
|
from textual.app import ComposeResult
|
||||||
from textual.widgets import (
|
from textual.containers import ScrollableContainer
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
Static,
|
|
||||||
Button,
|
|
||||||
Markdown,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer, ScrollableContainer
|
|
||||||
from textual.screen import Screen
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import Button, Footer, Header, Label, Markdown, Static
|
||||||
|
|
||||||
import heurams.services.version as version
|
import heurams.services.version as version
|
||||||
from heurams.context import *
|
from heurams.context import *
|
||||||
|
|||||||
@@ -1,26 +1,20 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
import pathlib
|
||||||
|
|
||||||
from textual.app import ComposeResult
|
from textual.app import ComposeResult
|
||||||
from textual.widgets import (
|
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
ListView,
|
|
||||||
ListItem,
|
|
||||||
Button,
|
|
||||||
Static,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer
|
from textual.containers import ScrollableContainer
|
||||||
from textual.screen import Screen
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import (Button, Footer, Header, Label, ListItem, ListView,
|
||||||
|
Static)
|
||||||
|
|
||||||
from heurams.kernel.particles import *
|
|
||||||
from heurams.context import *
|
|
||||||
import heurams.services.version as version
|
|
||||||
import heurams.services.timer as timer
|
import heurams.services.timer as timer
|
||||||
from .preparation import PreparationScreen
|
import heurams.services.version as version
|
||||||
from .about import AboutScreen
|
from heurams.context import *
|
||||||
|
from heurams.kernel.particles import *
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
import pathlib
|
from .about import AboutScreen
|
||||||
|
from .preparation import PreparationScreen
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
@@ -51,8 +45,8 @@ class DashboardScreen(Screen):
|
|||||||
res = dict()
|
res = dict()
|
||||||
filestem = pathlib.Path(filename).stem
|
filestem = pathlib.Path(filename).stem
|
||||||
res[0] = f"{filename}\0"
|
res[0] = f"{filename}\0"
|
||||||
from heurams.kernel.particles.loader import load_electron
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
|
from heurams.kernel.particles.loader import load_electron
|
||||||
|
|
||||||
electron_file_path = pathlib.Path(config_var.get()["paths"]["electron_dir"]) / (
|
electron_file_path = pathlib.Path(config_var.get()["paths"]["electron_dir"]) / (
|
||||||
filestem + ".json"
|
filestem + ".json"
|
||||||
@@ -80,7 +74,7 @@ class DashboardScreen(Screen):
|
|||||||
is_activated = 1
|
is_activated = 1
|
||||||
nextdate = min(nextdate, i.nextdate())
|
nextdate = min(nextdate, i.nextdate())
|
||||||
res[1] = f"下一次复习: {nextdate}\n"
|
res[1] = f"下一次复习: {nextdate}\n"
|
||||||
res[1] += f"{is_due if "需要复习" else "当前无需复习"}"
|
res[1] += f"{"需要复习" if is_due else "当前无需复习"}"
|
||||||
if not is_activated:
|
if not is_activated:
|
||||||
res[1] = " 尚未激活"
|
res[1] = " 尚未激活"
|
||||||
return res
|
return res
|
||||||
|
|||||||
@@ -1,16 +1,18 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from textual.app import ComposeResult
|
|
||||||
from textual.widgets import Header, Footer, Label, Static, Button
|
|
||||||
from textual.containers import Center, ScrollableContainer
|
|
||||||
from textual.screen import Screen
|
|
||||||
from textual.reactive import reactive
|
|
||||||
from enum import Enum, auto
|
from enum import Enum, auto
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from textual.app import ComposeResult
|
||||||
from heurams.context import config_var
|
from textual.containers import Center, ScrollableContainer
|
||||||
from heurams.kernel.reactor import *
|
from textual.reactive import reactive
|
||||||
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import Button, Footer, Header, Label, Static
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.kernel.puzzles as pz
|
import heurams.kernel.puzzles as pz
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.kernel.reactor import *
|
||||||
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
from .. import shim
|
from .. import shim
|
||||||
|
|
||||||
|
|
||||||
@@ -44,14 +46,14 @@ class MemScreen(Screen):
|
|||||||
) -> None:
|
) -> None:
|
||||||
super().__init__(name, id, classes)
|
super().__init__(name, id, classes)
|
||||||
self.atoms = atoms
|
self.atoms = atoms
|
||||||
|
for i in self.atoms:
|
||||||
|
i.do_eval()
|
||||||
self.phaser = Phaser(atoms)
|
self.phaser = Phaser(atoms)
|
||||||
# logger.debug(self.phaser.state)
|
# logger.debug(self.phaser.state)
|
||||||
self.procession: Procession = self.phaser.current_procession() # type: ignore
|
self.procession: Procession = self.phaser.current_procession() # type: ignore
|
||||||
self.atom: pt.Atom = self.procession.current_atom
|
self.atom: pt.Atom = self.procession.current_atom
|
||||||
# logger.debug(self.phaser.state)
|
# logger.debug(self.phaser.state)
|
||||||
# self.procession.forward(1)
|
# self.procession.forward(1)
|
||||||
for i in atoms:
|
|
||||||
i.do_eval()
|
|
||||||
|
|
||||||
def on_mount(self):
|
def on_mount(self):
|
||||||
self.load_puzzle()
|
self.load_puzzle()
|
||||||
@@ -142,8 +144,21 @@ class MemScreen(Screen):
|
|||||||
self.atom.lock(1)
|
self.atom.lock(1)
|
||||||
|
|
||||||
def action_play_voice(self):
|
def action_play_voice(self):
|
||||||
|
self.run_worker(self.play_voice, exclusive=True, thread=True)
|
||||||
|
|
||||||
|
def play_voice(self):
|
||||||
"""朗读当前内容"""
|
"""朗读当前内容"""
|
||||||
pass
|
from heurams.services.audio_service import play_by_path
|
||||||
|
from pathlib import Path
|
||||||
|
from heurams.services.hasher import get_md5
|
||||||
|
path = Path(config_var.get()['paths']["cache_dir"])
|
||||||
|
path = path / f"{get_md5(self.atom.registry['nucleon'].metadata["formation"]["tts_text"])}.wav"
|
||||||
|
if path.exists():
|
||||||
|
play_by_path(path)
|
||||||
|
else:
|
||||||
|
from heurams.services.tts_service import convertor
|
||||||
|
convertor(self.atom.registry['nucleon'].metadata["formation"]["tts_text"], path)
|
||||||
|
play_by_path(path)
|
||||||
|
|
||||||
def action_toggle_dark(self):
|
def action_toggle_dark(self):
|
||||||
self.app.action_toggle_dark()
|
self.app.action_toggle_dark()
|
||||||
|
|||||||
@@ -1,21 +1,15 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import toml
|
||||||
from textual.app import ComposeResult
|
from textual.app import ComposeResult
|
||||||
from textual.widgets import (
|
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
Input,
|
|
||||||
Select,
|
|
||||||
Button,
|
|
||||||
Markdown,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer
|
from textual.containers import ScrollableContainer
|
||||||
from textual.screen import Screen
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import (Button, Footer, Header, Input, Label, Markdown,
|
||||||
|
Select)
|
||||||
|
|
||||||
from heurams.services.version import ver
|
|
||||||
import toml
|
|
||||||
from pathlib import Path
|
|
||||||
from heurams.context import config_var
|
from heurams.context import config_var
|
||||||
|
from heurams.services.version import ver
|
||||||
|
|
||||||
|
|
||||||
class NucleonCreatorScreen(Screen):
|
class NucleonCreatorScreen(Screen):
|
||||||
@@ -27,6 +21,7 @@ class NucleonCreatorScreen(Screen):
|
|||||||
|
|
||||||
def search_templates(self):
|
def search_templates(self):
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from heurams.context import config_var
|
from heurams.context import config_var
|
||||||
|
|
||||||
template_dir = Path(config_var.get()["paths"]["template_dir"])
|
template_dir = Path(config_var.get()["paths"]["template_dir"])
|
||||||
|
|||||||
@@ -1,22 +1,15 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from textual.app import ComposeResult
|
|
||||||
from textual.widgets import (
|
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
Button,
|
|
||||||
Static,
|
|
||||||
ProgressBar,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer, Horizontal
|
|
||||||
from textual.containers import ScrollableContainer
|
|
||||||
from textual.screen import Screen
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
|
from textual.app import ComposeResult
|
||||||
|
from textual.containers import Horizontal, ScrollableContainer
|
||||||
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import Button, Footer, Header, Label, ProgressBar, Static
|
||||||
|
from textual.worker import get_current_worker
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.services.hasher as hasher
|
import heurams.services.hasher as hasher
|
||||||
from heurams.context import *
|
from heurams.context import *
|
||||||
from textual.worker import get_current_worker
|
|
||||||
|
|
||||||
|
|
||||||
class PrecachingScreen(Screen):
|
class PrecachingScreen(Screen):
|
||||||
@@ -46,7 +39,9 @@ class PrecachingScreen(Screen):
|
|||||||
self.desc = desc
|
self.desc = desc
|
||||||
for i in nucleons:
|
for i in nucleons:
|
||||||
i: pt.Nucleon
|
i: pt.Nucleon
|
||||||
i.do_eval()
|
atom = pt.Atom()
|
||||||
|
atom.link("nucleon", i)
|
||||||
|
atom.do_eval()
|
||||||
# print("完成 EVAL")
|
# print("完成 EVAL")
|
||||||
|
|
||||||
def compose(self) -> ComposeResult:
|
def compose(self) -> ComposeResult:
|
||||||
@@ -96,17 +91,15 @@ class PrecachingScreen(Screen):
|
|||||||
|
|
||||||
def precache_by_text(self, text: str):
|
def precache_by_text(self, text: str):
|
||||||
"""预缓存单段文本的音频"""
|
"""预缓存单段文本的音频"""
|
||||||
from heurams.context import rootdir, workdir, config_var
|
from heurams.context import config_var, rootdir, workdir
|
||||||
|
|
||||||
cache_dir = pathlib.Path(config_var.get()["paths"]["cache_dir"])
|
cache_dir = pathlib.Path(config_var.get()["paths"]["cache_dir"])
|
||||||
cache_dir.mkdir(parents=True, exist_ok=True)
|
cache_dir.mkdir(parents=True, exist_ok=True)
|
||||||
cache_file = cache_dir / f"{hasher.get_md5(text)}.wav"
|
cache_file = cache_dir / f"{hasher.get_md5(text)}.wav"
|
||||||
if not cache_file.exists():
|
if not cache_file.exists():
|
||||||
try: # TODO: 调用模块消除tts耦合
|
try:
|
||||||
import edge_tts as tts
|
from heurams.services.tts_service import convertor
|
||||||
|
convertor(text, cache_file)
|
||||||
communicate = tts.Communicate(text, "zh-CN-XiaoxiaoNeural")
|
|
||||||
communicate.save_sync(str(cache_file))
|
|
||||||
return 1
|
return 1
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"预缓存失败 '{text}': {e}")
|
print(f"预缓存失败 '{text}': {e}")
|
||||||
@@ -166,7 +159,7 @@ class PrecachingScreen(Screen):
|
|||||||
|
|
||||||
def precache_all_files(self):
|
def precache_all_files(self):
|
||||||
"""预缓存所有文件"""
|
"""预缓存所有文件"""
|
||||||
from heurams.context import rootdir, workdir, config_var
|
from heurams.context import config_var, rootdir, workdir
|
||||||
|
|
||||||
nucleon_path = pathlib.Path(config_var.get()["paths"]["nucleon_dir"])
|
nucleon_path = pathlib.Path(config_var.get()["paths"]["nucleon_dir"])
|
||||||
nucleon_files = [
|
nucleon_files = [
|
||||||
@@ -185,7 +178,9 @@ class PrecachingScreen(Screen):
|
|||||||
self.total = len(nu)
|
self.total = len(nu)
|
||||||
for i in nu:
|
for i in nu:
|
||||||
i: pt.Nucleon
|
i: pt.Nucleon
|
||||||
i.do_eval()
|
atom = pt.Atom()
|
||||||
|
atom.link("nucleon", i)
|
||||||
|
atom.do_eval()
|
||||||
return self.precache_by_list(nu)
|
return self.precache_by_list(nu)
|
||||||
|
|
||||||
def on_button_pressed(self, event: Button.Pressed) -> None:
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||||||
@@ -220,7 +215,8 @@ class PrecachingScreen(Screen):
|
|||||||
# 清空缓存
|
# 清空缓存
|
||||||
try:
|
try:
|
||||||
import shutil
|
import shutil
|
||||||
from heurams.context import rootdir, workdir, config_var
|
|
||||||
|
from heurams.context import config_var, rootdir, workdir
|
||||||
|
|
||||||
shutil.rmtree(
|
shutil.rmtree(
|
||||||
f"{config_var.get()["paths"]["cache_dir"]}", ignore_errors=True
|
f"{config_var.get()["paths"]["cache_dir"]}", ignore_errors=True
|
||||||
|
|||||||
@@ -1,21 +1,15 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from textual.app import ComposeResult
|
from textual.app import ComposeResult
|
||||||
from textual.widgets import (
|
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
Static,
|
|
||||||
Button,
|
|
||||||
Markdown,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer
|
from textual.containers import ScrollableContainer
|
||||||
|
from textual.reactive import reactive
|
||||||
from textual.screen import Screen
|
from textual.screen import Screen
|
||||||
from heurams.context import config_var
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Footer, Header, Label, Markdown, Static
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.services.hasher as hasher
|
import heurams.services.hasher as hasher
|
||||||
from heurams.context import *
|
from heurams.context import *
|
||||||
from textual.reactive import reactive
|
from heurams.context import config_var
|
||||||
from textual.widget import Widget
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,22 +1,15 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
from textual.app import ComposeResult
|
|
||||||
from textual.widgets import (
|
|
||||||
Header,
|
|
||||||
Footer,
|
|
||||||
Label,
|
|
||||||
Button,
|
|
||||||
Static,
|
|
||||||
ProgressBar,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer, Horizontal
|
|
||||||
from textual.containers import ScrollableContainer
|
|
||||||
from textual.screen import Screen
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
|
from textual.app import ComposeResult
|
||||||
|
from textual.containers import Horizontal, ScrollableContainer
|
||||||
|
from textual.screen import Screen
|
||||||
|
from textual.widgets import Button, Footer, Header, Label, ProgressBar, Static
|
||||||
|
from textual.worker import get_current_worker
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.services.hasher as hasher
|
import heurams.services.hasher as hasher
|
||||||
from heurams.context import *
|
from heurams.context import *
|
||||||
from textual.worker import get_current_worker
|
|
||||||
|
|
||||||
|
|
||||||
class SyncScreen(Screen):
|
class SyncScreen(Screen):
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
"""Kernel 操作辅助函数库"""
|
"""Kernel 操作辅助函数库"""
|
||||||
|
|
||||||
import random
|
import random
|
||||||
|
from typing import TypedDict
|
||||||
|
|
||||||
|
import heurams.interface.widgets as pzw
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.kernel.puzzles as pz
|
import heurams.kernel.puzzles as pz
|
||||||
import heurams.interface.widgets as pzw
|
|
||||||
from typing import TypedDict
|
|
||||||
|
|
||||||
staging = {} # 细粒度缓存区, 是 ident -> quality 的封装
|
staging = {} # 细粒度缓存区, 是 ident -> quality 的封装
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,8 @@
|
|||||||
from typing import Iterable
|
from typing import Iterable
|
||||||
|
|
||||||
from textual.app import ComposeResult
|
from textual.app import ComposeResult
|
||||||
from textual.widget import Widget
|
from textual.widget import Widget
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,13 +1,11 @@
|
|||||||
from textual.widgets import (
|
from textual.containers import Horizontal, ScrollableContainer
|
||||||
Label,
|
|
||||||
Static,
|
|
||||||
Button,
|
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer, Horizontal
|
|
||||||
from textual.widget import Widget
|
|
||||||
import heurams.kernel.particles as pt
|
|
||||||
from .base_puzzle_widget import BasePuzzleWidget
|
|
||||||
from textual.message import Message
|
from textual.message import Message
|
||||||
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label, Static
|
||||||
|
|
||||||
|
import heurams.kernel.particles as pt
|
||||||
|
|
||||||
|
from .base_puzzle_widget import BasePuzzleWidget
|
||||||
|
|
||||||
|
|
||||||
class BasicEvaluation(BasePuzzleWidget):
|
class BasicEvaluation(BasePuzzleWidget):
|
||||||
@@ -51,7 +49,7 @@ class BasicEvaluation(BasePuzzleWidget):
|
|||||||
# 显示主要内容
|
# 显示主要内容
|
||||||
yield Label(self.atom.registry["nucleon"]["content"], id="main")
|
yield Label(self.atom.registry["nucleon"]["content"], id="main")
|
||||||
|
|
||||||
# 显示评估说明(可选)
|
# 显示评估说明(可选)
|
||||||
yield Static("请评估你对这个内容的记忆程度: ", classes="instruction")
|
yield Static("请评估你对这个内容的记忆程度: ", classes="instruction")
|
||||||
|
|
||||||
# 按钮容器
|
# 按钮容器
|
||||||
|
|||||||
@@ -1,17 +1,17 @@
|
|||||||
from textual.widgets import (
|
|
||||||
Label,
|
|
||||||
Button,
|
|
||||||
)
|
|
||||||
from textual.widget import Widget
|
|
||||||
import heurams.kernel.particles as pt
|
|
||||||
import heurams.kernel.puzzles as pz
|
|
||||||
from .base_puzzle_widget import BasePuzzleWidget
|
|
||||||
import copy
|
import copy
|
||||||
import random
|
import random
|
||||||
|
from typing import TypedDict
|
||||||
|
|
||||||
from textual.containers import Container
|
from textual.containers import Container
|
||||||
from textual.message import Message
|
from textual.message import Message
|
||||||
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label
|
||||||
|
|
||||||
|
import heurams.kernel.particles as pt
|
||||||
|
import heurams.kernel.puzzles as pz
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
from typing import TypedDict
|
|
||||||
|
from .base_puzzle_widget import BasePuzzleWidget
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
from textual.widgets import (
|
|
||||||
Label,
|
|
||||||
Button,
|
|
||||||
)
|
|
||||||
from textual.widget import Widget
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label
|
||||||
|
|
||||||
|
|
||||||
class Finished(Widget):
|
class Finished(Widget):
|
||||||
|
|||||||
@@ -1,17 +1,17 @@
|
|||||||
# 单项选择题
|
# 单项选择题
|
||||||
from textual.widgets import (
|
from typing import TypedDict
|
||||||
Label,
|
|
||||||
Button,
|
from textual.containers import Container, ScrollableContainer
|
||||||
)
|
|
||||||
from textual.containers import ScrollableContainer, Container
|
|
||||||
from textual.widget import Widget
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.kernel.puzzles as pz
|
import heurams.kernel.puzzles as pz
|
||||||
from .base_puzzle_widget import BasePuzzleWidget
|
|
||||||
from typing import TypedDict
|
|
||||||
from heurams.services.hasher import hash
|
from heurams.services.hasher import hash
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base_puzzle_widget import BasePuzzleWidget
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -61,11 +61,16 @@ class MCQPuzzle(BasePuzzleWidget):
|
|||||||
self.puzzle.refresh()
|
self.puzzle.refresh()
|
||||||
|
|
||||||
def compose(self):
|
def compose(self):
|
||||||
self.atom.registry["nucleon"].do_eval()
|
|
||||||
setting: Setting = self.atom.registry["nucleon"].metadata["orbital"]["puzzles"][
|
setting: Setting = self.atom.registry["nucleon"].metadata["orbital"]["puzzles"][
|
||||||
self.alia
|
self.alia
|
||||||
]
|
]
|
||||||
logger.debug(f"Puzzle Setting: {setting}")
|
logger.debug(f"Puzzle Setting: {setting}")
|
||||||
|
logger.debug(f"WIRED INDEX: {len(self.inputlist)}")
|
||||||
|
if len(self.inputlist) > len(self.puzzle.options):
|
||||||
|
logger.debug("ERR IDX")
|
||||||
|
logger.debug(self.inputlist)
|
||||||
|
logger.debug(self.puzzle.options)
|
||||||
|
else:
|
||||||
current_options = self.puzzle.options[len(self.inputlist)]
|
current_options = self.puzzle.options[len(self.inputlist)]
|
||||||
yield Label(setting["primary"], id="sentence")
|
yield Label(setting["primary"], id="sentence")
|
||||||
yield Label(self.puzzle.wording[len(self.inputlist)], id="puzzle")
|
yield Label(self.puzzle.wording[len(self.inputlist)], id="puzzle")
|
||||||
@@ -116,7 +121,7 @@ class MCQPuzzle(BasePuzzleWidget):
|
|||||||
|
|
||||||
self.screen.rating = rating # type: ignore
|
self.screen.rating = rating # type: ignore
|
||||||
self.handler(rating)
|
self.handler(rating)
|
||||||
# 重置输入(如果回答错误)
|
# 重置输入(如果回答错误)
|
||||||
if not is_correct:
|
if not is_correct:
|
||||||
self.inputlist = []
|
self.inputlist = []
|
||||||
self.refresh_buttons()
|
self.refresh_buttons()
|
||||||
@@ -127,7 +132,7 @@ class MCQPuzzle(BasePuzzleWidget):
|
|||||||
self.update_display()
|
self.update_display()
|
||||||
|
|
||||||
def refresh_buttons(self):
|
def refresh_buttons(self):
|
||||||
"""刷新按钮显示(用于题目切换)"""
|
"""刷新按钮显示(用于题目切换)"""
|
||||||
# 移除所有选项按钮
|
# 移除所有选项按钮
|
||||||
logger.debug("刷新按钮")
|
logger.debug("刷新按钮")
|
||||||
self.cursor += 1
|
self.cursor += 1
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
from textual.widgets import (
|
|
||||||
Label,
|
|
||||||
Button,
|
|
||||||
)
|
|
||||||
from textual.widget import Widget
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label
|
||||||
|
|
||||||
|
|
||||||
class Placeholder(Widget):
|
class Placeholder(Widget):
|
||||||
|
|||||||
@@ -1,20 +1,17 @@
|
|||||||
from textual.reactive import reactive
|
|
||||||
from textual.widgets import (
|
|
||||||
Markdown,
|
|
||||||
Label,
|
|
||||||
Static,
|
|
||||||
Button,
|
|
||||||
)
|
|
||||||
from textual.containers import Center
|
|
||||||
from textual.widget import Widget
|
|
||||||
from typing import Dict
|
|
||||||
import heurams.kernel.particles as pt
|
|
||||||
import re
|
import re
|
||||||
from .base_puzzle_widget import BasePuzzleWidget
|
from typing import Dict, List, TypedDict
|
||||||
from typing import TypedDict, List
|
|
||||||
|
from textual.containers import Center
|
||||||
from textual.message import Message
|
from textual.message import Message
|
||||||
|
from textual.reactive import reactive
|
||||||
|
from textual.widget import Widget
|
||||||
|
from textual.widgets import Button, Label, Markdown, Static
|
||||||
|
|
||||||
|
import heurams.kernel.particles as pt
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base_puzzle_widget import BasePuzzleWidget
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -52,6 +49,10 @@ class Recognition(BasePuzzleWidget):
|
|||||||
self.alia = alia
|
self.alia = alia
|
||||||
|
|
||||||
def compose(self):
|
def compose(self):
|
||||||
|
from heurams.context import config_var
|
||||||
|
autovoice = config_var.get()['interface']['memorizor']['autovoice']
|
||||||
|
if autovoice:
|
||||||
|
self.screen.action_play_voice() # type: ignore
|
||||||
cfg: RecognitionConfig = self.atom.registry["orbital"]["puzzles"][self.alia]
|
cfg: RecognitionConfig = self.atom.registry["orbital"]["puzzles"][self.alia]
|
||||||
delim = self.atom.registry["nucleon"].metadata["formation"]["delimiter"]
|
delim = self.atom.registry["nucleon"].metadata["formation"]["delimiter"]
|
||||||
replace_dict = {
|
replace_dict = {
|
||||||
@@ -71,7 +72,8 @@ class Recognition(BasePuzzleWidget):
|
|||||||
primary = cfg["primary"]
|
primary = cfg["primary"]
|
||||||
|
|
||||||
with Center():
|
with Center():
|
||||||
yield Static(f"[dim]{cfg['top_dim']}[/]")
|
for i in cfg['top_dim']:
|
||||||
|
yield Static(f"[dim]{i}[/]")
|
||||||
yield Label("")
|
yield Label("")
|
||||||
|
|
||||||
for old, new in replace_dict.items():
|
for old, new in replace_dict.items():
|
||||||
|
|||||||
@@ -1,6 +1,8 @@
|
|||||||
from .sm2 import SM2Algorithm
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .sm2 import SM2Algorithm
|
||||||
|
from .sm15m import SM15MAlgorithm
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
@@ -9,7 +11,8 @@ __all__ = [
|
|||||||
|
|
||||||
algorithms = {
|
algorithms = {
|
||||||
"SM-2": SM2Algorithm,
|
"SM-2": SM2Algorithm,
|
||||||
"supermemo2": SM2Algorithm,
|
"SM-15M": SM15MAlgorithm,
|
||||||
|
# "SM-15M": SM15MAlgorithm,
|
||||||
}
|
}
|
||||||
|
|
||||||
logger.debug("算法模块初始化完成, 注册的算法: %s", list(algorithms.keys()))
|
logger.debug("算法模块初始化完成, 注册的算法: %s", list(algorithms.keys()))
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import heurams.services.timer as timer
|
|
||||||
from typing import TypedDict
|
from typing import TypedDict
|
||||||
|
|
||||||
|
import heurams.services.timer as timer
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
283
src/heurams/kernel/algorithms/sm15m.py
Normal file
283
src/heurams/kernel/algorithms/sm15m.py
Normal file
@@ -0,0 +1,283 @@
|
|||||||
|
"""
|
||||||
|
SM-15 接口兼容实现, 基于 SM-15 算法的逆向工程
|
||||||
|
全局状态保存在文件中, 项目状态通过 algodata 字典传递
|
||||||
|
|
||||||
|
基于: https://github.com/slaypni/sm.js
|
||||||
|
原始 CoffeeScript 代码: (c) 2014 Kazuaki Tanida
|
||||||
|
MIT 许可证
|
||||||
|
"""
|
||||||
|
|
||||||
|
import datetime
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
from typing import TypedDict
|
||||||
|
import pathlib
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.kernel.algorithms.sm15m_calc import (MAX_AF, MIN_AF, NOTCH_AF,
|
||||||
|
RANGE_AF, RANGE_REPETITION,
|
||||||
|
SM, THRESHOLD_RECALL, Item)
|
||||||
|
|
||||||
|
# 全局状态文件路径
|
||||||
|
_GLOBAL_STATE_FILE = os.path.expanduser(pathlib.Path(config_var.get()['paths']['global_dir']) / 'sm15m_global_state.json')
|
||||||
|
|
||||||
|
|
||||||
|
def _get_global_sm():
|
||||||
|
"""获取全局 SM 实例, 从文件加载或创建新的"""
|
||||||
|
if os.path.exists(_GLOBAL_STATE_FILE):
|
||||||
|
try:
|
||||||
|
with open(_GLOBAL_STATE_FILE, "r", encoding="utf-8") as f:
|
||||||
|
data = json.load(f)
|
||||||
|
sm_instance = SM.load(data)
|
||||||
|
return sm_instance
|
||||||
|
except Exception:
|
||||||
|
# 如果加载失败, 创建新的实例
|
||||||
|
pass
|
||||||
|
|
||||||
|
# 创建新的 SM 实例
|
||||||
|
sm_instance = SM()
|
||||||
|
# 保存初始状态
|
||||||
|
_save_global_sm(sm_instance)
|
||||||
|
return sm_instance
|
||||||
|
|
||||||
|
|
||||||
|
def _save_global_sm(sm_instance):
|
||||||
|
"""保存全局 SM 实例到文件"""
|
||||||
|
try:
|
||||||
|
data = sm_instance.data()
|
||||||
|
with open(_GLOBAL_STATE_FILE, "w", encoding="utf-8") as f:
|
||||||
|
json.dump(data, f, indent=2)
|
||||||
|
except Exception:
|
||||||
|
# 忽略保存错误
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class SM15MAlgorithm:
|
||||||
|
algo_name = "SM-15M"
|
||||||
|
|
||||||
|
class AlgodataDict(TypedDict):
|
||||||
|
efactor: float
|
||||||
|
real_rept: int
|
||||||
|
rept: int
|
||||||
|
interval: int
|
||||||
|
last_date: int
|
||||||
|
next_date: int
|
||||||
|
is_activated: int
|
||||||
|
last_modify: float
|
||||||
|
|
||||||
|
defaults = {
|
||||||
|
"efactor": 2.5,
|
||||||
|
"real_rept": 0,
|
||||||
|
"rept": 0,
|
||||||
|
"interval": 0,
|
||||||
|
"last_date": 0,
|
||||||
|
"next_date": 0,
|
||||||
|
"is_activated": 0,
|
||||||
|
"last_modify": 0.0,
|
||||||
|
}
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _get_timestamp(cls):
|
||||||
|
"""获取当前时间戳(秒)"""
|
||||||
|
return datetime.datetime.now().timestamp()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _get_daystamp(cls):
|
||||||
|
"""获取当前天数戳(从某个纪元开始的天数)"""
|
||||||
|
# 使用与原始 SM-2 相同的纪元:1970-01-01
|
||||||
|
now = datetime.datetime.now()
|
||||||
|
epoch = datetime.datetime(1970, 1, 1)
|
||||||
|
delta = now - epoch
|
||||||
|
return delta.days
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _algodata_to_item(cls, algodata, sm_instance):
|
||||||
|
"""将 algodata 转换为 Item 实例"""
|
||||||
|
# 从 algodata 获取 SM-2 数据
|
||||||
|
sm15_data = algodata.get(cls.algo_name, cls.defaults.copy())
|
||||||
|
|
||||||
|
# 创建 Item 实例
|
||||||
|
item = Item(sm_instance)
|
||||||
|
|
||||||
|
# 映射字段
|
||||||
|
# efactor -> A-Factor (需要转换)
|
||||||
|
efactor = sm15_data.get("efactor", 2.5)
|
||||||
|
# SM-2 的 efactor 范围 [1.3, 2.5+], SM-15 的 A-Factor 范围 [1.2, 6.9]
|
||||||
|
# 简单线性映射:af = (efactor - 1.3) * (MAX_AF - MIN_AF) / (2.5 - 1.3) + MIN_AF
|
||||||
|
# 但 efactor 可能大于 2.5, 所以需要限制
|
||||||
|
af = max(MIN_AF, min(MAX_AF, efactor * 2.0)) # 粗略映射
|
||||||
|
# 调试
|
||||||
|
# print(f"DEBUG: efactor={efactor}, af before set={af}")
|
||||||
|
item.af(af)
|
||||||
|
# print(f"DEBUG: item.af() after set={item.af()}")
|
||||||
|
|
||||||
|
# rept -> repetition (成功回忆次数)
|
||||||
|
rept = sm15_data.get("rept", 0)
|
||||||
|
item.repetition = (
|
||||||
|
rept - 1 if rept > 0 else -1
|
||||||
|
) # SM-15 中 repetition=-1 表示新项目
|
||||||
|
|
||||||
|
# real_rept -> lapse? 或者忽略
|
||||||
|
real_rept = sm15_data.get("real_rept", 0)
|
||||||
|
# 可以存储在 value 中或忽略
|
||||||
|
|
||||||
|
# interval -> optimum_interval (需要从天数转换为毫秒)
|
||||||
|
interval_days = sm15_data.get("interval", 0)
|
||||||
|
if interval_days == 0:
|
||||||
|
item.optimum_interval = sm_instance.interval_base
|
||||||
|
else:
|
||||||
|
item.optimum_interval = interval_days * 24 * 60 * 60 * 1000 # 天转毫秒
|
||||||
|
|
||||||
|
# last_date -> previous_date
|
||||||
|
last_date_days = sm15_data.get("last_date", 0)
|
||||||
|
if last_date_days > 0:
|
||||||
|
epoch = datetime.datetime(1970, 1, 1)
|
||||||
|
item.previous_date = epoch + datetime.timedelta(days=last_date_days)
|
||||||
|
|
||||||
|
# next_date -> due_date
|
||||||
|
next_date_days = sm15_data.get("next_date", 0)
|
||||||
|
if next_date_days > 0:
|
||||||
|
epoch = datetime.datetime(1970, 1, 1)
|
||||||
|
item.due_date = epoch + datetime.timedelta(days=next_date_days)
|
||||||
|
|
||||||
|
# is_activated 和 last_modify 忽略
|
||||||
|
|
||||||
|
# 将原始 algodata 保存在 value 中以便恢复
|
||||||
|
item.value = {
|
||||||
|
"front": "SM-15 item",
|
||||||
|
"back": "SM-15 item",
|
||||||
|
"_sm15_data": sm15_data,
|
||||||
|
}
|
||||||
|
|
||||||
|
return item
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _item_to_algodata(cls, item, algodata):
|
||||||
|
"""将 Item 实例状态写回 algodata"""
|
||||||
|
if cls.algo_name not in algodata:
|
||||||
|
algodata[cls.algo_name] = cls.defaults.copy()
|
||||||
|
|
||||||
|
sm15_data = algodata[cls.algo_name]
|
||||||
|
|
||||||
|
# A-Factor -> efactor (反向映射)
|
||||||
|
af = item.af()
|
||||||
|
if af is None:
|
||||||
|
af = MIN_AF
|
||||||
|
# 反向粗略映射
|
||||||
|
efactor = max(1.3, min(af / 2.0, 10.0)) # 限制范围
|
||||||
|
# 调试
|
||||||
|
# print(f"DEBUG: item.af()={af}, computed efactor={efactor}")
|
||||||
|
sm15_data["efactor"] = efactor
|
||||||
|
|
||||||
|
# repetition -> rept
|
||||||
|
rept = item.repetition + 1 if item.repetition >= 0 else 0
|
||||||
|
sm15_data["rept"] = rept
|
||||||
|
|
||||||
|
# real_rept: 递增在 revisor 中处理, 这里保持不变
|
||||||
|
# 但如果没有 real_rept 字段, 则初始化为0
|
||||||
|
if "real_rept" not in sm15_data:
|
||||||
|
sm15_data["real_rept"] = 0
|
||||||
|
|
||||||
|
# optimum_interval -> interval (毫秒转天)
|
||||||
|
interval_ms = item.optimum_interval
|
||||||
|
if interval_ms == item.sm.interval_base:
|
||||||
|
sm15_data["interval"] = 0
|
||||||
|
else:
|
||||||
|
interval_days = max(0, round(interval_ms / (24 * 60 * 60 * 1000)))
|
||||||
|
sm15_data["interval"] = interval_days
|
||||||
|
|
||||||
|
# previous_date -> last_date
|
||||||
|
if item.previous_date:
|
||||||
|
epoch = datetime.datetime(1970, 1, 1)
|
||||||
|
last_date_days = (item.previous_date - epoch).days
|
||||||
|
sm15_data["last_date"] = last_date_days
|
||||||
|
else:
|
||||||
|
sm15_data["last_date"] = 0
|
||||||
|
|
||||||
|
# due_date -> next_date
|
||||||
|
if item.due_date:
|
||||||
|
epoch = datetime.datetime(1970, 1, 1)
|
||||||
|
next_date_days = (item.due_date - epoch).days
|
||||||
|
sm15_data["next_date"] = next_date_days
|
||||||
|
else:
|
||||||
|
sm15_data["next_date"] = 0
|
||||||
|
|
||||||
|
# is_activated: 保持不变或设为1
|
||||||
|
if "is_activated" not in sm15_data:
|
||||||
|
sm15_data["is_activated"] = 1
|
||||||
|
|
||||||
|
# last_modify: 更新时间戳
|
||||||
|
sm15_data["last_modify"] = cls._get_timestamp()
|
||||||
|
|
||||||
|
return algodata
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def revisor(
|
||||||
|
cls, algodata: dict, feedback: int = 5, is_new_activation: bool = False
|
||||||
|
):
|
||||||
|
"""SM-15 算法迭代决策机制实现"""
|
||||||
|
# 获取全局 SM 实例
|
||||||
|
sm_instance = _get_global_sm()
|
||||||
|
|
||||||
|
# 将 algodata 转换为 Item
|
||||||
|
item = cls._algodata_to_item(algodata, sm_instance)
|
||||||
|
|
||||||
|
# 处理 is_new_activation
|
||||||
|
if is_new_activation:
|
||||||
|
# 重置为初始状态
|
||||||
|
item.repetition = -1
|
||||||
|
item.lapse = 0
|
||||||
|
item.optimum_interval = sm_instance.interval_base
|
||||||
|
item.previous_date = None
|
||||||
|
item.due_date = datetime.datetime.fromtimestamp(0)
|
||||||
|
item.af(2.5) # 重置 efactor
|
||||||
|
|
||||||
|
# 将项目临时添加到 SM 实例(以便 answer 更新共享状态)
|
||||||
|
sm_instance.q.append(item)
|
||||||
|
|
||||||
|
# 处理反馈(评分)
|
||||||
|
# SM-2 的 feedback 是 0-5, SM-15 的 grade 也是 0-5
|
||||||
|
grade = feedback
|
||||||
|
now = datetime.datetime.now()
|
||||||
|
|
||||||
|
# 调用 answer 方法
|
||||||
|
item.answer(grade, now)
|
||||||
|
|
||||||
|
# 更新共享状态(FI-Graph, ForgettingCurves, OFM)
|
||||||
|
if item.repetition >= 0:
|
||||||
|
sm_instance.forgetting_curves.register_point(grade, item, now)
|
||||||
|
sm_instance.ofm.update()
|
||||||
|
sm_instance.fi_g.update(grade, item, now)
|
||||||
|
|
||||||
|
# 从队列中移除项目
|
||||||
|
sm_instance.q.remove(item)
|
||||||
|
|
||||||
|
# 保存全局状态
|
||||||
|
_save_global_sm(sm_instance)
|
||||||
|
|
||||||
|
# 将更新后的 Item 状态写回 algodata
|
||||||
|
cls._item_to_algodata(item, algodata)
|
||||||
|
|
||||||
|
# 更新 real_rept(总复习次数)
|
||||||
|
algodata[cls.algo_name]["real_rept"] += 1
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def is_due(cls, algodata):
|
||||||
|
"""检查项目是否到期"""
|
||||||
|
sm15_data = algodata.get(cls.algo_name, cls.defaults.copy())
|
||||||
|
next_date_days = sm15_data.get("next_date", 0)
|
||||||
|
current_daystamp = cls._get_daystamp()
|
||||||
|
return next_date_days <= current_daystamp
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def rate(cls, algodata):
|
||||||
|
"""获取项目的评分(返回 efactor 字符串)"""
|
||||||
|
sm15_data = algodata.get(cls.algo_name, cls.defaults.copy())
|
||||||
|
efactor = sm15_data.get("efactor", 2.5)
|
||||||
|
return str(efactor)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def nextdate(cls, algodata) -> int:
|
||||||
|
"""获取下次复习日期(天数戳)"""
|
||||||
|
sm15_data = algodata.get(cls.algo_name, cls.defaults.copy())
|
||||||
|
next_date_days = sm15_data.get("next_date", 0)
|
||||||
|
return next_date_days
|
||||||
1573
src/heurams/kernel/algorithms/sm15m_calc.py
Normal file
1573
src/heurams/kernel/algorithms/sm15m_calc.py
Normal file
File diff suppressed because it is too large
Load Diff
@@ -1,8 +1,10 @@
|
|||||||
from .base import BaseAlgorithm
|
|
||||||
import heurams.services.timer as timer
|
|
||||||
from typing import TypedDict
|
from typing import TypedDict
|
||||||
|
|
||||||
|
import heurams.services.timer as timer
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base import BaseAlgorithm
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -9,12 +9,12 @@ from heurams.services.logger import get_logger
|
|||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
logger.debug("粒子模块已加载")
|
logger.debug("粒子模块已加载")
|
||||||
|
|
||||||
|
from .atom import Atom, atom_registry
|
||||||
from .electron import Electron
|
from .electron import Electron
|
||||||
|
from .loader import load_electron, load_nucleon
|
||||||
from .nucleon import Nucleon
|
from .nucleon import Nucleon
|
||||||
from .orbital import Orbital
|
from .orbital import Orbital
|
||||||
from .atom import Atom, atom_registry
|
|
||||||
from .probe import probe_all, probe_by_filename
|
from .probe import probe_all, probe_by_filename
|
||||||
from .loader import load_nucleon, load_electron
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"Electron",
|
"Electron",
|
||||||
|
|||||||
@@ -1,14 +1,17 @@
|
|||||||
|
import json
|
||||||
|
import pathlib
|
||||||
|
import typing
|
||||||
|
from typing import TypedDict
|
||||||
|
|
||||||
|
import bidict
|
||||||
|
import toml
|
||||||
|
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
from .electron import Electron
|
from .electron import Electron
|
||||||
from .nucleon import Nucleon
|
from .nucleon import Nucleon
|
||||||
from .orbital import Orbital
|
from .orbital import Orbital
|
||||||
from typing import TypedDict
|
|
||||||
import pathlib
|
|
||||||
import typing
|
|
||||||
import toml
|
|
||||||
import json
|
|
||||||
import bidict
|
|
||||||
from heurams.context import config_var
|
|
||||||
from heurams.services.logger import get_logger
|
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
@@ -59,7 +62,6 @@ class Atom:
|
|||||||
"orbital_fmt": "toml",
|
"orbital_fmt": "toml",
|
||||||
"runtime": {"locked": False, "min_rate": 0x3F3F3F3F, "newact": False},
|
"runtime": {"locked": False, "min_rate": 0x3F3F3F3F, "newact": False},
|
||||||
}
|
}
|
||||||
self.do_eval()
|
|
||||||
logger.debug("Atom 初始化完成")
|
logger.debug("Atom 初始化完成")
|
||||||
|
|
||||||
def link(self, key, value):
|
def link(self, key, value):
|
||||||
@@ -67,14 +69,63 @@ class Atom:
|
|||||||
if key in self.registry.keys():
|
if key in self.registry.keys():
|
||||||
self.registry[key] = value
|
self.registry[key] = value
|
||||||
logger.debug("键 '%s' 已链接, 触发 do_eval", key)
|
logger.debug("键 '%s' 已链接, 触发 do_eval", key)
|
||||||
self.do_eval()
|
if key == "electron":
|
||||||
if key == 'electron':
|
if self.registry["electron"].is_activated() == 0:
|
||||||
if self.registry['electron'].is_activated() == 0:
|
self.registry["runtime"]["newact"] = True
|
||||||
self.registry['runtime']['newact'] = True
|
|
||||||
else:
|
else:
|
||||||
logger.error("尝试链接不受支持的键: '%s'", key)
|
logger.error("尝试链接不受支持的键: '%s'", key)
|
||||||
raise ValueError("不受支持的原子元数据链接操作")
|
raise ValueError("不受支持的原子元数据链接操作")
|
||||||
|
|
||||||
|
def do_eval(self):
|
||||||
|
"""
|
||||||
|
执行并以结果替换当前单元的所有 eval 语句
|
||||||
|
TODO: 带有限制的 eval, 异步/多线程执行避免堵塞
|
||||||
|
"""
|
||||||
|
logger.debug("EVAL 开始")
|
||||||
|
|
||||||
|
# eval 环境设置
|
||||||
|
def eval_with_env(s: str):
|
||||||
|
default = config_var.get()["puzzles"]
|
||||||
|
payload = self.registry['nucleon'].payload
|
||||||
|
metadata = self.registry['nucleon'].metadata
|
||||||
|
eval_value = eval(s)
|
||||||
|
if isinstance(eval_value, (int, float)):
|
||||||
|
ret = str(eval_value)
|
||||||
|
else:
|
||||||
|
ret = eval_value
|
||||||
|
logger.debug(
|
||||||
|
"eval 执行成功: '%s' -> '%s'",
|
||||||
|
s,
|
||||||
|
str(ret)[:50] + "..." if len(ret) > 50 else ret,
|
||||||
|
)
|
||||||
|
return ret
|
||||||
|
|
||||||
|
def traverse(data, modifier):
|
||||||
|
if isinstance(data, dict):
|
||||||
|
for key, value in data.items():
|
||||||
|
data[key] = traverse(value, modifier)
|
||||||
|
return data
|
||||||
|
elif isinstance(data, list):
|
||||||
|
for i, item in enumerate(data):
|
||||||
|
data[i] = traverse(item, modifier)
|
||||||
|
return data
|
||||||
|
elif isinstance(data, tuple):
|
||||||
|
return tuple(traverse(item, modifier) for item in data)
|
||||||
|
else:
|
||||||
|
if isinstance(data, str):
|
||||||
|
if data.startswith("eval:"):
|
||||||
|
logger.debug("发现 eval 表达式: '%s'", data[5:])
|
||||||
|
return modifier(data[5:])
|
||||||
|
return data
|
||||||
|
try:
|
||||||
|
traverse(self.registry['nucleon'].payload, eval_with_env)
|
||||||
|
traverse(self.registry['nucleon'].metadata, eval_with_env)
|
||||||
|
traverse(self.registry['orbital'], eval_with_env)
|
||||||
|
except Exception as e:
|
||||||
|
ret = f"此 eval 实例发生错误: {e}"
|
||||||
|
logger.warning(ret)
|
||||||
|
logger.debug("EVAL 完成")
|
||||||
|
|
||||||
def minimize(self, rating):
|
def minimize(self, rating):
|
||||||
"""效果等同于 self.registry['runtime']['min_rate'] = min(rating, self.registry['runtime']['min_rate'])
|
"""效果等同于 self.registry['runtime']['min_rate'] = min(rating, self.registry['runtime']['min_rate'])
|
||||||
|
|
||||||
@@ -104,84 +155,13 @@ class Atom:
|
|||||||
"""
|
"""
|
||||||
if self.registry["runtime"]["locked"]:
|
if self.registry["runtime"]["locked"]:
|
||||||
logger.debug(f"允许总评分: {self.registry['runtime']['min_rate']}")
|
logger.debug(f"允许总评分: {self.registry['runtime']['min_rate']}")
|
||||||
self.registry["electron"].revisor(self.registry["runtime"]["min_rate"], is_new_activation=self.registry["runtime"]["newact"])
|
self.registry["electron"].revisor(
|
||||||
|
self.registry["runtime"]["min_rate"],
|
||||||
|
is_new_activation=self.registry["runtime"]["newact"],
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
logger.debug("禁止总评分")
|
logger.debug("禁止总评分")
|
||||||
|
|
||||||
def do_eval(self):
|
|
||||||
"""
|
|
||||||
执行并以结果替换当前单元的所有 eval 语句
|
|
||||||
TODO: 带有限制的 eval, 异步/多线程执行避免堵塞
|
|
||||||
"""
|
|
||||||
logger.debug("Atom.do_eval 开始")
|
|
||||||
|
|
||||||
# eval 环境设置
|
|
||||||
def eval_with_env(s: str):
|
|
||||||
# 初始化默认值
|
|
||||||
nucleon = self.registry["nucleon"]
|
|
||||||
default = {}
|
|
||||||
metadata = {}
|
|
||||||
try:
|
|
||||||
default = config_var.get()["puzzles"]
|
|
||||||
metadata = nucleon.metadata
|
|
||||||
except Exception:
|
|
||||||
# 如果无法获取配置或元数据, 使用空字典
|
|
||||||
logger.debug("无法获取配置或元数据, 使用空字典")
|
|
||||||
pass
|
|
||||||
try:
|
|
||||||
eval_value = eval(s)
|
|
||||||
if isinstance(eval_value, (list, dict)):
|
|
||||||
ret = eval_value
|
|
||||||
else:
|
|
||||||
ret = str(eval_value)
|
|
||||||
logger.debug(
|
|
||||||
"eval 执行成功: '%s' -> '%s'",
|
|
||||||
s,
|
|
||||||
str(ret)[:50] + "..." if len(ret) > 50 else ret,
|
|
||||||
)
|
|
||||||
except Exception as e:
|
|
||||||
ret = f"此 eval 实例发生错误: {e}"
|
|
||||||
logger.warning("eval 执行错误: '%s' -> %s", s, e)
|
|
||||||
return ret
|
|
||||||
|
|
||||||
def traverse(data, modifier):
|
|
||||||
if isinstance(data, dict):
|
|
||||||
for key, value in data.items():
|
|
||||||
data[key] = traverse(value, modifier)
|
|
||||||
return data
|
|
||||||
elif isinstance(data, list):
|
|
||||||
for i, item in enumerate(data):
|
|
||||||
data[i] = traverse(item, modifier)
|
|
||||||
return data
|
|
||||||
elif isinstance(data, tuple):
|
|
||||||
return tuple(traverse(item, modifier) for item in data)
|
|
||||||
else:
|
|
||||||
if isinstance(data, str):
|
|
||||||
if data.startswith("eval:"):
|
|
||||||
logger.debug("发现 eval 表达式: '%s'", data[5:])
|
|
||||||
return modifier(data[5:])
|
|
||||||
return data
|
|
||||||
|
|
||||||
# 如果 nucleon 存在且有 do_eval 方法, 调用它
|
|
||||||
nucleon = self.registry["nucleon"]
|
|
||||||
if nucleon is not None and hasattr(nucleon, "do_eval"):
|
|
||||||
nucleon.do_eval()
|
|
||||||
logger.debug("已调用 nucleon.do_eval")
|
|
||||||
|
|
||||||
# 如果 electron 存在且其 algodata 包含 eval 字符串, 遍历它
|
|
||||||
electron = self.registry["electron"]
|
|
||||||
if electron is not None and hasattr(electron, "algodata"):
|
|
||||||
traverse(electron.algodata, eval_with_env)
|
|
||||||
logger.debug("已处理 electron algodata eval")
|
|
||||||
|
|
||||||
# 如果 orbital 存在且是字典, 遍历它
|
|
||||||
orbital = self.registry["orbital"]
|
|
||||||
if orbital is not None and isinstance(orbital, dict):
|
|
||||||
traverse(orbital, eval_with_env)
|
|
||||||
logger.debug("orbital eval 完成")
|
|
||||||
|
|
||||||
logger.debug("Atom.do_eval 完成")
|
|
||||||
|
|
||||||
def persist(self, key):
|
def persist(self, key):
|
||||||
logger.debug("Atom.persist: key='%s'", key)
|
logger.debug("Atom.persist: key='%s'", key)
|
||||||
path: pathlib.Path | None = self.registry[key + "_path"]
|
path: pathlib.Path | None = self.registry[key + "_path"]
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ logger = get_logger(__name__)
|
|||||||
class Electron:
|
class Electron:
|
||||||
"""电子: 记忆分析元数据及算法"""
|
"""电子: 记忆分析元数据及算法"""
|
||||||
|
|
||||||
def __init__(self, ident: str, algodata: dict = {}, algo_name: str = "supermemo2"):
|
def __init__(self, ident: str, algodata: dict = {}, algo_name: str = ""):
|
||||||
"""初始化电子对象 (记忆数据)
|
"""初始化电子对象 (记忆数据)
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
@@ -17,19 +17,21 @@ class Electron:
|
|||||||
algodata: 算法数据字典, 包含算法的各项参数和设置
|
algodata: 算法数据字典, 包含算法的各项参数和设置
|
||||||
algo: 使用的算法模块标识
|
algo: 使用的算法模块标识
|
||||||
"""
|
"""
|
||||||
|
if algo_name == "":
|
||||||
|
algo_name = config_var.get()['algorithm']['default']
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"创建 Electron 实例, ident: '%s', algo_name: '%s'", ident, algo_name
|
"创建 Electron 实例, ident: '%s', algo_name: '%s', algodata: %s", ident, algo_name, algodata
|
||||||
)
|
)
|
||||||
self.algodata = algodata
|
self.algodata = algodata
|
||||||
self.ident = ident
|
self.ident = ident
|
||||||
self.algo = algorithms[algo_name]
|
self.algo = algorithms[algo_name]
|
||||||
logger.debug("使用的算法类: %s", self.algo.__name__)
|
logger.debug("使用的算法类: %s", self.algo.__name__)
|
||||||
|
|
||||||
if self.algo not in self.algodata.keys():
|
if self.algo.algo_name not in self.algodata.keys():
|
||||||
self.algodata[self.algo.algo_name] = {}
|
self.algodata[self.algo.algo_name] = {}
|
||||||
logger.debug("算法键 '%s' 不存在, 已创建空字典", self.algo)
|
logger.debug("算法键 '%s' 不存在, 已创建空字典", self.algo)
|
||||||
if not self.algodata[self.algo.algo_name]:
|
if not self.algodata[self.algo.algo_name]:
|
||||||
logger.debug("算法数据为空, 使用默认值初始化")
|
logger.debug(f"算法数据为空, 使用默认值初始化{self.algodata[self.algo.algo_name]}")
|
||||||
self._default_init(self.algo.defaults)
|
self._default_init(self.algo.defaults)
|
||||||
else:
|
else:
|
||||||
logger.debug("算法数据已存在, 跳过默认初始化")
|
logger.debug("算法数据已存在, 跳过默认初始化")
|
||||||
|
|||||||
@@ -1,12 +1,15 @@
|
|||||||
from .nucleon import Nucleon
|
|
||||||
from .electron import Electron
|
|
||||||
import heurams.services.hasher as hasher
|
|
||||||
import pathlib
|
|
||||||
import toml
|
|
||||||
import json
|
import json
|
||||||
|
import pathlib
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
|
|
||||||
|
import toml
|
||||||
|
|
||||||
|
import heurams.services.hasher as hasher
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .electron import Electron
|
||||||
|
from .nucleon import Nucleon
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -65,7 +68,7 @@ def load_electron(path: pathlib.Path, fmt="json") -> dict:
|
|||||||
logger.debug("JSON 解析成功, keys: %s", list(dictdata.keys()))
|
logger.debug("JSON 解析成功, keys: %s", list(dictdata.keys()))
|
||||||
dic = dict()
|
dic = dict()
|
||||||
for item, attr in dictdata.items():
|
for item, attr in dictdata.items():
|
||||||
logger.debug("处理电子项目: %s", item)
|
logger.debug("处理电子项目: %s, %s", item, attr)
|
||||||
dic[item] = Electron(item, attr)
|
dic[item] = Electron(item, attr)
|
||||||
logger.debug("load_electron 完成, 加载了 %d 个 Electron 对象", len(dic))
|
logger.debug("load_electron 完成, 加载了 %d 个 Electron 对象", len(dic))
|
||||||
return dic
|
return dic
|
||||||
|
|||||||
@@ -49,54 +49,6 @@ class Nucleon:
|
|||||||
def __hash__(self):
|
def __hash__(self):
|
||||||
return hash(self.ident)
|
return hash(self.ident)
|
||||||
|
|
||||||
def do_eval(self):
|
|
||||||
"""
|
|
||||||
执行并以结果替换当前单元的所有 eval 语句
|
|
||||||
TODO: 带有限制的 eval, 异步/多线程执行避免堵塞
|
|
||||||
"""
|
|
||||||
logger.debug("Nucleon.do_eval 开始")
|
|
||||||
|
|
||||||
# eval 环境设置
|
|
||||||
def eval_with_env(s: str):
|
|
||||||
try:
|
|
||||||
nucleon = self
|
|
||||||
eval_value = eval(s)
|
|
||||||
if isinstance(eval_value, (int, float)):
|
|
||||||
ret = str(eval_value)
|
|
||||||
else:
|
|
||||||
ret = eval_value
|
|
||||||
logger.debug(
|
|
||||||
"eval 执行成功: '%s' -> '%s'",
|
|
||||||
s,
|
|
||||||
str(ret)[:50] + "..." if len(ret) > 50 else ret,
|
|
||||||
)
|
|
||||||
except Exception as e:
|
|
||||||
ret = f"此 eval 实例发生错误: {e}"
|
|
||||||
logger.warning("eval 执行错误: '%s' -> %s", s, e)
|
|
||||||
return ret
|
|
||||||
|
|
||||||
def traverse(data, modifier):
|
|
||||||
if isinstance(data, dict):
|
|
||||||
for key, value in data.items():
|
|
||||||
data[key] = traverse(value, modifier)
|
|
||||||
return data
|
|
||||||
elif isinstance(data, list):
|
|
||||||
for i, item in enumerate(data):
|
|
||||||
data[i] = traverse(item, modifier)
|
|
||||||
return data
|
|
||||||
elif isinstance(data, tuple):
|
|
||||||
return tuple(traverse(item, modifier) for item in data)
|
|
||||||
else:
|
|
||||||
if isinstance(data, str):
|
|
||||||
if data.startswith("eval:"):
|
|
||||||
logger.debug("发现 eval 表达式: '%s'", data[5:])
|
|
||||||
return modifier(data[5:])
|
|
||||||
return data
|
|
||||||
|
|
||||||
traverse(self.payload, eval_with_env)
|
|
||||||
traverse(self.metadata, eval_with_env)
|
|
||||||
logger.debug("Nucleon.do_eval 完成")
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def placeholder():
|
def placeholder():
|
||||||
"""生成一个占位原子核"""
|
"""生成一个占位原子核"""
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
from typing import TypedDict
|
from typing import TypedDict
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
from heurams.context import config_var
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
|
from heurams.context import config_var
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ class BasePuzzle:
|
|||||||
"""谜题基类"""
|
"""谜题基类"""
|
||||||
|
|
||||||
def refresh(self):
|
def refresh(self):
|
||||||
logger.debug("BasePuzzle.refresh 被调用(未实现)")
|
logger.debug("BasePuzzle.refresh 被调用(未实现)")
|
||||||
raise NotImplementedError("谜题对象未实现 refresh 方法")
|
raise NotImplementedError("谜题对象未实现 refresh 方法")
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
|
|||||||
@@ -1,7 +1,9 @@
|
|||||||
from .base import BasePuzzle
|
|
||||||
import random
|
import random
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base import BasePuzzle
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
# mcq.py
|
# mcq.py
|
||||||
from .base import BasePuzzle
|
|
||||||
import random
|
import random
|
||||||
from typing import List, Dict, Optional, Union
|
from typing import Dict, List, Optional, Union
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base import BasePuzzle
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
# mcq.py
|
# mcq.py
|
||||||
from .base import BasePuzzle
|
|
||||||
import random
|
import random
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .base import BasePuzzle
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -14,5 +16,5 @@ class RecognitionPuzzle(BasePuzzle):
|
|||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
def refresh(self):
|
def refresh(self):
|
||||||
logger.debug("RecognitionPuzzle.refresh(空实现)")
|
logger.debug("RecognitionPuzzle.refresh(空实现)")
|
||||||
pass
|
pass
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
from .states import PhaserState, ProcessionState
|
from heurams.services.logger import get_logger
|
||||||
from .procession import Procession
|
|
||||||
from .fission import Fission
|
from .fission import Fission
|
||||||
from .phaser import Phaser
|
from .phaser import Phaser
|
||||||
from heurams.services.logger import get_logger
|
from .procession import Procession
|
||||||
|
from .states import PhaserState, ProcessionState
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
|
import random
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
import heurams.kernel.puzzles as puz
|
import heurams.kernel.puzzles as puz
|
||||||
import random
|
|
||||||
from .states import PhaserState
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .states import PhaserState
|
||||||
|
|
||||||
|
|
||||||
class Fission:
|
class Fission:
|
||||||
"""裂变器: 单原子调度展开器"""
|
"""裂变器: 单原子调度展开器"""
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
# 移相器类定义
|
# 移相器类定义
|
||||||
|
|
||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
from .states import PhaserState, ProcessionState
|
|
||||||
from .procession import Procession
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .procession import Procession
|
||||||
|
from .states import PhaserState, ProcessionState
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,8 @@
|
|||||||
import heurams.kernel.particles as pt
|
import heurams.kernel.particles as pt
|
||||||
from .states import PhaserState, ProcessionState
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from .states import PhaserState, ProcessionState
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -51,7 +52,7 @@ class Procession:
|
|||||||
self.queue.append(atom)
|
self.queue.append(atom)
|
||||||
logger.debug("原子已追加到队列, 新队列长度=%d", len(self.queue))
|
logger.debug("原子已追加到队列, 新队列长度=%d", len(self.queue))
|
||||||
else:
|
else:
|
||||||
logger.debug("原子未追加(重复或队列长度<=1)")
|
logger.debug("原子未追加(重复或队列长度<=1)")
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
length = len(self.queue) - self.cursor
|
length = len(self.queue) - self.cursor
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
from enum import Enum, auto
|
from enum import Enum, auto
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
# 音频播放器, 必须基于文件操作
|
# 音频播放器, 必须基于文件操作
|
||||||
from . import termux_audio
|
|
||||||
from . import playsound_audio
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
from . import playsound_audio, termux_audio
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
|
|||||||
@@ -5,7 +5,9 @@
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
import playsound
|
import playsound
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
from typing import Protocol
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
from typing import Protocol
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -5,6 +5,7 @@
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -2,4 +2,4 @@ from heurams.services.logger import get_logger
|
|||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
logger.debug("OpenAI provider 模块已加载(未实现)")
|
logger.debug("OpenAI provider 模块已加载(未实现)")
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
from .base import BaseTTS
|
from .base import BaseTTS
|
||||||
from .edge_tts import EdgeTTS
|
from .edge_tts import EdgeTTS
|
||||||
from heurams.services.logger import get_logger
|
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,7 +1,11 @@
|
|||||||
from .base import BaseTTS
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
import edge_tts
|
import edge_tts
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
from heurams.context import config_var
|
||||||
|
|
||||||
|
from .base import BaseTTS
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
@@ -15,7 +19,7 @@ class EdgeTTS(BaseTTS):
|
|||||||
try:
|
try:
|
||||||
communicate = edge_tts.Communicate(
|
communicate = edge_tts.Communicate(
|
||||||
text,
|
text,
|
||||||
"zh-CN-YunjianNeural",
|
config_var.get()['providers']['tts']['edgetts']["voice"],
|
||||||
)
|
)
|
||||||
logger.debug("EdgeTTS 通信对象创建成功, 正在保存音频")
|
logger.debug("EdgeTTS 通信对象创建成功, 正在保存音频")
|
||||||
communicate.save_sync(str(path))
|
communicate.save_sync(str(path))
|
||||||
|
|||||||
@@ -1,12 +1,13 @@
|
|||||||
# 音频服务
|
# 音频服务
|
||||||
|
from typing import Callable
|
||||||
|
|
||||||
from heurams.context import config_var
|
from heurams.context import config_var
|
||||||
from heurams.providers.audio import providers as prov
|
from heurams.providers.audio import providers as prov
|
||||||
from typing import Callable
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
play_by_path: Callable = prov[config_var.get()["services"]["audio"]].play_by_path
|
play_by_path: Callable = prov[config_var.get()["services"]["audio"]].play_by_path
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"音频服务初始化完成, 使用 provider: %s", config_var.get()["services"]["audio"]
|
"音频服务初始化完成, 使用 Provider: %s", config_var.get()["services"]["audio"]
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,7 +1,9 @@
|
|||||||
# 配置文件服务
|
# 配置文件服务
|
||||||
import pathlib
|
import pathlib
|
||||||
import toml
|
|
||||||
import typing
|
import typing
|
||||||
|
|
||||||
|
import toml
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
# 哈希服务
|
# 哈希服务
|
||||||
import hashlib
|
import hashlib
|
||||||
|
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -45,7 +45,7 @@ def setup_logging(
|
|||||||
# 创建formatter
|
# 创建formatter
|
||||||
formatter = logging.Formatter(log_format, date_format)
|
formatter = logging.Formatter(log_format, date_format)
|
||||||
|
|
||||||
# 创建文件handler(使用RotatingFileHandler防止日志过大)
|
# 创建文件handler(使用RotatingFileHandler防止日志过大)
|
||||||
file_handler = logging.handlers.RotatingFileHandler(
|
file_handler = logging.handlers.RotatingFileHandler(
|
||||||
filename=log_path,
|
filename=log_path,
|
||||||
maxBytes=max_bytes,
|
maxBytes=max_bytes,
|
||||||
@@ -55,7 +55,7 @@ def setup_logging(
|
|||||||
file_handler.setFormatter(formatter)
|
file_handler.setFormatter(formatter)
|
||||||
file_handler.setLevel(log_level)
|
file_handler.setLevel(log_level)
|
||||||
|
|
||||||
# 配置root logger - 设置为 WARNING 级别(只记录重要信息)
|
# 配置root logger - 设置为 WARNING 级别(只记录重要信息)
|
||||||
root_logger = logging.getLogger()
|
root_logger = logging.getLogger()
|
||||||
root_logger.setLevel(logging.WARNING) # 这里改为 WARNING
|
root_logger.setLevel(logging.WARNING) # 这里改为 WARNING
|
||||||
|
|
||||||
@@ -63,7 +63,7 @@ def setup_logging(
|
|||||||
for handler in root_logger.handlers[:]:
|
for handler in root_logger.handlers[:]:
|
||||||
root_logger.removeHandler(handler)
|
root_logger.removeHandler(handler)
|
||||||
|
|
||||||
# 创建自己的应用logger(单独设置DEBUG级别)
|
# 创建自己的应用logger(单独设置DEBUG级别)
|
||||||
app_logger = logging.getLogger("heurams")
|
app_logger = logging.getLogger("heurams")
|
||||||
app_logger.setLevel(log_level) # 保持DEBUG级别
|
app_logger.setLevel(log_level) # 保持DEBUG级别
|
||||||
app_logger.addHandler(file_handler)
|
app_logger.addHandler(file_handler)
|
||||||
@@ -146,7 +146,7 @@ def exception(msg: str, *args, **kwargs) -> None:
|
|||||||
get_logger().exception(msg, *args, **kwargs)
|
get_logger().exception(msg, *args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
# 初始化日志系统(硬编码配置)
|
# 初始化日志系统(硬编码配置)
|
||||||
setup_logging()
|
setup_logging()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
# 时间服务
|
# 时间服务
|
||||||
from heurams.context import config_var
|
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
from heurams.context import config_var
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|||||||
@@ -1,12 +1,13 @@
|
|||||||
# 文本转语音服务
|
# 文本转语音服务
|
||||||
from heurams.context import config_var
|
|
||||||
from heurams.providers.tts import TTSs
|
|
||||||
from typing import Callable
|
from typing import Callable
|
||||||
|
|
||||||
|
from heurams.context import config_var
|
||||||
|
from heurams.providers.tts import providers as prov
|
||||||
from heurams.services.logger import get_logger
|
from heurams.services.logger import get_logger
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
convert: Callable = TTSs[config_var.get().get("tts_provider")]
|
convertor: Callable = prov[config_var.get()["services"]["tts"]].convert
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"TTS服务初始化完成, 使用 provider: %s", config_var.get().get("tts_provider")
|
"TTS服务初始化完成, 使用 provider: %s", config_var.get()["services"]["tts"]
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ from heurams.services.logger import get_logger
|
|||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
|
||||||
ver = "0.4.0"
|
ver = "0.4.2"
|
||||||
stage = "prototype"
|
stage = "prototype"
|
||||||
codename = "fledge" # 雏鸟, 0.4.x 版本
|
codename = "fledge" # 雏鸟, 0.4.x 版本
|
||||||
|
|
||||||
|
|||||||
@@ -2,21 +2,22 @@
|
|||||||
"""
|
"""
|
||||||
DashboardScreen 的测试, 包括单元测试和 pilot 测试.
|
DashboardScreen 的测试, 包括单元测试和 pilot 测试.
|
||||||
"""
|
"""
|
||||||
import unittest
|
|
||||||
import tempfile
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
import tempfile
|
||||||
import time
|
import time
|
||||||
from unittest.mock import patch, MagicMock
|
import unittest
|
||||||
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from textual.pilot import Pilot
|
from textual.pilot import Pilot
|
||||||
|
|
||||||
from heurams.context import ConfigContext
|
from heurams.context import ConfigContext
|
||||||
from heurams.services.config import ConfigFile
|
|
||||||
from heurams.interface.__main__ import HeurAMSApp
|
from heurams.interface.__main__ import HeurAMSApp
|
||||||
from heurams.interface.screens.dashboard import DashboardScreen
|
from heurams.interface.screens.dashboard import DashboardScreen
|
||||||
|
from heurams.services.config import ConfigFile
|
||||||
|
|
||||||
|
|
||||||
class TestDashboardScreenUnit(unittest.TestCase):
|
class TestDashboardScreenUnit(unittest.TestCase):
|
||||||
"""DashboardScreen 的单元测试(不启动完整应用). """
|
"""DashboardScreen 的单元测试(不启动完整应用)."""
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
"""在每个测试之前运行, 设置临时目录和配置."""
|
"""在每个测试之前运行, 设置临时目录和配置."""
|
||||||
@@ -66,7 +67,7 @@ class TestDashboardScreenUnit(unittest.TestCase):
|
|||||||
result = screen.compose()
|
result = screen.compose()
|
||||||
widgets = list(result)
|
widgets = list(result)
|
||||||
# 检查是否包含 Header 和 Footer
|
# 检查是否包含 Header 和 Footer
|
||||||
from textual.widgets import Header, Footer
|
from textual.widgets import Footer, Header
|
||||||
|
|
||||||
header_present = any(isinstance(w, Header) for w in widgets)
|
header_present = any(isinstance(w, Header) for w in widgets)
|
||||||
footer_present = any(isinstance(w, Footer) for w in widgets)
|
footer_present = any(isinstance(w, Footer) for w in widgets)
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from heurams.kernel.algorithms.sm2 import SM2Algorithm
|
from heurams.kernel.algorithms.sm2 import SM2Algorithm
|
||||||
|
|
||||||
@@ -94,7 +94,7 @@ class TestSM2Algorithm(unittest.TestCase):
|
|||||||
SM2Algorithm.revisor(algodata, feedback=5, is_new_activation=True)
|
SM2Algorithm.revisor(algodata, feedback=5, is_new_activation=True)
|
||||||
self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 0)
|
self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 0)
|
||||||
self.assertEqual(algodata[SM2Algorithm.algo_name]["efactor"], 2.5)
|
self.assertEqual(algodata[SM2Algorithm.algo_name]["efactor"], 2.5)
|
||||||
# interval 应为 1(因为 rept=0)
|
# interval 应为 1(因为 rept=0)
|
||||||
self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 1)
|
self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 1)
|
||||||
|
|
||||||
def test_revisor_efactor_calculation(self):
|
def test_revisor_efactor_calculation(self):
|
||||||
|
|||||||
@@ -1,15 +1,16 @@
|
|||||||
import unittest
|
import json
|
||||||
from unittest.mock import patch, MagicMock
|
|
||||||
import pathlib
|
import pathlib
|
||||||
import tempfile
|
import tempfile
|
||||||
import toml
|
import unittest
|
||||||
import json
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
|
import toml
|
||||||
|
|
||||||
|
from heurams.context import ConfigContext
|
||||||
from heurams.kernel.particles.atom import Atom, atom_registry
|
from heurams.kernel.particles.atom import Atom, atom_registry
|
||||||
from heurams.kernel.particles.electron import Electron
|
from heurams.kernel.particles.electron import Electron
|
||||||
from heurams.kernel.particles.nucleon import Nucleon
|
from heurams.kernel.particles.nucleon import Nucleon
|
||||||
from heurams.kernel.particles.orbital import Orbital
|
from heurams.kernel.particles.orbital import Orbital
|
||||||
from heurams.context import ConfigContext
|
|
||||||
from heurams.services.config import ConfigFile
|
from heurams.services.config import ConfigFile
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
import unittest
|
|
||||||
from unittest.mock import patch, MagicMock
|
|
||||||
import sys
|
import sys
|
||||||
|
import unittest
|
||||||
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from heurams.kernel.particles.electron import Electron
|
|
||||||
from heurams.kernel.algorithms import algorithms
|
from heurams.kernel.algorithms import algorithms
|
||||||
|
from heurams.kernel.particles.electron import Electron
|
||||||
|
|
||||||
|
|
||||||
class TestElectron(unittest.TestCase):
|
class TestElectron(unittest.TestCase):
|
||||||
@@ -27,7 +27,7 @@ class TestElectron(unittest.TestCase):
|
|||||||
self.assertEqual(electron.algo, algorithms["supermemo2"])
|
self.assertEqual(electron.algo, algorithms["supermemo2"])
|
||||||
self.assertIn(electron.algo, electron.algodata)
|
self.assertIn(electron.algo, electron.algodata)
|
||||||
self.assertIsInstance(electron.algodata[electron.algo], dict)
|
self.assertIsInstance(electron.algodata[electron.algo], dict)
|
||||||
# 检查默认值(排除动态字段)
|
# 检查默认值(排除动态字段)
|
||||||
defaults = electron.algo.defaults
|
defaults = electron.algo.defaults
|
||||||
for key, value in defaults.items():
|
for key, value in defaults.items():
|
||||||
if key == "last_modify":
|
if key == "last_modify":
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from heurams.kernel.particles.nucleon import Nucleon
|
from heurams.kernel.particles.nucleon import Nucleon
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from heurams.kernel.puzzles.cloze import ClozePuzzle
|
from heurams.kernel.puzzles.cloze import ClozePuzzle
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import patch, MagicMock, call
|
from unittest.mock import MagicMock, call, patch
|
||||||
|
|
||||||
from heurams.kernel.puzzles.mcq import MCQPuzzle
|
from heurams.kernel.puzzles.mcq import MCQPuzzle
|
||||||
|
|
||||||
@@ -48,7 +48,7 @@ class TestMCQPuzzle(unittest.TestCase):
|
|||||||
# 模拟 random.sample 返回前两个映射项
|
# 模拟 random.sample 返回前两个映射项
|
||||||
mock_sample.side_effect = [
|
mock_sample.side_effect = [
|
||||||
[("q1", "a1"), ("q2", "a2")], # 选择问题
|
[("q1", "a1"), ("q2", "a2")], # 选择问题
|
||||||
["j1", "j2", "j3"], # 为每个问题选择干扰项(实际调用两次)
|
["j1", "j2", "j3"], # 为每个问题选择干扰项(实际调用两次)
|
||||||
]
|
]
|
||||||
puzzle.refresh()
|
puzzle.refresh()
|
||||||
|
|
||||||
@@ -59,7 +59,7 @@ class TestMCQPuzzle(unittest.TestCase):
|
|||||||
self.assertEqual(puzzle.answer, ["a1", "a2"])
|
self.assertEqual(puzzle.answer, ["a1", "a2"])
|
||||||
# 检查 options 列表
|
# 检查 options 列表
|
||||||
self.assertEqual(len(puzzle.options), 2)
|
self.assertEqual(len(puzzle.options), 2)
|
||||||
# 每个选项列表应包含 4 个选项(正确答案 + 3 个干扰项)
|
# 每个选项列表应包含 4 个选项(正确答案 + 3 个干扰项)
|
||||||
self.assertEqual(len(puzzle.options[0]), 4)
|
self.assertEqual(len(puzzle.options[0]), 4)
|
||||||
self.assertEqual(len(puzzle.options[1]), 4)
|
self.assertEqual(len(puzzle.options[1]), 4)
|
||||||
# random.shuffle 应被调用
|
# random.shuffle 应被调用
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import Mock, patch, MagicMock
|
from unittest.mock import MagicMock, Mock, patch
|
||||||
|
|
||||||
from heurams.kernel.reactor.phaser import Phaser
|
|
||||||
from heurams.kernel.reactor.states import PhaserState, ProcessionState
|
|
||||||
from heurams.kernel.particles.atom import Atom
|
from heurams.kernel.particles.atom import Atom
|
||||||
from heurams.kernel.particles.electron import Electron
|
from heurams.kernel.particles.electron import Electron
|
||||||
|
from heurams.kernel.reactor.phaser import Phaser
|
||||||
|
from heurams.kernel.reactor.states import PhaserState, ProcessionState
|
||||||
|
|
||||||
|
|
||||||
class TestPhaser(unittest.TestCase):
|
class TestPhaser(unittest.TestCase):
|
||||||
|
|||||||
Reference in New Issue
Block a user