642 lines
27 KiB
Python
642 lines
27 KiB
Python
import asyncio
|
||
import base64
|
||
import os
|
||
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
|
||
|
||
from .base import *
|
||
from .protect import protector
|
||
from ..errors import *
|
||
|
||
|
||
class MessageMixin(WechatAPIClientBase):
|
||
def __init__(self, ip: str, port: int):
|
||
# 初始化消息队列
|
||
super().__init__(ip, port)
|
||
self._message_queue = Queue()
|
||
self._is_processing = False
|
||
|
||
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:
|
||
"""撤回消息。
|
||
|
||
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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "ClientMsgId": client_msg_id, "CreateTime": create_time,
|
||
"NewMsgId": new_msg_id}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/RevokeMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.info("消息撤回成功: 对方wxid:{} ClientMsgId:{} CreateTime:{} NewMsgId:{}",
|
||
wxid,
|
||
client_msg_id,
|
||
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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendTextMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
if json_resp.get("Success"):
|
||
logger.info("发送文字消息: 对方wxid:{} at:{} 内容:{}", 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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
if isinstance(image, str):
|
||
pass
|
||
elif isinstance(image, bytes):
|
||
image = base64.b64encode(image).decode()
|
||
elif isinstance(image, os.PathLike):
|
||
with open(image, 'rb') as f:
|
||
image = base64.b64encode(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}/SendImageMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
logger.info("发送图片消息: 对方wxid:{} 图片base64略", wxid)
|
||
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_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处理错误
|
||
"""
|
||
if not image:
|
||
image = Path(os.path.join(Path(__file__).resolve().parent, "fallback.png"))
|
||
# 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))
|
||
elif isinstance(video, os.PathLike):
|
||
with open(video, "rb") as f:
|
||
file_len = len(f.read())
|
||
vid_base64 = base64.b64encode(f.read()).decode()
|
||
media_info = MediaInfo.parse(video)
|
||
else:
|
||
raise ValueError("video should be str, bytes, or path")
|
||
duration = media_info.tracks[0].duration
|
||
|
||
# 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):
|
||
with open(image, "rb") as f:
|
||
image_base64 = base64.b64encode(f.read()).decode()
|
||
else:
|
||
raise ValueError("image should be str, bytes, or path")
|
||
|
||
# 打印预估时间，300KB/s
|
||
predict_time = int(file_len / 1024 / 300)
|
||
logger.info("开始发送视频: 对方wxid:{} 视频base64略 图片base64略 预计耗时:{}秒", wxid, predict_time)
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": vid_base64, "ImageBase64": image_base64,
|
||
"PlayLength": duration}
|
||
async with session.post(f'http://{self.ip}:{self.port}/SendVideoMsg', json=json_param) as resp:
|
||
json_resp = await resp.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
json_param.pop('ImageBase64')
|
||
logger.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 = "amr") -> \
|
||
tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
elif format not in ["amr", "wav", "mp3"]:
|
||
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):
|
||
with open(voice, "rb") as f:
|
||
voice_byte = f.read()
|
||
else:
|
||
raise ValueError("voice should be str, bytes, or path")
|
||
|
||
# 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)
|
||
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()
|
||
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()
|
||
else:
|
||
raise ValueError("format must be one of amr, wav, mp3")
|
||
|
||
duration = len(audio)
|
||
|
||
format_dict = {"amr": 0, "wav": 4, "mp3": 4}
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": voice_base64, "VoiceTime": duration,
|
||
"Type": format_dict[format]}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/SendVoiceMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param.pop('Base64')
|
||
logger.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_message(self, wxid: str, url: str, title: str = "", description: str = "",
|
||
thumb_url: str = "") -> tuple[str, int, int]:
|
||
"""发送链接消息。
|
||
|
||
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_message(self, wxid: str, url: str, title: str = "", description: str = "",
|
||
thumb_url: str = "") -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendShareLink', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Md5": md5, "TotalLen": total_length}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/SendEmojiMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.info("发送表情消息: 对方wxid:{} md5:{} 总长度:{}", wxid, md5, total_length)
|
||
return json_resp.get("Data").get("emojiItem")
|
||
else:
|
||
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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
json_param = {"Wxid": self.wxid, "ToWxid": wxid, "CardWxid": card_wxid, "CardAlias": card_alias,
|
||
"CardNickname": card_nickname}
|
||
response = await session.post(f'http://{self.ip}:{self.port}/SendCardMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.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) -> tuple[int, int, int]:
|
||
if not self.wxid:
|
||
raise UserLoggedOut("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendAppMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
json_param["Xml"] = json_param["Xml"].replace("\n", "")
|
||
logger.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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendCDNFileMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendCDNImgMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.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("请先登录")
|
||
elif not self.ignore_protect and protector.check(14400):
|
||
raise BanProtection("风控保护: 新设备登录后4小时内请挂机")
|
||
|
||
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}/SendCDNVideoMsg', json=json_param)
|
||
json_resp = await response.json()
|
||
|
||
if json_resp.get("Success"):
|
||
logger.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}/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)
|