@acedatacloud/skills 2026.714.4 → 2026.715.0

package/skills/xiaohongshu/scripts/vendor/xhs/like_favorite.py

@@ -1,188 +0,0 @@
-"""点赞/收藏操作，对应 Go xiaohongshu/like_favorite.go。"""
-
-from __future__ import annotations
-
-import json
-import logging
-import time
-
-from .cdp import Page
-from .errors import NoFeedDetailError
-from .selectors import COLLECT_BUTTON, LIKE_BUTTON
-from .types import ActionResult
-from .urls import make_feed_detail_url
-
-logger = logging.getLogger(__name__)
-
-# 从 __INITIAL_STATE__ 读取互动状态的 JS
-_GET_INTERACT_STATE_JS = """
-(() => {
-    if (window.__INITIAL_STATE__ &&
-        window.__INITIAL_STATE__.note &&
-        window.__INITIAL_STATE__.note.noteDetailMap) {
-        return JSON.stringify(window.__INITIAL_STATE__.note.noteDetailMap);
-    }
-    return "";
-})()
-"""
-
-
-def _get_interact_state(page: Page, feed_id: str) -> tuple[bool, bool]:
-    """读取笔记的点赞/收藏状态。
-
-    Returns:
-        (liked, collected)
-
-    Raises:
-        NoFeedDetailError: 无法获取状态。
-    """
-    result = page.evaluate(_GET_INTERACT_STATE_JS)
-    if not result:
-        raise NoFeedDetailError()
-
-    note_detail_map = json.loads(result)
-    detail = note_detail_map.get(feed_id)
-    if not detail and len(note_detail_map) == 1:
-        detail = next(iter(note_detail_map.values()))
-
-    if not detail:
-        raise NoFeedDetailError()
-
-    interact = detail.get("note", {}).get("interactInfo", {})
-    return interact.get("liked", False), interact.get("collected", False)
-
-
-def _prepare_page(page: Page, feed_id: str, xsec_token: str, xsec_source: str) -> None:
-    """导航到 feed 详情页。"""
-    url = make_feed_detail_url(feed_id, xsec_token, xsec_source)
-    page.navigate(url)
-    page.wait_for_load()
-    page.wait_dom_stable()
-    time.sleep(1)
-
-
-# ========== 点赞 ==========
-
-
-def like_feed(page: Page, feed_id: str, xsec_token: str, xsec_source: str = "pc_feed") -> ActionResult:
-    """点赞笔记（幂等：已点赞则跳过）。"""
-    _prepare_page(page, feed_id, xsec_token, xsec_source)
-    return _toggle_like(page, feed_id, target_liked=True)
-
-
-def unlike_feed(page: Page, feed_id: str, xsec_token: str, xsec_source: str = "pc_feed") -> ActionResult:
-    """取消点赞（幂等：未点赞则跳过）。"""
-    _prepare_page(page, feed_id, xsec_token, xsec_source)
-    return _toggle_like(page, feed_id, target_liked=False)
-
-
-def _toggle_like(page: Page, feed_id: str, target_liked: bool) -> ActionResult:
-    """执行点赞/取消点赞操作。"""
-    action_name = "点赞" if target_liked else "取消点赞"
-
-    try:
-        liked, _ = _get_interact_state(page, feed_id)
-    except NoFeedDetailError:
-        logger.error("无法读取点赞状态，拒绝切换")
-        return ActionResult(feed_id=feed_id, success=False, message=f"{action_name}失败：无法读取当前状态")
-
-    # 幂等检查
-    if liked == target_liked:
-        logger.info("feed %s 已%s，跳过", feed_id, action_name)
-        return ActionResult(feed_id=feed_id, success=True, message=f"已{action_name}")
-
-    # 点击
-    page.click_element(LIKE_BUTTON)
-    time.sleep(3)
-
-    # 验证
-    try:
-        liked, _ = _get_interact_state(page, feed_id)
-        if liked == target_liked:
-            logger.info("feed %s %s成功", feed_id, action_name)
-            return ActionResult(feed_id=feed_id, success=True, message=f"{action_name}成功")
-    except NoFeedDetailError:
-        pass
-
-    logger.error("feed %s %s未确认成功", feed_id, action_name)
-    return ActionResult(feed_id=feed_id, success=False, message=f"{action_name}失败：状态未变化")
-
-
-# ========== 收藏 ==========
-
-
-def favorite_feed(page: Page, feed_id: str, xsec_token: str, xsec_source: str = "pc_feed") -> ActionResult:
-    """收藏笔记（幂等：已收藏则跳过）。"""
-    _prepare_page(page, feed_id, xsec_token, xsec_source)
-    return _toggle_favorite(page, feed_id, target_collected=True)
-
-
-def unfavorite_feed(page: Page, feed_id: str, xsec_token: str, xsec_source: str = "pc_feed") -> ActionResult:
-    """取消收藏（幂等：未收藏则跳过）。"""
-    _prepare_page(page, feed_id, xsec_token, xsec_source)
-    return _toggle_favorite(page, feed_id, target_collected=False)
-
-
-def _wait_collect_button(page: Page, timeout: float = 5.0, interval: float = 0.2) -> bool:
-    """等待收藏按钮出现。"""
-    deadline = time.monotonic() + timeout
-    while time.monotonic() < deadline:
-        if page.has_element(COLLECT_BUTTON):
-            return True
-        time.sleep(interval)
-    return False
-
-
-def _wait_collected_state(
-    page: Page,
-    feed_id: str,
-    target_collected: bool,
-    timeout: float = 3.0,
-    interval: float = 0.3,
-) -> bool:
-    """短轮询验证收藏状态是否达到目标。"""
-    deadline = time.monotonic() + timeout
-    while time.monotonic() < deadline:
-        try:
-            _, collected = _get_interact_state(page, feed_id)
-            if collected == target_collected:
-                return True
-        except NoFeedDetailError:
-            pass
-        time.sleep(interval)
-    return False
-
-
-def _toggle_favorite(page: Page, feed_id: str, target_collected: bool) -> ActionResult:
-    """执行收藏/取消收藏操作。"""
-    action_name = "收藏" if target_collected else "取消收藏"
-
-    try:
-        _, collected = _get_interact_state(page, feed_id)
-    except NoFeedDetailError:
-        logger.error("无法读取收藏状态，拒绝切换")
-        return ActionResult(feed_id=feed_id, success=False, message=f"{action_name}失败：无法读取当前状态")
-
-    # 幂等检查
-    if collected == target_collected:
-        logger.info("feed %s 已%s，跳过", feed_id, action_name)
-        return ActionResult(feed_id=feed_id, success=True, message=f"已{action_name}")
-
-    if not _wait_collect_button(page, timeout=5.0):
-        logger.error("feed %s 未找到收藏按钮: %s", feed_id, COLLECT_BUTTON)
-        return ActionResult(
-            feed_id=feed_id, success=False, message=f"{action_name}失败：未找到收藏按钮"
-        )
-
-    try:
-        page.click_element(COLLECT_BUTTON)
-    except Exception as error:
-        logger.warning("feed %s 点击收藏按钮失败: %s", feed_id, error)
-        return ActionResult(feed_id=feed_id, success=False, message=f"{action_name}失败：点击失败")
-
-    if _wait_collected_state(page, feed_id, target_collected, timeout=3.0, interval=0.3):
-        logger.info("feed %s %s成功", feed_id, action_name)
-        return ActionResult(feed_id=feed_id, success=True, message=f"{action_name}成功")
-
-    logger.error("feed %s %s未确认成功", feed_id, action_name)
-    return ActionResult(feed_id=feed_id, success=False, message=f"{action_name}失败：状态未变化")

