Files
planka/server/utils/send_notifications.py
2025-11-19 18:19:31 +01:00

50 lines
1.4 KiB
Python

import sys
import json
import logging
import apprise
last_apprise_message = None
class CaptureWarningHandler(logging.Handler):
def emit(self, record):
global last_apprise_message
last_apprise_message = record.getMessage()
capture_warning_handler = CaptureWarningHandler()
capture_warning_handler.setLevel(logging.WARNING)
apprise_logger = logging.getLogger('apprise')
apprise_logger.setLevel(logging.WARNING)
apprise_logger.propagate = False
apprise_logger.addHandler(capture_warning_handler)
def send_notification(url, title, body, body_format):
app = apprise.Apprise()
app.add(url)
return app.notify(title=title, body=body, body_format=body_format)
if __name__ == '__main__':
services = json.loads(sys.argv[1])
title = sys.argv[2]
body_by_format = json.loads(sys.argv[3])
for service in services:
url = service['url']
body_format = service['format']
body = body_by_format[body_format]
last_apprise_message = None
if not send_notification(url, title, body, body_format):
if last_apprise_message:
if last_apprise_message == 'There are no service(s) to notify':
sys.stderr.write('Unknown service URL')
else:
sys.stderr.write(last_apprise_message)
else:
sys.stderr.write('Unknown error')
sys.exit(1)