141 lines
3.9 KiB
Python
141 lines
3.9 KiB
Python
from wechatauto.utils.tools import (
|
|
detect_message_direction
|
|
)
|
|
from wechatauto import uia
|
|
from .mattr import (
|
|
SystemMessage,
|
|
FriendMessage,
|
|
SelfMessage
|
|
)
|
|
from .mtype import *
|
|
from . import self as selfmsg
|
|
from . import friend as friendmsg
|
|
from typing import (
|
|
TYPE_CHECKING,
|
|
Literal,
|
|
Dict,
|
|
Any
|
|
)
|
|
import os
|
|
import re
|
|
|
|
if TYPE_CHECKING:
|
|
from wechatauto.ui.chatbox import ChatBox
|
|
|
|
|
|
def parse_msg_attr(
|
|
control: uia.Control,
|
|
parent: 'ChatBox'
|
|
):
|
|
msg_direction_hash = {
|
|
'left': 'friend',
|
|
'right': 'self'
|
|
}
|
|
if control.AutomationId:
|
|
msg_screenshot = control.ScreenShot()
|
|
msg_direction, msg_direction_distence = detect_message_direction(msg_screenshot)
|
|
msg_attr = msg_direction_hash.get(msg_direction)
|
|
os.remove(msg_screenshot)
|
|
|
|
additonal_attr = {
|
|
'direction': msg_direction,
|
|
'direction_distence': msg_direction_distence
|
|
}
|
|
|
|
else:
|
|
msg_attr = 'system'
|
|
|
|
if msg_attr == 'system':
|
|
return SystemMessage(control, parent)
|
|
elif msg_attr == 'friend':
|
|
return parse_msg_type(control, parent, 'Friend', additonal_attr)
|
|
elif msg_attr == 'self':
|
|
return parse_msg_type(control, parent, 'Self', additonal_attr)
|
|
|
|
|
|
def parse_msg_type(
|
|
control: uia.Control,
|
|
parent,
|
|
attr: Literal['Self', 'Friend'],
|
|
additonal_attr: Dict[str, Any]
|
|
):
|
|
"""
|
|
多层次消息类型识别算法
|
|
基于ClassName、Name等多重验证确保识别准确性
|
|
"""
|
|
if attr == 'Friend':
|
|
msgtype = friendmsg
|
|
else:
|
|
msgtype = selfmsg
|
|
|
|
msg_text = control.Name
|
|
msg_classname = control.ClassName
|
|
msg_automation_id = control.AutomationId
|
|
|
|
# 第一层：ClassName强特征识别（最可靠）
|
|
classname_result = _classify_by_classname(msg_classname)
|
|
if classname_result:
|
|
return getattr(msgtype, f'{attr}{classname_result}')(control, parent, additonal_attr)
|
|
|
|
# 第二层：基于ClassName分类后的详细识别
|
|
if msg_classname == "mmui::ChatBubbleItemView":
|
|
# Name前缀特征识别
|
|
prefix_result = _classify_by_name_prefix(msg_text)
|
|
if prefix_result:
|
|
return getattr(msgtype, f'{attr}{prefix_result}')(control, parent, additonal_attr)
|
|
|
|
# 图片消息处理
|
|
if msg_text == '图片':
|
|
return getattr(msgtype, f'{attr}ImageMessage')(control, parent, additonal_attr)
|
|
|
|
# 如果都不匹配，归类为其他消息
|
|
return getattr(msgtype, f'{attr}OtherMessage')(control, parent, additonal_attr)
|
|
|
|
elif msg_classname == "mmui::ChatTextItemView":
|
|
# 第三层：引用消息处理
|
|
if _is_quote_message(msg_text):
|
|
return getattr(msgtype, f'{attr}QuoteMessage')(control, parent, additonal_attr)
|
|
else:
|
|
return getattr(msgtype, f'{attr}TextMessage')(control, parent, additonal_attr)
|
|
|
|
return getattr(msgtype, f'{attr}OtherMessage')(control, parent, additonal_attr)
|
|
|
|
|
|
def _classify_by_classname(classname: str) -> str:
|
|
classname_mapping = {
|
|
"mmui::ChatVoiceItemView": "VoiceMessage",
|
|
"mmui::ChatPersonalCardItemView": "PersonalCardMessage",
|
|
}
|
|
return classname_mapping.get(classname, "")
|
|
|
|
|
|
def _classify_by_name_prefix(name: str) -> str:
|
|
if name.startswith("[动画表情]") or name.startswith("[动画表情"):
|
|
return "EmojiMessage"
|
|
|
|
if name.startswith("[链接]"):
|
|
return "LinkMessage"
|
|
|
|
elif name.startswith("位置"):
|
|
return "LocationMessage"
|
|
|
|
elif name.startswith("文件\n"):
|
|
return "FileMessage"
|
|
|
|
elif name.startswith("视频"):
|
|
return "VideoMessage"
|
|
|
|
return ""
|
|
|
|
|
|
def _is_quote_message(name: str) -> bool:
|
|
quote_pattern = r'^(.*?)\s*\n引用\s+(.+?)\s+的消息\s*:\s*(.*)$'
|
|
return bool(re.search(quote_pattern, name, re.DOTALL))
|
|
|
|
|
|
def parse_msg(
|
|
control: uia.Control,
|
|
parent
|
|
):
|
|
return parse_msg_attr(control, parent)
|