73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""NapCat 通知事件资料补全器。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, Optional
|
|
|
|
from ...services import NapCatQueryService
|
|
from .helpers import normalize_optional_string
|
|
|
|
|
|
class NapCatNoticeEntityResolver:
|
|
"""为通知事件补全用户和群资料。"""
|
|
|
|
def __init__(self, query_service: NapCatQueryService) -> None:
|
|
"""初始化实体补全器。
|
|
|
|
Args:
|
|
query_service: NapCat 查询服务。
|
|
"""
|
|
self._query_service = query_service
|
|
|
|
async def build_user_info(self, group_id: str, user_id: str) -> Dict[str, Optional[str]]:
|
|
"""构造通知消息的用户信息。
|
|
|
|
Args:
|
|
group_id: 群号；私聊或系统通知时为空字符串。
|
|
user_id: 事件关联用户号。
|
|
|
|
Returns:
|
|
Dict[str, Optional[str]]: 规范化后的用户信息字典。
|
|
"""
|
|
if not user_id:
|
|
return {
|
|
"user_id": "notice",
|
|
"user_nickname": "系统通知",
|
|
"user_cardname": None,
|
|
}
|
|
|
|
member_info: Optional[Dict[str, Any]]
|
|
if group_id:
|
|
member_info = await self._query_service.get_group_member_info(group_id, user_id)
|
|
else:
|
|
member_info = await self._query_service.get_stranger_info(user_id)
|
|
|
|
if member_info is None:
|
|
return {
|
|
"user_id": user_id,
|
|
"user_nickname": user_id,
|
|
"user_cardname": None,
|
|
}
|
|
|
|
return {
|
|
"user_id": user_id,
|
|
"user_nickname": str(member_info.get("nickname") or user_id),
|
|
"user_cardname": normalize_optional_string(member_info.get("card")),
|
|
}
|
|
|
|
async def build_group_info(self, group_id: str) -> Optional[Dict[str, str]]:
|
|
"""构造通知消息的群信息。
|
|
|
|
Args:
|
|
group_id: 群号。
|
|
|
|
Returns:
|
|
Optional[Dict[str, str]]: 群信息字典；若不是群通知则返回 ``None``。
|
|
"""
|
|
if not group_id:
|
|
return None
|
|
|
|
group_info = await self._query_service.get_group_info(group_id)
|
|
group_name = str(group_info.get("group_name") or f"group_{group_id}") if group_info else f"group_{group_id}"
|
|
return {"group_id": group_id, "group_name": group_name}
|