""" 小红书登录辅助模块 借鉴 ai_mip 项目的优秀实践，提供增强的验证码登录功能 """ import asyncio import random import time from typing import Optional, List, Dict, Any from playwright.async_api import Page, ElementHandle from loguru import logger class XHSLoginHelper: """小红书登录辅助工具类 - 借鉴ai_mip项目""" def __init__(self, page: Page): """ 初始化登录辅助器 Args: page: Playwright Page 对象 """ self.page = page async def human_type(self, selector: str, text: str, clear_first: bool = True) -> bool: """ 模拟人类打字速度输入文本（借鉴ai_mip） Args: selector: 输入框选择器 text: 要输入的文本 clear_first: 是否先清空输入框 Returns: 是否输入成功 """ try: # 查找元素 element = await self._find_element_smart(selector) if not element: logger.error(f"[人类输入] 未找到元素: {selector}") return False # 滚动到可见 await element.scroll_into_view_if_needed() await asyncio.sleep(random.uniform(0.1, 0.3)) # 聚焦输入框 await element.focus() await asyncio.sleep(random.uniform(0.1, 0.2)) # 先清空 if clear_first: await element.fill('') await asyncio.sleep(random.uniform(0.1, 0.3)) # 模拟人类打字（逐字符输入，随机延迟） for char in text: await self.page.keyboard.type(char) # 随机延迟 50ms - 150ms await asyncio.sleep(random.uniform(0.05, 0.15)) # 触发change事件 await element.evaluate('el => el.dispatchEvent(new Event("input", { bubbles: true }))') await element.evaluate('el => el.dispatchEvent(new Event("change", { bubbles: true }))') logger.success(f"[人类输入] 已输入 {len(text)} 个字符: {text}") return True except Exception as e: logger.error(f"[人类输入] 输入失败: {e}") return False async def human_click(self, selector: str, wait_after: float = 0.5) -> bool: """ 模拟人类点击行为（借鉴ai_mip） Args: selector: 元素选择器 wait_after: 点击后等待时间 Returns: 是否点击成功 """ try: # 查找元素 element = await self._find_element_smart(selector) if not element: logger.error(f"[人类点击] 未找到元素: {selector}") return False # 滚动到可见 await element.scroll_into_view_if_needed() await asyncio.sleep(random.uniform(0.1, 0.3)) # 获取元素位置 box = await element.bounding_box() if box: # 在元素范围内随机一个点击位置（避免总是点击中心） x = box['x'] + random.uniform(box['width'] * 0.3, box['width'] * 0.7) y = box['y'] + random.uniform(box['height'] * 0.3, box['height'] * 0.7) # 移动鼠标（模拟人类移动轨迹） await self.page.mouse.move(x, y) await asyncio.sleep(random.uniform(0.1, 0.3)) # 点击 await self.page.mouse.click(x, y) logger.success(f"[人类点击] 点击位置: ({x:.0f}, {y:.0f})") else: # 直接点击（降级方案） await element.click() logger.success(f"[人类点击] 直接点击元素") await asyncio.sleep(wait_after) return True except Exception as e: logger.error(f"[人类点击] 点击失败: {e}") return False async def _find_element_smart(self, selector: str, timeout: int = 5000) -> Optional[ElementHandle]: """ 智能查找元素（支持多种选择器格式） Args: selector: 元素选择器（CSS/XPath/text等） timeout: 超时时间（毫秒） Returns: 找到的元素，失败返回None """ try: # 尝试等待元素 element = await self.page.wait_for_selector(selector, timeout=timeout, state='visible') return element except Exception: return None async def find_input_with_fallback(self, primary_selectors: List[str], fallback_selectors: List[str] = None) -> Optional[ElementHandle]: """ 查找输入框（多选择器降级策略，借鉴ai_mip） Args: primary_selectors: 主要选择器列表 fallback_selectors: 降级选择器列表 Returns: 找到的输入框元素 """ try: logger.info("[智能查找] 开始查找输入框...") # 第一轮：尝试主要选择器 for selector in primary_selectors: try: elements = await self.page.query_selector_all(selector) logger.debug(f"[智能查找] 选择器 '{selector}' 找到 {len(elements)} 个元素") for elem in elements: if await elem.is_visible(): logger.success(f"[智能查找] 找到可见输入框: {selector}") return elem except Exception as e: logger.debug(f"[智能查找] 选择器 '{selector}' 失败: {str(e)}") continue # 第二轮：尝试降级选择器 if fallback_selectors: logger.warning("[智能查找] 主要选择器未找到，尝试降级选择器...") for selector in fallback_selectors: try: elements = await self.page.query_selector_all(selector) logger.debug(f"[智能查找] 降级选择器 '{selector}' 找到 {len(elements)} 个元素") for elem in elements: if await elem.is_visible(): logger.warning(f"[智能查找] 使用降级选择器找到: {selector}") return elem except Exception as e: logger.debug(f"[智能查找] 降级选择器 '{selector}' 失败: {str(e)}") continue logger.error("[智能查找] 所有选择器均未找到可见输入框") return None except Exception as e: logger.error(f"[智能查找] 查找异常: {str(e)}") return None async def find_button_with_fallback(self, primary_selectors: List[str], expected_texts: List[str] = None) -> Optional[ElementHandle]: """ 查找按钮（多选择器降级策略，支持文本验证） Args: primary_selectors: 主要选择器列表 expected_texts: 期望的按钮文本列表（用于验证） Returns: 找到的按钮元素 """ try: logger.info("[智能查找] 开始查找按钮...") for selector in primary_selectors: try: elements = await self.page.query_selector_all(selector) logger.debug(f"[智能查找] 选择器 '{selector}' 找到 {len(elements)} 个按钮") for elem in elements: if not await elem.is_visible(): continue # 验证按钮文本（如果指定） if expected_texts: try: btn_text = await elem.inner_text() btn_text = btn_text.strip() if btn_text else "" if not any(expected in btn_text for expected in expected_texts): logger.debug(f"[智能查找] 按钮文本不匹配: '{btn_text}', 期望: {expected_texts}") continue logger.success(f"[智能查找] 找到匹配按钮: {selector}, 文本: '{btn_text}'") return elem except Exception: # 无法获取文本，跳过验证 logger.success(f"[智能查找] 找到可见按钮: {selector}") return elem else: logger.success(f"[智能查找] 找到可见按钮: {selector}") return elem except Exception as e: logger.debug(f"[智能查找] 选择器 '{selector}' 失败: {str(e)}") continue logger.error("[智能查找] 所有选择器均未找到可见按钮") return None except Exception as e: logger.error(f"[智能查找] 查找按钮异常: {str(e)}") return None async def wait_for_button_active(self, element: ElementHandle, timeout: int = 5) -> bool: """ 等待按钮激活状态（小红书特有逻辑） Args: element: 按钮元素 timeout: 超时时间（秒） Returns: 是否激活成功 """ try: logger.info("[按钮激活] 等待按钮激活...") start_time = time.time() while time.time() - start_time < timeout: try: class_name = await element.get_attribute('class') or "" if 'active' in class_name or 'enabled' in class_name: logger.success(f"[按钮激活] 按钮已激活: class={class_name}") return True except Exception: pass await asyncio.sleep(0.2) logger.warning(f"[按钮激活] 等待超时（{timeout}秒）") return False except Exception as e: logger.error(f"[按钮激活] 检查失败: {e}") return False async def check_button_countdown(self, element: ElementHandle) -> Optional[str]: """ 检查按钮是否处于倒计时状态 Args: element: 按钮元素 Returns: 倒计时文本（如果处于倒计时），否则返回None """ try: btn_text = await element.inner_text() btn_text = btn_text.strip() if btn_text else "" # 检查是否包含倒计时标识 if btn_text and (btn_text[-1] == 's' or '秒' in btn_text or btn_text.isdigit()): logger.warning(f"[倒计时检测] 按钮处于倒计时: {btn_text}") return btn_text return None except Exception as e: logger.error(f"[倒计时检测] 检查失败: {e}") return None async def random_delay(self, min_seconds: float = 0.5, max_seconds: float = 1.5): """ 随机延迟，模拟人工操作（借鉴ai_mip） Args: min_seconds: 最小延迟（秒） max_seconds: 最大延迟（秒） """ delay = random.uniform(min_seconds, max_seconds) await asyncio.sleep(delay) async def scroll_to_element(self, element: ElementHandle): """ 平滑滚动到元素位置（模拟人类滚动行为） Args: element: 目标元素 """ try: # 获取元素位置 box = await element.bounding_box() if box: # 计算滚动目标（元素在视口中间位置） viewport = self.page.viewport_size target_y = box['y'] - (viewport['height'] / 2) + (box['height'] / 2) # 分步滚动（模拟人类滚动） current_scroll = await self.page.evaluate('window.pageYOffset') distance = target_y - current_scroll steps = max(3, int(abs(distance) / 100)) # 根据距离计算步数 for i in range(steps): progress = (i + 1) / steps scroll_y = current_scroll + distance * progress await self.page.evaluate(f'window.scrollTo(0, {scroll_y})') await asyncio.sleep(random.uniform(0.05, 0.1)) logger.success(f"[平滑滚动] 已滚动到元素位置") else: # 降级方案：直接滚动到可见 await element.scroll_into_view_if_needed() logger.success(f"[平滑滚动] 使用降级方案滚动") except Exception as e: logger.error(f"[平滑滚动] 滚动失败: {e}") # 最终降级方案 try: await element.scroll_into_view_if_needed() except Exception: pass async def debug_print_inputs(self): """ 调试：打印页面上所有输入框信息（借鉴ai_mip的调试逻辑） """ try: logger.info("=" * 50) logger.info("[调试] 打印页面所有输入框...") logger.info("=" * 50) inputs = await self.page.query_selector_all('input') logger.info(f"[调试] 页面上找到 {len(inputs)} 个input元素") for i, inp in enumerate(inputs[:10]): # 只打印前10个 try: placeholder = await inp.get_attribute('placeholder') input_type = await inp.get_attribute('type') name = await inp.get_attribute('name') class_name = await inp.get_attribute('class') is_visible = await inp.is_visible() logger.info(f"[调试] Input {i+1}:") logger.info(f" - type: {input_type}") logger.info(f" - placeholder: {placeholder}") logger.info(f" - name: {name}") logger.info(f" - class: {class_name}") logger.info(f" - visible: {is_visible}") except Exception as e: logger.debug(f"[调试] 获取Input {i+1}信息失败: {e}") logger.info("=" * 50) except Exception as e: logger.error(f"[调试] 打印输入框信息失败: {e}") async def debug_print_buttons(self): """ 调试：打印页面上所有按钮信息 """ try: logger.info("=" * 50) logger.info("[调试] 打印页面所有按钮...") logger.info("=" * 50) buttons = await self.page.query_selector_all('button, div[role="button"], span[role="button"]') logger.info(f"[调试] 页面上找到 {len(buttons)} 个按钮元素") for i, btn in enumerate(buttons[:10]): # 只打印前10个 try: text = await btn.inner_text() class_name = await btn.get_attribute('class') is_visible = await btn.is_visible() logger.info(f"[调试] Button {i+1}:") logger.info(f" - text: {text}") logger.info(f" - class: {class_name}") logger.info(f" - visible: {is_visible}") except Exception as e: logger.debug(f"[调试] 获取Button {i+1}信息失败: {e}") logger.info("=" * 50) except Exception as e: logger.error(f"[调试] 打印按钮信息失败: {e}") # 定义常用的选择器配置（借鉴ai_mip的结构化选择器管理） class XHSSelectors: """小红书登录页面选择器配置""" # 手机号输入框选择器（创作者中心） PHONE_INPUT_CREATOR = [ 'input[placeholder="手机号"]', 'input.css-nt440g', 'input[placeholder*="手机号"]', 'input[type="tel"]', ] # 手机号输入框选择器（小红书首页） PHONE_INPUT_HOME = [ 'input[placeholder="输入手机号"]', 'label.phone input', 'input[name="blur"]', ] # 手机号输入框降级选择器 PHONE_INPUT_FALLBACK = [ 'input[type="text"]', 'input', ] # 验证码输入框选择器（创作者中心） CODE_INPUT_CREATOR = [ 'input[placeholder="验证码"]', 'input.css-1ge5flv', 'input[placeholder*="验证码"]', 'input[type="text"]:not([placeholder*="手机"])', ] # 验证码输入框选择器（小红书首页） CODE_INPUT_HOME = [ 'input[placeholder="输入验证码"]', 'label.auth-code input', 'input[type="number"]', 'input[placeholder*="验证码"]', ] # 发送验证码按钮选择器（创作者中心） SEND_CODE_BTN_CREATOR = [ 'div.css-uyobdj', 'text="发送验证码"', 'div:has-text("发送验证码")', 'text="重新发送"', 'text="获取验证码"', ] # 发送验证码按钮选择器（小红书首页） SEND_CODE_BTN_HOME = [ 'span.code-button', '.code-button', 'text="获取验证码"', 'span:has-text("获取验证码")', ] # 登录按钮选择器 LOGIN_BTN = [ 'button:has-text("登录")', 'text="登录"', 'div:has-text("登录")', '.login-button', 'button.login', ] # 协议复选框选择器（小红书首页） AGREEMENT_CHECKBOX = [ '.agree-icon', '.agreements .icon-wrapper', 'span.agree-icon', '.icon-wrapper', ] # 导出便捷函数 def get_login_helper(page: Page) -> XHSLoginHelper: """ 获取登录辅助器实例 Args: page: Playwright Page 对象 Returns: XHSLoginHelper实例 """ return XHSLoginHelper(page)