mirror of
https://forgejo.altau.su/lego/lego-monitoring.git
synced 2026-03-09 20:31:10 +00:00
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
import argparse
|
|
import asyncio
|
|
import datetime
|
|
import logging
|
|
import signal
|
|
from typing import Coroutine
|
|
|
|
from . import checks
|
|
from .alerting import sender
|
|
from .alerting.commands import CommandHandlerManager
|
|
from .checks.temp.sensors import print_readings
|
|
from .config import enums as config_enums
|
|
from .config import load_config
|
|
from .core import cvars
|
|
from .core.checkers import BaseChecker, IntervalChecker, ScheduledChecker
|
|
|
|
stopping = False
|
|
|
|
|
|
def stop_gracefully(signum, frame):
|
|
global stopping
|
|
stopping = True
|
|
|
|
|
|
def main() -> None:
|
|
asyncio.run(async_main())
|
|
|
|
|
|
async def async_main():
|
|
parser = argparse.ArgumentParser(
|
|
prog="lego-monitoring",
|
|
description="Lego-monitoring service",
|
|
)
|
|
parser.add_argument("-c", "--config", help="config file")
|
|
parser.add_argument("--print-temp", help="print temp sensor readings and exit", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
if args.config:
|
|
config_path = parser.parse_args().config
|
|
config = load_config(config_path)
|
|
cvars.config.set(config)
|
|
|
|
if args.print_temp:
|
|
print_readings()
|
|
raise SystemExit
|
|
|
|
if not args.config:
|
|
raise RuntimeError("--config must be specified in standard operating mode")
|
|
|
|
logging.basicConfig(level=config.log_level)
|
|
|
|
tg_client = await sender.get_client()
|
|
|
|
cvars.tg_client.set(tg_client)
|
|
my_username = (await tg_client.get_me()).username
|
|
logging.info(f"Logged in as @{my_username}")
|
|
check_sets = config_enums.CheckSet
|
|
|
|
checker_sets: dict[config_enums.CheckSet, list[Coroutine | BaseChecker]] = {
|
|
check_sets.START: [sender.send_start_alert()],
|
|
check_sets.STOP: [], # this is checked later
|
|
check_sets.CPU: [IntervalChecker(checks.cpu_check, interval=datetime.timedelta(minutes=3), persistent=True)],
|
|
check_sets.RAM: [IntervalChecker(checks.ram_check, interval=datetime.timedelta(minutes=1), persistent=True)],
|
|
check_sets.TEMP: [IntervalChecker(checks.temp_check, interval=datetime.timedelta(minutes=5), persistent=True)],
|
|
check_sets.VULNIX: [
|
|
IntervalChecker(
|
|
checks.vulnix_check,
|
|
interval=datetime.timedelta(days=3),
|
|
persistent=True,
|
|
send_any_state=True,
|
|
# As those are checked less often than daily, reminds could lead to awkward situations
|
|
# when the vuln is fixed but you still get reminders about it for 2 more days.
|
|
remind=False,
|
|
)
|
|
],
|
|
check_sets.REMIND: [
|
|
ScheduledChecker(
|
|
checks.remind_check,
|
|
period=datetime.timedelta(days=1),
|
|
when=datetime.time(hour=0, minute=0),
|
|
persistent=False,
|
|
is_reminder=True,
|
|
)
|
|
],
|
|
}
|
|
|
|
checkers = []
|
|
for enabled_set in config.enabled_check_sets:
|
|
for checker in checker_sets[enabled_set]:
|
|
checkers.append(checker)
|
|
|
|
checker_sets[check_sets.REMIND][0].check_args = [checkers]
|
|
|
|
command_manager = CommandHandlerManager(checkers)
|
|
await command_manager.attach_handlers(tg_client)
|
|
|
|
signal.signal(signal.SIGTERM, stop_gracefully)
|
|
|
|
async with asyncio.TaskGroup() as tg:
|
|
checker_tasks: set[asyncio.Task] = set()
|
|
for c in checkers:
|
|
if isinstance(c, BaseChecker):
|
|
c = c.run_checker()
|
|
task = tg.create_task(c)
|
|
checker_tasks.add(task)
|
|
while True:
|
|
if stopping:
|
|
if "stop" in config.enabled_check_sets:
|
|
await sender.send_stop_alert()
|
|
await tg_client.disconnect()
|
|
raise SystemExit
|
|
else:
|
|
await asyncio.sleep(3)
|