748 lines
32 KiB
Python
748 lines
32 KiB
Python
import asyncio
|
||
import base64
|
||
import os
|
||
import time
|
||
from asyncio import Future
|
||
from asyncio import Queue, sleep
|
||
from io import BytesIO
|
||
from pathlib import Path
|
||
from typing import Union
|
||
|
||
import aiohttp
|
||
import pysilk
|
||
from loguru import logger
|
||
from pydub import AudioSegment
|
||
from pymediainfo import MediaInfo
|
||
import aiofiles
|
||
|
||
from utils.video_utils import get_first_frame, get_first_frame_bytes
|
||
from utils.trace_context import format_trace_prefix
|
||
from wechat_ipad import UserLoggedOut
|
||
from wechat_ipad.client.base import WechatAPIClientBase
|
||
|
||
|
||
class MessageMixin(WechatAPIClientBase):
|
||
def __init__(self, ip: str, port: int):
|
||
# 初始化消息队列
|
||
super().__init__(ip, port)
|
||
self._message_queue = Queue()
|
||
self._is_processing = False
|
||
self.logging = logger
|
||
|
||
async def _process_message_queue(self):
|
||
"""
|
||
处理消息队列的异步方法
|
||
"""
|
||
if self._is_processing:
|
||
return
|
||
|
||
self._is_processing = True
|
||
while True:
|
||
if self._message_queue.empty():
|
||
self._is_processing = False
|
||
break
|
||
|
||
func, args, kwargs, future = await self._message_queue.get()
|
||
try:
|
||
result = await func(*args, **kwargs)
|
||
future.set_result(result)
|
||
except Exception as e:
|
||
future.set_exception(e)
|
||
finally:
|
||
self._message_queue.task_done()
|
||
await sleep(1) # 消息发送间隔1秒
|
||
|
||
async def _queue_message(self, func, *args, **kwargs):
|
||
"""
|
||
将消息添加到队列
|
||
"""
|
||
future = Future()
|
||
await self._message_queue.put((func, args, kwargs, future))
|
||
|
||
if not self._is_processing:
|
||
asyncio.create_task(self._process_message_queue())
|
||
|
||
return await future
|
||
|
||
async def revoke_message(self, wxid: str, client_msg_id: int, create_time: int, new_msg_id: int) -> bool:
|
||
"""撤回消息。
|
||
{
|
||
"ClientMsgId": 0,
|
||
"CreateTime": 0,
|
||
"NewMsgId": 0,
|
||
"ToUserName": "string",
|
||
"Wxid": "string"
|
||
}
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
client_msg_id (int): 发送消息的返回值
|
||
create_time (int): 发送消息的返回值
|
||
new_msg_id (int): 发送消息的返回值
|
||
|
||
Returns:
|
||
bool: 成功返回True，失败返回False
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToUserName": wxid, "ClientMsgId": client_msg_id,
|
||
"CreateTime": create_time,
|
||
"NewMsgId": new_msg_id}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/Revoke', json=json_param)
|
||
json_resp = await response.json()
|
||
if json_resp.get("Success"):
|
||
self.logging.info("消息撤回成功: 对方wxid:{} ClientMsgId:{} CreateTime:{} NewMsgId:{}",
|
||
wxid,
|
||
client_msg_id,
|
||
create_time,
|
||
new_msg_id) # 确保四个参数都正确传入
|
||
return True
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_text_message(self, wxid: str, content: str, at: Union[list, str] = "") -> tuple[int, int, int]:
|
||
"""发送文本消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
content (str): 消息内容
|
||
at (list, str, optional): 要@的用户
|
||
|
||
Returns:
|
||
tuple[int, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_text_message, wxid, content, at)
|
||
|
||
async def _send_text_message(self, wxid: str, content: str, at: list[str] = None) -> tuple[int, int, int]:
|
||
"""
|
||
实际发送文本消息的方法
|
||
"""
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
if isinstance(at, str):
|
||
at_str = at
|
||
elif isinstance(at, list):
|
||
if at is None:
|
||
at = []
|
||
at_str = ",".join(at)
|
||
else:
|
||
raise ValueError("Argument 'at' should be str or list")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Content": content, "Type": 1, "At": at_str}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendTxt', json=json_param)
|
||
json_resp = await response.json()
|
||
if json_resp.get("Success"):
|
||
# 发送动作也带上 trace_id，便于把“某条入站消息最终发了什么”直接串起来。
|
||
self.logging.info("{}发送文字消息: 对方wxid:{} at:{} 内容:{}",
|
||
format_trace_prefix(), wxid, at, content)
|
||
data = json_resp.get("Data")
|
||
return data.get("List")[0].get("ClientMsgId"), data.get("List")[0].get("CreateTime"), data.get("List")[
|
||
0].get("NewMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_image_message(self, wxid: str, image: Union[str, bytes, os.PathLike]) -> tuple[int, int, int]:
|
||
"""发送图片消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
image (str, byte, os.PathLike): 图片，支持base64字符串，图片byte，图片路径
|
||
|
||
Returns:
|
||
tuple[int, int, int]: 返回(ClientImgId, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
ValueError: image_path和image_base64都为空或都不为空时
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_image_message, wxid, image)
|
||
|
||
async def _send_image_message(self, wxid: str, image: Union[str, bytes, os.PathLike]) -> tuple[
|
||
int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
if isinstance(image, str):
|
||
pass
|
||
elif isinstance(image, bytes):
|
||
image = base64.b64encode(image).decode()
|
||
elif isinstance(image, os.PathLike):
|
||
async with aiofiles.open(image, 'rb') as f:
|
||
image = base64.b64encode(await f.read()).decode()
|
||
else:
|
||
raise ValueError("Argument 'image' can only be str, bytes, or os.PathLike")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": image}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/UploadImg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
# 图片日志不打印 base64 内容，但保留 trace_id，便于关联具体发送动作。
|
||
self.logging.info("{}发送图片消息: 对方wxid:{} 图片base64略",
|
||
format_trace_prefix(), wxid)
|
||
data = json_resp.get("Data")
|
||
self.logging.debug("发送图片消息成功,返回：{}", data)
|
||
return data.get("ClientImgId").get("string"), data.get("CreateTime"), data.get("NewMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_video_message(self, wxid: str, video: Union[str, bytes, os.PathLike],
|
||
image: [str, bytes, os.PathLike] = None):
|
||
"""发送视频消息。不推荐使用，上传速度很慢300KB/s。如要使用，可压缩视频，或者发送链接卡片而不是视频。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
video (str, bytes, os.PathLike): 视频 接受base64字符串，字节，文件路径
|
||
image (str, bytes, os.PathLike): 视频封面图片 接受base64字符串，字节，文件路径
|
||
|
||
Returns:
|
||
tuple[int, int]: 返回(ClientMsgid, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
ValueError: 视频或图片参数都为空或都不为空时
|
||
根据error_handler处理错误
|
||
"""
|
||
has_image = False
|
||
if not image:
|
||
image = Path(os.path.join(Path(__file__).resolve().parent, "fallback.png"))
|
||
else:
|
||
has_image = True
|
||
# get video base64 and duration
|
||
if isinstance(video, str):
|
||
vid_base64 = video
|
||
video = base64.b64decode(video)
|
||
file_len = len(video)
|
||
media_info = MediaInfo.parse(BytesIO(video))
|
||
elif isinstance(video, bytes):
|
||
vid_base64 = base64.b64encode(video).decode()
|
||
file_len = len(video)
|
||
media_info = MediaInfo.parse(BytesIO(video))
|
||
# 如果没有传入首帧，则自己提取一次
|
||
if not has_image:
|
||
first_frame = get_first_frame_bytes(video, f"frame_{int(time.time())}.jpg")
|
||
if first_frame:
|
||
image = Path(first_frame)
|
||
elif isinstance(video, os.PathLike):
|
||
video_path = Path(video)
|
||
if not video_path.exists():
|
||
raise ValueError(f"Video file does not exist: {video_path}")
|
||
async with aiofiles.open(video_path, "rb") as f:
|
||
video_bytes = await f.read()
|
||
file_len = len(video_bytes)
|
||
vid_base64 = base64.b64encode(video_bytes).decode()
|
||
media_info = MediaInfo.parse(video_path)
|
||
# 如果没有传入首帧，则自己提取一次
|
||
if not has_image:
|
||
first_frame = get_first_frame(video_path, f"frame_{int(time.time())}.jpg")
|
||
if first_frame:
|
||
image = Path(first_frame)
|
||
else:
|
||
raise ValueError("video should be str, bytes, or path")
|
||
# 获取视频时长
|
||
duration = None
|
||
for track in media_info.tracks:
|
||
if track.track_type == "Video" and track.duration is not None:
|
||
duration = int(track.duration / 1000) # 将毫秒转换为秒
|
||
break
|
||
if duration is None:
|
||
duration = 1
|
||
self.logging.error(f"无法从视频文件获取时长: {video}")
|
||
# get image base64
|
||
if isinstance(image, str):
|
||
image_base64 = image
|
||
elif isinstance(image, bytes):
|
||
image_base64 = base64.b64encode(image).decode()
|
||
elif isinstance(image, os.PathLike):
|
||
async with aiofiles.open(image, "rb") as f:
|
||
image_base64 = base64.b64encode(await f.read()).decode()
|
||
else:
|
||
raise ValueError("image should be str, bytes, or path")
|
||
# self.logging.debug(f"vid_base64:{vid_base64}")
|
||
# self.logging.debug(f"images_base64:{image_base64}")
|
||
# 打印预估时间，300KB/s
|
||
predict_time = int(file_len / 1024 / 300)
|
||
self.logging.debug("开始发送视频: 对方wxid:{} 视频base64略 图片base64略 预计耗时:{}秒", wxid, predict_time)
|
||
# self.logging.debug(f"image：{image};image_base64:{image_base64}")
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": "data:video/mp4;base64," + vid_base64,
|
||
"ImageBase64": "data:image/jpeg;base64," + image_base64,
|
||
"PlayLength": duration}
|
||
# self.logging.debug(f"json_param：:{json_param}")
|
||
async with session.post(f'http://{self.ip}:{self.port}/api/Msg/SendVideo', json=json_param) as resp:
|
||
json_resp = await resp.json()
|
||
# self.logging.debug(f"json_resp:{json_resp}")
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
json_param.pop('ImageBase64')
|
||
self.logging.info("发送视频成功: 对方wxid:{} 时长:{} 视频base64略 图片base64略", wxid, duration)
|
||
data = json_resp.get("Data")
|
||
return data.get("clientMsgId"), data.get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_voice_message(self, wxid: str, voice: Union[str, bytes, os.PathLike], format: str = "amr") -> \
|
||
tuple[int, int, int]:
|
||
"""发送语音消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
voice (str, bytes, os.PathLike): 语音 接受base64字符串，字节，文件路径
|
||
format (str, optional): 语音格式，支持amr/wav/mp3. Defaults to "amr".
|
||
|
||
Returns:
|
||
tuple[int, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
ValueError: voice_path和voice_base64都为空或都不为空时，或format不支持时
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_voice_message, wxid, voice, format)
|
||
|
||
async def _send_voice_message(self, wxid: str, voice: Union[str, bytes, os.PathLike], format: str = "mar") -> \
|
||
tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
elif format not in ["amr", "wav", "mp3", "silk", "speex"]:
|
||
raise ValueError("format must be one of amr, wav, mp3")
|
||
|
||
# read voice to byte
|
||
if isinstance(voice, str):
|
||
voice_byte = base64.b64decode(voice)
|
||
elif isinstance(voice, bytes):
|
||
voice_byte = voice
|
||
elif isinstance(voice, os.PathLike):
|
||
async with aiofiles.open(voice, "rb") as f:
|
||
voice_byte = await f.read()
|
||
else:
|
||
raise ValueError("voice should be str, bytes, or path")
|
||
voice_type = 0
|
||
# get voice duration and b64
|
||
if format.lower() == "amr":
|
||
audio = AudioSegment.from_file(BytesIO(voice_byte), format="amr")
|
||
voice_base64 = base64.b64encode(voice_byte).decode()
|
||
elif format.lower() == "wav":
|
||
audio = AudioSegment.from_file(BytesIO(voice_byte), format="wav").set_channels(1)
|
||
self.logging.debug(f"1audio.frame_rate: {audio.frame_rate}")
|
||
audio = audio.set_frame_rate(self._get_closest_frame_rate(audio.frame_rate))
|
||
self.logging.debug(f"2audio.frame_rate: {audio.frame_rate}")
|
||
audio = audio.set_channels(1).set_sample_width(2) # 16-bit PCM
|
||
logger.info(
|
||
f"音频处理: 格式={format}, 采样率={audio.frame_rate}, 声道数={audio.channels}, 时长={len(audio) / 1000}s")
|
||
voice_base64 = base64.b64encode(
|
||
await pysilk.async_encode(audio.raw_data, sample_rate=audio.frame_rate)).decode()
|
||
voice_type = 4
|
||
elif format.lower() == "mp3":
|
||
audio = AudioSegment.from_file(BytesIO(voice_byte), format="mp3").set_channels(1)
|
||
audio = audio.set_frame_rate(self._get_closest_frame_rate(audio.frame_rate))
|
||
voice_base64 = base64.b64encode(
|
||
await pysilk.async_encode(audio.raw_data, sample_rate=audio.frame_rate)).decode()
|
||
voice_type = 4
|
||
else:
|
||
raise ValueError("format must be one of amr, wav, mp3")
|
||
|
||
duration = len(audio)
|
||
# Type： AMR = 0, MP3 = 2, SILK = 4, SPEEX = 1, WAVE = 3 VoiceTime ：音频长度 1000为一秒
|
||
format_dict = {"amr": 0, "wav": 3, "mp3": 2, "silk": 4, "speex": 1}
|
||
# {
|
||
# "Base64": "string",
|
||
# "ToWxid": "string",
|
||
# "Type": 0,
|
||
# "VoiceTime": 0,
|
||
# "Wxid": "string"
|
||
# }
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": voice_base64, "VoiceTime": duration,
|
||
"Type": voice_type}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendVoice', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
self.logging.info("发送语音消息: 对方wxid:{} 时长:{} 格式:{} 音频base64略", wxid, duration, format)
|
||
data = json_resp.get("Data")
|
||
return int(data.get("ClientMsgId")), data.get("CreateTime"), data.get("NewMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
@staticmethod
|
||
def _get_closest_frame_rate(frame_rate: int) -> int:
|
||
supported = [8000, 12000, 16000, 24000]
|
||
closest_rate = None
|
||
smallest_diff = float('inf')
|
||
for num in supported:
|
||
diff = abs(frame_rate - num)
|
||
if diff < smallest_diff:
|
||
smallest_diff = diff
|
||
closest_rate = num
|
||
|
||
return closest_rate
|
||
|
||
async def send_link_xml_message(self, xml: str, towxid: str) -> tuple[str, int, int]:
|
||
"""发送链接消息。
|
||
{
|
||
"ToWxid": "string",
|
||
"Type": 0,
|
||
"Wxid": "string",
|
||
"Xml": "string"
|
||
}
|
||
Args:
|
||
xml (str): 发送的内容
|
||
towxid (str):接收人
|
||
|
||
Returns:
|
||
tuple[str, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
|
||
return await self._queue_message(self._send_link_xml_message, xml, towxid)
|
||
|
||
async def send_link_message(self, wxid: str, url: str, title: str = "", description: str = "",
|
||
thumb_url: str = "") -> tuple[str, int, int]:
|
||
"""发送链接消息。
|
||
{
|
||
"ToWxid": "string",
|
||
"Type": 0,
|
||
"Wxid": "string",
|
||
"Xml": "string"
|
||
}
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
url (str): 跳转链接
|
||
title (str, optional): 标题. Defaults to "".
|
||
description (str, optional): 描述. Defaults to "".
|
||
thumb_url (str, optional): 缩略图链接. Defaults to "".
|
||
|
||
Returns:
|
||
tuple[str, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
|
||
return await self._queue_message(self._send_link_message, wxid, url, title, description, thumb_url)
|
||
|
||
async def _send_link_xml_message(self, xml: str, towxid: str) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": towxid, "Xml": xml, "Type": 0}
|
||
logger.debug(f"_send_link_xml_message：{xml}")
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/ShareLink', json=json_param)
|
||
json_resp = await response.json()
|
||
logger.info(f"_send_link_xml_message resp:{json_resp}")
|
||
if json_resp.get("Success"):
|
||
data = json_resp.get("Data")
|
||
return data.get("clientMsgId"), data.get("createTime"), data.get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def _send_link_message(self, wxid: str, url: str, title: str = "", description: str = "",
|
||
thumb_url: str = "") -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Url": url, "Title": title, "Desc": description,
|
||
"ThumbUrl": thumb_url}
|
||
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/ShareLink', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
self.logging.info("发送链接消息: 对方wxid:{} 链接:{} 标题:{} 描述:{} 缩略图链接:{}",
|
||
wxid,
|
||
url,
|
||
title,
|
||
description,
|
||
thumb_url)
|
||
data = json_resp.get("Data")
|
||
return data.get("clientMsgId"), data.get("createTime"), data.get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_emoji_message(self, wxid: str, md5: str, total_length: int) -> list[dict]:
|
||
"""发送表情消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
md5 (str): 表情md5值
|
||
total_length (int): 表情总长度
|
||
|
||
Returns:
|
||
list[dict]: 返回表情项列表(list of emojiItem)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_emoji_message, wxid, md5, total_length)
|
||
|
||
async def _send_emoji_message(self, wxid: str, md5: str, total_length: int) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
# 表情发送接口历史上最容易出现“接口长时间不返回，导致整个消息队列被拖住”的问题，
|
||
# 因此这里单独加总超时和更细的日志，方便区分“参数错误”和“接口无响应”两类故障。
|
||
timeout = aiohttp.ClientTimeout(total=20)
|
||
async with aiohttp.ClientSession(timeout=timeout) as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Md5": md5, "TotalLen": total_length}
|
||
try:
|
||
self.logging.info("开始发送表情消息: 对方wxid:{} md5:{} 总长度:{}", wxid, md5, total_length)
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendEmoji', json=json_param)
|
||
json_resp = await response.json(content_type=None)
|
||
except asyncio.TimeoutError as exc:
|
||
self.logging.error("发送表情消息超时: 对方wxid:{} md5:{} 总长度:{}", wxid, md5, total_length)
|
||
raise TimeoutError("SendEmoji 接口调用超时") from exc
|
||
|
||
if json_resp.get("Success"):
|
||
data = json_resp.get("Data") or {}
|
||
self.logging.info("发送表情消息成功: 对方wxid:{} md5:{} 总长度:{}", wxid, md5, total_length)
|
||
return data.get("emojiItem") or data.get("EmojiItem") or data
|
||
else:
|
||
self.logging.error("发送表情消息失败: 对方wxid:{} md5:{} 总长度:{} resp:{}",
|
||
wxid, md5, total_length, json_resp)
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_card_message(self, wxid: str, card_wxid: str, card_nickname: str, card_alias: str = "") -> tuple[
|
||
int, int, int]:
|
||
"""发送名片消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
card_wxid (str): 名片用户的wxid
|
||
card_nickname (str): 名片用户的昵称
|
||
card_alias (str, optional): 名片用户的备注. Defaults to "".
|
||
|
||
Returns:
|
||
tuple[int, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_card_message, wxid, card_wxid, card_nickname, card_alias)
|
||
|
||
async def _send_card_message(self, wxid: str, card_wxid: str, card_nickname: str, card_alias: str = "") -> tuple[
|
||
int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with (aiohttp.ClientSession() as session):
|
||
|
||
json_param = {
|
||
"CardAlias": card_alias,
|
||
"CardNickName": card_nickname,
|
||
"CardWxId": card_wxid,
|
||
"ToWxid": wxid,
|
||
"Wxid": self.wxid
|
||
}
|
||
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/ShareCard', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
self.logging.info("发送名片消息: 对方wxid:{} 名片wxid:{} 名片备注:{} 名片昵称:{}", wxid,
|
||
card_wxid,
|
||
card_alias,
|
||
card_nickname)
|
||
data = json_resp.get("Data")
|
||
return data.get("List")[0].get("ClientMsgid"), data.get("List")[0].get("Createtime"), data.get("List")[
|
||
0].get("NewMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_app_message(self, wxid: str, xml: str, type: int) -> tuple[str, int, int]:
|
||
"""发送应用消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
xml (str): 应用消息的xml内容
|
||
type (int): 应用消息类型
|
||
|
||
Returns:
|
||
tuple[str, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_app_message, wxid, xml, type)
|
||
|
||
async def _send_app_message(self, wxid: str, xml: str, type: int = 0) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
# {
|
||
# "ToWxid": "string",
|
||
# "Type": 0,
|
||
# "Wxid": "string",
|
||
# "Xml": "string"
|
||
# }
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Xml": xml, "Type": type}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendApp', json=json_param)
|
||
json_resp = await response.json()
|
||
logger.info(f"json_resp: {json_resp}")
|
||
if json_resp.get("Success"):
|
||
json_param["Xml"] = json_param["Xml"].replace("\n", "")
|
||
self.logging.info("发送app消息: 对方wxid:{} 类型:{} xml:{}", wxid, type, json_param["Xml"])
|
||
return json_resp.get("Data").get("clientMsgId"), json_resp.get("Data").get(
|
||
"createTime"), json_resp.get("Data").get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_cdn_file_msg(self, wxid: str, xml: str) -> tuple[str, int, int]:
|
||
"""转发文件消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
xml (str): 要转发的文件消息xml内容
|
||
|
||
Returns:
|
||
tuple[str, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_cdn_file_msg, wxid, xml)
|
||
|
||
async def _send_cdn_file_msg(self, wxid: str, xml: str) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Content": xml}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendCDNFile', json=json_param)
|
||
json_resp = await response.json()
|
||
self.logging.debug("json_resp: %s", json_resp)
|
||
if json_resp.get("Success"):
|
||
self.logging.info("转发文件消息: 对方wxid:{} xml:{}", wxid, xml)
|
||
data = json_resp.get("Data")
|
||
return data.get("clientMsgId"), data.get("createTime"), data.get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_cdn_img_msg(self, wxid: str, xml: str) -> tuple[str, int, int]:
|
||
"""转发图片消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
xml (str): 要转发的图片消息xml内容
|
||
|
||
Returns:
|
||
tuple[str, int, int]: 返回(ClientImgId, CreateTime, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_cdn_img_msg, wxid, xml)
|
||
|
||
async def _send_cdn_img_msg(self, wxid: str, xml: str) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Content": xml}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendCDNImg', json=json_param)
|
||
json_resp = await response.json()
|
||
self.logging.debug("json_resp: %s", json_resp)
|
||
|
||
if json_resp.get("Success"):
|
||
self.logging.info("转发图片消息: 对方wxid:{} xml:{}", wxid, xml)
|
||
data = json_resp.get("Data")
|
||
return data.get("ClientImgId").get("string"), data.get("CreateTime"), data.get("Newmsgid")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def send_cdn_video_msg(self, wxid: str, xml: str) -> tuple[str, int]:
|
||
"""转发视频消息。
|
||
|
||
Args:
|
||
wxid (str): 接收人wxid
|
||
xml (str): 要转发的视频消息xml内容
|
||
|
||
Returns:
|
||
tuple[str, int]: 返回(ClientMsgid, NewMsgId)
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
BanProtection: 登录新设备后4小时内操作
|
||
根据error_handler处理错误
|
||
"""
|
||
return await self._queue_message(self._send_cdn_video_msg, wxid, xml)
|
||
|
||
async def _send_cdn_video_msg(self, wxid: str, xml: str) -> tuple[int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Content": xml}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/SendCDNVideo', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
self.logging.debug("json_resp: %s", json_resp)
|
||
if json_resp.get("Success"):
|
||
self.logging.info("转发视频消息: 对方wxid:{} xml:{}", wxid, xml)
|
||
data = json_resp.get("Data")
|
||
return data.get("clientMsgId"), data.get("newMsgId")
|
||
else:
|
||
self.error_handler(json_resp)
|
||
|
||
async def sync_message(self) -> dict:
|
||
"""同步消息。
|
||
|
||
Returns:
|
||
dict: 返回同步到的消息数据
|
||
|
||
Raises:
|
||
UserLoggedOut: 未登录时调用
|
||
根据error_handler处理错误
|
||
"""
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
|
||
async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10)) as session:
|
||
json_param = {"Wxid": self.wxid, "Scene": 0, "Synckey": ""}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/api/Msg/Sync', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
return json_resp.get("Data")
|
||
else:
|
||
self.error_handler(json_resp)
|