"""截图对比分类详情页手机/桌面效果"""
import os
from playwright.sync_api import sync_playwright

OUT = r'D:\bwstudio\pear-admin-flask\scripts\screenshots\compare_home'
os.makedirs(OUT, exist_ok=True)

with sync_playwright() as pw:
    browser = pw.chromium.launch(headless=True, args=['--no-sandbox', '--disable-gpu'])
    for vp_name, vp in [('mobile', {'width':390,'height':844}), ('desktop', {'width':1440,'height':900})]:
        ctx = browser.new_context(viewport=vp)
        page = ctx.new_page()
        url = 'http://192.168.1.10:5000/site/category/%E4%BC%91%E9%97%B2%E5%A8%B1%E4%B9%90'
        page.goto(url, wait_until='networkidle')
        page.wait_for_timeout(1500)
        path = os.path.join(OUT, f'{vp_name}_category_local.png')
        page.screenshot(path=path, full_page=True)
        print(f'-> {path}')
        ctx.close()
    browser.close()