基于 Playwright 的 Python RPA 录制脚本生成与优化指南

3. 高级录制技巧

录制上下文感知操作

录制时携带认证信息

playwright codegen --load-storage=auth.json https://admin.site.com

保存认证信息供后续使用

playwright codegen --save-storage=auth.json https://admin.site.com

录制自定义操作

录制时生成 Pytest 测试代码

playwright codegen --target=python-pytest https://example.com

录制 JavaScript 代码

playwright codegen --target=javascript https://example.com

4. 优化录制生成的代码

自动优化脚本

from playwright.sync_api import sync_playwright
import time

class RecordedScriptOptimizer:
    """优化录制生成的代码"""
    
    def __init__(self):
        self.playwright = sync_playwright().start()
        self.browser = self.playwright.chromium.launch(
            headless=False,
            args=['--disable-blink-features=AutomationControlled', '--start-maximized']
        )
    
    def optimize_iframe_operations(self, url, iframe_selector):
        """优化 iframe 操作"""
        context = self.browser.new_context(
            ignore_https_errors=True,
            java_script_enabled=True
        )
        page = context.new_page()
        try:
            # 1. 增加智能等待
            page.goto(url, wait_until="networkidle")
            
            # 2. 等待 iframe 加载的多种策略
            page.wait_for_selector(iframe_selector, state="attached")
            
            # 3. 自动重试机制
            max_retries = 3
            for attempt in range(max_retries):
                try:
                    # 4. 多种方式定位 iframe
                    frame = page.frame_locator(iframe_selector)
                    
                    # 5. 验证 iframe 内容已加载
                    frame.locator("body").wait_for(state="visible")
                    
                    # 这里是录制的主要操作
                    # 例如点击按钮
                    frame.locator("button#confirm").click()
                    
                    # 6. 等待操作反馈
                    page.wait_for_timeout(1000)
                    
                    # 7. 验证操作成功
                    success_indicator = frame.locator(".success-message")
                    expect(success_indicator).to_be_visible()
                    print(f"操作成功，尝试次数：{attempt + 1}")
                    break
                except Exception as e:
                    if attempt == max_retries - 1:
                        raise
                    print(f"尝试 {attempt + 1} 失败，重试...")
                    page.wait_for_timeout(2000)
        except Exception as e:
            # 8. 错误处理和截图
            page.screenshot(path=f"error_{int(time.time())}.png")
            raise
        finally:
            context.close()
    
    def close(self):
        self.browser.close()
        self.playwright.stop()

if __name__ == "__main__":
    optimizer = RecordedScriptOptimizer()
    optimizer.optimize_iframe_operations(
        url="https://your-app.com",
        iframe_selector="iframe.main-content"
    )
    optimizer.close()

企业级录制模板

import logging
from datetime import datetime
from pathlib import Path
from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('automation.log'),
        logging.StreamHandler()
    ]
)
logger = logging.getLogger(__name__)

