xhs-note-extractor 0.1.dev2__py3-none-any.whl → 0.1.4__py3-none-any.whl

xhs_note_extractor/login_propmt.py

@@ -0,0 +1,145 @@
+phone_agent_protocol_v3_t1="""
+自动化验证码登录协议 - 任务 1：触发验证码
+你是一个强执行型 Agent，必须严格按照以下协议执行，不得有任何偏差。
+
+【最高优先级：异常检测与处理】
+- 在执行**每一个操作前**，必须首先检查是否出现异常情况
+- 一旦发现异常，必须**立即终止所有操作**，直接返回指定JSON格式错误信息
+- 异常处理优先于任何其他任务步骤，不得延迟或忽略
+
+核心任务
+目标：在小红书 App 中触发手机号 {0} 的验证码发送。
+注意：本任务仅负责触发验证码，**只有看到纯数字验证码输入框（可直接输入数字的界面）出现即视为完成**。
+【重要】：图片验证码（包括需要点击文字、旋转图片、滑块、拼图、拖拽等任何包含图片元素的验证机制）**不属于数字验证码输入框**，遇到此类情况应立即终止任务。
+
+执行流程
+
+=== 操作前强制检查 ===
+在执行任何操作前，必须先检查当前界面是否存在以下异常：
+1. 检查是否存在**图片验证码**（包括但不限于：需要点击文字的验证码、旋转图片的验证码、滑块验证码、拼图验证码、拖拽验证码等任何包含图片元素的验证机制）
+2. 检查是否存在**需要人工介入**的提示或界面（包括但不限于：扫码验证、人脸识别、指纹验证等）
+3. 检查是否存在**发送次数限制**的提示（包括但不限于："已超过今天发送短信条数限"、"发送次数过多"、"请稍后再试"等）
+
+【触发finish指令】若发现上述任何异常，立即终止任务并返回对应JSON格式错误信息。
+【特别注意】：图片验证码不属于任务完成标准中的数字验证码输入框，遇到图片验证码必须立即终止任务。
+
+=== 1. 登出清理 ===
+1. 打开小红书 App
+2. **检查异常**：操作前先执行【操作前强制检查】
+3. 检查登录状态：
+   - 若已登录，执行：我 -> 设置 -> 向下滑到底部 -> 点击退出登录
+   - 若未登录，直接进入下一步
+4. 输出：`{{"status":"info","message":"当前已处于未登录状态"}}`
+
+=== 2. 触发验证码 ===
+1. **检查异常**：操作前先执行【操作前强制检查】
+2. 在登录页选择【手机号验证码登录】
+3. **检查异常**：操作前先执行【操作前强制检查】
+4. 输入手机号 {0}，输入后必须确认手机号完整（忽略空格）
+5. **检查异常**：操作前先执行【操作前强制检查】
+6. 点击【获取验证码】按钮
+7. **立即检查异常**：点击后立即执行【操作前强制检查】
+
+=== 3. 异常处理（触发后立即执行） ===
+【必须严格执行，不得有任何延迟】
+
+- **情况1：出现任何形式的图片验证码**
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"账号出现异常","reason":"检测到图片验证码","code":"CAPTCHA_DETECTED"}}`
+
+- **情况2：出现需要人工介入的情况**
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"账号出现异常","reason":"需要人工介入验证","code":"HUMAN_VERIFICATION_REQUIRED"}}`
+
+- **情况3：出现发送次数限制提示**
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"今日短信发送次数已达上限","reason":"验证码发送次数超过限制","code":"SMS_LIMIT_EXCEEDED"}}`
+
+=== 4. 任务完成检查 ===
+- 【强制检查】：在判断任务是否完成前，必须先执行【操作前强制检查】
+- 只有当界面显示**纯数字验证码输入框**（可直接输入数字的界面，不包含任何图片元素）且无任何异常时，任务才算完成
+- 【重要】：若界面显示的是图片验证码（包括需要点击文字、旋转图片、滑块、拼图、拖拽等任何包含图片元素的验证机制），即使显示了文字验证输入框，也**不视为任务完成**，应立即终止任务
+- 【触发finish指令】若满足完成条件，立即停止所有操作，返回：`{{"status":"success","message":"任务已完成","reason":"成功触发验证码发送","code":"CAPTCHA_SENT"}}`
+
+=== 禁止的行为 ===
+1. 禁止查看任何协议的具体内容（包括但不限于隐私协议、用户协议等），但可以直接同意或确认协议
+2. 禁止点击【账号与安全】等与账号安全相关的操作
+3. 禁止等待验证码完全生成并显示
+4. 禁止点击任何复选框
+5. 禁止执行任何人工行为或请求人工介入
+6. 禁止在发现异常后继续执行任何操作
+
+=== 完成标准 ===
+- **成功标准**：必须同时满足以下条件
+  1. 界面显示**纯数字验证码输入框**（可直接输入数字的界面，不包含任何图片元素）
+  2. 没有检测到任何异常情况（图片验证码、人工介入提示、发送次数限制等）
+  3. 成功返回标准化JSON格式成功信息
+
+- **失败标准**：满足以下任一条件
+  1. 检测到任何异常情况（图片验证码、人工介入提示、发送次数限制等）
+  2. 界面显示的是图片验证码（包括需要点击文字、旋转图片、滑块、拼图、拖拽等任何包含图片元素的验证机制）
+  3. 没有看到纯数字验证码输入框
+  4. 返回标准化JSON格式错误信息
+
+"""
+
+phone_agent_protocol_v3_t3="""
+自动化验证码登录协议 - 任务 3：完成登录
+你是一个强执行型 Agent，必须严格按照以下协议执行，不得有任何偏差。
+
+【最高优先级：异常检测与处理】
+- 在执行**每一个操作前**，必须首先检查是否出现异常情况
+- 一旦发现异常，必须**立即终止所有操作**，直接返回指定JSON格式错误信息
+- 异常处理优先于任何其他任务步骤，不得延迟或忽略
+
+核心任务
+目标：使用验证码 {1} 完成手机号 {0} 的登录。
+注意：本任务仅处理纯数字验证码登录流程，**遇到图片验证码（包括需要点击文字、旋转图片、滑块、拼图、拖拽等任何包含图片元素的验证机制）应立即终止任务**。
+
+执行流程
+
+=== 操作前强制检查 ===
+在执行任何操作前，必须先检查当前界面是否存在以下异常：
+1. 检查是否存在**图片验证码**（包括但不限于：需要点击文字的验证码、旋转图片的验证码、滑块验证码、拼图验证码、拖拽验证码等任何包含图片元素的验证机制）
+2. 检查是否存在**验证码错误/失效**的提示（包括但不限于："验证码错误"、"验证码已失效"、"验证码过期"、"验证码不正确"等）
+3. 检查是否存在**需要人工介入**的提示或界面（包括但不限于：扫码验证、人脸识别、指纹验证等）
+
+【触发finish指令】若发现上述任何异常，立即终止任务并返回对应JSON格式错误信息。
+【特别注意】：本任务仅处理纯数字验证码登录流程，图片验证码不属于正常登录流程，遇到此类情况应立即终止任务。
+
+=== 1. 回到小红书 ===
+1. 打开小红书 App（应处于等待输入验证码界面）
+2. **检查异常**：操作前先执行【操作前强制检查】
+
+=== 2. 输入与登录 ===
+1. **检查异常**：操作前先执行【操作前强制检查】
+2. 点击验证码输入框
+3. **检查异常**：操作前先执行【操作前强制检查】
+4. 输入验证码 {1}
+5. **检查异常**：操作前先执行【操作前强制检查】
+6. 点击登录按钮
+7. **立即检查异常**：点击后立即执行【操作前强制检查】
+
+=== 3. 异常处理（触发后立即执行） ===
+【必须严格执行，不得有任何延迟】
+
+- **情况1：提示验证码错误/失效**（包括但不限于：验证码错误、验证码已失效、验证码过期、验证码不正确等）
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"验证码失效，请重新执行全流程","reason":"验证码验证失败","code":"VERIFICATION_CODE_FAILED"}}`
+
+- **情况2：出现任何形式的图片验证码**（包括但不限于：需要点击特定文字的验证码、旋转图片的验证码、滑块验证码、拼图验证码、拖拽验证码等）
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"验证码失效，请重新执行全流程","reason":"检测到图片验证码","code":"CAPTCHA_DETECTED"}}`
+
+- **情况3：出现需要人工介入的情况**（包括但不限于：扫码验证、人脸识别、指纹验证等）
+  → 【触发finish指令】立即终止任务并返回：`{{"status":"error","message":"验证码失效，请重新执行全流程","reason":"需要人工介入验证","code":"HUMAN_VERIFICATION_REQUIRED"}}`
+
+=== 4. 完成标准检查 ===
+1. 检查是否成功进入小红书首页
+2. 若成功进入首页，立即停止所有操作【触发finish指令】，返回：`{{"status":"success","message":"登录成功","reason":"成功完成小红书账号登录","code":"LOGIN_SUCCESS"}}`
+3. 若未进入首页，检查是否存在异常，若有异常则返回对应错误信息的JSON格式
+
+=== 禁止的行为 ===
+1. 禁止查看任何协议的具体内容（包括但不限于隐私协议、用户协议等），但可以直接同意或确认协议
+2. 禁止在发现异常后继续执行任何操作
+3. 禁止执行任何人工行为或请求人工介入
+4. 禁止修改或伪造验证码
+
+完成标准
+成功进入小红书首页并返回标准化的JSON格式成功信息：`{{"status":"success","message":"登录成功","reason":"成功完成小红书账号登录","code":"LOGIN_SUCCESS"}}`
+"""

