mobile-mcp-ai 2.1.2__py3-none-any.whl → 2.5.8__py3-none-any.whl

core/ai/test_generator_from_history.py

@@ -1,391 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-基于操作历史生成测试脚本 - 智能定位 + 自动降级
-
-功能：
-1. 从操作历史（operation_history）生成脚本
-2. 优先使用MCP验证过的定位方式（快速、准确）
-3. 定位失败时自动降级到智能定位（自愈能力）
-4. 页面改版后大部分用例能自动适应
-
-用法:
-    generator = TestGeneratorFromHistory()
-    script = generator.generate_from_history(
-        test_name="测试用例",
-        package_name="com.im30.way",
-        operation_history=client.operation_history
-    )
-    generator.save("test_generated.py", script)
-"""
-import sys
-import re
-from pathlib import Path
-from typing import List, Dict
-from datetime import datetime
-
-
-class TestGeneratorFromHistory:
-    """
-    基于操作历史生成测试脚本
-    
-    特点：
-    - 优先使用MCP验证过的定位方式（性能最优）
-    - 定位失败时自动降级到智能定位（自愈能力）
-    - 页面改版后大部分用例能自动适应
-    """
-    
-    def __init__(self, output_dir: str = "tests"):
-        """
-        初始化生成器
-        
-        Args:
-            output_dir: 生成的测试文件输出目录（默认tests，用于pytest）
-        """
-        self.output_dir = Path(output_dir)
-        self.output_dir.mkdir(exist_ok=True)
-        
-        # 🎯 弹窗关键词（用于识别可选操作）
-        self.popup_keywords = [
-            "允许", "取消", "确定", "同意", "拒绝", "关闭", "跳过", 
-            "知道了", "我知道了", "好的", "稍后", "暂不", "以后再说",
-            "Allow", "Cancel", "OK", "Agree", "Deny", "Close", "Skip",
-            "Got it", "Later", "Not now"
-        ]
-        
-        # 🎯 弹窗resource-id特征
-        self.popup_id_patterns = ["permission", "dialog", "alert", "popup", "grant"]
-    
-    def _is_popup_element(self, element: str, ref: str) -> bool:
-        """
-        判断是否是弹窗元素（可选操作）
-        
-        Args:
-            element: 元素描述
-            ref: 元素定位方式
-            
-        Returns:
-            True表示是弹窗元素
-        """
-        # 检查元素描述是否包含弹窗关键词
-        for keyword in self.popup_keywords:
-            if keyword in element:
-                return True
-        
-        # 检查resource-id是否包含弹窗特征
-        ref_lower = ref.lower()
-        for pattern in self.popup_id_patterns:
-            if pattern in ref_lower:
-                return True
-        
-        return False
-    
-    def _is_dropdown_scenario(self, operations: List[Dict], index: int) -> bool:
-        """
-        判断是否是下拉框场景
-        
-        Args:
-            operations: 操作历史列表
-            index: 当前操作的索引
-            
-        Returns:
-            True表示当前操作是下拉框选择的第二步（需要等待）
-        """
-        # 检查：当前是click，且前一个也是click
-        if index > 0:
-            current = operations[index]
-            previous = operations[index - 1]
-            
-            if current.get('action') == 'click' and previous.get('action') == 'click':
-                current_element = current.get('element', '')
-                
-                # 🎯 排除明显的按钮关键词
-                button_keywords = ["按钮", "button", "btn", "继续", "下一步", "跳过", "完成"]
-                for keyword in button_keywords:
-                    if keyword in current_element.lower():
-                        return False
-                
-                # 🎯 选项通常是1-5个字符（排除按钮后）
-                # 例如："北京"(2)、"男"(1)、"确定"(2)、"China"(5)
-                if 1 <= len(current_element) <= 5:
-                    return True
-        
-        return False
-    
-    def generate_from_history(
-        self, 
-        test_name: str, 
-        package_name: str,
-        operation_history: List[Dict]
-    ) -> str:
-        """
-        从操作历史生成测试脚本
-        
-        Args:
-            test_name: 测试用例名称
-            package_name: App包名
-            operation_history: 操作历史列表
-            
-        Returns:
-            生成的测试脚本内容
-        """
-        # 生成文件名（中文转拼音或直接使用）
-        safe_name = re.sub(r'[^\w\s-]', '', test_name).strip().replace(' ', '_')
-        
-        # 生成脚本内容（pytest格式）
-        script_lines = [
-            "#!/usr/bin/env python3",
-            "# -*- coding: utf-8 -*-",
-            f'"""',
-            f"移动端测试用例: {test_name}",
-            f"生成时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
-            f"",
-            f"✨ 特性：智能定位 + 自动降级",
-            f"   - 优先使用MCP验证过的定位方式（快速）",
-            f"   - 定位失败时自动降级到智能定位（自愈）",
-            f"   - 页面改版后大部分用例能自动适应",
-            f"",
-            f"运行方式:",
-            f"    pytest {safe_name}.py -v",
-            f"    pytest {safe_name}.py --alluredir=./allure-results  # 生成allure报告",
-            f'"""',
-            "import asyncio",
-            "import pytest",
-            "import sys",
-            "from pathlib import Path",
-            "",
-            "# 添加backend目录到路径",
-            "# tests目录结构: backend/mobile_mcp/tests/test_xxx.py",
-            "# 需要导入: backend/mobile_mcp/core/mobile_client.py",
-            "sys.path.insert(0, str(Path(__file__).parent.parent))",
-            "",
-            "from mobile_mcp.core.mobile_client import MobileClient",
-            "from mobile_mcp.core.locator.mobile_smart_locator import MobileSmartLocator",
-            "",
-            "",
-            f"PACKAGE_NAME = \"{package_name}\"",
-            "",
-            "",
-            "@pytest.fixture(scope='function')",
-            "async def mobile_client():",
-            "    \"\"\"",
-            "    pytest fixture: 创建并返回MobileClient实例",
-            "    scope='function': 每个测试函数都会创建一个新的client",
-            "    \"\"\"",
-            "    client = MobileClient(device_id=None)",
-            "    ",
-            "    # 🎯 附加智能定位器（用于降级场景）",
-            "    client.smart_locator = MobileSmartLocator(client)",
-            "    ",
-            "    # 启动App",
-            "    print(f\"\\n📱 启动App: {{PACKAGE_NAME}}\", file=sys.stderr)",
-            "    result = await client.launch_app(PACKAGE_NAME, wait_time=5)",
-            "    if not result.get('success'):",
-            "        raise Exception(f\"启动App失败: {{result.get('reason')}}\")",
-            "    ",
-            "    await asyncio.sleep(2)  # 等待页面加载",
-            "    ",
-            "    yield client",
-            "    ",
-            "    # 清理",
-            "    client.device_manager.disconnect()",
-            "",
-            "",
-            f"@pytest.mark.asyncio",
-            f"async def test_{safe_name.lower()}(mobile_client):",
-            f'    """',
-            f"    测试用例: {test_name}",
-            f"    ",
-            f"    Args:",
-            f"        mobile_client: pytest fixture，已启动App的MobileClient实例",
-            f'    """',
-            f"    client = mobile_client",
-            f"    ",
-            f"    print(\"=\" * 60, file=sys.stderr)",
-            f"    print(f\"🚀 {test_name}\", file=sys.stderr)",
-            f"    print(\"=\" * 60, file=sys.stderr)",
-            f"    ",
-            f"    try:",
-        ]
-        
-        # 根据操作历史生成测试步骤
-        step_index = 1
-        for op_index, operation in enumerate(operation_history):
-            action = operation.get('action')
-            element = operation.get('element', '')
-            ref = operation.get('ref', '')
-            
-            if action == 'click':
-                # 🎯 判断是否是弹窗元素（可选操作）
-                is_popup = self._is_popup_element(element, ref)
-                
-                script_lines.append(f"        # 步骤{step_index}: 点击 {element}")
-                script_lines.append(f"        print(f\"\\n步骤{step_index}: 点击 {element}\", file=sys.stderr)")
-                
-                # 🎯 弹窗元素：可选操作（不出现也不报错）
-                if is_popup:
-                    script_lines.append(f"        # 🎯 可选操作：弹窗/权限请求（不一定出现）")
-                    script_lines.append(f"        try:")
-                else:
-                    script_lines.append(f"        try:")
-                
-                # 根据ref类型生成不同的优先定位代码
-                if ref.startswith('vision_coord_'):
-                    # 视觉识别坐标：vision_coord_x_y
-                    parts = ref.replace('vision_coord_', '').split('_')
-                    if len(parts) >= 2:
-                        x, y = parts[0], parts[1]
-                        script_lines.append(f"            # 优先使用MCP验证过的坐标")
-                        script_lines.append(f"            client.u2.click({x}, {y})")
-                        script_lines.append(f"            print(f\"✅ 点击成功（坐标: {x}, {y}）\", file=sys.stderr)")
-                elif ref.startswith('[') and '][' in ref:
-                    # bounds坐标：[x1,y1][x2,y2]
-                    script_lines.append(f"            # 优先使用MCP验证过的bounds")
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                    script_lines.append(f"            print(f\"✅ 点击成功（bounds: {ref}）\", file=sys.stderr)")
-                elif ref.startswith('com.') or ':' in ref:
-                    # resource-id定位
-                    script_lines.append(f"            # 优先使用MCP验证过的resource-id")
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                    script_lines.append(f"            print(f\"✅ 点击成功（resource-id: {ref}）\", file=sys.stderr)")
-                else:
-                    # text/description定位
-                    script_lines.append(f"            # 优先使用MCP验证过的text/description")
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                    script_lines.append(f"            print(f\"✅ 点击成功（text: {ref}）\", file=sys.stderr)")
-                
-                # 添加降级逻辑（区分弹窗和普通元素）
-                if is_popup:
-                    # 🎯 弹窗：失败不报错，只打印提示
-                    script_lines.append(f"        except Exception as e:")
-                    script_lines.append(f"            # 弹窗未出现，跳过")
-                    script_lines.append(f"            print(f\"ℹ️  '{element}'未出现，跳过（可能已授权或无需操作）\", file=sys.stderr)")
-                else:
-                    # 🎯 普通元素：失败后启用智能定位
-                    script_lines.append(f"        except Exception as e:")
-                    script_lines.append(f"            # 🎯 原定位失效，启用智能定位（自愈）")
-                    script_lines.append(f"            print(f\"⚠️  原定位失效: {{e}}\", file=sys.stderr)")
-                    script_lines.append(f"            print(f\"🔍 启用智能定位重新查找'{element}'...\", file=sys.stderr)")
-                    script_lines.append(f"            ")
-                    script_lines.append(f"            locate_result = await client.smart_locator.locate(\"{element}\")")
-                    script_lines.append(f"            if locate_result:")
-                    script_lines.append(f"                await client.click(\"{element}\", ref=locate_result['ref'], verify=False)")
-                    script_lines.append(f"                print(f\"✅ 智能定位成功: {{locate_result['ref']}}\", file=sys.stderr)")
-                    script_lines.append(f"            else:")
-                    script_lines.append(f"                raise Exception(f\"❌ 智能定位也失败了，元素'{element}'可能已被删除或页面结构大幅改变\")")
-                script_lines.append(f"        ")
-                
-                # 🎯 下拉框场景：添加等待
-                if self._is_dropdown_scenario(operation_history, op_index):
-                    script_lines.append(f"        await asyncio.sleep(0.5)  # 🎯 等待下拉选项加载")
-                else:
-                    script_lines.append(f"        await asyncio.sleep(1.5)  # 等待页面响应")
-                
-                step_index += 1
-            
-            elif action == 'type':
-                text = operation.get('text', '')
-                script_lines.append(f"        # 步骤{step_index}: 在{element}输入 {text}")
-                script_lines.append(f"        print(f\"\\n步骤{step_index}: 在{element}输入 {text}\", file=sys.stderr)")
-                
-                # 🎯 生成智能定位 + 自动降级代码
-                script_lines.append(f"        try:")
-                
-                # 🎯 输入前先清空（避免内容累加）
-                script_lines.append(f"            # 🎯 先点击输入框聚焦")
-                if ref.startswith('[') and '][' in ref:
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                elif ref.startswith('com.') or ':' in ref:
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                else:
-                    script_lines.append(f"            await client.click(\"{element}\", ref=\"{ref}\", verify=False)")
-                
-                script_lines.append(f"            await asyncio.sleep(0.3)")
-                script_lines.append(f"            ")
-                script_lines.append(f"            # 🎯 清空输入框（避免内容累加）")
-                script_lines.append(f"            if client.platform == 'android':")
-                script_lines.append(f"                client.u2.clear_text()")
-                script_lines.append(f"            elif client.platform == 'ios':")
-                script_lines.append(f"                # iOS清空逻辑")
-                script_lines.append(f"                pass")
-                script_lines.append(f"            await asyncio.sleep(0.2)")
-                script_lines.append(f"            ")
-                
-                # 根据ref类型生成不同的优先定位代码
-                if ref.startswith('[') and '][' in ref:
-                    # bounds坐标
-                    script_lines.append(f"            # 优先使用MCP验证过的bounds")
-                    script_lines.append(f"            await client.type_text(\"{element}\", \"{text}\", ref=\"{ref}\")")
-                    script_lines.append(f"            print(f\"✅ 输入成功（bounds: {ref}）\", file=sys.stderr)")
-                elif ref.startswith('com.') or ':' in ref:
-                    # resource-id定位
-                    script_lines.append(f"            # 优先使用MCP验证过的resource-id")
-                    script_lines.append(f"            await client.type_text(\"{element}\", \"{text}\", ref=\"{ref}\")")
-                    script_lines.append(f"            print(f\"✅ 输入成功（resource-id: {ref}）\", file=sys.stderr)")
-                else:
-                    # text定位
-                    script_lines.append(f"            # 优先使用MCP验证过的text")
-                    script_lines.append(f"            await client.type_text(\"{element}\", \"{text}\", ref=\"{ref}\")")
-                    script_lines.append(f"            print(f\"✅ 输入成功（text: {ref}）\", file=sys.stderr)")
-                
-                # 添加降级逻辑
-                script_lines.append(f"        except Exception as e:")
-                script_lines.append(f"            # 🎯 原定位失效，启用智能定位（自愈）")
-                script_lines.append(f"            print(f\"⚠️  原定位失效: {{e}}\", file=sys.stderr)")
-                script_lines.append(f"            print(f\"🔍 启用智能定位重新查找'{element}'...\", file=sys.stderr)")
-                script_lines.append(f"            ")
-                script_lines.append(f"            locate_result = await client.smart_locator.locate(\"{element}\")")
-                script_lines.append(f"            if locate_result:")
-                script_lines.append(f"                # 重新点击聚焦")
-                script_lines.append(f"                await client.click(\"{element}\", ref=locate_result['ref'], verify=False)")
-                script_lines.append(f"                await asyncio.sleep(0.3)")
-                script_lines.append(f"                # 清空")
-                script_lines.append(f"                if client.platform == 'android':")
-                script_lines.append(f"                    client.u2.clear_text()")
-                script_lines.append(f"                await asyncio.sleep(0.2)")
-                script_lines.append(f"                # 输入")
-                script_lines.append(f"                await client.type_text(\"{element}\", \"{text}\", ref=locate_result['ref'])")
-                script_lines.append(f"                print(f\"✅ 智能定位成功: {{locate_result['ref']}}\", file=sys.stderr)")
-                script_lines.append(f"            else:")
-                script_lines.append(f"                raise Exception(f\"❌ 智能定位也失败了，元素'{element}'可能已被删除或页面结构大幅改变\")")
-                script_lines.append(f"        ")
-                script_lines.append(f"        await asyncio.sleep(1)  # 等待输入完成")
-                
-                step_index += 1
-        
-        # 添加结尾（pytest格式）
-        script_lines.extend([
-            f"        ",
-            f"        print(\"\\n✅ 测试完成！\", file=sys.stderr)",
-            f"        ",
-            f"    except AssertionError as e:",
-            f"        print(f\"\\n❌ 断言失败: {{e}}\", file=sys.stderr)",
-            f"        # 打印当前页面快照以便调试",
-            f"        snapshot = await client.snapshot()",
-            f"        print(f\"\\n当前页面快照:\\n{{snapshot[:500]}}...\", file=sys.stderr)",
-            f"        raise",
-            f"    except Exception as e:",
-            f"        print(f\"\\n❌ 测试失败: {{e}}\", file=sys.stderr)",
-            f"        import traceback",
-            f"        traceback.print_exc()",
-            f"        raise",
-        ])
-        
-        return '\n'.join(script_lines)
-    
-    def save(self, filename: str, script: str):
-        """
-        保存生成的测试脚本
-        
-        Args:
-            filename: 文件名（会自动添加.py后缀）
-            script: 脚本内容
-        """
-        if not filename.endswith('.py'):
-            filename += '.py'
-        
-        file_path = self.output_dir / filename
-        file_path.write_text(script, encoding='utf-8')
-        print(f"✅ 测试用例已保存: {file_path}", file=sys.stderr)
-        return file_path
-

