添加虎牙自动注册

2026-07-06 00:31:34 +08:00
parent 3f50d962a0
commit af98ea8416
12 changed files with 1060 additions and 19 deletions
@@ -0,0 +1,199 @@
+"""虎牙手机号自动注册流程。"""
+
+from __future__ import annotations
+
+import threading
+import time
+from dataclasses import dataclass
+from datetime import datetime
+
+from core.sms_provider import SmsLine, SmsProviderClient
+
+from .login import HuyaLoginError
+from .sms_login import login_huya_sms, normalize_huya_phone, send_huya_sms_code
+
+
+@dataclass
+class HuyaAutoRegisterResult:
+    """单个手机号自动注册结果。"""
+
+    phone: str
+    provider: str
+    success: bool
+    status: str
+    message: str
+    cookie: str = ""
+    code: str = ""
+    sdid: str = ""
+    normalized_phone: str = ""
+    attempts: int = 0
+
+
+def _sleep_or_stop(stop_event: threading.Event | None, seconds: float) -> bool:
+    """等待一小段时间；返回 False 表示收到停止信号。"""
+    if seconds <= 0:
+        return not (stop_event and stop_event.is_set())
+    if stop_event:
+        return not stop_event.wait(seconds)
+    time.sleep(seconds)
+    return True
+
+
+def register_huya_with_sms_line(
+    item: SmsLine,
+    wait_seconds: float = 180,
+    poll_interval: float = 5,
+    stop_event: threading.Event | None = None,
+) -> HuyaAutoRegisterResult:
+    """使用固定手机号和接码地址完成虎牙短信注册/登录。"""
+    phone = item.phone.strip()
+    normalized_phone = normalize_huya_phone(phone)
+    wait_seconds = max(15.0, float(wait_seconds or 180))
+    poll_interval = max(1.0, float(poll_interval or 5))
+
+    if stop_event and stop_event.is_set():
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="stopped",
+            message="已停止",
+            normalized_phone=normalized_phone,
+        )
+
+    sent_at = datetime.now()
+    try:
+        code_result = send_huya_sms_code(phone=phone)
+    except HuyaLoginError as exc:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=str(exc),
+            normalized_phone=normalized_phone,
+        )
+    except Exception as exc:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=f"发送虎牙短信失败: {exc}",
+            normalized_phone=normalized_phone,
+        )
+
+    if not code_result.success or not code_result.state:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=code_result.message or "发送虎牙短信失败",
+            sdid=code_result.sdid,
+            normalized_phone=normalized_phone,
+        )
+
+    client = SmsProviderClient()
+    deadline = time.monotonic() + wait_seconds
+    attempts = 0
+    last_message = "等待验证码"
+
+    while time.monotonic() < deadline:
+        if stop_event and stop_event.is_set():
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="stopped",
+                message="已停止",
+                sdid=code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+        attempts += 1
+        poll_result = client.poll(item, after_time=sent_at)
+        last_message = poll_result.message or last_message
+        if poll_result.has_code:
+            try:
+                login_result = login_huya_sms(
+                    authcode=poll_result.code,
+                    state=code_result.state,
+                    phone=phone,
+                )
+            except HuyaLoginError as exc:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=False,
+                    status="error",
+                    message=str(exc),
+                    code=poll_result.code,
+                    sdid=code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+            except Exception as exc:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=False,
+                    status="error",
+                    message=f"提交虎牙短信登录失败: {exc}",
+                    code=poll_result.code,
+                    sdid=code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+
+            if login_result.success and login_result.cookie:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=True,
+                    status="success",
+                    message="注册/登录成功",
+                    cookie=login_result.cookie,
+                    code=poll_result.code,
+                    sdid=login_result.sdid or code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="error",
+                message=login_result.message or "虎牙短信登录失败",
+                code=poll_result.code,
+                sdid=login_result.sdid or code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+        remaining = deadline - time.monotonic()
+        if remaining <= 0:
+            break
+        if not _sleep_or_stop(stop_event, min(poll_interval, remaining)):
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="stopped",
+                message="已停止",
+                sdid=code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+    return HuyaAutoRegisterResult(
+        phone=phone,
+        provider=item.provider,
+        success=False,
+        status="error",
+        message=f"等待验证码超时: {last_message}",
+        sdid=code_result.sdid,
+        normalized_phone=normalized_phone,
+        attempts=attempts,
+    )
@@ -54,7 +54,12 @@ class HuyaSmsCodeResult:

 def normalize_huya_phone(phone: str) -> str:
    """归一化虎牙短信接口手机号格式。"""
-    digits = "".join(ch for ch in str(phone or "") if ch.isdigit())
+    raw = str(phone or "").strip()
+    digits = "".join(ch for ch in raw if ch.isdigit())
+    if raw.startswith("+1") and len(digits) == 11:
+        return f"001{digits[1:]}"
+    if digits.startswith("001") and len(digits) == 13:
+        return digits
    if digits.startswith("086"):
        return digits
    if digits.startswith("86") and len(digits) == 13:
