71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
from __future__ import annotations
|
||
|
||
import html
|
||
import xml.etree.ElementTree as ET
|
||
from typing import Any, Callable, Dict
|
||
|
||
from wechat_ipad.models.message import MessageType
|
||
|
||
|
||
def parse_quote_context(full_msg: Any, room_id: str, get_sender_name: Callable[[str, str], str]) -> Dict[str, str]:
|
||
if not full_msg or not getattr(full_msg, "content", None):
|
||
return {}
|
||
xml_content = getattr(full_msg.content, "xml_content", "") or ""
|
||
if not xml_content:
|
||
return {}
|
||
try:
|
||
root = ET.fromstring(xml_content)
|
||
except ET.ParseError:
|
||
return {}
|
||
|
||
appmsg = root.find(".//appmsg")
|
||
if appmsg is None or appmsg.findtext("type", "").strip() != "57":
|
||
return {}
|
||
|
||
refer = appmsg.find("refermsg")
|
||
if refer is None:
|
||
return {}
|
||
|
||
title = html.unescape(appmsg.findtext("title", "") or "").strip()
|
||
quote_sender_name = html.unescape(refer.findtext("displayname", "") or "").strip()
|
||
if not quote_sender_name:
|
||
quote_sender = html.unescape(refer.findtext("chatusr", "") or "").strip()
|
||
quote_sender_name = get_sender_name(room_id, quote_sender) if quote_sender else "未知成员"
|
||
ref_type = int(refer.findtext("type", "0") or 0)
|
||
ref_content = html.unescape(refer.findtext("content", "") or "").strip()
|
||
quote_type_label = quote_type_label_for(ref_type)
|
||
quote_body = build_quote_body(ref_type, ref_content, title)
|
||
return {
|
||
"title": title,
|
||
"quote_sender_name": quote_sender_name,
|
||
"quote_type_label": quote_type_label,
|
||
"quote_body": quote_body,
|
||
"raw_ref_content": ref_content,
|
||
}
|
||
|
||
|
||
def quote_type_label_for(ref_type: int) -> str:
|
||
mapping = {
|
||
MessageType.TEXT.value: "引用文本",
|
||
MessageType.IMAGE.value: "引用图片",
|
||
MessageType.VIDEO.value: "引用视频",
|
||
MessageType.APP.value: "引用应用消息",
|
||
MessageType.EMOTICON.value: "引用表情",
|
||
}
|
||
return mapping.get(ref_type, f"引用消息[{ref_type}]")
|
||
|
||
|
||
def build_quote_body(ref_type: int, ref_content: str, title: str) -> str:
|
||
if ref_type == MessageType.TEXT.value:
|
||
return ref_content[:220].strip()
|
||
if ref_type == MessageType.IMAGE.value:
|
||
details = []
|
||
if title:
|
||
details.append(f"当前追问文案：{title}")
|
||
if ref_content:
|
||
details.append("被引用的是一张图片")
|
||
return "；".join(details) or "被引用的是一张图片"
|
||
if title:
|
||
return title[:220].strip()
|
||
return ref_content[:220].strip()
|