iframe = page.frame('frame-name')  # 通過name/id/url定位iframe
element = iframe.locator('button#submit')  # 在iframe內(nèi)定位元素

iframe_element = page.locator('iframe#my-iframe')
element = iframe_element.locator(':scope >> button#submit')

from playwright.sync_api import sync_playwright

def demonstrate_iframe_handling():
    with sync_playwright() as p:
        # 啟動瀏覽器
        browser = p.chromium.launch(headless=False)
        page = browser.new_page()
        
        # 導(dǎo)航到測試頁面（包含iframe的示例頁面）
        page.goto('https://www.w3schools.com/tags/tryit.asp?filename=tryhtml_iframe')
        
        # 示例1：通過name屬性定位iframe
        try:
            iframe = page.frame(name="iframeResult")
            h1_element = iframe.locator("h1")
            print("通過name定位 - h1內(nèi)容:", h1_element.inner_text())
        except Exception as e:
            print("通過name定位失敗:", str(e))
        
        # 示例2：通過iframe元素定位
        iframe_element = page.locator("iframe#iframeResult")
        frame = iframe_element.content_frame()
        if frame:
            print("通過iframe元素定位 - 頁面標(biāo)題:", frame.title())
        
        # 示例3：自動檢測元素所在的iframe
        def find_element_context(page, selector):
            # 首先在主frame中查找
            element = page.locator(selector)
            if element.count() > 0:
                return {"element": element, "frame": page.main_frame}
            
            # 檢查所有iframe
            for frame in page.frames[1:]:
                try:
                    element = frame.locator(selector)
                    if element.count() > 0:
                        return {"element": element, "frame": frame}
                except:
                    continue
            return None
        
        # 查找h1元素所在的上下文
        result = find_element_context(page, "h1")
        if result:
            print("\n自動檢測結(jié)果:")
            print("元素文本:", result["element"].first.inner_text())
            print("所在frame URL:", result["frame"].url)
            print("frame名稱:", result["frame"].name or "無")
        
        # 示例4：處理嵌套iframe
        # 假設(shè)有二級嵌套iframe: page > iframe1 > iframe2
        # 定位方法:
        # iframe1 = page.frame("iframe1")
        # iframe2 = iframe1.frame("iframe2")
        # element = iframe2.locator("button")
        
        browser.close()

if __name__ == "__main__":
    demonstrate_iframe_handling()

# 等待iframe加載并定位元素
with page.expect_frame(url=lambda url: "login" in url) as frame_info:
    page.click("button#load-iframe")  # 觸發(fā)iframe加載
login_frame = frame_info.value
login_frame.fill("#username", "admin")

# 保存主frame上下文
main_frame = page.main_frame

# 切換到iframe操作
iframe = page.frame("content")
iframe.click("button")

# 切換回主frame
main_frame.click("home-link")

def get_frame_info(frame):
    return {
        "url": frame.url,
        "name": frame.name,
        "title": frame.title(),
        "parent_url": frame.parent_frame.url if frame.parent_frame else None,
        "child_count": len(frame.child_frames),
        "element_attributes": get_iframe_element_attrs(frame)
    }

def get_iframe_element_attr(frame):
    element = frame.frame_element()
    return {
        "id": element.get_attribute("id"),
        "class": element.get_attribute("class"),
        "src": element.get_attribute("src"),
        "width": element.get_attribute("width"),
        "height": element.get_attribute("height")
    }

from playwright.sync_api import sync_playwright
import time

def find_element_with_iframe_context(page, selector, timeout=10, verbose=False):
    """
    查找元素并確定它所在的iframe，同時收集iframe的詳細(xì)信息
    
    參數(shù):
        page: Playwright頁面對象
        selector: 要查找的元素選擇器
        timeout: 等待元素出現(xiàn)的超時時間(秒)
        verbose: 是否打印詳細(xì)過程信息
        
    返回:
        包含元素和iframe信息的字典，如果找不到返回None
    """
    start_time = time.time()
    last_frame_count = 0
    
    while time.time() - start_time < timeout:
        # 獲取當(dāng)前所有frame(包括主frame和iframe)
        frames = page.frames
        
        if verbose:
            print(f"\n檢查幀... 當(dāng)前幀數(shù): {len(frames)}")
            if len(frames) != last_frame_count:
                print("幀數(shù)量變化，重新掃描")
                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的詳細(xì)信息
                    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的詳細(xì)信息"""
    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"標(biāo)題: {frame_info['frame_info']['title']}")
    
    if frame_info['frame_type'] == 'iframe':
        print("\niframe詳細(xì)信息:")
        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()
        
        # 導(dǎo)航到測試頁面（這里用包含iframe的示例頁面）
        page.goto('https://www.w3schools.com/tags/tryit.asp?filename=tryhtml_iframe')
        
        # 等待頁面加載
        page.wait_for_load_state('networkidle')
        
        # 要查找的元素選擇器（這里選擇iframe內(nèi)的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"元素文本內(nèi)容: {result['element'].inner_text()}")
            
            # 打印frame的詳細(xì)信息
            print_frame_info(result)
            
            # 現(xiàn)在你可以使用result['frame']來操作這個frame
            # 例如: result['frame'].click(target_selector)
        else:
            print(f"\n未找到元素: '{target_selector}'")
        
        browser.close()

if __name__ == '__main__':
    main()