gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🤖 自动格式化代码 [skip ci]

Browse Source
This commit is contained in:
github-actions[bot]
2025-07-01 06:47:24 +00:00
parent 0c0ae96655
commit 7efe17a9c8
5 changed files with 102 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/person_info/relationship_builder_manager.py
View File

@@ -4,37 +4,37 @@ from .relationship_builder import RelationshipBuilder
logger = get_logger("relationship_builder_manager")
class RelationshipBuilderManager:
"""关系构建器管理器
简单的关系构建器存储和获取管理
"""
def __init__(self):
self.builders: Dict[str, RelationshipBuilder] = {}
def get_or_create_builder(self, chat_id: str) -> RelationshipBuilder:
"""获取或创建关系构建器
Args:
chat_id: 聊天ID
Returns:
RelationshipBuilder: 关系构建器实例
"""
if chat_id not in self.builders:
self.builders[chat_id] = RelationshipBuilder(chat_id)
logger.info(f"创建聊天 {chat_id} 的关系构建器")
return self.builders[chat_id]
def get_builder(self, chat_id: str) -> Optional[RelationshipBuilder]:
"""获取关系构建器
Args:
chat_id: 聊天ID
Returns:
Optional[RelationshipBuilder]: 关系构建器实例或None
"""
@@ -42,10 +42,10 @@ class RelationshipBuilderManager:
def remove_builder(self, chat_id: str) -> bool:
"""移除关系构建器
Args:
chat_id: 聊天ID
Returns:
bool: 是否成功移除
"""
@@ -57,7 +57,7 @@ class RelationshipBuilderManager:
def get_all_chat_ids(self) -> List[str]:
"""获取所有管理的聊天ID列表
Returns:
List[str]: 聊天ID列表
"""
@@ -65,7 +65,7 @@ class RelationshipBuilderManager:
def get_status(self) -> Dict[str, any]:
"""获取管理器状态
Returns:
Dict[str, any]: 状态信息
"""
@@ -76,7 +76,7 @@ class RelationshipBuilderManager:
async def process_chat_messages(self, chat_id: str):
"""处理指定聊天的消息
Args:
chat_id: 聊天ID
"""
@@ -85,11 +85,11 @@ class RelationshipBuilderManager:
async def force_cleanup_user(self, chat_id: str, person_id: str) -> bool:
"""强制清理指定用户的关系构建缓存
Args:
chat_id: 聊天ID
person_id: 用户ID
Returns:
bool: 是否成功清理
"""
@@ -100,4 +100,4 @@ class RelationshipBuilderManager:
# 全局管理器实例
relationship_builder_manager = RelationshipBuilderManager()
relationship_builder_manager = RelationshipBuilderManager()