#!/usr/bin/env python3
import subprocess
import tempfile
from pathlib import Path
try:
from playwright.sync_api import sync_playwright # type: ignore[import-not-found]
except ImportError:
msg = """
playwright not found. Install it with:
pip install playwright
playwright install chromium
Or, run this script with:
nox -s update_how_it_works_image
"""
raise SystemExit(msg) from None
def main() -> None:
with tempfile.TemporaryDirectory() as tmp_dir_str:
tmp_dir = Path(tmp_dir_str)
subprocess.run(["mkdocs", "build", "--site-dir", tmp_dir], check=True)
html_str = Path("docs/diagram.html").read_text()
css_tags = f"""
"""
html_str = f"{css_tags}{html_str}"
html_path = Path(tmp_dir) / "diagram_screenshot.html"
html_path.write_text(html_str)
dest_path = Path("docs/data/how-it-works.png")
with sync_playwright() as p:
browser = p.chromium.launch()
page = browser.new_page(device_scale_factor=2, viewport={"width": 830, "height": 600})
page.goto(html_path.as_uri())
page.wait_for_load_state("networkidle")
height = page.evaluate("document.body.scrollHeight")
page.set_viewport_size({"width": 830, "height": height})
page.screenshot(path=str(dest_path), full_page=True)
browser.close()
if __name__ == "__main__":
main()