ram check, configurable loglevel

This commit is contained in:
Alex Tau 2025-05-13 14:15:56 +03:00
parent 5095057a13
commit da85a566c4
10 changed files with 180 additions and 18 deletions

View file

@ -7,6 +7,7 @@ import signal
from . import checks
from .alerting import alerts
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 interval_checker
@ -20,8 +21,6 @@ def stop_gracefully(signum, frame):
def main() -> None:
logging.basicConfig(level=logging.INFO)
asyncio.run(async_main())
@ -46,17 +45,20 @@ async def async_main():
if not args.config:
raise RuntimeError("--config must be specified in standard operating mode")
logging.basicConfig(level=config.log_level)
tg_client = await alerts.get_client()
cvars.tg_client.set(tg_client)
check_sets = config_enums.CheckSet
checker_sets = {
"start": [
alerts.send_start_alert(),
],
"stop": [], # this is checked later
"temp": [interval_checker(checks.temp_check, datetime.timedelta(minutes=5))],
"vulnix": [interval_checker(checks.vulnix_check, datetime.timedelta(days=3))],
"cpu": [interval_checker(checks.cpu_check, datetime.timedelta(minutes=5))],
check_sets.START: [alerts.send_start_alert()],
check_sets.STOP: [], # this is checked later
check_sets.CPU: [interval_checker(checks.cpu_check, datetime.timedelta(minutes=5))],
check_sets.RAM: [interval_checker(checks.ram_check, datetime.timedelta(minutes=1))],
check_sets.TEMP: [interval_checker(checks.temp_check, datetime.timedelta(minutes=5))],
check_sets.VULNIX: [interval_checker(checks.vulnix_check, datetime.timedelta(days=3))],
}
checkers = []