core/ai/test_generator_standalone.py

@@ -1,293 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-生成独立的测试脚本 - 不依赖 mobile_mcp 包
-
-生成纯粹基于 uiautomator2 的测试脚本，用户可以直接运行
-"""
-import re
-from pathlib import Path
-from typing import List, Dict
-from datetime import datetime
-
-
-class StandaloneTestGenerator:
-    """
-    生成独立的测试脚本（不依赖 mobile_mcp 包）
-    
-    特点：
-    1. 只依赖 uiautomator2（用户常用库）
-    2. 使用 MCP 验证过的坐标/bounds/resource-id
-    3. 无需安装 mobile-mcp-ai 包即可运行
-    """
-    
-    def __init__(self, output_dir: str = "./tests"):
-        """
-        初始化生成器
-        
-        Args:
-            output_dir: 输出目录（默认为当前目录的tests子目录）
-        """
-        self.output_dir = Path(output_dir)
-        self.output_dir.mkdir(parents=True, exist_ok=True)
-    
-    def generate_from_history(
-        self, 
-        test_name: str, 
-        package_name: str,
-        operation_history: List[Dict],
-        device_id: str = None
-    ) -> str:
-        """
-        从操作历史生成独立的测试脚本
-        
-        Args:
-            test_name: 测试用例名称
-            package_name: App包名
-            operation_history: 操作历史列表
-            device_id: 设备ID（可选）
-            
-        Returns:
-            生成的测试脚本内容
-        """
-        safe_name = re.sub(r'[^\w\s-]', '', test_name).strip().replace(' ', '_')
-        
-        # 生成脚本头部
-        script_lines = self._generate_header(test_name, safe_name)
-        
-        # 生成导入部分
-        script_lines.extend(self._generate_imports())
-        
-        # 生成常量
-        script_lines.extend([
-            f'PACKAGE_NAME = "{package_name}"',
-            f'DEVICE_ID = {repr(device_id)}  # None表示自动选择第一个设备',
-            "",
-            ""
-        ])
-        
-        # 生成 fixture
-        script_lines.extend(self._generate_fixture())
-        
-        # 生成测试函数
-        script_lines.extend(self._generate_test_function(
-            test_name, safe_name, operation_history
-        ))
-        
-        return "\n".join(script_lines)
-    
-    def _generate_header(self, test_name: str, safe_name: str) -> List[str]:
-        """生成文件头部"""
-        return [
-            "#!/usr/bin/env python3",
-            "# -*- coding: utf-8 -*-",
-            f'"""',
-            f"移动端自动化测试: {test_name}",
-            f"生成时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
-            f"",
-            f"依赖: pip install uiautomator2 pytest pytest-asyncio",
-            f"",
-            f"运行方式:",
-            f"    pytest test_{safe_name}.py -v -s",
-            f"    pytest test_{safe_name}.py --alluredir=./allure-results  # 生成allure报告",
-            f'"""',
-            ""
-        ]
-    
-    def _generate_imports(self) -> List[str]:
-        """生成导入部分"""
-        return [
-            "import time",
-            "import pytest",
-            "import uiautomator2 as u2",
-            "",
-            ""
-        ]
-    
-    def _generate_fixture(self) -> List[str]:
-        """生成 pytest fixture"""
-        return [
-            "@pytest.fixture(scope='function')",
-            "def device():",
-            '    """',
-            "    pytest fixture: 创建并返回设备连接",
-            "    scope='function': 每个测试函数都会创建一个新的连接",
-            '    """',
-            "    # 连接设备",
-            "    d = u2.connect(DEVICE_ID)  # None表示自动选择第一个设备",
-            "    print(f\"\\n📱 连接设备: {d.device_info}\")",
-            "    ",
-            "    # 启动App",
-            "    print(f\"🚀 启动App: {PACKAGE_NAME}\")",
-            "    d.app_start(PACKAGE_NAME, stop=True)",
-            "    ",
-            "    # 🎯 智能等待：App启动+首页加载（5-8秒）",
-            "    print(\"⏳ 等待App启动和首页加载...\")",
-            "    time.sleep(2)  # 等待进程启动",
-            "    ",
-            "    # 等待页面稳定（检测连续2次页面内容相同）",
-            "    last_xml = None",
-            "    stable_count = 0",
-            "    max_wait = 8  # 最多等待8秒",
-            "    start_time = time.time()",
-            "    ",
-            "    while time.time() - start_time < max_wait:",
-            "        try:",
-            "            current_xml = d.dump_hierarchy()",
-            "            if current_xml == last_xml:",
-            "                stable_count += 1",
-            "                if stable_count >= 2:",
-            "                    print(f\"✅ 首页加载完成（{time.time() - start_time:.1f}秒）\")",
-            "                    break",
-            "            else:",
-            "                stable_count = 0",
-            "            last_xml = current_xml",
-            "            time.sleep(0.5)",
-            "        except:",
-            "            time.sleep(0.5)",
-            "    ",
-            "    yield d",
-            "    ",
-            "    # 清理（可选：关闭App）",
-            "    # d.app_stop(PACKAGE_NAME)",
-            "",
-            ""
-        ]
-    
-    def _generate_test_function(
-        self, 
-        test_name: str, 
-        safe_name: str, 
-        operations: List[Dict]
-    ) -> List[str]:
-        """生成测试函数"""
-        lines = [
-            f"def test_{safe_name.lower()}(device):",
-            f'    """',
-            f"    测试用例: {test_name}",
-            f"    ",
-            f"    Args:",
-            f"        device: pytest fixture，已启动App的设备连接",
-            f'    """',
-            f"    d = device",
-            f"    ",
-        ]
-        
-        step_index = 1
-        for op in operations:
-            action = op.get('action')
-            element = op.get('element', '')
-            ref = op.get('ref', '')
-            
-            if action == 'click':
-                lines.extend(self._generate_click_code(element, ref, step_index))
-                step_index += 1
-            elif action == 'type':
-                text = op.get('text', '')
-                lines.extend(self._generate_input_code(element, ref, text, step_index))
-                step_index += 1
-        
-        # 添加断言（可选）
-        lines.extend([
-            "    ",
-            "    # ✅ 测试完成",
-            "    print(\"✅ 测试通过\")",
-        ])
-        
-        return lines
-    
-    def _generate_click_code(self, element: str, ref: str, step: int) -> List[str]:
-        """生成点击代码"""
-        lines = [
-            f"    # 步骤{step}: 点击 {element}",
-            f"    print(f\"\\n步骤{step}: 点击 {element}\")",
-        ]
-        
-        # 🎯 判断是否需要更长等待（页面跳转类操作）
-        is_navigation = any(keyword in element.lower() for keyword in [
-            '首页', '搜索', '返回', '确定', '提交', '登录', '注册', 
-            'home', 'search', 'back', 'submit', 'login', 'register'
-        ])
-        wait_time = 2.0 if is_navigation else 1.5
-        
-        # 根据ref类型生成不同的点击代码
-        if ref.startswith('[') and '][' in ref:
-            # bounds坐标：[x1,y1][x2,y2]
-            import re
-            match = re.search(r'\[(\d+),(\d+)\]\[(\d+),(\d+)\]', ref)
-            if match:
-                x1, y1, x2, y2 = match.groups()
-                x = (int(x1) + int(x2)) // 2
-                y = (int(y1) + int(y2)) // 2
-                lines.extend([
-                    f"    d.click({x}, {y})  # 使用MCP验证过的坐标",
-                    f"    time.sleep({wait_time})  # 等待页面响应",
-                ])
-        elif ref.startswith('com.') or ':id/' in ref:
-            # resource-id
-            lines.extend([
-                f"    d(resourceId=\"{ref}\").click()  # 使用MCP验证过的resource-id",
-                f"    time.sleep({wait_time})  # 等待页面响应",
-            ])
-        else:
-            # text
-            lines.extend([
-                f"    d(text=\"{ref}\").click()  # 使用MCP验证过的text",
-                f"    time.sleep({wait_time})  # 等待页面响应",
-            ])
-        
-        lines.append("")
-        return lines
-    
-    def _generate_input_code(self, element: str, ref: str, text: str, step: int) -> List[str]:
-        """生成输入代码"""
-        lines = [
-            f"    # 步骤{step}: 在{element}输入 {text}",
-            f"    print(f\"\\n步骤{step}: 在{element}输入 {text}\")",
-        ]
-        
-        if ref.startswith('com.') or ':id/' in ref:
-            # resource-id
-            lines.extend([
-                f"    d(resourceId=\"{ref}\").click()  # 先点击聚焦",
-                f"    time.sleep(0.5)  # 等待键盘弹出",
-                f"    d(resourceId=\"{ref}\").clear_text()  # 清空",
-                f"    time.sleep(0.3)",
-                f"    d(resourceId=\"{ref}\").set_text(\"{text}\")  # 输入",
-                f"    time.sleep(1.5)  # 等待输入完成",
-            ])
-        else:
-            # text
-            lines.extend([
-                f"    d(text=\"{ref}\").click()  # 先点击聚焦",
-                f"    time.sleep(0.5)  # 等待键盘弹出",
-                f"    d.clear_text()  # 清空",
-                f"    time.sleep(0.3)",
-                f"    d.send_keys(\"{text}\")  # 输入",
-                f"    time.sleep(1.5)  # 等待输入完成",
-            ])
-        
-        lines.append("")
-        return lines
-    
-    def save(self, filename: str, script: str) -> Path:
-        """
-        保存脚本到文件
-        
-        Args:
-            filename: 文件名（不含.py后缀）
-            script: 脚本内容
-            
-        Returns:
-            保存的文件路径
-        """
-        if not filename.endswith('.py'):
-            filename = f"{filename}.py"
-        
-        file_path = self.output_dir / filename
-        with open(file_path, 'w', encoding='utf-8') as f:
-            f.write(script)
-        
-        print(f"✅ 测试用例已保存: {file_path}")
-        return file_path
-