Fix lint: make scripts/render_terminal.py flake8-clean

scripts/render_terminal.py

@@ -5,44 +5,91 @@ Usage (as a module):
     render(title, command, lines, png='out.png', gif='out.gif', width=1000, rules=[(regex, css_color), ...])
 `lines` is a list of plain-text output lines. No real terminal is needed, so output can be sanitized first.
 """
-import asyncio, html, io, re
+
+# flake8: noqa: E501  (the HTML/CSS page template below contains long literal lines)
+
+import asyncio
+import html
+import io
+import re
 from PIL import Image
 from playwright.async_api import async_playwright
 
+ANSI_COL = {
+    30: "#6e7681",
+    31: "#ff5c6c",
+    32: "#3ddc84",
+    33: "#ffb454",
+    34: "#58a6ff",
+    35: "#d2a8ff",
+    36: "#7fdbff",
+    37: "#e6edf3",
+    90: "#8b949e",
+    91: "#ff7b86",
+    92: "#56f09f",
+    93: "#ffd479",
+    94: "#79c0ff",
+    95: "#e2c5ff",
+    96: "#a5f3ff",
+    97: "#ffffff",
+}
+
 
-ANSI_COL={30:'#6e7681',31:'#ff5c6c',32:'#3ddc84',33:'#ffb454',34:'#58a6ff',35:'#d2a8ff',36:'#7fdbff',37:'#e6edf3',
-          90:'#8b949e',91:'#ff7b86',92:'#56f09f',93:'#ffd479',94:'#79c0ff',95:'#e2c5ff',96:'#a5f3ff',97:'#ffffff'}
 def ansi_to_html(line):
     """Convert ANSI SGR colour/bold sequences to <span> tags (everything else is stripped)."""
-    out=[]; open_spans=0; pos=0
-    for m in re.finditer(r'\x1b\[([0-9;]*)m|\x1b\[[0-9;?]*[A-Za-z]',line):
-        out.append(html.escape(line[pos:m.start()])); pos=m.end()
-        if m.group(1) is None: continue
-        codes=[int(c) if c else 0 for c in m.group(1).split(';')]
+    out = []
+    open_spans = 0
+    pos = 0
+    for m in re.finditer(r"\x1b\[([0-9;]*)m|\x1b\[[0-9;?]*[A-Za-z]", line):
+        out.append(html.escape(line[pos : m.start()]))
+        pos = m.end()
+        if m.group(1) is None:
+            continue
+        codes = [int(c) if c else 0 for c in m.group(1).split(";")]
         for c in codes:
-            if c==0: out.append('</span>'*open_spans); open_spans=0
-            elif c==1: out.append('<span style="font-weight:700">'); open_spans+=1
-            elif c in ANSI_COL: out.append(f'<span style="color:{ANSI_COL[c]}">'); open_spans+=1
-    out.append(html.escape(line[pos:])); out.append('</span>'*open_spans)
-    return ''.join(out)
+            if c == 0:
+                out.append("</span>" * open_spans)
+                open_spans = 0
+            elif c == 1:
+                out.append('<span style="font-weight:700">')
+                open_spans += 1
+            elif c in ANSI_COL:
+                out.append(f'<span style="color:{ANSI_COL[c]}">')
+                open_spans += 1
+    out.append(html.escape(line[pos:]))
+    out.append("</span>" * open_spans)
+    return "".join(out)
+
 
 def mask(lines, serial_cols=True, ips=True):
     """Sanitize output before publishing: hide drive serials and private IPs."""
-    res=[]
-    for l in lines:
-        if ips: l=re.sub(r'\b(10|192\.168|172\.(1[6-9]|2\d|3[01]))\.(\d+)\.(\d+)\.(\d+)\b','x.x.x.x',l) if not l.startswith('x') else l
-        if serial_cols and re.match(r'^(\d+|m2-\d)\s+/dev/',l):
-            parts=re.split(r'(\s{2,})',l)
+    res = []
+    for ln in lines:
+        if ips:
+            ln = (
+                re.sub(r"\b(10|192\.168|172\.(1[6-9]|2\d|3[01]))\.(\d+)\.(\d+)\.(\d+)\b", "x.x.x.x", ln)
+                if not ln.startswith("x")
+                else ln
+            )
+        if serial_cols and re.match(r"^(\d+|m2-\d)\s+/dev/", ln):
+            parts = re.split(r"(\s{2,})", ln)
             # fields alternate with whitespace; serial is the 8th field (index 14 in the split list)
-            if len(parts)>14:
-                v=parts[14]; parts[14]=v[:2]+'\u2022'*max(0,len(v)-2)
-            l=''.join(parts)
-        res.append(l)
+            if len(parts) > 14:
+                v = parts[14]
+                parts[14] = v[:2] + "\u2022" * max(0, len(v) - 2)
+            ln = "".join(parts)
+        res.append(ln)
     return res
 
-DEFAULT_RULES=[(r'✓|\bOK\b|\bNORMAL\b|\bPASS(ED)?\b|HEALTH_OK','#3ddc84'),(r'⚠️?|\bWARN(ING)?\b|HEALTH_WARN|\[WARN\]','#ffb454'),(r'\bERROR\b|\bFAIL(ED)?\b|\bCRITICAL\b|✗|HEALTH_ERR','#ff5c6c'),(r'^=+.*=+$|^┏.*|^┃.*|^┣.*|^┗.*','#7fdbff')]
 
-PAGE='''<!doctype html><meta charset=utf-8>
+DEFAULT_RULES = [
+    (r"✓|\bOK\b|\bNORMAL\b|\bPASS(ED)?\b|HEALTH_OK", "#3ddc84"),
+    (r"⚠️?|\bWARN(ING)?\b|HEALTH_WARN|\[WARN\]", "#ffb454"),
+    (r"\bERROR\b|\bFAIL(ED)?\b|\bCRITICAL\b|✗|HEALTH_ERR", "#ff5c6c"),
+    (r"^=+.*=+$|^┏.*|^┃.*|^┣.*|^┗.*", "#7fdbff"),
+]
+
+PAGE = """<!doctype html><meta charset=utf-8>
 <link href="https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;700&display=swap" rel="stylesheet">
 <style>
  body{margin:0;background:#0b0f14;font-family:'JetBrains Mono','DejaVu Sans Mono',monospace}
@@ -53,52 +100,90 @@ PAGE='''<!doctype html><meta charset=utf-8>
  pre{margin:0;padding:16px 18px;color:#c9d1d9;font-size:14px;line-height:1.42;white-space:pre-wrap;min-height:%(h)dpx}
  .p{color:#ff8c1a;font-weight:700}.c{color:#e6edf3}.cur{background:#c9d1d9;color:#0d1117}
 </style>
-<div id=win><div id=bar><span class=dot style=background:#ff5f56></span><span class=dot style=background:#ffbd2e></span><span class=dot style=background:#27c93f></span><span id=ttl>%(t)s</span></div><pre id=t></pre></div>'''
+<div id=win><div id=bar><span class=dot style=background:#ff5f56></span><span class=dot style=background:#ffbd2e></span><span class=dot style=background:#27c93f></span><span id=ttl>%(t)s</span></div><pre id=t></pre></div>"""
+
 
 def colorize(line, rules):
-    if '\x1b[' in line: return ansi_to_html(line)
-    s=html.escape(line)
-    for pat,col in rules:
-        if pat.startswith('^'):
-            if re.search(pat,line): return f'<span style="color:{col}">{s}</span>'
+    if "\x1b[" in line:
+        return ansi_to_html(line)
+    s = html.escape(line)
+    for pat, col in rules:
+        if pat.startswith("^"):
+            if re.search(pat, line):
+                return f'<span style="color:{col}">{s}</span>'
         else:
-            s=re.sub('('+pat+')',lambda m:f'<span style="color:{col}">{m.group(0)}</span>',s)
+            s = re.sub("(" + pat + ")", lambda m: f'<span style="color:{col}">{m.group(0)}</span>', s)
     return s
 
-async def _render(title,command,lines,png,gif,width,rules,min_h):
-    rules=rules or DEFAULT_RULES
-    prompt='<span class=p>root@node</span><span style="color:#8b949e">:~#</span> '
+
+async def _render(title, command, lines, png, gif, width, rules, min_h):
+    rules = rules or DEFAULT_RULES
+    prompt = '<span class=p>root@node</span><span style="color:#8b949e">:~#</span> '
     async with async_playwright() as p:
-        b=await p.chromium.launch(); pg=await b.new_page(viewport={'width':width+2,'height':600})
-        await pg.set_content(PAGE%{'w':width,'h':min_h,'t':html.escape(title)}); await pg.wait_for_timeout(1200)
-        async def show(body,cursor=False):
-            await pg.evaluate("(h)=>{document.getElementById('t').innerHTML=h}",body+('<span class=cur> </span>' if cursor else ''))
+        b = await p.chromium.launch()
+        pg = await b.new_page(viewport={"width": width + 2, "height": 600})
+        await pg.set_content(PAGE % {"w": width, "h": min_h, "t": html.escape(title)})
+        await pg.wait_for_timeout(1200)
+
+        async def show(body, cursor=False):
+            await pg.evaluate(
+                "(h)=>{document.getElementById('t').innerHTML=h}", body + ("<span class=cur> </span>" if cursor else "")
+            )
+
         async def shot():
-            el=pg.locator('#win'); return await el.screenshot()
-        full=prompt+'<span class=c>'+html.escape(command)+'</span>\n'+'\n'.join(colorize(l,rules) for l in lines)
+            el = pg.locator("#win")
+            return await el.screenshot()
+
+        full = (
+            prompt
+            + "<span class=c>"
+            + html.escape(command)
+            + "</span>\n"
+            + "\n".join(colorize(ln, rules) for ln in lines)
+        )
         await show(full)
-        if png: open(png,'wb').write(await shot()); print('png',png)
+        if png:
+            open(png, "wb").write(await shot())
+            print("png", png)
         if gif:
-            frames=[]; dur=[]
-            def add(img,d): frames.append(Image.open(io.BytesIO(img)).convert('RGB')); dur.append(d)
-            await show(prompt,True); add(await shot(),500)
-            for i in range(1,len(command)+1,3):
-                await show(prompt+'<span class=c>'+html.escape(command[:i])+'</span>',True); add(await shot(),45)
-            await show(prompt+'<span class=c>'+html.escape(command)+'</span>',True); add(await shot(),500)
-            head=prompt+'<span class=c>'+html.escape(command)+'</span>\n'; out=[]
-            step=max(1,len(lines)//20)
-            for i in range(0,len(lines),step):
-                out=lines[:i+step]; await show(head+'\n'.join(colorize(l,rules) for l in out)); add(await shot(),120)
-            await show(full); add(await shot(),3500)
-            w0=frames[0].width
-            scale=min(1.0,780/w0); frames=[f.resize((int(f.width*scale),int(f.height*scale)),Image.LANCZOS) for f in frames]
+            frames = []
+            dur = []
+
+            def add(img, d):
+                frames.append(Image.open(io.BytesIO(img)).convert("RGB"))
+                dur.append(d)
+
+            await show(prompt, True)
+            add(await shot(), 500)
+            for i in range(1, len(command) + 1, 3):
+                await show(prompt + "<span class=c>" + html.escape(command[:i]) + "</span>", True)
+                add(await shot(), 45)
+            await show(prompt + "<span class=c>" + html.escape(command) + "</span>", True)
+            add(await shot(), 500)
+            head = prompt + "<span class=c>" + html.escape(command) + "</span>\n"
+            out = []
+            step = max(1, len(lines) // 20)
+            for i in range(0, len(lines), step):
+                out = lines[: i + step]
+                await show(head + "\n".join(colorize(ln, rules) for ln in out))
+                add(await shot(), 120)
+            await show(full)
+            add(await shot(), 3500)
+            w0 = frames[0].width
+            scale = min(1.0, 780 / w0)
+            frames = [f.resize((int(f.width * scale), int(f.height * scale)), Image.LANCZOS) for f in frames]
             # all frames must share size: pad to the tallest
-            H=max(f.height for f in frames); W=frames[0].width
-            norm=[]
+            H = max(f.height for f in frames)
+            W = frames[0].width
+            norm = []
             for f in frames:
-                c=Image.new('RGB',(W,H),(13,17,23)); c.paste(f,(0,0)); norm.append(c.quantize(colors=32,method=Image.MEDIANCUT,dither=Image.NONE))
-            norm[0].save(gif,save_all=True,append_images=norm[1:],duration=dur,loop=0,optimize=True); print('gif',gif,len(norm))
+                c = Image.new("RGB", (W, H), (13, 17, 23))
+                c.paste(f, (0, 0))
+                norm.append(c.quantize(colors=32, method=Image.MEDIANCUT, dither=Image.NONE))
+            norm[0].save(gif, save_all=True, append_images=norm[1:], duration=dur, loop=0, optimize=True)
+            print("gif", gif, len(norm))
         await b.close()
 
-def render(title,command,lines,png=None,gif=None,width=1000,rules=None,min_h=200):
-    asyncio.run(_render(title,command,lines,png,gif,width,rules,min_h))
+
+def render(title, command, lines, png=None, gif=None, width=1000, rules=None, min_h=200):
+    asyncio.run(_render(title, command, lines, png, gif, width, rules, min_h))