99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
"""
|
||
event class
|
||
"""
|
||
from dataclasses import dataclass, field
|
||
from typing import List, Optional
|
||
import uuid
|
||
import time
|
||
from datetime import datetime
|
||
|
||
from src.classes.calendar import Month, Year, MonthStamp
|
||
|
||
@dataclass
|
||
class Event:
|
||
month_stamp: MonthStamp
|
||
content: str
|
||
# 相关角色ID列表；若与任何角色无关则为 None
|
||
related_avatars: Optional[List[str]] = None
|
||
# 是否为大事（长期记忆），默认False（小事/短期记忆）
|
||
is_major: bool = False
|
||
# 是否为故事事件（不进入记忆索引），默认False
|
||
is_story: bool = False
|
||
# 唯一ID，用于去重
|
||
id: str = field(default_factory=lambda: str(uuid.uuid4()))
|
||
# 创建时间戳 (Unix timestamp float)
|
||
created_at: float = field(default_factory=time.time)
|
||
|
||
def __str__(self) -> str:
|
||
year = self.month_stamp.get_year()
|
||
month = self.month_stamp.get_month()
|
||
return f"{year}年{month}月: {self.content}"
|
||
|
||
def to_dict(self) -> dict:
|
||
"""转换为可序列化的字典"""
|
||
return {
|
||
"month_stamp": int(self.month_stamp),
|
||
"content": self.content,
|
||
"related_avatars": self.related_avatars,
|
||
"is_major": self.is_major,
|
||
"is_story": self.is_story,
|
||
"id": self.id,
|
||
"created_at": self.created_at
|
||
}
|
||
|
||
@classmethod
|
||
def from_dict(cls, data: dict) -> "Event":
|
||
"""从字典重建Event"""
|
||
return cls(
|
||
month_stamp=MonthStamp(data["month_stamp"]),
|
||
content=data["content"],
|
||
related_avatars=data.get("related_avatars"),
|
||
is_major=data.get("is_major", False),
|
||
is_story=data.get("is_story", False),
|
||
id=data.get("id", str(uuid.uuid4())),
|
||
created_at=data.get("created_at", time.time())
|
||
)
|
||
|
||
class NullEvent:
|
||
"""
|
||
空事件单例类，保持与 Event 相同的最小接口，避免调用方访问属性时报错。
|
||
"""
|
||
_instance = None
|
||
|
||
def __new__(cls):
|
||
if cls._instance is None:
|
||
cls._instance = super().__new__(cls)
|
||
# 初始化一次即可
|
||
cls._instance.month_stamp = MonthStamp(0)
|
||
cls._instance.content = ""
|
||
cls._instance.related_avatars = None
|
||
cls._instance.is_major = False
|
||
cls._instance.is_story = False
|
||
cls._instance.id = "NULL_EVENT"
|
||
return cls._instance
|
||
|
||
def __str__(self) -> str:
|
||
return ""
|
||
|
||
def __bool__(self) -> bool:
|
||
"""使NullEvent实例在布尔上下文中为False"""
|
||
return False
|
||
|
||
def to_dict(self) -> dict:
|
||
"""保持序列化接口"""
|
||
return {
|
||
"month_stamp": int(self.month_stamp),
|
||
"content": self.content,
|
||
"related_avatars": self.related_avatars,
|
||
"is_major": self.is_major,
|
||
"is_story": self.is_story,
|
||
"id": self.id,
|
||
}
|
||
|
||
# 全局单例实例
|
||
NULL_EVENT = NullEvent()
|
||
|
||
def is_null_event(event) -> bool:
|
||
"""检查事件是否为空事件的便捷函数"""
|
||
return event is NULL_EVENT
|