alerting service

service.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python3
+import asyncio
+import logging
+import signal
+from typing import Callable, Coroutine
+
+import nio
+
+from alerting import alerts
+from misc import checks
+
+logging.basicConfig(level=logging.INFO)
+
+stopping = False
+
+
+def stop_gracefully(signum, frame):
+    global stopping
+    stopping = True
+
+
+async def checker(check: Callable | Coroutine, interval_secs: int, client: nio.AsyncClient, *args, **kwargs):
+    logging.info(f"Calling {check.__name__}")
+    while True:
+        if isinstance(check, Callable):
+            result = check(*args, **kwargs)
+            if isinstance(result, Coroutine):
+                result = await result
+        elif isinstance(check, Coroutine):
+            result = await check
+        else:
+            raise TypeError(f"check is {type(check)}, neither function nor coroutine")
+        logging.info(f"Got {len(result)} alerts")
+        for alert in result:
+            await alerts.send_alert(alert, client)
+        await asyncio.sleep(interval_secs)
+
+
+async def main():
+    signal.signal(signal.SIGTERM, stop_gracefully)
+    client = await alerts.get_client()
+    checkers = (checker(checks.temp_check, 5 * 60, client),)
+    async with asyncio.TaskGroup() as tg:
+        checker_tasks: set[asyncio.Task] = set()
+        for c in checkers:
+            task = tg.create_task(c)
+            checker_tasks.add(task)
+        while True:
+            if stopping:
+                await client.close()
+                raise SystemExit
+            else:
+                await asyncio.sleep(3)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())