Genesis commit

lib.py

@@ -0,0 +1,38 @@
+from typing import AsyncIterator
+from pydantic import BaseModel
+import asyncio
+
+
+class CommandModel(BaseModel):
+    password: str
+    cmd: str
+
+
+class LogsModel(BaseModel):
+    password: str
+
+
+def sse_event(data: str, event: str | None = None, id: str | None = None) -> bytes:
+    parts = []
+    if id is not None:
+        parts.append(f"id: {id}")
+    if event is not None:
+        parts.append(f"event: {event}")
+    for line in data.rstrip("\n").split("\n"):
+        parts.append(f"data: {line}")
+    parts.append("")
+    return ("\n".join(parts) + "\n").encode("utf-8")
+
+
+async def sse_from_iterator(iterator: AsyncIterator[str], keep_alive: float = 15):
+    ait = iterator.__aiter__()
+    while True:
+        try:
+            item = await asyncio.wait_for(ait.__anext__(), timeout=keep_alive)
+        except asyncio.TimeoutError:
+            yield b": keep-alive\n\n"
+            continue
+        except StopAsyncIteration:
+            break
+        yield sse_event(item)
+    yield sse_event("stream closed", event="close")