From 7505d35f14176d623be2b19fe0e2e3be2d8033cc Mon Sep 17 00:00:00 2001
From: blacktwin <blacktwin@users.noreply.github.com>
Date: Sat, 21 Oct 2017 10:48:53 -0400
Subject: [PATCH] Create wait_kill_pause_notify_main.py

---
 killstream/wait_kill_pause_notify_main.py | 120 ++++++++++++++++++++++
 1 file changed, 120 insertions(+)
 create mode 100644 killstream/wait_kill_pause_notify_main.py

diff --git a/killstream/wait_kill_pause_notify_main.py b/killstream/wait_kill_pause_notify_main.py
new file mode 100644
index 0000000..d2fe337
--- /dev/null
+++ b/killstream/wait_kill_pause_notify_main.py
@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+
+'''
+PlexPy > Settings > Notification Agents > Scripts > Bell icon:
+        [X] Notify on pause
+PlexPy > Settings > Notification Agents > Scripts > Gear icon:
+        Playback Pause: wait_kill_pause_notify_main.py
+PlexPy > Settings > Notifications > Script > Script Arguments:
+        {session_key}
+
+wait_kill_pause_notify_main.py & wait_kill_pause_notify_sub.py should be in the same directory.
+wait_kill_pause_notify_main.py executes sub_script wait_kill_pause_notify_sub.py.
+
+PlexPy will timeout wait_kill_pause_notify_main.py after 30 seconds (default)
+    but wait_kill_pause_notify_sub.py will continue.
+
+wait_kill_pause_notify_sub will check if the stream's session_id is still paused or if playing as restarted.
+If playback is restarted then wait_kill_pause_notify_sub will stop and delete itself.
+If stream remains paused then it will be killed and wait_kill_pause_notify_sub will stop.
+Set TIMEOUT to max time before killing stream
+Set INTERVAL to how often you want to check the stream status
+'''
+
+import os
+import sys
+import requests
+import subprocess
+from plexapi.server import PlexServer
+
+
+## EDIT THESE SETTINGS ##
+PLEX_TOKEN = 'xxxx'
+PLEX_URL = 'http://localhost:32400'
+PLEXPY_APIKEY = 'xxxx'  # Your PlexPy API key
+PLEXPY_URL = 'http://localhost:8182/'  # Your PlexPy URL
+
+TIMEOUT = '120'
+INTERVAL = '20'
+
+KILL_MESSAGE = 'This stream has ended due to being paused and transcoding.'
+
+USER_IGNORE = ('') # ('Username','User2')
+
+SUBJECT_TEXT = "Killed Paused Transcoded Stream."
+BODY_TEXT = "Killed {user}'s paused transcoded stream of {title}."
+
+AGENT_ID = 10  # Notification agent ID for PlexPy
+# Find Notification agent ID here:
+# https://github.com/JonnyWong16/plexpy/blob/master/API.md#notify
+# AGENT = '' to disable notification
+
+sub_script = 'wait_kill_trans_notify_sub.py'
+##/EDIT THESE SETTINGS ##
+
+sess = requests.Session()
+sess.verify = False
+plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
+
+sessionKey = sys.argv[1]
+
+
+def send_notification(subject_text, body_text):
+    # Send the notification through PlexPy
+    payload = {'apikey': PLEXPY_APIKEY,
+               'cmd': 'notify',
+               'agent_id': AGENT_ID,
+               'subject': subject_text,
+               'body': body_text}
+
+    try:
+        r = requests.post(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload)
+        response = r.json()
+
+        if response['response']['result'] == 'success':
+            sys.stdout.write("Successfully sent PlexPy notification.")
+        else:
+            raise Exception(response['response']['message'])
+    except Exception as e:
+        sys.stderr.write("PlexPy API 'notify' request failed: {0}.".format(e))
+        return None
+
+
+def check_session(sessionKey):
+    for session in plex.sessions():
+        if session.sessionKey == int(sessionKey):
+            return session
+
+
+def kill_stream(session, xtime, ntime):
+    state = session.players[0].state
+    videoDecision = session.transcodeSessions[0].videoDecision
+    username = session.usernames[0]
+    title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+
+    if state == 'paused' and xtime == ntime and videoDecision == 'transcode':
+        if AGENT_ID:
+            send_notification(SUBJECT_TEXT, BODY_TEXT.format(user=username, title=title))
+        return ntime
+    elif state in ('playing', 'buffering'):
+        sys.stdout.write("{user}'s stream of {title} is now {state}".format(user=username, title=title,
+                                                                            state=state))
+        return None
+    else:
+        return xtime
+
+
+if __name__ == '__main__':
+
+    startupinfo = None
+    if os.name == 'nt':
+        startupinfo = subprocess.STARTUPINFO()
+        startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
+
+    fileDir = os.path.dirname(os.path.realpath(__file__))
+    sub_path = os.path.join(fileDir, sub_script)
+
+    for session in plex.sessions():
+        if session.sessionKey == int(sessionKey) and session.usernames[0] not in USER_IGNORE:
+            subprocess.Popen([sys.executable, sub_path, sessionKey, TIMEOUT, INTERVAL],
+                             startupinfo=startupinfo)