mirror of
https://forgejo.altau.su/lego/lego-monitoring.git
synced 2026-03-10 04:41:10 +00:00
78 lines
2.3 KiB
Python
Executable file
78 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import asyncio
|
|
import datetime
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import signal
|
|
|
|
from alerting import alerts, delayed
|
|
from misc import checks, cvars
|
|
from misc.checkers import interval_checker, scheduled_checker
|
|
from misc.common import TMP_DIR
|
|
from misc.config import get_config
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
stopping = False
|
|
|
|
|
|
def stop_gracefully(signum, frame):
|
|
global stopping
|
|
stopping = True
|
|
|
|
|
|
def create_temp_dir() -> None:
|
|
try:
|
|
os.mkdir(str(TMP_DIR))
|
|
except FileExistsError:
|
|
pass
|
|
os.chmod(str(TMP_DIR), 0o770)
|
|
try:
|
|
shutil.chown(str(TMP_DIR), user="root", group="alerting")
|
|
except LookupError:
|
|
shutil.chown(str(TMP_DIR), user="root", group="root")
|
|
with open(str(TMP_DIR / "pid"), "w") as pf:
|
|
pf.write(str(os.getpid()))
|
|
|
|
|
|
async def main():
|
|
create_temp_dir()
|
|
|
|
signal.signal(signal.SIGTERM, stop_gracefully)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
loop.add_signal_handler(signal.SIGUSR1, lambda: asyncio.create_task(delayed.handler()))
|
|
|
|
cvars.config.set(get_config())
|
|
|
|
client = await alerts.get_client()
|
|
cvars.matrix_client.set(client)
|
|
checkers = (
|
|
interval_checker(checks.temp_check, datetime.timedelta(minutes=5)),
|
|
interval_checker(checks.cpu_check, datetime.timedelta(minutes=5)),
|
|
interval_checker(checks.ups_check, datetime.timedelta(minutes=5)),
|
|
interval_checker(checks.ram_check, datetime.timedelta(minutes=1)),
|
|
interval_checker(checks.vuln_check, datetime.timedelta(days=1)),
|
|
interval_checker(checks.raid_check, datetime.timedelta(days=1)),
|
|
interval_checker(checks.disk_wearout_check, datetime.timedelta(days=1)),
|
|
scheduled_checker(
|
|
checks.docker_registry_check, period=datetime.timedelta(days=1), when=datetime.time(hour=0, minute=0)
|
|
),
|
|
)
|
|
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()
|
|
shutil.rmtree(str(TMP_DIR))
|
|
raise SystemExit
|
|
else:
|
|
await asyncio.sleep(3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|