102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
from copy import deepcopy
|
|
from typing import TypedDict
|
|
|
|
import heurams.kernel.algorithms as algolib
|
|
import heurams.services.timer as timer
|
|
from heurams.kernel.algorithms import algorithms
|
|
from heurams.services.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class Electron:
|
|
"""电子: 单算法支持的记忆数据包装"""
|
|
|
|
def __init__(self, ident: str, algodata: dict = {}, algo_name: str = ""):
|
|
"""初始化电子对象 (记忆数据)
|
|
|
|
Args:
|
|
ident: 算法的唯一标识符, 用于区分不同的算法实例, 使用 algodata[ident] 获取
|
|
algodata: 算法数据字典引用, 包含算法的各项参数和设置
|
|
algo_name: 使用的算法模块标识
|
|
"""
|
|
if algo_name == "":
|
|
algo_name = "sm-2"
|
|
self.algodata = algodata
|
|
self.ident = ident
|
|
self.algo: algolib.BaseAlgorithm = algorithms[algo_name]
|
|
|
|
if not self.algo.check_integrity(self.algodata):
|
|
self.algodata[self.algo.algo_name] = deepcopy(self.algo.defaults)
|
|
|
|
def activate(self):
|
|
"""激活此电子"""
|
|
self.algodata[self.algo.algo_name]["is_activated"] = 1
|
|
self.algodata[self.algo.algo_name]["last_modify"] = timer.get_timestamp()
|
|
|
|
def modify(self, key, value):
|
|
"""修改 algodata[algo] 中子字典数据"""
|
|
if key in self.algodata[self.algo.algo_name]:
|
|
self.algodata[self.algo.algo_name][key] = value
|
|
self.algodata[self.algo.algo_name]["last_modify"] = timer.get_timestamp()
|
|
else:
|
|
raise AttributeError("不存在的子键")
|
|
|
|
def is_due(self):
|
|
"""是否应该复习"""
|
|
result = self.algo.is_due(self.algodata)
|
|
return result and self.is_activated()
|
|
|
|
def is_activated(self):
|
|
result = self.algodata[self.algo.algo_name]["is_activated"]
|
|
return result
|
|
|
|
def get_rating(self):
|
|
try:
|
|
result = self.algo.get_rating(self.algodata)
|
|
return result
|
|
except:
|
|
return 0
|
|
|
|
def nextdate(self) -> int:
|
|
result = self.algo.nextdate(self.algodata)
|
|
return result
|
|
|
|
def revisor(self, quality: int = 5, is_new_activation: bool = False):
|
|
"""算法迭代决策机制实现
|
|
|
|
Args:
|
|
quality (int): 记忆保留率量化参数 (0-5)
|
|
is_new_activation (bool): 是否为初次激活
|
|
"""
|
|
self.algo.revisor(self.algodata, quality, is_new_activation)
|
|
|
|
def __hash__(self):
|
|
return hash(self.ident)
|
|
|
|
def __getitem__(self, key):
|
|
if key == "ident":
|
|
return self.ident
|
|
if key in self.algodata[self.algo.algo_name]:
|
|
return self.algodata[self.algo.algo_name][key]
|
|
else:
|
|
raise KeyError(f"键 '{key}' 未在 algodata[self.algo] 中")
|
|
|
|
def __setitem__(self, key, value):
|
|
if key == "ident":
|
|
raise AttributeError("ident 应为只读")
|
|
self.algodata[self.algo.algo_name][key] = value
|
|
self.algodata[self.algo.algo_name]["last_modify"] = timer.get_timestamp()
|
|
|
|
def __len__(self):
|
|
"""仅返回当前算法的配置数量"""
|
|
return len(self.algodata[self.algo.algo_name])
|
|
|
|
@staticmethod
|
|
def create_on_electonic_data(electronic_data: tuple, algo_name: str = ""):
|
|
_data = electronic_data
|
|
ident = _data[0]
|
|
algodata = _data[1]
|
|
ident = ident
|
|
return Electron(ident, algodata, algo_name)
|