import os
from playwright.sync_api import sync_playwright
import time
out = os.path.join(os.environ.get('TEMP', 'C:/tmp'), 'qiwei-group-ops-prompt-modal3.png')
with sync_playwright() as p:
    browser = p.chromium.launch(headless=True)
    page = browser.new_page(viewport={'width': 1440, 'height': 900})
    page.goto('http://127.0.0.1:4320/')
    time.sleep(2)
    page.click('text=社群运营')
    time.sleep(2)
    page.click('text=SOP 版本')
    time.sleep(1)
    btn = page.query_selector('[data-group-ops-new-version]')
    print('new version button found:', btn is not None)
    if btn:
        btn.click()
        time.sleep(1)
        page.screenshot(path=out, full_page=False)
        print('screenshot saved to', out)
    browser.close()
print('done')