\nimport asyncio\nimport base64\nimport os\nfrom asyncio import Future\nfrom asyncio import Queue, sleep\nfrom io import BytesIO\nfrom pathlib import Path\nfrom typing import Union\n\nimport aiohttp\nimport pysilk\nfrom loguru import logger\nfrom pydub import AudioSegment\nfrom pymediainfo import MediaInfo\n\nfrom .base import *\nfrom .protect import protector\nfrom ..errors import *\n\n\n\n
[文档]\n
class MessageMixin(WechatAPIClientBase):\n def __init__(self, ip: str, port: int):\n # 初始化消息队列\n super().__init__(ip, port)\n self._message_queue = Queue()\n self._is_processing = False\n\n\n
[文档]\n
async def _process_message_queue(self):\n """\n 处理消息队列的异步方法\n """\n if self._is_processing:\n return\n\n self._is_processing = True\n while True:\n if self._message_queue.empty():\n self._is_processing = False\n break\n\n func, args, kwargs, future = await self._message_queue.get()\n try:\n result = await func(*args, **kwargs)\n future.set_result(result)\n except Exception as e:\n future.set_exception(e)\n finally:\n self._message_queue.task_done()\n await sleep(1) # 消息发送间隔1秒\n
[文档]\n
async def _queue_message(self, func, *args, **kwargs):\n """\n 将消息添加到队列\n """\n future = Future()\n await self._message_queue.put((func, args, kwargs, future))\n\n if not self._is_processing:\n asyncio.create_task(self._process_message_queue())\n\n return await future\n
[文档]\n
async def revoke_message(self, wxid: str, client_msg_id: int, create_time: int, new_msg_id: int) -> bool:\n """撤回消息。\n\n Args:\n wxid (str): 接收人wxid\n client_msg_id (int): 发送消息的返回值\n create_time (int): 发送消息的返回值\n new_msg_id (int): 发送消息的返回值\n\n Returns:\n bool: 成功返回True，失败返回False\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n 根据error_handler处理错误\n """\n if not self.wxid:\n raise UserLoggedOut("请先登录")\n elif not self.ignore_protect and protector.check(14400):\n raise BanProtection("风控保护: 新设备登录后4小时内请挂机")\n\n async with aiohttp.ClientSession() as session:\n json_param = {"Wxid": self.wxid, "ToWxid": wxid, "ClientMsgId": client_msg_id, "CreateTime": create_time,\n "NewMsgId": new_msg_id}\n response = await session.post(f'http://{self.ip}:{self.port}/RevokeMsg', json=json_param)\n json_resp = await response.json()\n\n if json_resp.get("Success"):\n logger.info("消息撤回成功: 对方wxid:{} ClientMsgId:{} CreateTime:{} NewMsgId:{}",\n wxid,\n client_msg_id,\n new_msg_id)\n return True\n else:\n self.error_handler(json_resp)\n
[文档]\n
async def send_text_message(self, wxid: str, content: str, at: Union[list, str] = "") -> tuple[int, int, int]:\n """发送文本消息。\n\n Args:\n wxid (str): 接收人wxid\n content (str): 消息内容\n at (list, str, optional): 要@的用户\n\n Returns:\n tuple[int, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n 根据error_handler处理错误\n """\n return await self._queue_message(self._send_text_message, wxid, content, at)\n
[文档]\n
async def _send_text_message(self, wxid: str, content: str, at: list[str] = None) -> tuple[int, int, int]:\n """\n 实际发送文本消息的方法\n """\n if not self.wxid:\n raise UserLoggedOut("请先登录")\n elif not self.ignore_protect and protector.check(14400):\n raise BanProtection("风控保护: 新设备登录后4小时内请挂机")\n\n if isinstance(at, str):\n at_str = at\n elif isinstance(at, list):\n if at is None:\n at = []\n at_str = ",".join(at)\n else:\n raise ValueError("Argument 'at' should be str or list")\n\n async with aiohttp.ClientSession() as session:\n json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Content": content, "Type": 1, "At": at_str}\n response = await session.post(f'http://{self.ip}:{self.port}/SendTextMsg', json=json_param)\n json_resp = await response.json()\n if json_resp.get("Success"):\n logger.info("发送文字消息: 对方wxid:{} at:{} 内容:{}", wxid, at, content)\n data = json_resp.get("Data")\n return data.get("List")[0].get("ClientMsgid"), data.get("List")[0].get("Createtime"), data.get("List")[\n 0].get("NewMsgId")\n else:\n self.error_handler(json_resp)\n
[文档]\n
async def send_image_message(self, wxid: str, image: Union[str, bytes, os.PathLike]) -> tuple[int, int, int]:\n """发送图片消息。\n\n Args:\n wxid (str): 接收人wxid\n image (str, byte, os.PathLike): 图片，支持base64字符串，图片byte，图片路径\n\n Returns:\n tuple[int, int, int]: 返回(ClientImgId, CreateTime, NewMsgId)\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n ValueError: image_path和image_base64都为空或都不为空时\n 根据error_handler处理错误\n """\n return await self._queue_message(self._send_image_message, wxid, image)\n
[文档]\n
async def send_video_message(self, wxid: str, video: Union[str, bytes, os.PathLike],\n image: [str, bytes, os.PathLike] = None):\n """发送视频消息。不推荐使用，上传速度很慢300KB/s。如要使用，可压缩视频，或者发送链接卡片而不是视频。\n\n Args:\n wxid (str): 接收人wxid\n video (str, bytes, os.PathLike): 视频 接受base64字符串，字节，文件路径\n image (str, bytes, os.PathLike): 视频封面图片 接受base64字符串，字节，文件路径\n\n Returns:\n tuple[int, int]: 返回(ClientMsgid, NewMsgId)\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n ValueError: 视频或图片参数都为空或都不为空时\n 根据error_handler处理错误\n """\n if not image:\n image = Path(os.path.join(Path(__file__).resolve().parent, "fallback.png"))\n # get video base64 and duration\n if isinstance(video, str):\n vid_base64 = video\n video = base64.b64decode(video)\n file_len = len(video)\n media_info = MediaInfo.parse(BytesIO(video))\n elif isinstance(video, bytes):\n vid_base64 = base64.b64encode(video).decode()\n file_len = len(video)\n media_info = MediaInfo.parse(BytesIO(video))\n elif isinstance(video, os.PathLike):\n with open(video, "rb") as f:\n file_len = len(f.read())\n vid_base64 = base64.b64encode(f.read()).decode()\n media_info = MediaInfo.parse(video)\n else:\n raise ValueError("video should be str, bytes, or path")\n duration = media_info.tracks[0].duration\n\n # get image base64\n if isinstance(image, str):\n image_base64 = image\n elif isinstance(image, bytes):\n image_base64 = base64.b64encode(image).decode()\n elif isinstance(image, os.PathLike):\n with open(image, "rb") as f:\n image_base64 = base64.b64encode(f.read()).decode()\n else:\n raise ValueError("image should be str, bytes, or path")\n\n # 打印预估时间，300KB/s\n predict_time = int(file_len / 1024 / 300)\n logger.info("开始发送视频: 对方wxid:{} 视频base64略 图片base64略 预计耗时:{}秒", wxid, predict_time)\n\n async with aiohttp.ClientSession() as session:\n json_param = {"Wxid": self.wxid, "ToWxid": wxid, "Base64": vid_base64, "ImageBase64": image_base64,\n "PlayLength": duration}\n async with session.post(f'http://{self.ip}:{self.port}/SendVideoMsg', json=json_param) as resp:\n json_resp = await resp.json()\n\n if json_resp.get("Success"):\n json_param.pop('Base64')\n json_param.pop('ImageBase64')\n logger.info("发送视频成功: 对方wxid:{} 时长:{} 视频base64略 图片base64略", wxid, duration)\n data = json_resp.get("Data")\n return data.get("clientMsgId"), data.get("newMsgId")\n else:\n self.error_handler(json_resp)\n
[文档]\n
async def send_voice_message(self, wxid: str, voice: Union[str, bytes, os.PathLike], format: str = "amr") -> \\\n tuple[int, int, int]:\n """发送语音消息。\n\n Args:\n wxid (str): 接收人wxid\n voice (str, bytes, os.PathLike): 语音 接受base64字符串，字节，文件路径\n format (str, optional): 语音格式，支持amr/wav/mp3. Defaults to "amr".\n\n Returns:\n tuple[int, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n ValueError: voice_path和voice_base64都为空或都不为空时，或format不支持时\n 根据error_handler处理错误\n """\n return await self._queue_message(self._send_voice_message, wxid, voice, format)\n
[文档]\n
async def send_link_message(self, wxid: str, url: str, title: str = "", description: str = "",\n thumb_url: str = "") -> tuple[str, int, int]:\n """发送链接消息。\n\n Args:\n wxid (str): 接收人wxid\n url (str): 跳转链接\n title (str, optional): 标题. Defaults to "".\n description (str, optional): 描述. Defaults to "".\n thumb_url (str, optional): 缩略图链接. Defaults to "".\n\n Returns:\n tuple[str, int, int]: 返回(ClientMsgid, CreateTime, NewMsgId)\n\n Raises:\n UserLoggedOut: 未登录时调用\n BanProtection: 登录新设备后4小时内操作\n 根据error_handler处理错误\n """\n return await self._queue_message(self._send_link_message, wxid, url, title, description, thumb_url)
![\"微信收款码\"](\"sponsor1.jpg\")
\n ![\"微信收款码\"](\"sponsor2.jpg\")
\n