added health check support #47
| @@ -81,6 +81,10 @@ class Qbt: | |||||||
|         #ignored_tags |         #ignored_tags | ||||||
|         self.ignored_tags = self.config["ignored_domains"] |         self.ignored_tags = self.config["ignored_domains"] | ||||||
|  |  | ||||||
|  |         #healthcheck | ||||||
|  |         self.use_healthcheck = self.config["healthcheck"]["use_healthcheck"] | ||||||
|  |         self.healthcheck_url = self.config["healthcheck"]["healthcheck_url"] | ||||||
|  |  | ||||||
|         # Calling log and notify functions |         # Calling log and notify functions | ||||||
|         tor_log(self) |         tor_log(self) | ||||||
|         tor_notify(self) |         tor_notify(self) | ||||||
| @@ -122,6 +126,8 @@ class Qbt: | |||||||
|             tor_notify_summary(self) |             tor_notify_summary(self) | ||||||
|         if self.use_apprise: |         if self.use_apprise: | ||||||
|             tor_notify_apprise(self, r, apprise_notify) |             tor_notify_apprise(self, r, apprise_notify) | ||||||
|  |         if self.use_healthcheck: | ||||||
|  |             send_ping(self, r, self.healthcheck_url) | ||||||
| # Run | # Run | ||||||
| if  __name__== "__main__": | if  __name__== "__main__": | ||||||
|     Qbt() |     Qbt() | ||||||
| @@ -84,4 +84,10 @@ def get_script_runtime(self): | |||||||
|     if self.use_pushover: |     if self.use_pushover: | ||||||
|         self.extm = f"Execution time: [{elapsed_time}]" |         self.extm = f"Execution time: [{elapsed_time}]" | ||||||
|     if self.use_apprise: |     if self.use_apprise: | ||||||
|         self.extm = f"Execution time: [{elapsed_time}]" |         self.extm = f"Execution time: [{elapsed_time}]" | ||||||
|  |  | ||||||
|  | def send_ping(self, req_obj, healthcheck_url): | ||||||
|  |     try: | ||||||
|  |         req_obj.get(healthcheck_url, timeout=10) | ||||||
|  |     except req_obj.RequestException as e: | ||||||
|  |         self.tl.info(f"Ping failed: {e}") | ||||||
		Reference in New Issue
	
	Block a user