855 协议版本

wechat_ipad/client/tools.py

@@ -0,0 +1,357 @@
+import base64
+import io
+import os
+
+import aiohttp
+import pysilk
+from pydub import AudioSegment
+
+from wechat_ipad import UserLoggedOut
+from wechat_ipad.client.base import WechatAPIClientBase, Proxy
+
+
+class ToolMixin(WechatAPIClientBase):
+    async def download_image(self, aeskey: str, cdnmidimgurl: str) -> str:
+        """CDN下载高清图片。
+
+        Args:
+            aeskey (str): 图片的AES密钥
+            cdnmidimgurl (str): 图片的CDN URL
+
+        Returns:
+            str: 图片的base64编码字符串
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid, "AesKey": aeskey, "Cdnmidimgurl": cdnmidimgurl}
+            response = await session.post(f'http://{self.ip}:{self.port}/CdnDownloadImg', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data")
+            else:
+                self.error_handler(json_resp)
+
+    async def download_voice(self, msg_id: str, voiceurl: str, length: int) -> str:
+        """下载语音文件。
+
+        Args:
+            msg_id (str): 消息的msgid
+            voiceurl (str): 语音的url，从xml获取
+            length (int): 语音长度，从xml获取
+
+        Returns:
+            str: 语音的base64编码字符串
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid, "MsgId": msg_id, "Voiceurl": voiceurl, "Length": length}
+            response = await session.post(f'http://{self.ip}:{self.port}/DownloadVoice', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data").get("data").get("buffer")
+            else:
+                self.error_handler(json_resp)
+
+    async def download_attach(self, attach_id: str) -> dict:
+        """下载附件。
+
+        Args:
+            attach_id (str): 附件ID
+
+        Returns:
+            dict: 附件数据
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid, "AttachId": attach_id}
+            response = await session.post(f'http://{self.ip}:{self.port}/DownloadAttach', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data").get("data").get("buffer")
+            else:
+                self.error_handler(json_resp)
+
+    async def download_video(self, msg_id) -> str:
+        """下载视频。
+
+        Args:
+            msg_id (str): 消息的msg_id
+
+        Returns:
+            str: 视频的base64编码字符串
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid, "MsgId": msg_id}
+            response = await session.post(f'http://{self.ip}:{self.port}/DownloadVideo', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data").get("data").get("buffer")
+            else:
+                self.error_handler(json_resp)
+
+    async def set_step(self, count: int) -> bool:
+        """设置步数。
+
+        Args:
+            count (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, "StepCount": count}
+            response = await session.post(f'http://{self.ip}:{self.port}/SetStep', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return True
+            else:
+                self.error_handler(json_resp)
+
+    async def set_proxy(self, proxy: Proxy) -> bool:
+        """设置代理。
+
+        Args:
+            proxy (Proxy): 代理配置对象
+
+        Returns:
+            bool: 成功返回True，失败返回False
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid,
+                          "Proxy": {"ProxyIp": f"{proxy.ip}:{proxy.port}",
+                                    "ProxyUser": proxy.username,
+                                    "ProxyPassword": proxy.password}}
+            response = await session.post(f'http://{self.ip}:{self.port}/SetProxy', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return True
+            else:
+                self.error_handler(json_resp)
+
+    async def check_database(self) -> bool:
+        """检查数据库状态。
+
+        Returns:
+            bool: 数据库正常返回True，否则返回False
+        """
+        async with aiohttp.ClientSession() as session:
+            response = await session.get(f'http://{self.ip}:{self.port}/CheckDatabaseOK')
+            json_resp = await response.json()
+
+            if json_resp.get("Running"):
+                return True
+            else:
+                return False
+
+    @staticmethod
+    def base64_to_file(base64_str: str, file_name: str, file_path: str) -> bool:
+        """将base64字符串转换为文件并保存。
+
+        Args:
+            base64_str (str): base64编码的字符串
+            file_name (str): 要保存的文件名
+            file_path (str): 文件保存路径
+
+        Returns:
+            bool: 转换成功返回True，失败返回False
+        """
+        try:
+            os.makedirs(file_path, exist_ok=True)
+
+            # 拼接完整的文件路径
+            full_path = os.path.join(file_path, file_name)
+
+            # 移除可能存在的 base64 头部信息
+            if ',' in base64_str:
+                base64_str = base64_str.split(',')[1]
+
+            # 解码 base64 并写入文件
+            with open(full_path, 'wb') as f:
+                f.write(base64.b64decode(base64_str))
+
+            return True
+
+        except Exception as e:
+            return False
+
+    @staticmethod
+    def file_to_base64(file_path: str) -> str:
+        """将文件转换为base64字符串。
+
+        Args:
+            file_path (str): 文件路径
+
+        Returns:
+            str: base64编码的字符串
+        """
+        with open(file_path, 'rb') as f:
+            return base64.b64encode(f.read()).decode()
+
+    @staticmethod
+    def base64_to_byte(base64_str: str) -> bytes:
+        """将base64字符串转换为bytes。
+
+        Args:
+            base64_str (str): base64编码的字符串
+
+        Returns:
+            bytes: 解码后的字节数据
+        """
+        # 移除可能存在的 base64 头部信息
+        if ',' in base64_str:
+            base64_str = base64_str.split(',')[1]
+
+        return base64.b64decode(base64_str)
+
+    @staticmethod
+    def byte_to_base64(byte: bytes) -> str:
+        """将bytes转换为base64字符串。
+
+        Args:
+            byte (bytes): 字节数据
+
+        Returns:
+            str: base64编码的字符串
+        """
+        return base64.b64encode(byte).decode("utf-8")
+
+    @staticmethod
+    async def silk_byte_to_byte_wav_byte(silk_byte: bytes) -> bytes:
+        """将silk字节转换为wav字节。
+
+        Args:
+            silk_byte (bytes): silk格式的字节数据
+
+        Returns:
+            bytes: wav格式的字节数据
+        """
+        return await pysilk.async_decode(silk_byte, to_wav=True)
+
+    @staticmethod
+    def wav_byte_to_amr_byte(wav_byte: bytes) -> bytes:
+        """将WAV字节数据转换为AMR格式。
+
+        Args:
+            wav_byte (bytes): WAV格式的字节数据
+
+        Returns:
+            bytes: AMR格式的字节数据
+
+        Raises:
+            Exception: 转换失败时抛出异常
+        """
+        try:
+            # 从字节数据创建 AudioSegment 对象
+            audio = AudioSegment.from_wav(io.BytesIO(wav_byte))
+
+            # 设置 AMR 编码的标准参数
+            audio = audio.set_frame_rate(8000).set_channels(1)
+
+            # 创建一个字节缓冲区来存储 AMR 数据
+            output = io.BytesIO()
+
+            # 导出为 AMR 格式
+            audio.export(output, format="amr")
+
+            # 获取字节数据
+            return output.getvalue()
+
+        except Exception as e:
+            raise Exception(f"转换WAV到AMR失败: {str(e)}")
+
+    @staticmethod
+    def wav_byte_to_amr_base64(wav_byte: bytes) -> str:
+        """将WAV字节数据转换为AMR格式的base64字符串。
+
+        Args:
+            wav_byte (bytes): WAV格式的字节数据
+
+        Returns:
+            str: AMR格式的base64编码字符串
+        """
+        return base64.b64encode(ToolMixin.wav_byte_to_amr_byte(wav_byte)).decode()
+
+    @staticmethod
+    async def wav_byte_to_silk_byte(wav_byte: bytes) -> bytes:
+        """将WAV字节数据转换为silk格式。
+
+        Args:
+            wav_byte (bytes): WAV格式的字节数据
+
+        Returns:
+            bytes: silk格式的字节数据
+        """
+        # get pcm data
+        audio = AudioSegment.from_wav(io.BytesIO(wav_byte))
+        pcm = audio.raw_data
+        return await pysilk.async_encode(pcm, data_rate=audio.frame_rate, sample_rate=audio.frame_rate)
+
+    @staticmethod
+    async def wav_byte_to_silk_base64(wav_byte: bytes) -> str:
+        """将WAV字节数据转换为silk格式的base64字符串。
+
+        Args:
+            wav_byte (bytes): WAV格式的字节数据
+
+        Returns:
+            str: silk格式的base64编码字符串
+        """
+        return base64.b64encode(await ToolMixin.wav_byte_to_silk_byte(wav_byte)).decode()
+
+    @staticmethod
+    async def silk_base64_to_wav_byte(silk_base64: str) -> bytes:
+        """将silk格式的base64字符串转换为WAV字节数据。
+
+        Args:
+            silk_base64 (str): silk格式的base64编码字符串
+
+        Returns:
+            bytes: WAV格式的字节数据
+        """
+        return await ToolMixin.silk_byte_to_byte_wav_byte(base64.b64decode(silk_base64))