jared
4a1998046b
Lint / Python (flake8) (push) Failing after 2m9s
Lint / JS (eslint) (push) Successful in 11s
Lint / Deploy (push) Skipped
Security / Python Security (bandit) (push) Successful in 39s
Test / Python Tests (pytest) (push) Successful in 40s
Lint / Notify on failure (push) Successful in 3s
71 lines
5.2 KiB
Python
71 lines
5.2 KiB
Python
"""Regenerates docs/img/* from a LOCAL demo instance (fictional data). See README.md in this folder."""
|
|
import asyncio, io, sys
|
|
from PIL import Image
|
|
from playwright.async_api import async_playwright
|
|
import pathlib
|
|
OUT=str(pathlib.Path(__file__).resolve().parents[2]/'docs'/'img')
|
|
B='http://127.0.0.1:8500'
|
|
H={'Remote-User':'alex.rivera','Remote-Name':'Alex Rivera','Remote-Email':'alex@example.com','Remote-Groups':'admin'}
|
|
TOP="document.activeElement&&document.activeElement.blur();document.documentElement.style.scrollBehavior='auto';window.scrollTo(0,0)"
|
|
AV='<svg xmlns="http://www.w3.org/2000/svg" width="64" height="64"><rect width="64" height="64" fill="#050a10"/><text x="32" y="40" font-family="monospace" font-size="24" font-weight="700" fill="#00d4ff" text-anchor="middle">AR</text></svg>'
|
|
async def mk(p,theme='dark',w=1440,h=900):
|
|
b=await p.chromium.launch(); c=await b.new_context(viewport={'width':w,'height':h},extra_http_headers=H)
|
|
await c.add_init_script(f"try{{localStorage.setItem('lt_theme','{theme}')}}catch(e){{}}")
|
|
async def av(route): await route.fulfill(status=200,content_type='image/svg+xml',body=AV)
|
|
await c.route('**/api/avatar*',av)
|
|
return b,c
|
|
async def go(pg,path,wait=5500):
|
|
await pg.goto(B+path); await pg.wait_for_timeout(wait); await pg.evaluate(TOP); await pg.wait_for_timeout(300)
|
|
async def el(pg,sel,name,idx=0):
|
|
l=pg.locator(sel).nth(idx); await l.scroll_into_view_if_needed(); await pg.wait_for_timeout(300); await l.screenshot(path=f'{OUT}/{name}.png'); print(name,'ok')
|
|
async def select_port(pg,sw,idx):
|
|
await pg.locator(f'[data-action="select-port"][data-switch="{sw}"][data-port-idx="{idx}"]').first.click(); await pg.wait_for_timeout(1200)
|
|
async def shots():
|
|
async with async_playwright() as p:
|
|
b,c=await mk(p); pg=await c.new_page(); errs=[]; pg.on('pageerror',lambda e:errs.append(str(e)[:100]))
|
|
await go(pg,'/'); await pg.screenshot(path=f'{OUT}/dashboard.png'); print('dashboard ok')
|
|
await el(pg,'#topology-diagram','topology')
|
|
await el(pg,'.lt-section-header >> text=ALERT QUEUE','alerts-tmp') if False else None
|
|
full=Image.open(io.BytesIO(await pg.screenshot(full_page=True)))
|
|
# host cards + unifi table via element crops
|
|
hb=await pg.locator('#hosts-grid, .host-grid, .hosts-grid').first.bounding_box() if await pg.locator('#hosts-grid, .host-grid, .hosts-grid').count() else None
|
|
if hb: await pg.locator('#hosts-grid, .host-grid, .hosts-grid').first.screenshot(path=f'{OUT}/host-cards.png'); print('host-cards ok')
|
|
await go(pg,'/links'); await pg.screenshot(path=f'{OUT}/link-debug.png')
|
|
names=await pg.evaluate("[...document.querySelectorAll('.link-host-name')].map(e=>e.textContent.trim())"); print('hosts',names)
|
|
for nm,fn in (('large1','link-debug-sfp-warning'),('compute-storage-gpu-01','link-debug-sfp'),('Pro 24 PoE','link-debug-switch')):
|
|
i=next((k for k,n in enumerate(names) if nm in n),None)
|
|
if i is not None: await el(pg,'.link-host-panel',fn,i)
|
|
await go(pg,'/inspector'); await pg.screenshot(path=f'{OUT}/inspector.png')
|
|
await select_port(pg,'USW-Aggregation',4); await pg.screenshot(path=f'{OUT}/inspector-port.png',full_page=False)
|
|
await pg.locator('[data-action="run-diagnostic"]').first.click(); await pg.wait_for_timeout(9000)
|
|
await pg.evaluate("document.getElementById('diag-results')?.scrollIntoView({block:'start'})"); await pg.wait_for_timeout(600)
|
|
await pg.screenshot(path=f'{OUT}/inspector-diagnostics.png'); print('diagnostics ok')
|
|
await go(pg,'/suppressions'); await pg.screenshot(path=f'{OUT}/suppressions.png',full_page=True)
|
|
print('errors',errs[:3]); await b.close()
|
|
b,c=await mk(p,'light'); pg=await c.new_page(); await go(pg,'/'); await pg.screenshot(path=f'{OUT}/dashboard-light.png'); await b.close()
|
|
async def gif():
|
|
frames=[]
|
|
async with async_playwright() as p:
|
|
b,c=await mk(p,'dark',1280,720); pg=await c.new_page()
|
|
async def snap(n=1,ms=0):
|
|
if ms: await pg.wait_for_timeout(ms)
|
|
im=Image.open(io.BytesIO(await pg.screenshot())).convert('RGB').resize((960,540),Image.LANCZOS)
|
|
for _ in range(n): frames.append(im)
|
|
await go(pg,'/',5000); await snap(3)
|
|
await pg.mouse.wheel(0,620); await snap(3,600)
|
|
await pg.mouse.wheel(0,700); await snap(2,600)
|
|
await go(pg,'/links',4500); await snap(3)
|
|
await pg.mouse.wheel(0,500); await snap(2,500)
|
|
await go(pg,'/inspector',4000); await snap(2)
|
|
await select_port(pg,'USW-Aggregation',4); await snap(3)
|
|
await pg.locator('[data-action="run-diagnostic"]').first.click(); await snap(2,1500); await snap(2,7500)
|
|
await pg.evaluate("document.getElementById('diag-results')?.scrollIntoView({block:'start'})"); await snap(4,600)
|
|
await go(pg,'/suppressions',4000); await snap(3)
|
|
await b.close()
|
|
pal=[f.quantize(colors=96,method=Image.MEDIANCUT,dither=Image.NONE) for f in frames]
|
|
pal[0].save(f'{OUT}/demo.gif',save_all=True,append_images=pal[1:],duration=1100,loop=0,optimize=True); print('gif',len(frames))
|
|
async def main():
|
|
if 'gif' not in sys.argv: await shots()
|
|
if 'shots' not in sys.argv: await gif()
|
|
asyncio.run(main())
|