147 lines
6.5 KiB
Python
147 lines
6.5 KiB
Python
import qrcode
|
||
from urllib.parse import parse_qs, urlparse
|
||
|
||
from wechat_ipad.providers.server_864.base import Server864APIClientBase
|
||
|
||
|
||
class LoginMixin(Server864APIClientBase):
|
||
"""864 登录相关接口。"""
|
||
|
||
@staticmethod
|
||
def _normalize_login_way(login_way: str) -> str:
|
||
"""标准化 864 `GetLoginQrCodeNewX` 的 way 参数。"""
|
||
normalized_way = str(login_way or "").strip().lower()
|
||
return normalized_way if normalized_way in {"mac", "win", "harmony", "car", "watch"} else "mac"
|
||
|
||
def _extract_qr_response(self, data) -> tuple[str, str]:
|
||
"""从 864 的二维码返回结构中提取 uuid 与二维码地址。"""
|
||
qr_code_url = (
|
||
self._pick_first(data, "QrCodeUrl", "QRCodeUrl", "qrCodeUrl")
|
||
or self._pick_first(data, "QrUrl", "QRUrl", "qrUrl")
|
||
or self._pick_first(self._pick_first(data, "Qrcode", "QrCode", "qrcode") or {}, "Src", "src")
|
||
or ""
|
||
)
|
||
uuid = self._pick_first(data, "UUID", "Uuid", "uuid") or ""
|
||
if not uuid and qr_code_url:
|
||
# 864 的真实返回经常只给“二维码图片地址”，并不会直接带出 uuid：
|
||
# 1. 图床链接 query 中的 `data=http://weixin.qq.com/x/<uuid>` 才是真正扫码值；
|
||
# 2. Dashboard 当前以 uuid 作为刷新与历史记录主键，缺失后前端体验会明显变差；
|
||
# 3. 因此这里统一做一次反解，让 New / NewX 两种接口都复用同一份展示逻辑。
|
||
parsed_qs = parse_qs(urlparse(str(qr_code_url)).query)
|
||
scan_data = str((parsed_qs.get("data") or [""])[0] or "")
|
||
if "/x/" in scan_data:
|
||
uuid = scan_data.rsplit("/x/", 1)[-1].strip()
|
||
return str(uuid), str(qr_code_url)
|
||
|
||
async def get_qr_code(
|
||
self,
|
||
device_name: str = "",
|
||
device_id: str = "",
|
||
proxy=None,
|
||
print_qr: bool = False,
|
||
login_qr_api: str = "new_x",
|
||
login_way: str = "mac",
|
||
) -> tuple[str, str]:
|
||
"""获取 864 登录二维码。
|
||
|
||
说明：
|
||
1. 864 不依赖 855 的 `device_name/device_id` 入参，但保留参数签名以兼容上层调用；
|
||
2. `proxy` 当前仅保留兼容占位，后续如需补实际代理登录，可直接映射到 swagger 的 Proxy 字段；
|
||
3. 返回值继续保持 `(uuid, url)`，方便 Dashboard 与运行时共用同一套二维码展示逻辑。
|
||
"""
|
||
del device_name, device_id
|
||
proxy_value = ""
|
||
if proxy is not None:
|
||
proxy_value = getattr(proxy, "proxy", "") or ""
|
||
|
||
normalized_login_qr_api = str(login_qr_api or "new_x").strip().lower()
|
||
normalized_login_way = self._normalize_login_way(login_way)
|
||
|
||
if normalized_login_qr_api in {"new_x", "x", "newx"}:
|
||
try:
|
||
# NewX 是当前 864 联调里更完整的一条登录链路：
|
||
# 1. 它支持 `Way` 指定登录端形态，兼容更多 server 变体；
|
||
# 2. 用户实测也验证该接口可正常返回二维码；
|
||
# 3. 若某些旧版 864 仍未提供 NewX，则自动回退到旧接口，避免新配置直接打挂启动。
|
||
data = await self._request_data(
|
||
"post",
|
||
"/login/GetLoginQrCodeNewX",
|
||
json_body={"Proxy": proxy_value, "Check": False, "Way": normalized_login_way},
|
||
timeout=30,
|
||
)
|
||
uuid, qr_url = self._extract_qr_response(data)
|
||
except Exception:
|
||
data = await self._request_data(
|
||
"post",
|
||
"/login/GetLoginQrCodeNew",
|
||
json_body={"Proxy": proxy_value, "Check": False},
|
||
timeout=30,
|
||
)
|
||
uuid, qr_url = self._extract_qr_response(data)
|
||
else:
|
||
data = await self._request_data(
|
||
"post",
|
||
"/login/GetLoginQrCodeNew",
|
||
json_body={"Proxy": proxy_value, "Check": False},
|
||
timeout=30,
|
||
)
|
||
uuid, qr_url = self._extract_qr_response(data)
|
||
|
||
if print_qr and uuid:
|
||
qr = qrcode.QRCode(
|
||
version=1,
|
||
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
||
box_size=10,
|
||
border=4,
|
||
)
|
||
qr.add_data(f"http://weixin.qq.com/x/{uuid}")
|
||
qr.make(fit=True)
|
||
qr.print_ascii()
|
||
|
||
return str(uuid), str(qr_url)
|
||
|
||
async def check_login_status(self) -> tuple[bool, dict]:
|
||
"""检查当前二维码登录状态。"""
|
||
data = await self._request_data("get", "/login/CheckLoginStatus", timeout=20)
|
||
normalized = dict(data or {})
|
||
# 864 扫码后可能不会直接变成 online，而是先返回一个安全验证链接：
|
||
# 1. 该字段在 swagger / 实机联调里都存在，但大小写并不稳定；
|
||
# 2. 上层 runtime 只关心统一字段名，因此这里先做一层归一化；
|
||
# 3. 这样 Dashboard 只需要消费 `verification_url`，不用感知各 server 的原始差异。
|
||
verification_url = self._pick_first(
|
||
normalized,
|
||
"VerificationUrl",
|
||
"VerificationURL",
|
||
"verificationUrl",
|
||
"verification_url",
|
||
)
|
||
if verification_url:
|
||
normalized["verification_url"] = str(verification_url).strip()
|
||
state = int(normalized.get("state", 0) or 0)
|
||
login_state = str(normalized.get("loginState", "") or "").strip().lower()
|
||
return state == 2 or login_state == "online", normalized
|
||
|
||
async def get_init_status(self) -> bool:
|
||
"""检查 server 侧初始化是否完成。"""
|
||
data = await self._request_data("get", "/login/GetInItStatus", timeout=15)
|
||
return bool(data)
|
||
|
||
async def awaken_login(self, wxid: str = "") -> dict:
|
||
"""触发 864 的唤醒登录。"""
|
||
del wxid
|
||
return await self._request_data("post", "/login/WakeUpLogin", timeout=30)
|
||
|
||
async def get_login_status(self, auto_login: bool = True) -> dict:
|
||
"""获取 864 在线状态。"""
|
||
return await self._request_data(
|
||
"get",
|
||
"/login/GetLoginStatus",
|
||
params={"autoLogin": str(bool(auto_login)).lower()},
|
||
timeout=20,
|
||
)
|
||
|
||
async def log_out(self) -> bool:
|
||
"""退出当前 864 登录态。"""
|
||
await self._request_data("get", "/login/LogOutRequest", timeout=15)
|
||
return True
|