xhs_note_extractor/number_utils.py

@@ -0,0 +1,44 @@
+"""
+数字格式转换工具
+"""
+
+def parse_count_to_int(count_str: str) -> int:
+    """
+    将小红书的互动数字符串转换为整数
+    
+    支持格式:
+    - "3.1万" -> 31000
+    - "3.1w" / "3.1W" -> 31000
+    - "1234" -> 1234
+    - "12.5万" -> 125000
+    
+    Args:
+        count_str: 数字字符串
+        
+    Returns:
+        int: 转换后的整数
+    """
+    if not count_str or count_str == "0":
+        return 0
+    
+    count_str = count_str.strip()
+    
+    # 处理 "万" 或 "w/W"
+    if '万' in count_str:
+        num_part = count_str.replace('万', '').strip()
+        try:
+            return int(float(num_part) * 10000)
+        except ValueError:
+            return 0
+    elif 'w' in count_str.lower():
+        num_part = count_str.lower().replace('w', '').strip()
+        try:
+            return int(float(num_part) * 10000)
+        except ValueError:
+            return 0
+    else:
+        # 普通数字
+        try:
+            return int(float(count_str))
+        except ValueError:
+            return 0

xhs_note_extractor/sms_verification.py

@@ -0,0 +1,307 @@
+import asyncio
+import os
+import re
+from datetime import datetime
+from playwright.async_api import async_playwright
+
+# 从环境变量获取配置，允许用户自定义
+SMS_SYSTEM_URL = os.getenv('SMS_SYSTEM_URL', 'https://smssystem.dicbs.com/admin/smsrecord')
+
+async def parse_sms_table(page, phone_number, send_time):
+    """
+    解析短信记录表格，提取符合条件的验证码
+    
+    Args:
+        page: Playwright页面对象
+        phone_number: 目标手机号
+        send_time: 发送时间，只返回在此时间之后的验证码
+        
+    Returns:
+        str: 符合条件的6位数字验证码，如果没有找到则返回None
+    """
+    try:
+        # 查找表格元素 - 使用更通用的选择器
+        table_selectors = [
+            'tbody',       # 尝试tbody元素
+            '#dataList tbody',  # 尝试特定ID下的tbody
+            'table tbody',      # 尝试任何表格下的tbody
+        ]
+        
+        tbody = None
+        for selector in table_selectors:
+            try:
+                tbody = await page.wait_for_selector(selector, timeout=3000)
+                print(f'使用选择器 {selector} 找到表格')
+                break
+            except:
+                continue
+        
+        if not tbody:
+            print('未找到表格元素，使用替代方案提取验证码')
+            return await extract_codes_from_text(page, send_time)
+        
+        # 获取所有行
+        rows = await tbody.query_selector_all('tr')
+        print(f'找到表格行数量: {len(rows)}')
+        
+        # 只处理第一条数据行（最新的记录）
+        if len(rows) > 0:
+            row = rows[0]
+            try:
+                # 获取行内所有单元格
+                cells = await row.query_selector_all('td')
+                
+                if len(cells) < 5:  # 确保有足够的单元格（根据截图，表格有5列）
+                    print(f'单元格数量不足（实际: {len(cells)}），跳过该行')
+                    return await extract_codes_from_text(page, send_time)
+                
+                # 根据截图调整列索引：
+                # 第0列：ID
+                # 第1列：短信内容
+                # 第2列：发送号码
+                # 第3列：接收号码
+                # 第4列：创建时间
+                
+                # 获取手机号（第4列）
+                phone_cell = cells[3]
+                phone_text = await phone_cell.inner_text()
+                print(f'检查手机号: {phone_text}')
+                
+                # 检查是否是目标手机号
+                if phone_number not in phone_text:
+                    print('手机号不匹配，尝试从页面文本提取')
+                    return await extract_codes_from_text(page, send_time)
+                
+                # 获取短信内容（第2列）
+                content_cell = cells[1]
+                content_text = await content_cell.inner_text()
+                print(f'短信内容: {content_text}')
+                
+                # 查找6位数字验证码
+                codes = re.findall(r'\d{6}', content_text)
+                if not codes:
+                    print('未找到6位验证码，尝试从页面文本提取')
+                    return await extract_codes_from_text(page, send_time)
+                
+                # 获取发送时间（第5列）
+                time_cell = cells[4]
+                time_text = await time_cell.inner_text()
+                print(f'发送时间: {time_text}')
+                
+                # 解析时间格式
+                try:
+                    sms_time = datetime.strptime(time_text, '%Y-%m-%d %H:%M:%S')
+                except ValueError:
+                    print(f'无法解析时间格式: {time_text}')
+                    return await extract_codes_from_text(page, send_time)
+                
+                # 检查时间是否在发送时间之后
+                if sms_time > send_time:
+                    print(f'找到符合条件的验证码: {codes[0]} (时间: {time_text})')
+                    return codes[0]
+                else:
+                    print(f'验证码时间 {time_text} 早于发送时间 {send_time.strftime("%Y-%m-%d %H:%M:%S")}')
+                    return await extract_codes_from_text(page, send_time)
+                    
+            except Exception as e:
+                print(f'解析表格行失败: {e}')
+                return await extract_codes_from_text(page, send_time)
+        
+        print('表格中未找到符合条件的验证码，尝试从页面文本提取')
+        # 如果表格中没有找到，尝试从整个页面文本提取
+        return await extract_codes_from_text(page, send_time)
+        
+    except Exception as e:
+        print(f'解析表格失败: {e}')
+        # 失败时回退到文本提取
+        return await extract_codes_from_text(page, send_time)
+
+async def extract_codes_from_text(page, send_time):
+    """
+    从页面文本中提取验证码的备用方案
+    
+    Args:
+        page: Playwright页面对象
+        send_time: 发送时间
+        
+    Returns:
+        str: 符合条件的验证码，如果没有找到则返回None
+    """
+    try:
+        # 提取所有文本内容
+        all_text = await page.inner_text('body')
+        print(f'页面文本内容长度: {len(all_text)} 字符')
+        
+        # 查找所有6位数字验证码
+        all_codes = re.findall(r'\d{6}', all_text)
+        print(f'找到验证码数量: {len(all_codes)} 个')
+        
+        if all_codes:
+            # 返回最新的验证码（页面上的最后一个）
+            verification_code = all_codes[-1]
+            print(f'获取到验证码：[{verification_code}]')
+            return verification_code
+        else:
+            return None
+            
+    except Exception as e:
+        print(f'从文本提取验证码失败: {e}')
+        return None
+
+async def get_verification_code(phone_number, send_time=None, max_retries=5, retry_interval=5):
+    """
+    获取指定手机号的最新验证码
+    
+    Args:
+        phone_number (str): 目标手机号
+        send_time (datetime, optional): 发送时间，只返回在此时间之后的验证码
+            默认使用当前时间
+        max_retries (int, optional): 最大重试次数，默认5次
+        retry_interval (int, optional): 重试间隔时间（秒），默认5秒
+        
+    Returns:
+        str: 6位数字验证码
+        
+    Raises:
+        Exception: 如果经过多次尝试后仍未找到验证码
+    """
+    # 如果未指定发送时间，使用当前时间
+    if send_time is None:
+        send_time = datetime.now()
+        print(f'未指定发送时间，默认使用: {send_time.strftime("%Y-%m-%d %H:%M:%S")}')
+    
+    for retry in range(max_retries):
+        try:
+            print(f'\n=== 第 {retry + 1}/{max_retries} 次尝试 ===')
+            # 启动Playwright
+            async with async_playwright() as p:
+                # 启动浏览器
+                browser = await p.chromium.launch(
+                    headless=False,
+                    args=['--no-sandbox', '--disable-setuid-sandbox'],
+                    timeout=60000
+                )
+                
+                # 创建页面
+                page = await browser.new_page()
+                
+                # 访问登录页面
+                await page.goto('https://smssystem.dicbs.com/admin/user/login', 
+                              wait_until='domcontentloaded', timeout=60000)
+                
+                print(f'当前页面: {page.url}')
+                
+                # 检查是否需要登录
+                if 'login' in page.url:
+                    print('检测到未登录状态，开始登录...')
+                    
+                    # 输入账号密码
+                    await page.fill('input[name="username"]', '京东xhs')
+                    await page.fill('input[name="password"]', '123456')
+                    await page.fill('input[name="captcha"]', '1234')
+                    
+                    # 点击登录按钮
+                    await page.click('button')
+                    await page.wait_for_load_state('domcontentloaded', timeout=30000)
+                    
+                    # 检查是否登录成功
+                    if 'login' not in page.url:
+                        print('登录成功！')
+                    else:
+                        print('登录失败，可能是验证码错误，但继续尝试...')
+                
+                # 确保在目标页面
+                if SMS_SYSTEM_URL not in page.url:
+                    await page.goto(SMS_SYSTEM_URL, wait_until='domcontentloaded', timeout=60000)
+                
+                # 尝试搜索手机号
+                print(f'尝试搜索手机号：{phone_number}...')
+                try:
+                    # 查找搜索输入框
+                    search_box = await page.wait_for_selector('input[name*="receiver"], input[id*="receiver"]', timeout=10000)
+                    if search_box:
+                        await search_box.fill(phone_number)
+                        await page.click('#memberSearch')
+                        await page.wait_for_load_state('domcontentloaded', timeout=15000)
+                        print('搜索完成')
+                    else:
+                        print('未找到搜索框，直接提取表格内容')
+                except Exception as e:
+                    print(f'搜索失败: {e}')
+                
+                # 解析表格内容
+                print('解析表格内容...')
+                verification_code = await parse_sms_table(page, phone_number, send_time)
+                
+                if verification_code:
+                    await browser.close()
+                    return verification_code
+                else:
+                    print('未找到符合条件的验证码')
+                    await browser.close()
+            
+        except Exception as e:
+            print(f'获取验证码失败: {type(e).__name__}: {str(e)}')
+            
+        # 如果不是最后一次尝试，等待一段时间后重试
+        if retry < max_retries - 1:
+            print(f'等待 {retry_interval} 秒后重试...')
+            await asyncio.sleep(retry_interval)
+    
+    raise Exception(f'经过 {max_retries} 次尝试后仍未找到符合条件的验证码')
+
+
+def get_verification_code_sync(phone_number, send_time=None, max_retries=5, retry_interval=5):
+    """
+    同步版本的获取验证码函数
+    
+    Args:
+        phone_number (str): 目标手机号
+        send_time (datetime, optional): 发送时间，只返回在此时间之后的验证码
+            默认使用当前时间
+        max_retries (int, optional): 最大重试次数，默认5次
+        retry_interval (int, optional): 重试间隔时间（秒），默认5秒
+        
+    Returns:
+        str: 6位数字验证码
+        
+    Raises:
+        Exception: 如果经过多次尝试后仍未找到验证码
+    """
+    return asyncio.run(get_verification_code(
+        phone_number=phone_number,
+        send_time=send_time,
+        max_retries=max_retries,
+        retry_interval=retry_interval
+    ))
+
+
+if __name__ == '__main__':
+    """
+    示例用法
+    """
+    import sys
+    
+    if len(sys.argv) < 2 or len(sys.argv) > 3:
+        print('用法: python sms_verification.py <手机号> [<发送时间>]')
+        print('发送时间格式: YYYY-MM-DD HH:MM:SS')
+        sys.exit(1)
+    
+    phone_number = sys.argv[1]
+    
+    if len(sys.argv) == 3:
+        try:
+            send_time = datetime.strptime(sys.argv[2], '%Y-%m-%d %H:%M:%S')
+        except ValueError:
+            print('发送时间格式错误，请使用YYYY-MM-DD HH:MM:SS格式')
+            sys.exit(1)
+    else:
+        send_time = None
+    
+    try:
+        # 使用同步版本调用
+        code = get_verification_code_sync(phone_number, send_time)
+        print(f'\n最终结果：获取到验证码：[{code}]')
+    except Exception as e:
+        print(f'\n执行失败: {str(e)}')
+        sys.exit(1)

