Merge pull request 'intergrate-drone' (#51) from intergrate-drone into main

Reviewed-on: https://git.jbranan.com/jblu/qbit-maid/pulls/51

.dockerignore

@@ -11,6 +11,12 @@ Dockerfile
 *.csv
 *.toml
 *.git*
+.dockerignore
+.DS_Store
+.vscode/*
+thunder-tests/*
+.drone.yml
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

.drone.yml

@@ -0,0 +1,19 @@
+kind: pipeline
+name: default
+
+steps:
+- name: docker
+  image: plugins/docker
+  settings:
+    registry: git.jbranan.com
+    dry_run: true
+    username: jblu
+    password: 
+      from_secret: gittea_drone
+    repo: jblu/qbit-maid
+    tags:
+    - latest
+    - main
+  when:
+    branch:
+      - main

qbit-maid.py

@@ -12,14 +12,15 @@ from collections import Counter
 import csv
 import requests as r
 import os
-import sys
+
 class Qbt:
     def __init__(self):
         """Main object, should be calling functions from qlist.py, qlogging.py and qprocess.py"""
         # Open the config. Needs a json file with the data in config.json.example
         self.st = datetime.datetime.now()
         
-        config_file_path=os.environ["toml_path"]
+        if os.getenv("toml_path"):
+            config_file_path=os.getenv("toml_path")
             with open(config_file_path, 'rb') as c:
                 self.config = load(c)
         if os.path.exists('./config.toml'):
@@ -80,6 +81,10 @@ class Qbt:
         #ignored_tags
         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
         tor_log(self)
         tor_notify(self)
@@ -121,6 +126,8 @@ class Qbt:
             tor_notify_summary(self)
         if self.use_apprise:
             tor_notify_apprise(self, r, apprise_notify)
+        if self.use_healthcheck:
+            send_ping(self, r, self.healthcheck_url)
 # Run
 if  __name__== "__main__":
     Qbt()

qlogging.py

@@ -85,3 +85,9 @@ def get_script_runtime(self):
         self.extm = f"Execution time: [{elapsed_time}]"
     if self.use_apprise:
         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}")