mirror of
https://github.com/Tautulli/Tautulli.git
synced 2025-07-11 07:46:07 -07:00
Add code to filter available triggers for notification agents
This commit is contained in:
JonnyWong16
parent
79cd2ca9b9
commit
6ccf801ee6
1 changed files with 63 additions and 30 deletions
|
|
@ -101,103 +101,128 @@ DEFAULT_CUSTOM_CONDITIONS = [{'parameter': '', 'operator': '', 'value': ''}]
|
||||||
def available_notification_agents():
|
def available_notification_agents():
|
||||||
agents = [{'label': 'Tautulli Remote Android App',
|
agents = [{'label': 'Tautulli Remote Android App',
|
||||||
'name': 'androidapp',
|
'name': 'androidapp',
|
||||||
'id': AGENT_IDS['androidapp']
|
'id': AGENT_IDS['androidapp'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Boxcar',
|
{'label': 'Boxcar',
|
||||||
'name': 'boxcar',
|
'name': 'boxcar',
|
||||||
'id': AGENT_IDS['boxcar']
|
'id': AGENT_IDS['boxcar'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Browser',
|
{'label': 'Browser',
|
||||||
'name': 'browser',
|
'name': 'browser',
|
||||||
'id': AGENT_IDS['browser']
|
'id': AGENT_IDS['browser'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Discord',
|
{'label': 'Discord',
|
||||||
'name': 'discord',
|
'name': 'discord',
|
||||||
'id': AGENT_IDS['discord'],
|
'id': AGENT_IDS['discord'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Email',
|
{'label': 'Email',
|
||||||
'name': 'email',
|
'name': 'email',
|
||||||
'id': AGENT_IDS['email']
|
'id': AGENT_IDS['email'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Facebook',
|
{'label': 'Facebook',
|
||||||
'name': 'facebook',
|
'name': 'facebook',
|
||||||
'id': AGENT_IDS['facebook']
|
'id': AGENT_IDS['facebook'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'GroupMe',
|
{'label': 'GroupMe',
|
||||||
'name': 'groupme',
|
'name': 'groupme',
|
||||||
'id': AGENT_IDS['groupme']
|
'id': AGENT_IDS['groupme'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Growl',
|
{'label': 'Growl',
|
||||||
'name': 'growl',
|
'name': 'growl',
|
||||||
'id': AGENT_IDS['growl']
|
'id': AGENT_IDS['growl'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Hipchat',
|
{'label': 'Hipchat',
|
||||||
'name': 'hipchat',
|
'name': 'hipchat',
|
||||||
'id': AGENT_IDS['hipchat']
|
'id': AGENT_IDS['hipchat'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'IFTTT',
|
{'label': 'IFTTT',
|
||||||
'name': 'ifttt',
|
'name': 'ifttt',
|
||||||
'id': AGENT_IDS['ifttt']
|
'id': AGENT_IDS['ifttt'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Join',
|
{'label': 'Join',
|
||||||
'name': 'join',
|
'name': 'join',
|
||||||
'id': AGENT_IDS['join']
|
'id': AGENT_IDS['join'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Kodi',
|
{'label': 'Kodi',
|
||||||
'name': 'xbmc',
|
'name': 'xbmc',
|
||||||
'id': AGENT_IDS['xbmc']
|
'id': AGENT_IDS['xbmc'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
# {'label': 'Notify My Android',
|
# {'label': 'Notify My Android',
|
||||||
# 'name': 'nma',
|
# 'name': 'nma',
|
||||||
# 'id': AGENT_IDS['nma']
|
# 'id': AGENT_IDS['nma'],
|
||||||
|
# 'action_types': ('all',)
|
||||||
# },
|
# },
|
||||||
{'label': 'MQTT',
|
{'label': 'MQTT',
|
||||||
'name': 'mqtt',
|
'name': 'mqtt',
|
||||||
'id': AGENT_IDS['mqtt']
|
'id': AGENT_IDS['mqtt'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Plex Home Theater',
|
{'label': 'Plex Home Theater',
|
||||||
'name': 'plex',
|
'name': 'plex',
|
||||||
'id': AGENT_IDS['plex']
|
'id': AGENT_IDS['plex'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Prowl',
|
{'label': 'Prowl',
|
||||||
'name': 'prowl',
|
'name': 'prowl',
|
||||||
'id': AGENT_IDS['prowl']
|
'id': AGENT_IDS['prowl'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
# {'label': 'Pushalot',
|
# {'label': 'Pushalot',
|
||||||
# 'name': 'pushalot',
|
# 'name': 'pushalot',
|
||||||
# 'id': AGENT_IDS['pushalot']
|
# 'id': AGENT_IDS['pushalot'],
|
||||||
|
# 'action_types': ('all',)
|
||||||
# },
|
# },
|
||||||
{'label': 'Pushbullet',
|
{'label': 'Pushbullet',
|
||||||
'name': 'pushbullet',
|
'name': 'pushbullet',
|
||||||
'id': AGENT_IDS['pushbullet']
|
'id': AGENT_IDS['pushbullet'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Pushover',
|
{'label': 'Pushover',
|
||||||
'name': 'pushover',
|
'name': 'pushover',
|
||||||
'id': AGENT_IDS['pushover']
|
'id': AGENT_IDS['pushover'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Script',
|
{'label': 'Script',
|
||||||
'name': 'scripts',
|
'name': 'scripts',
|
||||||
'id': AGENT_IDS['scripts']
|
'id': AGENT_IDS['scripts'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Slack',
|
{'label': 'Slack',
|
||||||
'name': 'slack',
|
'name': 'slack',
|
||||||
'id': AGENT_IDS['slack']
|
'id': AGENT_IDS['slack'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Telegram',
|
{'label': 'Telegram',
|
||||||
'name': 'telegram',
|
'name': 'telegram',
|
||||||
'id': AGENT_IDS['telegram']
|
'id': AGENT_IDS['telegram'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Twitter',
|
{'label': 'Twitter',
|
||||||
'name': 'twitter',
|
'name': 'twitter',
|
||||||
'id': AGENT_IDS['twitter']
|
'id': AGENT_IDS['twitter'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Webhook',
|
{'label': 'Webhook',
|
||||||
'name': 'webhook',
|
'name': 'webhook',
|
||||||
'id': AGENT_IDS['webhook']
|
'id': AGENT_IDS['webhook'],
|
||||||
|
'action_types': ('all',)
|
||||||
},
|
},
|
||||||
{'label': 'Zapier',
|
{'label': 'Zapier',
|
||||||
'name': 'zapier',
|
'name': 'zapier',
|
||||||
'id': AGENT_IDS['zapier']
|
'id': AGENT_IDS['zapier'],
|
||||||
|
'action_types': ('all',)
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
@ -205,13 +230,14 @@ def available_notification_agents():
|
||||||
if OSX().validate():
|
if OSX().validate():
|
||||||
agents.append({'label': 'macOS Notification Center',
|
agents.append({'label': 'macOS Notification Center',
|
||||||
'name': 'osx',
|
'name': 'osx',
|
||||||
'id': AGENT_IDS['osx']
|
'id': AGENT_IDS['osx'],
|
||||||
|
'action_types': ('all',)
|
||||||
})
|
})
|
||||||
|
|
||||||
return agents
|
return agents
|
||||||
|
|
||||||
|
|
||||||
def available_notification_actions():
|
def available_notification_actions(agent_id=None):
|
||||||
actions = [{'label': 'Playback Start',
|
actions = [{'label': 'Playback Start',
|
||||||
'name': 'on_play',
|
'name': 'on_play',
|
||||||
'description': 'Trigger a notification when a stream is started.',
|
'description': 'Trigger a notification when a stream is started.',
|
||||||
|
|
@ -350,6 +376,11 @@ def available_notification_actions():
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
|
||||||
|
if agent_id:
|
||||||
|
action_types = get_notify_agents(return_dict=True).get(agent_id, {}).get('action_types', [])
|
||||||
|
if 'all' not in action_types:
|
||||||
|
actions = [a for a in actions if a['name'] in action_types]
|
||||||
|
|
||||||
return actions
|
return actions
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -415,13 +446,15 @@ def get_agent_class(agent_id=None, config=None):
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
def get_notify_agents():
|
def get_notify_agents(return_dict=False):
|
||||||
|
if return_dict:
|
||||||
|
return {a['id']: a for a in available_notification_agents()}
|
||||||
return tuple(a['name'] for a in sorted(available_notification_agents(), key=lambda k: k['label']))
|
return tuple(a['name'] for a in sorted(available_notification_agents(), key=lambda k: k['label']))
|
||||||
|
|
||||||
|
|
||||||
def get_notify_actions(return_dict=False):
|
def get_notify_actions(return_dict=False):
|
||||||
if return_dict:
|
if return_dict:
|
||||||
return {a.pop('name'): a for a in available_notification_actions()}
|
return {a['name']: a for a in available_notification_actions()}
|
||||||
return tuple(a['name'] for a in available_notification_actions())
|
return tuple(a['name'] for a in available_notification_actions())
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -529,7 +562,7 @@ def add_notifier_config(agent_id=None, **kwargs):
|
||||||
% agent_id)
|
% agent_id)
|
||||||
return False
|
return False
|
||||||
|
|
||||||
agent = next((a for a in available_notification_agents() if a['id'] == agent_id), None)
|
agent = get_notify_agents(return_dict=True).get(agent_id, None)
|
||||||
|
|
||||||
if not agent:
|
if not agent:
|
||||||
logger.error(u"Tautulli Notifiers :: Unable to retrieve new notification agent: invalid agent_id %s."
|
logger.error(u"Tautulli Notifiers :: Unable to retrieve new notification agent: invalid agent_id %s."
|
||||||
|
|
@ -578,7 +611,7 @@ def set_notifier_config(notifier_id=None, agent_id=None, **kwargs):
|
||||||
% agent_id)
|
% agent_id)
|
||||||
return False
|
return False
|
||||||
|
|
||||||
agent = next((a for a in available_notification_agents() if a['id'] == agent_id), None)
|
agent = get_notify_agents(return_dict=True).get(agent_id, None)
|
||||||
|
|
||||||
if not agent:
|
if not agent:
|
||||||
logger.error(u"Tautulli Notifiers :: Unable to retrieve existing notification agent: invalid agent_id %s."
|
logger.error(u"Tautulli Notifiers :: Unable to retrieve existing notification agent: invalid agent_id %s."
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue