add healthchecks.io integration

lib/config.py

@@ -27,6 +27,7 @@ def __init__(self) -> None:
         self.notification_level = NotificationLevel.INFO
         self.notification_urls = []
         self.retrieval_interval = 24 * 60 * 60
+        self.healthchecks_url = None
 
     def create(self, config_json: JSON, global_config: "GlobalConfig") -> None:
         self._merge_globals(global_config)
@@ -56,6 +57,14 @@ def _parse_config(self, config: JSON) -> None:
             if not isinstance(self.check_fares, bool):
                 raise ConfigError("'check_fares' must be a boolean")
 
+        if "healthchecks_url" in config:
+            self.healthchecks_url = config["healthchecks_url"]
+
+            if not isinstance(self.healthchecks_url, str):
+                raise ConfigError("'healthchecks_url' must be a string")
+
+            logger.debug("A Healthchecks URL has been provided")
+
         if "notification_level" in config:
             notification_level = config["notification_level"]
             try:

lib/notification_handler.py

@@ -3,6 +3,7 @@
 from typing import TYPE_CHECKING, Any, Dict, List
 
 import apprise
+import requests
 
 from .flight import Flight
 from .log import get_logger
@@ -118,5 +119,13 @@ def lower_fare(self, flight: Flight, price_info: str) -> None:
         logger.debug("Sending lower fare notification...")
         self.send_notification(message, NotificationLevel.INFO)
 
+    def healthchecks_success(self, data: Any) -> None:
+        if self.reservation_monitor.config.healthchecks_url is not None:
+            requests.post(self.reservation_monitor.config.healthchecks_url, data=data)
+
+    def healthchecks_fail(self, data: Any) -> None:
+        if self.reservation_monitor.config.healthchecks_url is not None:
+            requests.post(self.reservation_monitor.config.healthchecks_url + "/fail", data=data)
+
     def _get_account_name(self) -> str:
         return f"{self.reservation_monitor.first_name} {self.reservation_monitor.last_name}"

lib/reservation_monitor.py

@@ -98,11 +98,14 @@ def _check_flight_fares(self) -> None:
             # and continue
             try:
                 fare_checker.check_flight_price(flight)
+                self.notification_handler.healthchecks_success("Successful fare check, confirmation number=" + flight.confirmation_number)
             except RequestError as err:
+                self.notification_handler.healthchecks_fail("Failed fare check, confirmation number=" + flight.confirmation_number)
                 logger.error("Requesting error during fare check. %s. Skipping...", err)
             except FlightChangeError as err:
                 logger.debug("%s. Skipping fare check", err)
             except Exception as err:
+                self.notification_handler.healthchecks_fail("Failed fare check, confirmation number=" + flight.confirmation_number)
                 logger.exception("Unexpected error during fare check: %s", repr(err))
 
     def _smart_sleep(self, previous_time: datetime) -> None: