107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
from dataclasses import dataclass
|
|
|
|
from WechatAPI.errors import *
|
|
|
|
|
|
@dataclass
|
|
class Proxy:
|
|
"""代理(无效果，别用！)
|
|
|
|
Args:
|
|
ip (str): 代理服务器IP地址
|
|
port (int): 代理服务器端口
|
|
username (str, optional): 代理认证用户名. 默认为空字符串
|
|
password (str, optional): 代理认证密码. 默认为空字符串
|
|
"""
|
|
ip: str
|
|
port: int
|
|
username: str = ""
|
|
password: str = ""
|
|
|
|
|
|
@dataclass
|
|
class Section:
|
|
"""数据段配置类
|
|
|
|
Args:
|
|
data_len (int): 数据长度
|
|
start_pos (int): 起始位置
|
|
"""
|
|
data_len: int
|
|
start_pos: int
|
|
|
|
|
|
class WechatAPIClientBase:
|
|
"""微信API客户端基类
|
|
|
|
Args:
|
|
ip (str): 服务器IP地址
|
|
port (int): 服务器端口
|
|
|
|
Attributes:
|
|
wxid (str): 微信ID
|
|
nickname (str): 昵称
|
|
alias (str): 别名
|
|
phone (str): 手机号
|
|
ignore_protect (bool): 是否忽略保护机制
|
|
"""
|
|
def __init__(self, ip: str, port: int):
|
|
self.ip = ip
|
|
self.port = port
|
|
|
|
self.wxid = ""
|
|
self.nickname = ""
|
|
self.alias = ""
|
|
self.phone = ""
|
|
|
|
self.ignore_protect = False
|
|
|
|
# 调用所有 Mixin 的初始化方法
|
|
super().__init__()
|
|
|
|
@staticmethod
|
|
def error_handler(json_resp):
|
|
"""处理API响应中的错误码
|
|
|
|
Args:
|
|
json_resp (dict): API响应的JSON数据
|
|
|
|
Raises:
|
|
ValueError: 参数错误时抛出
|
|
MarshallingError: 序列化错误时抛出
|
|
UnmarshallingError: 反序列化错误时抛出
|
|
MMTLSError: MMTLS初始化错误时抛出
|
|
PacketError: 数据包长度错误时抛出
|
|
UserLoggedOut: 用户已退出登录时抛出
|
|
ParsePacketError: 解析数据包错误时抛出
|
|
DatabaseError: 数据库错误时抛出
|
|
Exception: 其他类型错误时抛出
|
|
"""
|
|
code = json_resp.get("Code")
|
|
if code == -1: # 参数错误
|
|
raise ValueError(json_resp.get("Message"))
|
|
elif code == -2: # 其他错误
|
|
raise Exception(json_resp.get("Message"))
|
|
elif code == -3: # 序列化错误
|
|
raise MarshallingError(json_resp.get("Message"))
|
|
elif code == -4: # 反序列化错误
|
|
raise UnmarshallingError(json_resp.get("Message"))
|
|
elif code == -5: # MMTLS初始化错误
|
|
raise MMTLSError(json_resp.get("Message"))
|
|
elif code == -6: # 收到的数据包长度错误
|
|
raise PacketError(json_resp.get("Message"))
|
|
elif code == -7: # 已退出登录
|
|
raise UserLoggedOut("Already logged out")
|
|
elif code == -8: # 链接过期
|
|
raise Exception(json_resp.get("Message"))
|
|
elif code == -9: # 解析数据包错误
|
|
raise ParsePacketError(json_resp.get("Message"))
|
|
elif code == -10: # 数据库错误
|
|
raise DatabaseError(json_resp.get("Message"))
|
|
elif code == -11: # 登陆异常
|
|
raise UserLoggedOut(json_resp.get("Message"))
|
|
elif code == -12: # 操作过于频繁
|
|
raise Exception(json_resp.get("Message"))
|
|
elif code == -13: # 上传失败
|
|
raise Exception(json_resp.get("Message"))
|