855 协议版本

wechat_ipad/client/user.py

@@ -0,0 +1,81 @@
+import aiohttp
+
+from wechat_ipad import UserLoggedOut
+from wechat_ipad.client.base import WechatAPIClientBase
+
+from loguru import logger
+
+
+class UserMixin(WechatAPIClientBase):
+    async def get_profile(self, wxid: str = None) -> dict:
+        """获取用户信息。
+
+        Args:
+            wxid (str, optional): 用户wxid. Defaults to None.
+
+        Returns:
+            dict: 用户信息字典
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            根据error_handler处理错误
+        """
+        if not self.wxid and not wxid:
+            raise UserLoggedOut("请先登录")
+
+        if not wxid:
+            wxid = self.wxid
+
+        async with aiohttp.ClientSession() as session:
+            response = await session.post(f'http://{self.ip}:{self.port}/api/User/GetContractProfile?wxid={wxid}')
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data").get("userInfo")
+            else:
+                self.error_handler(json_resp)
+
+    async def get_my_qrcode(self, style: int = 0) -> str:
+        """获取个人二维码。
+
+        Args:
+            style (int, optional): 二维码样式. Defaults to 0.
+
+        Returns:
+            str: 图片的base64编码字符串
+
+        Raises:
+            UserLoggedOut: 未登录时调用
+            BanProtection: 风控保护: 新设备登录后4小时内请挂机
+            根据error_handler处理错误
+        """
+        if not self.wxid:
+            raise UserLoggedOut("请先登录")
+
+        async with aiohttp.ClientSession() as session:
+            json_param = {"Wxid": self.wxid, "Style": style}
+            response = await session.post(f'http://{self.ip}:{self.port}/GetMyQRCode', json=json_param)
+            json_resp = await response.json()
+
+            if json_resp.get("Success"):
+                return json_resp.get("Data").get("qrcode").get("buffer")
+            else:
+                self.error_handler(json_resp)
+
+    async def is_logged_in(self, wxid: str = None) -> bool:
+        """检查是否登录。
+
+        Args:
+            wxid (str, optional): 用户wxid. Defaults to None.
+
+        Returns:
+            bool: 已登录返回True，未登录返回False
+        """
+        if not wxid:
+            wxid = self.wxid
+        try:
+            await self.get_profile(wxid)
+            return True
+        except Exception as e:
+            logger.error("is_logged_in:{}", e)
+            return False