import logging import re import sys import requests from uptime_kuma_api import MonitorType logger = logging.getLogger(__name__) def hydrate_http_probes(tree, excludes=[]): for probe in tree.get("monitors", []): if "type" not in probe.kwargs: logger.error("Fatal: probes must have a 'type' parameter") sys.exit(1) if ( probe.kwargs["type"] == MonitorType.KEYWORD and probe.kwargs.get("keyword", None) is None ): logger.debug(f"Hydrating {probe.name}") if "url" not in probe.kwargs: logger.error("Fatal: http probe must provide an url") sys.exit(1) if "method" not in probe.kwargs: logger.error("Fatal: http probe must provide a method") sys.exit(1) url = probe.kwargs["url"] method = probe.kwargs["method"] headers = probe.kwargs.get("headers", None) body = probe.kwargs.get("body", None) content = requests.request(method, url, headers=headers, data=body).text m = re.search("(.*?)", content) if m is None: logger.info(f"Didn't find keywords for probe {probe.name}, skipping") probe.kwargs["keyword"] = m.group(1) logger.debug(f"New keyword: {m.group(1)}")