xhs_note_extractor/test_device_retry.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+设备重试机制测试脚本
+
+该脚本用于测试小红书笔记提取器的设备重试功能，
+当某个设备需要登录时，会自动尝试其他设备。
+"""
+
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from extractor import XHSNoteExtractor, extract_note_from_url
+import logging
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+def test_device_retry():
+    """测试设备重试机制"""
+    print("=" * 60)
+    print("小红书笔记提取器 - 设备重试机制测试")
+    print("=" * 60)
+    
+    # 测试URL（请替换为实际的小红书笔记URL）
+    test_url = "https://www.xiaohongshu.com/explore/65a1b2c3d4e5f6g7h8i9j0k1l2m3n4o5p6q7r8s9t0"
+    
+    try:
+        # 方式1：使用便捷函数
+        print("\n1. 使用便捷函数测试...")
+        result = extract_note_from_url(test_url, enable_time_logging=True)
+        
+        if result:
+            print(f"✓ 成功提取笔记数据")
+            print(f"  作者: {result.get('author_name', '未知')}")
+            print(f"  点赞数: {result.get('likes', '0')}")
+            print(f"  收藏数: {result.get('collects', '0')}")
+            print(f"  评论数: {result.get('comments', '0')}")
+            print(f"  图片数: {len(result.get('image_urls', []))}")
+            print(f"  正文预览: {result.get('content', '')[:100]}...")
+        else:
+            print("✗ 提取失败，所有设备都需要登录或不可用")
+        
+        # 方式2：使用类实例
+        print("\n2. 使用类实例测试...")
+        extractor = XHSNoteExtractor(enable_time_logging=True)
+        
+        # 显示可用设备
+        print(f"可用设备: {extractor.available_devices}")
+        
+        result = extractor.extract_note_data(url=test_url)
+        
+        if result:
+            print(f"✓ 成功提取笔记数据")
+            print(f"  作者: {result.get('author_name', '未知')}")
+            print(f"  点赞数: {result.get('likes', '0')}")
+            print(f"  收藏数: {result.get('collects', '0')}")
+            print(f"  评论数: {result.get('comments', '0')}")
+            print(f"  图片数: {len(result.get('image_urls', []))}")
+        else:
+            print("✗ 提取失败，所有设备都需要登录或不可用")
+            
+    except Exception as e:
+        print(f"✗ 测试过程中发生错误: {e}")
+        logger.exception("测试异常")
+
+def test_device_switching():
+    """测试设备切换功能"""
+    print("\n" + "=" * 60)
+    print("设备切换功能测试")
+    print("=" * 60)
+    
+    try:
+        extractor = XHSNoteExtractor(enable_time_logging=True)
+        
+        print(f"初始设备: {extractor.device.serial if extractor.device else '无'}")
+        print(f"可用设备列表: {extractor.available_devices}")
+        
+        if len(extractor.available_devices) > 1:
+            print("\n测试设备切换...")
+            success = extractor.switch_to_next_device()
+            if success:
+                print(f"✓ 成功切换到设备: {extractor.device.serial}")
+            else:
+                print("✗ 设备切换失败")
+        else:
+            print("只有一个或没有可用设备，无法测试切换功能")
+            
+    except Exception as e:
+        print(f"✗ 测试过程中发生错误: {e}")
+        logger.exception("测试异常")
+
+if __name__ == "__main__":
+    test_device_retry()
+    test_device_switching()

xhs_note_extractor/test_initialization_fix.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+"""
+测试设备重试机制的初始化修复
+"""
+
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from extractor import XHSNoteExtractor
+
+def test_initialization():
+    """测试初始化是否正常"""
+    print("开始测试初始化...")
+    
+    try:
+        # 测试初始化
+        extractor = XHSNoteExtractor()
+        print("✓ 初始化成功")
+        
+        # 测试属性是否存在adb connect 11.138.141.113:7121
+        print(f"可用设备: {extractor.available_devices}")
+        print(f"当前设备索引: {extractor.current_device_index}")
+        print(f"设备对象: {extractor.device}")
+        
+        # 测试连接设备
+        if extractor.connect_device():
+            print("✓ 设备连接成功")
+        else:
+            print("⚠ 设备连接失败（可能无设备连接）")
+            
+        return True
+        
+    except Exception as e:
+        print(f"✗ 测试失败: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+if __name__ == "__main__":
+    success = test_initialization()
+    if success:
+        print("\n✓ 所有测试通过")
+    else:
+        print("\n✗ 测试失败")
+        sys.exit(1)

{xhs_note_extractor-0.1.dev2.dist-info → xhs_note_extractor-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xhs-note-extractor
-Version: 0.1.dev2
+Version: 0.1.4
 Summary: A Python package for extracting Xiaohongshu (Little Red Book) note data from URLs
 Author-email: JoyCode Agent <agent@joycode.com>
 License: MIT
@@ -27,6 +27,9 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: uiautomator2>=2.16.17
 Requires-Dist: requests>=2.25.0
+Requires-Dist: playwright>=1.48.0
+Requires-Dist: typing-extensions>=4.13.2
+Requires-Dist: pydantic>=2.10.6
 Provides-Extra: dev
 Requires-Dist: pytest>=6.0; extra == "dev"
 Requires-Dist: pytest-cov>=2.0; extra == "dev"