package/skills/xiaohongshu/scripts/vendor/xhs/login.py

@@ -1,377 +0,0 @@
-"""登录管理，对应 Go xiaohongshu/login.go。"""
-
-from __future__ import annotations
-
-import json
-import logging
-import os
-import tempfile
-import time
-
-_QR_DIR = os.path.join(tempfile.gettempdir(), "xhs")
-_QR_FILE = os.path.join(_QR_DIR, "login_qrcode.png")
-
-from .cdp import Page
-from .errors import RateLimitError
-from .human import sleep_random
-from .selectors import (
-    AGREE_CHECKBOX,
-    AGREE_CHECKBOX_CHECKED,
-    CODE_INPUT,
-    GET_CODE_BUTTON,
-    LOGIN_CONTAINER,
-    LOGIN_ERR_MSG,
-    LOGIN_STATUS,
-    LOGOUT_MENU_ITEM,
-    LOGOUT_MORE_BUTTON,
-    PHONE_INPUT,
-    PHONE_LOGIN_SUBMIT,
-    QRCODE_IMG,
-    USER_NICKNAME,
-    USER_PROFILE_NAV_LINK,
-)
-from .urls import EXPLORE_URL
-
-logger = logging.getLogger(__name__)
-
-
-def _wait_for_countdown(page: Page, timeout: float = 5.0) -> None:
-    """等待"获取验证码"按钮出现倒计时数字，确认验证码已发送。
-
-    轮询按钮文字直到包含数字（如 "60s"），超时则抛出 RateLimitError。
-    """
-    deadline = time.monotonic() + timeout
-    while time.monotonic() < deadline:
-        btn_text = page.get_element_text(GET_CODE_BUTTON) or ""
-        if any(ch.isdigit() for ch in btn_text):
-            return
-        time.sleep(0.3)
-    raise RateLimitError()
-
-
-
-def get_current_user_nickname(page: Page) -> str:
-    """获取当前登录用户的真实昵称，失败时返回空字符串（best-effort）。
-
-    流程：首页导航栏取个人主页 href → 导航过去 → 读 .user-name 文字。
-    """
-    try:
-        page.navigate(EXPLORE_URL)
-        page.wait_for_load()
-        if not check_login_status(page):
-            return ""
-
-        # 从导航栏"我"的链接取个人主页 URL（含 /user/profile/<user_id>）
-        profile_href = page.evaluate(
-            f"document.querySelector({json.dumps(USER_PROFILE_NAV_LINK)})?.getAttribute('href') || ''"
-        )
-        if not profile_href:
-            return ""
-
-        # 导航到个人主页读取真实昵称
-        profile_url = f"https://www.xiaohongshu.com{profile_href}"
-        page.navigate(profile_url)
-        page.wait_for_load()
-        page.wait_dom_stable()
-
-        nickname = page.evaluate(
-            f"document.querySelector({json.dumps(USER_NICKNAME)})?.innerText?.trim() || ''"
-        )
-        return nickname or ""
-    except Exception:
-        logger.warning("获取用户昵称失败")
-        return ""
-
-
-def check_login_status(page: Page) -> bool:
-    """检查登录状态。
-
-    Returns:
-        True 已登录，False 未登录。
-    """
-    # 如果当前页面已在 explore，跳过重复导航
-    current_url = page.evaluate("location.href") or ""
-    if "explore" not in current_url:
-        page.navigate(EXPLORE_URL)
-        page.wait_for_load()
-
-    # 直接等待登录状态或登录容器出现，替代 _wait_for_auth_ui
-    deadline = time.monotonic() + 10.0
-    while time.monotonic() < deadline:
-        if page.has_element(LOGIN_STATUS):
-            return True
-        if page.has_element(LOGIN_CONTAINER):
-            return False
-        time.sleep(0.2)
-    return False
-
-
-def fetch_qrcode(page: Page) -> tuple[bytes, str, bool]:
-    """获取登录二维码图片。
-
-    直接读取 img.src（data:image/png;base64,...），跳过 Canvas 绘制。
-
-    Returns:
-        (png_bytes, b64_str, already_logged_in)
-        - 如果已登录，返回 (b"", "", True)
-        - 如果未登录，返回 (png_bytes, b64_str, False)
-    """
-    # 如果当前页面已在 explore（如 check-login 刚导航过），跳过重复导航
-    current_url = page.evaluate("location.href") or ""
-    if "explore" not in current_url:
-        page.navigate(EXPLORE_URL)
-        page.wait_for_load()
-
-    # 快速检查是否已登录，避免无谓等待二维码
-    if page.has_element(LOGIN_STATUS):
-        return b"", "", True
-
-    # 直接等待二维码元素出现，合并了 _wait_for_auth_ui 的逻辑
-    page.wait_for_element(QRCODE_IMG, timeout=15.0)
-
-    # img.src 本身就是 data:image/png;base64,...，直接读取
-    src = page.evaluate(
-        f"document.querySelector({json.dumps(QRCODE_IMG)})?.src || ''"
-    )
-    if not src or "base64," not in src:
-        raise RuntimeError("二维码图片 src 读取失败")
-
-    b64_str = src.split("base64,", 1)[1]
-
-    import base64
-    png_bytes = base64.b64decode(b64_str)
-
-    return png_bytes, b64_str, False
-
-
-def _decode_qr_content(png_bytes: bytes) -> str | None:
-    """通过 goqr.me read API 解码二维码内容。
-
-    Returns:
-        解码后的文本（通常是登录 URL），失败返回 None。
-    """
-    import http.client
-
-    boundary = "----XhsQrBoundary"
-    body = (
-        f"--{boundary}\r\n"
-        f'Content-Disposition: form-data; name="file";'
-        f' filename="qr.png"\r\n'
-        f"Content-Type: image/png\r\n\r\n"
-    ).encode() + png_bytes + f"\r\n--{boundary}--\r\n".encode()
-
-    try:
-        conn = http.client.HTTPSConnection(
-            "api.qrserver.com", timeout=5
-        )
-        conn.request(
-            "POST",
-            "/v1/read-qr-code/",
-            body=body,
-            headers={
-                "Content-Type": (
-                    f"multipart/form-data; boundary={boundary}"
-                ),
-            },
-        )
-        resp = conn.getresponse()
-        if resp.status != 200:
-            return None
-        result = json.loads(resp.read().decode())
-        data = result[0]["symbol"][0].get("data")
-        return data if data else None
-    except Exception:
-        logger.debug("goqr.me 解码失败，将使用 base64 fallback")
-        return None
-
-
-def make_qrcode_url(
-    png_bytes: bytes,
-) -> tuple[str, str | None]:
-    """生成二维码展示 URL 和登录链接。
-
-    通过 goqr.me read API 解码 QR 内容，构造 API 图片 URL
-    （~270 字符）和小红书官方登录链接。
-
-    Returns:
-        (image_url, login_url)
-        - image_url: 可用于 markdown 图片的 URL
-        - login_url: 小红书官方登录链接（解码失败时为 None）
-    """
-    import base64
-    import urllib.parse
-
-    qr_content = _decode_qr_content(png_bytes)
-    if qr_content:
-        image_url = (
-            "https://api.qrserver.com/v1/create-qr-code/"
-            "?size=300x300&data="
-            + urllib.parse.quote(qr_content, safe="")
-        )
-        return image_url, qr_content
-
-    # fallback: base64 data URL
-    b64 = base64.b64encode(png_bytes).decode()
-    return "data:image/png;base64," + b64, None
-
-
-def save_qrcode_to_file(png_bytes: bytes) -> str:
-    """将二维码 PNG 字节保存到临时文件，返回文件路径。
-
-    Args:
-        png_bytes: CDP 截图返回的 PNG 字节。
-
-    Returns:
-        file_path: 保存的 PNG 文件绝对路径。
-    """
-    os.makedirs(_QR_DIR, exist_ok=True)
-    with open(_QR_FILE, "wb") as f:
-        f.write(png_bytes)
-    logger.info("二维码已保存: %s", _QR_FILE)
-    return _QR_FILE
-
-
-def send_phone_code(page: Page, phone: str) -> bool:
-    """填写手机号并发送短信验证码。
-
-    适用于无界面服务器场景，全程通过 CDP 操作，无需扫码。
-
-    Args:
-        page: CDP 页面对象。
-        phone: 手机号（不含国家码，如 13800138000）。
-
-    Returns:
-        True 验证码已发送，False 已登录（无需再登录）。
-
-    Raises:
-        RuntimeError: 找不到登录表单或手机号输入框。
-    """
-    # 如果当前页面已在 explore，跳过重复导航
-    current_url = page.evaluate("location.href") or ""
-    if "explore" not in current_url:
-        page.navigate(EXPLORE_URL)
-        page.wait_for_load()
-
-    # 直接等待登录容器出现（合并了 _wait_for_auth_ui 的逻辑，避免重复等待）
-    try:
-        page.wait_for_element(LOGIN_CONTAINER, timeout=10.0)
-    except Exception as exc:
-        # 可能已登录（没有登录容器），检查登录状态
-        if page.has_element(LOGIN_STATUS):
-            return False
-        raise RuntimeError("找不到登录表单") from exc
-
-    if page.has_element(LOGIN_STATUS):
-        return False
-
-    sleep_random(200, 400)
-
-    # 点击手机号输入框并逐字输入
-    page.click_element(PHONE_INPUT)
-    sleep_random(200, 400)
-    page.type_text(phone, delay_ms=80)
-    sleep_random(200, 400)
-
-    # 先勾选用户协议，再点获取验证码
-    if not page.has_element(AGREE_CHECKBOX_CHECKED):
-        page.click_element(AGREE_CHECKBOX)
-        sleep_random(300, 600)
-
-    # 点击"获取验证码"
-    page.click_element(GET_CODE_BUTTON)
-
-    # 事件驱动：轮询按钮文字直到出现倒计时数字，替代固定 2-2.5s 等待
-    _wait_for_countdown(page)
-
-    logger.info("验证码已发送至 %s", phone[:3] + "****" + phone[-4:])
-    return True
-
-
-def submit_phone_code(page: Page, code: str) -> bool:
-    """填写短信验证码并提交登录。
-
-    Args:
-        page: CDP 页面对象。
-        code: 收到的短信验证码。
-
-    Returns:
-        True 登录成功，False 失败（超时或验证码错误）。
-    """
-    # 点击验证码输入框，先清空再用 CDP 键盘事件逐字输入（isTrusted=true，React 能识别）
-    page.click_element(CODE_INPUT)
-    sleep_random(100, 200)
-    page.evaluate(
-        f"""(() => {{
-            const el = document.querySelector({json.dumps(CODE_INPUT)});
-            if (el && el.value) {{
-                const setter = Object.getOwnPropertyDescriptor(
-                    window.HTMLInputElement.prototype, 'value'
-                ).set;
-                setter.call(el, '');
-                el.dispatchEvent(new Event('input', {{ bubbles: true }}));
-            }}
-        }})()"""
-    )
-    page.type_text(code, delay_ms=0)
-    sleep_random(100, 200)
-
-    # 点击登录按钮
-    page.click_element(PHONE_LOGIN_SUBMIT)
-    sleep_random(500, 1000)
-
-    # 检查是否有错误提示
-    err = page.get_element_text(LOGIN_ERR_MSG)
-    if err and err.strip():
-        logger.warning("登录失败: %s", err.strip())
-        return False
-
-    return wait_for_login(page, timeout=30.0)
-
-
-def logout(page: Page) -> bool:
-    """通过页面 UI 退出登录（点击"更多"→"退出登录"）。
-
-    Args:
-        page: CDP 页面对象。
-
-    Returns:
-        True 退出成功，False 未登录或操作失败。
-    """
-    page.navigate(EXPLORE_URL)
-    page.wait_for_load()
-    sleep_random(800, 1500)
-
-    if not page.has_element(LOGIN_STATUS):
-        logger.info("当前未登录，无需退出")
-        return False
-
-    # 点击"更多"按钮展开菜单
-    page.click_element(LOGOUT_MORE_BUTTON)
-    sleep_random(500, 800)
-
-    # 等待退出菜单项出现并点击
-    page.wait_for_element(LOGOUT_MENU_ITEM, timeout=5.0)
-    page.click_element(LOGOUT_MENU_ITEM)
-    sleep_random(1000, 1500)
-
-    logger.info("已退出登录")
-    return True
-
-
-def wait_for_login(page: Page, timeout: float = 120.0) -> bool:
-    """等待扫码登录完成。
-
-    Args:
-        page: CDP 页面对象。
-        timeout: 超时时间（秒）。
-
-    Returns:
-        True 登录成功，False 超时。
-    """
-    deadline = time.monotonic() + timeout
-    while time.monotonic() < deadline:
-        if page.has_element(LOGIN_STATUS):
-            logger.info("登录成功")
-            return True
-        time.sleep(0.3)
-    return False