mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-06 17:15:38 -08:00
145 lines
4.5 KiB
Python
Executable File
145 lines
4.5 KiB
Python
Executable File
|
|
#!/usr/bin/env python
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
import requests
|
|
from base64 import b64encode
|
|
|
|
# Register NetAlertX directories
|
|
INSTALL_PATH = os.getenv('NETALERTX_APP', '/app')
|
|
sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
|
|
|
|
import conf
|
|
from const import confFileName, logPath
|
|
from plugin_helper import Plugin_Objects, handleEmpty
|
|
from logger import mylog, Logger
|
|
from helper import timeNowTZ, get_setting_value
|
|
from models.notification_instance import NotificationInstance
|
|
from database import DB
|
|
from pytz import timezone
|
|
|
|
# Make sure the TIMEZONE for logging is correct
|
|
conf.tz = timezone(get_setting_value('TIMEZONE'))
|
|
|
|
# Make sure log level is initialized correctly
|
|
Logger(get_setting_value('LOG_LEVEL'))
|
|
|
|
pluginName = 'NTFY'
|
|
|
|
LOG_PATH = logPath + '/plugins'
|
|
RESULT_FILE = os.path.join(LOG_PATH, f'last_result.{pluginName}.log')
|
|
|
|
|
|
|
|
def main():
|
|
|
|
mylog('verbose', [f'[{pluginName}](publisher) In script'])
|
|
|
|
# Check if basic config settings supplied
|
|
if check_config() == False:
|
|
mylog('none', [f'[{pluginName}] ⚠ ERROR: Publisher notification gateway not set up correctly. Check your {confFileName} {pluginName}_* variables.'])
|
|
return
|
|
|
|
# Create a database connection
|
|
db = DB() # instance of class DB
|
|
db.open()
|
|
|
|
# Initialize the Plugin obj output file
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
# Create a NotificationInstance instance
|
|
notifications = NotificationInstance(db)
|
|
|
|
# Retrieve new notifications
|
|
new_notifications = notifications.getNew()
|
|
|
|
# Process the new notifications (see the Notifications DB table for structure or check the /php/server/query_json.php?file=table_notifications.json endpoint)
|
|
for notification in new_notifications:
|
|
|
|
# Send notification
|
|
response_text, response_status_code = send(notification["HTML"], notification["Text"])
|
|
|
|
# Log result
|
|
plugin_objects.add_object(
|
|
primaryId = pluginName,
|
|
secondaryId = timeNowTZ(),
|
|
watched1 = notification["GUID"],
|
|
watched2 = handleEmpty(response_text),
|
|
watched3 = response_status_code,
|
|
watched4 = 'null',
|
|
extra = 'null',
|
|
foreignKey = notification["GUID"]
|
|
)
|
|
|
|
plugin_objects.write_result_file()
|
|
|
|
|
|
|
|
#-------------------------------------------------------------------------------
|
|
def check_config():
|
|
if get_setting_value('NTFY_HOST') == '' or get_setting_value('NTFY_TOPIC') == '':
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
#-------------------------------------------------------------------------------
|
|
def send(html, text):
|
|
|
|
response_text = ''
|
|
response_status_code = ''
|
|
|
|
# settings
|
|
token = get_setting_value('NTFY_TOKEN')
|
|
user = get_setting_value('NTFY_USER')
|
|
pwd = get_setting_value('NTFY_PASSWORD')
|
|
verify_ssl = get_setting_value('NTFY_VERIFY_SSL')
|
|
|
|
# prepare request headers
|
|
headers = {
|
|
"Title": "NetAlertX Notification",
|
|
"Actions": "view, Open Dashboard, "+ get_setting_value('REPORT_DASHBOARD_URL'),
|
|
"Priority": get_setting_value('NTFY_PRIORITY'),
|
|
"Tags": "warning"
|
|
}
|
|
|
|
# if token or username and password are set generate hash and update header
|
|
if token != '':
|
|
headers["Authorization"] = "Bearer {}".format(token)
|
|
elif user != "" and pwd != "":
|
|
# Generate hash for basic auth
|
|
basichash = b64encode(bytes(user + ':' + pwd, "utf-8")).decode("ascii")
|
|
# add authorization header with hash
|
|
headers["Authorization"] = "Basic {}".format(basichash)
|
|
|
|
# call NTFY service
|
|
try:
|
|
response = requests.post("{}/{}".format( get_setting_value('NTFY_HOST'),
|
|
get_setting_value('NTFY_TOPIC')),
|
|
data = text,
|
|
headers = headers,
|
|
verify = verify_ssl)
|
|
|
|
response_status_code = response.status_code
|
|
|
|
# Check if the request was successful (status code 200)
|
|
if response_status_code == 200:
|
|
response_text = response.text # This captures the response body/message
|
|
else:
|
|
response_text = json.dumps(response.text)
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
mylog('none', [f'[{pluginName}] ⚠ ERROR: ', e])
|
|
|
|
response_text = e
|
|
|
|
return response_text, response_status_code
|
|
|
|
return response_text, response_status_code
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|
|
|