Compare commits
9 Commits
v2.0.0
...
e2cdca60c3
Author | SHA1 | Date | |
---|---|---|---|
e2cdca60c3 | |||
978e9326cc | |||
beee95caa4 | |||
6ce2add157 | |||
dc533a33b1 | |||
5e1c963b72 | |||
cf64321e49 | |||
a0ce03335d | |||
b6346ac335 |
@ -11,6 +11,12 @@ Dockerfile
|
|||||||
*.csv
|
*.csv
|
||||||
*.toml
|
*.toml
|
||||||
*.git*
|
*.git*
|
||||||
|
.dockerignore
|
||||||
|
.DS_Store
|
||||||
|
.vscode/*
|
||||||
|
thunder-tests/*
|
||||||
|
.drone.yml
|
||||||
|
|
||||||
# Byte-compiled / optimized / DLL files
|
# Byte-compiled / optimized / DLL files
|
||||||
__pycache__/
|
__pycache__/
|
||||||
*.py[cod]
|
*.py[cod]
|
||||||
|
19
.drone.yml
Normal file
19
.drone.yml
Normal file
@ -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
|
15
qbit-maid.py
15
qbit-maid.py
@ -12,16 +12,17 @@ from collections import Counter
|
|||||||
import csv
|
import csv
|
||||||
import requests as r
|
import requests as r
|
||||||
import os
|
import os
|
||||||
import sys
|
|
||||||
class Qbt:
|
class Qbt:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
"""Main object, should be calling functions from qlist.py, qlogging.py and qprocess.py"""
|
"""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
|
# Open the config. Needs a json file with the data in config.json.example
|
||||||
self.st = datetime.datetime.now()
|
self.st = datetime.datetime.now()
|
||||||
|
|
||||||
config_file_path=os.environ["toml_path"]
|
if os.getenv("toml_path"):
|
||||||
with open(config_file_path, 'rb') as c:
|
config_file_path=os.getenv("toml_path")
|
||||||
self.config = load(c)
|
with open(config_file_path, 'rb') as c:
|
||||||
|
self.config = load(c)
|
||||||
if os.path.exists('./config.toml'):
|
if os.path.exists('./config.toml'):
|
||||||
config_file_path = './config.toml'
|
config_file_path = './config.toml'
|
||||||
with open(config_file_path, 'rb') as c:
|
with open(config_file_path, 'rb') as c:
|
||||||
@ -80,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)
|
||||||
@ -121,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