class AutomatedWorkflow:
    def __init__(self, config):
        self.config = config
        self.screenshots_dir = Path("screenshots")
        self.screenshots_dir.mkdir(exist_ok=True)

    def take_screenshot(self, page, name):
        """记录关键步骤截图"""
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        path = self.screenshots_dir / f"{name}_{timestamp}.png"
        page.screenshot(path=path, full_page=True)
        logger.info(f"截图已保存：{path}")

    def execute_iframe_workflow(self):
        """执行录制的 iframe 工作流"""
        with sync_playwright() as p:
            # 启动浏览器
            browser = p.chromium.launch(
                headless=self.config.get("headless", False),
                slow_mo=500 # 减慢操作，便于观察
            )
            
            # 创建上下文
            context = browser.new_context(
                viewport={"width": 1920, "height": 1080},
                ignore_https_errors=True
            )
            page = context.new_page()
            try:
                logger.info("开始自动化流程")
                
                # 步骤 1：登录（如果需要）
                if self.config.get("login_required"):
                    self._login(page)
                    self.take_screenshot(page, "after_login")
                
                # 步骤 2：导航到目标页面
                page.goto(
                    self.config["target_url"],
                    wait_until="networkidle",
                    timeout=60000
                )
                self.take_screenshot(page, "page_loaded")
                
                # 步骤 3：处理 iframe
                logger.info("定位 iframe")
                iframe_selector = self.config["iframe_selector"]
                page.wait_for_selector(iframe_selector, timeout=30000)
                
                # 切换到 iframe
                frame = page.frame_locator(iframe_selector)
                
                # 步骤 4：在 iframe 内执行录制操作
                logger.info("在 iframe 内执行操作")
                # 示例操作：填写表单
                frame.locator("input[name='username']").fill("test_user")
                frame.locator("input[type='password']").fill("test_pass")
                self.take_screenshot(page, "form_filled")
                
                # 示例操作：点击提交按钮
                submit_button = frame.locator("button.submit-btn")
                submit_button.click()
                
                # 步骤 5：等待结果
                logger.info("等待操作结果")
                page.wait_for_timeout(3000)
                
                # 步骤 6：验证结果
                success_msg = frame.locator(".alert-success")
                if success_msg.count() > 0:
                    logger.info("操作成功完成")
                else:
                    logger.warning("未检测到成功提示")
                self.take_screenshot(page, "final_result")
                
                logger.info("自动化流程完成")
            except PlaywrightTimeoutError:
                logger.error("操作超时")
                self.take_screenshot(page, "timeout_error")
                raise
            except Exception as e:
                logger.error(f"执行失败：{e}")
                self.take_screenshot(page, "execution_error")
                raise
            finally:
                # 清理资源
                context.close()
                browser.close()

    def _login(self, page):
        """登录流程"""
        login_url = self.config.get("login_url")
        if login_url:
            page.goto(login_url)
            page.fill("#username", self.config["credentials"]["username"])
            page.fill("#password", self.config["credentials"]["password"])
            page.click("#login-btn")
            page.wait_for_url("**/dashboard")

配置示例

config = {
    "headless": False,
    "login_required": True,
    "login_url": "https://example.com/login",
    "credentials": {
        "username": "your_username",
        "password": "your_password"
    },
    "target_url": "https://example.com/app",
    "iframe_selector": "iframe.main-iframe",
    "actions": [
        {"type": "fill", "selector": "input.name", "value": "Test User"},
        {"type": "click", "selector": "button.submit"}
    ]
}

if __name__ == "__main__":
    workflow = AutomatedWorkflow(config)
    workflow.execute_iframe_workflow()

5. 录制最佳实践

录制前准备

1. 清理浏览器缓存：避免录制到旧的缓存数据。
2. 准备好测试数据：确保流程可完整执行。
3. 记录关键步骤：手动记录复杂操作流程。

录制时技巧

1. 慢速操作：不要操作太快，给录制器时间记录。
2. 明确选择器：尽量点击有明显特征的元素。
3. 验证步骤：在关键步骤后添加验证点。

录制后优化

1. 添加等待策略：替换硬编码的 wait_for_timeout。
2. 增强选择器：使用更稳定的 CSS 选择器。
3. 添加错误处理：完善异常处理逻辑。
4. 模块化代码：提取重复操作为函数。

6. 故障排除

常见问题解决

问题：录制时找不到 iframe

解决方案：

page.wait_for_function("""
() => {
    const iframe = document.querySelector('iframe');
    return iframe && iframe.contentWindow.document.readyState === 'complete';
}
""")

问题：iframe 跨域限制

**解决方案：**使用 try-catch 处理

try:
    frame = page.frame(name="target_frame")
except:
    # 备用方案：通过 URL 匹配
    frame = page.frame(url="**/target-path/**")

问题：动态 iframe ID

**解决方案：**使用部分属性匹配

frame = page.frame_locator("iframe[title*='目标']")

调试技巧

1. 启用调试模式

   playwright codegen --debug https://example.com
   

2. 生成带注释的代码

   playwright codegen --verbose https://example.com
   

3. 录制时查看选择器 在录制工具中，鼠标悬停元素会显示建议的选择器。

通过以上方法，你可以快速生成并优化操作 iframe 的自动化脚本。录制功能是一个很好的起点，结合手动优化可以创建出稳定可靠的企业级自动化脚本。