Playwright（Python） 解决iframe 上下文定位功能完整方案，附代码（下）

from playwright.sync_api import sync_playwrightimport timedef find_element_with_iframe_context(page, selector, timeout=10, verbose=False):    """    查找元素并确定它所在的iframe，同时收集iframe的详细信息    参数:        page: Playwright页面对象        selector: 要查找的元素选择器        timeout: 等待元素出现的超时时间(秒)        verbose: 是否打印详细过程信息    返回:        包含元素和iframe信息的字典，如果找不到返回None    """    start_time = time.time()    last_frame_count = 0    while time.time() - start_time < timeout:        # 获取当前所有frame(包括主frame和iframe)        frames = page.frames        if verbose:            print(f"\n检查帧... 当前帧数: {len(frames)}")            if len(frames) != last_frame_count:                print("帧数量变化，重新扫描")                last_frame_count = len(frames)        # 1. 首先在主frame中查找        if verbose:            print("检查主frame...")        main_frame = frames[0]        element = main_frame.query_selector(selector)        if element:            if verbose:                print("元素在主frame中找到")            return {                'element': element,                'frame_type': 'main_frame',                'frame': main_frame,                'frame_info': {                    'url': main_frame.url,                    'name': 'main_frame',                    'title': main_frame.title(),                    'parent_frame': None                }            }        # 2. 检查所有iframe        for i, frame in enumerate(frames[1:], start=1):            try:                if verbose:                    print(f"检查iframe #{i}...")                # 获取iframe元素句柄                frame_element = frame.frame_element()                # 尝试在iframe中查找元素                element = frame.query_selector(selector)                if element:                    if verbose:                        print(f"元素在iframe #{i}中找到")                    # 收集iframe的详细信息                    frame_info = {                        'url': frame.url,                        'name': frame.name or f"iframe_{i}",                        'title': frame.title(),                        'parent_frame': frame.parent_frame.url if frame.parent_frame else None,                        'html_attributes': {}                    }                    # 获取iframe元素的HTML属性                    attrs = ['id', 'class', 'src', 'width', 'height', 'title']                    for attr in attrs:                        value = frame_element.get_attribute(attr)                        if value:                            frame_info['html_attributes'][attr] = value                    return {                        'element': element,                        'frame_type': 'iframe',                        'frame': frame,                        'frame_info': frame_info                    }            except Exception as e:                if verbose:                    print(f"检查iframe #{i}时出错: {str(e)}")                continue        # 短暂等待后重试        time.sleep(0.5)    return None  # 超时后仍未找到元素def print_frame_info(frame_info):    """打印frame的详细信息"""    print("\n=== Frame信息 ===")    print(f"类型: {'主frame' if frame_info['frame_type'] == 'main_frame' else 'iframe'}")    print(f"URL: {frame_info['frame_info']['url']}")    print(f"标题: {frame_info['frame_info']['title']}")    if frame_info['frame_type'] == 'iframe':        print("\niframe详细信息:")        print(f"名称: {frame_info['frame_info']['name']}")        print(f"父frame URL: {frame_info['frame_info']['parent_frame']}")        print("HTML属性:")        for attr, value in frame_info['frame_info']['html_attributes'].items():            print(f"  {attr}: {value}")def main():    with sync_playwright() as p:        browser = p.chromium.launch(headless=False)        page = browser.new_page()        # 导航到测试页面（这里用包含iframe的示例页面）        page.goto('https://www.w3schools.com/tags/tryit.asp?filename=tryhtml_iframe')        # 等待页面加载        page.wait_for_load_state('networkidle')        # 要查找的元素选择器（这里选择iframe内的h1元素作为示例）        target_selector = 'h1'        # 查找元素并确定iframe上下文        result = find_element_with_iframe_context(            page,             selector=target_selector,            timeout=15,            verbose=True        )        if result:            print("\n=== 元素找到 ===")            print(f"元素选择器: '{target_selector}'")            print(f"元素文本内容: {result['element'].inner_text()}")            # 打印frame的详细信息            print_frame_info(result)            # 现在你可以使用result['frame']来操作这个frame            # 例如: result['frame'].click(target_selector)        else:            print(f"\n未找到元素: '{target_selector}'")        browser.close()if __name__ == '__main__':    main()