mirror of
https://forgejo.altau.su/lego/lego-monitoring.git
synced 2026-03-10 04:41:10 +00:00
67 lines
1.8 KiB
Python
Executable file
67 lines
1.8 KiB
Python
Executable file
#!/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):
|
|
while True:
|
|
logging.info(f"Calling {check.__name__}")
|
|
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():
|
|
MINUTE = 60
|
|
HOUR = 60 * MINUTE
|
|
DAY = 24 * HOUR
|
|
WEEK = 7 * DAY
|
|
|
|
signal.signal(signal.SIGTERM, stop_gracefully)
|
|
client = await alerts.get_client()
|
|
checkers = (
|
|
checker(checks.temp_check, 5 * MINUTE, client),
|
|
checker(checks.cpu_check, 5 * MINUTE, client),
|
|
checker(checks.ram_check, 1 * MINUTE, client),
|
|
checker(checks.vuln_check, 1 * DAY, 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())
|