94 lines
4.0 KiB
Python
94 lines
4.0 KiB
Python
import appdaemon.plugins.hass.hassapi as hass
|
|
import appdaemon.exceptions as ad_exception
|
|
import ad_toolbox.smartcondition as SmartCondition
|
|
import uuid
|
|
from expressionparser import ParsingException
|
|
|
|
class Notification:
|
|
def __init__(self,ad_api,config,name):
|
|
self.recipients = dict()
|
|
self.config = config
|
|
self.ad_api = ad_api
|
|
try: self.auto_clear = config['message']['auto_clear']
|
|
except KeyError: self.auto_clear = False
|
|
self.uuid = str(uuid.uuid4())
|
|
for recipient in config["recipients"]:
|
|
try: self.recipients[recipient] = SmartCondition.Evaluator(self.ad_api,self.config["recipients"][recipient]['conditions'],condition_name = name,on_succeed_cb = self.on_condition_succeed, on_fail_cb = self.on_conditon_fail,pass_condition_name_to_cb = recipient, trigger_callback_on_activation = False)
|
|
except ParsingException as e:
|
|
self.log_error(str(e))
|
|
continue
|
|
|
|
def on_conditon_fail(self, recipient):
|
|
if self.auto_clear:
|
|
try: tag = self.config['message']['data']['tag']
|
|
except KeyError: tag = self.uuid
|
|
|
|
self.clear_notification(recipient,tag)
|
|
|
|
def on_condition_succeed(self,recipient):
|
|
self.send_notification(recipient)
|
|
|
|
def on_timeout_expired(self,kwargs):
|
|
tag = kwargs['tag']
|
|
recipient = kwargs['recipient']
|
|
|
|
self.clear_notification(recipient,tag)
|
|
|
|
def clear_notification(self,recipient,tag):
|
|
message = 'clear_notification'
|
|
data = { 'tag' : tag }
|
|
|
|
def clear_notification(service,tag):
|
|
self.ad_api.log(f"Clear notification {data} from {service}")
|
|
try: self.ad_api.call_service(service,message = message, data = data)
|
|
except ad_exception.ServiceException: self.log_error(f"Invalid service: {service}")
|
|
|
|
recipient_yaml = self.config["recipients"][recipient]
|
|
if isinstance(recipient_yaml['services'], str):
|
|
clear_notification(recipient_yaml['services'],tag)
|
|
else:
|
|
for service in recipient_yaml['services']:
|
|
clear_notification(service,tag)
|
|
|
|
def send_notification(self,recipient):
|
|
try: title = self.config['message']['title']
|
|
except KeyError: title = ""
|
|
try: message = self.config['message']['content']
|
|
except KeyError: message = ""
|
|
try: data = self.config['message']['data']
|
|
except KeyError: data = dict()
|
|
|
|
#TODO : maybe it's better to not send tag when it's not needed
|
|
if not 'tag' in data:
|
|
data['tag'] = self.uuid
|
|
|
|
if 'timeout' in self.config['message']:
|
|
timeout = self.config['message']['timeout']
|
|
self.ad_api.run_in(self.on_timeout_expired,timeout,tag = data['tag'],recipient = recipient)
|
|
|
|
def send_notification(service):
|
|
self.ad_api.log(f"Sending notification '{message}' to {service}{f' with data {data}' if data else ''}")
|
|
try: self.ad_api.call_service(service,title = title,message = message, data = data)
|
|
except ad_exception.ServiceException: self.log_error(f"Invalid service: {service}")
|
|
|
|
recipient_yaml = self.config["recipients"][recipient]
|
|
if isinstance(recipient_yaml['services'], str):
|
|
send_notification(recipient_yaml['services'])
|
|
else:
|
|
for service in recipient_yaml['services']:
|
|
send_notification(service)
|
|
|
|
def log_error(self,message,stop_app = False):
|
|
self.ad_api.log(message,level = "ERROR", log = "main_log")
|
|
self.ad_api.log(message,level = "ERROR", log = "error_log")
|
|
if stop_app:
|
|
self.ad_api.stop_app(self.ad_api.name)
|
|
|
|
class NotificationsManager(hass.Hass):
|
|
|
|
def initialize(self):
|
|
self.notifications = dict()
|
|
|
|
if "notifications" in self.args:
|
|
for notification in self.args["notifications"]:
|
|
self.notifications[notification] = Notification(self,self.args["notifications"][notification],notification) |