From ae9da1c13a5d932469666a831bb4e429cff6a844 Mon Sep 17 00:00:00 2001
From: blacktwin <blacktwin@users.noreply.github.com>
Date: Thu, 28 Sep 2017 14:55:08 -0400
Subject: [PATCH] requested from reddit

https://www.reddit.com/r/plexpy/comments/6uimuf/any_custom_script_that_has_a_white_list_and_will/?ref=share&ref_source=link
---
 killstream/ip_whitelist.py | 82 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 killstream/ip_whitelist.py

diff --git a/killstream/ip_whitelist.py b/killstream/ip_whitelist.py
new file mode 100644
index 0000000..410ee03
--- /dev/null
+++ b/killstream/ip_whitelist.py
@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+
+'''
+PlexPy > Settings > Notification Agents > Scripts > Bell icon:
+        [X] Notify on playback start
+PlexPy > Settings > Notification Agents > Scripts > Gear icon:
+        Playback Start: ip_whitelist.py
+PlexPy > Settings > Notifications > Script > Script Arguments:
+        {session_key} {ip_address}
+
+'''
+
+import sys
+import requests
+from plexapi.server import PlexServer
+# pip install plexapi
+
+
+## EDIT THESE SETTINGS ##
+PLEX_TOKEN = 'xxxxx'
+PLEX_URL = 'http://localhost:32400'
+PLEXPY_APIKEY = 'xxxxx'  # Your PlexPy API key
+PLEXPY_URL = 'http://localhost:8182/'  # Your PlexPy URL
+
+IP_WHITELIST = ['10.10.0.12']
+REASON = 'IP Address: {} was not found in whitelist.'
+IGNORE_LST = ('Blacktwin')
+
+
+AGENT_ID = 14  # Notification agent ID for PlexPy
+# Find Notification agent ID here:
+# https://github.com/JonnyWong16/plexpy/blob/master/API.md#notify
+
+SUBJECT_TEXT = "IP Whitelist Violation"
+BODY_TEXT = "Killed {user}'s stream of {title}. IP: {ip} not in whitelist"
+
+##/EDIT THESE SETTINGS ##
+
+sessionKey = sys.argv[1]
+ip_address = sys.argv[2]
+
+plex = PlexServer(PLEX_URL, PLEX_TOKEN)
+
+
+def send_notification(SUBJECT_TEXT, BODY_TEXT):
+    # Format notification text
+    try:
+        subject = SUBJECT_TEXT.format()
+        body = BODY_TEXT.format()
+
+    except LookupError as e:
+        sys.stderr.write("Unable to substitute '{0}' in the notification subject or body".format(e))
+        return None
+    # Send the notification through PlexPy
+    payload = {'apikey': PLEXPY_APIKEY,
+               'cmd': 'notify',
+               'agent_id': AGENT_ID,
+               'subject': subject,
+               'body': body}
+
+    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
+
+
+if ip_address not in IP_WHITELIST:
+    for session in plex.sessions():
+        username = session.usernames[0]
+        title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+        if session.sessionKey == int(sessionKey) and username not in IGNORE_LST:
+            sys.stdout.write("Killing {user}'s stream of {title}. IP: {ip} not in whitelist".format(
+                user=username, title=title, ip=ip_address))
+            session.stop(reason=REASON.format(ip_address))
+            send_notification(SUBJECT_TEXT, BODY_TEXT.format(user=username, ip=ip_address, title=title))