@@ -0,0 +1,151 @@
+"""接码平台查询与验证码提取。"""
+
+from __future__ import annotations
+
+import json
+import re
+from dataclasses import dataclass
+from datetime import datetime
+
+import requests
+
+
+CODE_PATTERN = re.compile(
+    r"(?:verification\s+code|验证码|校验码|动态码|安全码)\D{0,20}(\d{4,8})",
+    re.IGNORECASE,
+)
+FALLBACK_CODE_PATTERN = re.compile(r"(?<!\d)(\d{4,8})(?!\d)")
+
+
+@dataclass
+class SmsLine:
+    """手机号和短信查询地址。"""
+
+    phone: str
+    url: str
+    provider: str
+    raw: str
+
+
+@dataclass
+class SmsPollResult:
+    """短信查询结果。"""
+
+    status: str
+    code: str = ""
+    message: str = ""
+    raw: str = ""
+
+    @property
+    def has_code(self) -> bool:
+        return self.status == "code" and bool(self.code)
+
+
+def parse_sms_lines(text: str) -> list[SmsLine]:
+    """解析手机号池，每行格式：手机号----查询URL。"""
+    rows: list[SmsLine] = []
+    for raw_line in (text or "").splitlines():
+        line = raw_line.strip()
+        if not line or line.startswith("#"):
+            continue
+        parts = [part.strip() for part in line.split("----", 1)]
+        if len(parts) != 2 or not parts[0] or not parts[1]:
+            continue
+        rows.append(SmsLine(
+            phone=parts[0],
+            url=parts[1],
+            provider=detect_provider(parts[1]),
+            raw=line,
+        ))
+    return rows
+
+
+def detect_provider(url: str) -> str:
+    """根据 URL 判断接码平台类型。"""
+    text = (url or "").lower()
+    if "sms8.net" in text:
+        return "sms8"
+    return "pipe"
+
+
+def extract_sms_code(text: str) -> str:
+    """从短信内容中提取验证码。"""
+    body = str(text or "")
+    match = CODE_PATTERN.search(body)
+    if match:
+        return match.group(1)
+    match = FALLBACK_CODE_PATTERN.search(body)
+    return match.group(1) if match else ""
+
+
+def _parse_sms8_time(value: str) -> datetime | None:
+    text = str(value or "").strip()
+    if not text:
+        return None
+    for fmt in ("%Y-%m-%d %H:%M:%S", "%Y/%m/%d %H:%M:%S"):
+        try:
+            return datetime.strptime(text, fmt)
+        except ValueError:
+            continue
+    return None
+
+
+class SmsProviderClient:
+    """固定手机号池的短信查询客户端。"""
+
+    def __init__(self, timeout: tuple[float, float] = (8, 20)):
+        self.timeout = timeout
+        self.session = requests.Session()
+        self.session.trust_env = False
+
+    def poll(self, item: SmsLine, after_time: datetime | None = None) -> SmsPollResult:
+        """查询一次短信。"""
+        try:
+            response = self.session.get(item.url, timeout=self.timeout)
+            response.raise_for_status()
+        except requests.RequestException as exc:
+            return SmsPollResult(status="error", message=f"查询短信失败: {exc}")
+
+        body = response.text.strip()
+        if item.provider == "sms8":
+            return self._parse_sms8(body, after_time=after_time)
+        return self._parse_pipe(body)
+
+    def _parse_pipe(self, body: str) -> SmsPollResult:
+        if not body:
+            return SmsPollResult(status="waiting", message="短信为空", raw=body)
+        low = body.lower()
+        if low.startswith("no|"):
+            return SmsPollResult(status="waiting", message=body, raw=body)
+        if low.startswith("yes|"):
+            code = extract_sms_code(body)
+            if code:
+                return SmsPollResult(status="code", code=code, message="收到验证码", raw=body)
+            return SmsPollResult(status="error", message=f"已收到短信但未识别验证码: {body}", raw=body)
+        code = extract_sms_code(body)
+        if code:
+            return SmsPollResult(status="code", code=code, message="收到验证码", raw=body)
+        return SmsPollResult(status="waiting", message=body, raw=body)
+
+    def _parse_sms8(self, body: str, after_time: datetime | None = None) -> SmsPollResult:
+        try:
+            payload = json.loads(body)
+        except json.JSONDecodeError:
+            code = extract_sms_code(body)
+            if code:
+                return SmsPollResult(status="code", code=code, message="收到验证码", raw=body)
+            return SmsPollResult(status="error", message=f"短信平台返回非 JSON: {body[:120]}", raw=body)
+
+        data = payload.get("data") if isinstance(payload.get("data"), dict) else {}
+        msg = str(payload.get("msg") or "")
+        code_text = str(data.get("code") or "")
+        code_time = _parse_sms8_time(str(data.get("code_time") or ""))
+        if after_time and code_time and code_time < after_time.replace(tzinfo=None):
+            return SmsPollResult(status="waiting", message="验证码时间早于本次发码", raw=body)
+
+        code = extract_sms_code(code_text)
+        if int(payload.get("code") or 0) == 1 and code:
+            return SmsPollResult(status="code", code=code, message="收到验证码", raw=body)
+        if code:
+            return SmsPollResult(status="code", code=code, message="收到验证码", raw=body)
+        return SmsPollResult(status="waiting", message=msg or "暂无验证码", raw=body)