112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
import json
|
|
|
|
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_profile_info_ext(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("userInfoExt")
|
|
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
|