Mirrored_Repos/JBOPS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Always send a notification if an agent is specified

Browse Source 
Whenever a stream is killed, send a notification for that stream if a 
notification agent has been specified. If no username is specified, 
simply list the session ID.
This commit is contained in:
Landon Abney
2018-06-16 23:15:59 -07:00
parent 2116082fd7
commit 4ebfaf17bd
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
killstream/kill_stream.py
View File

@ -50,7 +50,8 @@ TAUTULLI_URL = os.getenv('TAUTULLI_URL', TAUTULLI_URL)
TAUTULLI_APIKEY = os.getenv('TAUTULLI_APIKEY', TAUTULLI_APIKEY) TAUTULLI_APIKEY = os.getenv('TAUTULLI_APIKEY', TAUTULLI_APIKEY)
SUBJECT_TEXT = "Tautulli has killed a stream." SUBJECT_TEXT = "Tautulli has killed a stream."
BODY_TEXT = "Killed {user}'s stream. Reason: {message}." BODY_TEXT = "Killed session ID '{id}'. Reason: {message}"
BODY_TEXT_USER = "Killed {user}'s stream. Reason: {message}."
sess = requests.Session() sess = requests.Session()
# Ignore verifying the SSL certificate # Ignore verifying the SSL certificate
@ -142,7 +143,7 @@ def get_user_session_ids(user_id):
return user_streams return user_streams
def terminate_session(session_id, message): def terminate_session(session_id, message, notifier=None, username=None):
# Stop a streaming session. # Stop a streaming session.
payload = {'apikey': TAUTULLI_APIKEY, payload = {'apikey': TAUTULLI_APIKEY,
'cmd': 'terminate_session', 'cmd': 'terminate_session',
@ -156,6 +157,12 @@ def terminate_session(session_id, message):
if response['response']['result'] == 'success': if response['response']['result'] == 'success':
sys.stdout.write( sys.stdout.write(
"Successfully killed Plex session: {0}.".format(session_id)) "Successfully killed Plex session: {0}.".format(session_id))
if notifier:
if username:
body = BODY_TEXT.format(user=username, message=message)
else:
body = BODY_TEXT.format(id=session_id, message=message)
send_notification(SUBJECT_TEXT, body, notifier)
else: else:
raise Exception(response['response']['message']) raise Exception(response['response']['message'])
except Exception as e: except Exception as e:
@ -189,12 +196,8 @@ if __name__ == "__main__":
message = '' message = ''
if opts.jbop == 'stream': if opts.jbop == 'stream':
terminate_session(opts.sessionId, message) terminate_session(opts.sessionId, message, opts.notify, opts.username)
elif opts.jbop == 'allStreams': elif opts.jbop == 'allStreams':
streams = get_user_session_ids(opts.userId) streams = get_user_session_ids(opts.userId)
for session_id in streams: for session_id in streams:
terminate_session(session_id, message) terminate_session(session_id, message, opts.notify, opts.username)
if opts.notify:
BODY_TEXT = BODY_TEXT.format(user=opts.username, message=message)
send_notification(SUBJECT_TEXT, BODY_TEXT, opts.notify)