91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
'''
|
|
Receive session_key and IP from Tautulli when playback starts.
|
|
Use IP to check against whitelist.
|
|
If not in whitelist use session_key to determine stream and kill.
|
|
|
|
Tautulli > Settings > Notification Agents > Scripts > Bell icon:
|
|
[X] Notify on playback start
|
|
Tautulli > Settings > Notification Agents > Scripts > Gear icon:
|
|
Playback Start: ip_whitelist.py
|
|
Tautulli > Settings > Notifications > Script > Script Arguments:
|
|
{session_key} {ip_address}
|
|
|
|
'''
|
|
|
|
import sys
|
|
import requests
|
|
from plexapi.server import PlexServer
|
|
import configparser
|
|
|
|
|
|
## EDIT THESE SETTINGS ##
|
|
PLEX_TOKEN = 'xxxxxx'
|
|
PLEX_URL = 'http://localhost:32400'
|
|
|
|
TAUTULLI_APIKEY = 'xxxxxx' # Your Tautulli API key
|
|
TAUTULLI_URL = 'http://localhost:8182/' # Your Tautulli URL
|
|
|
|
IP_WHITELIST = ['10.10.0.12'] # List IP addresses.
|
|
IGNORE_LST = ('') # List usernames that should be ignored.
|
|
|
|
REASON = 'IP Address: {} was not found in whitelist.'
|
|
|
|
NOTIFIER_ID = 14 # Notification agent ID for Tautulli
|
|
# Find Notification agent ID here:
|
|
# Tautulli Settings -> NOTIFICATION AGENTS -> :bell: Agent (NotifierID - {Description)
|
|
|
|
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]
|
|
|
|
sess = requests.Session()
|
|
sess.verify = False
|
|
plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
|
|
|
|
def send_notification(subject_text, body_text):
|
|
# Format notification text
|
|
try:
|
|
subject = subject_text
|
|
body = body_text
|
|
|
|
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 Tautulli
|
|
payload = {'apikey': TAUTULLI_APIKEY,
|
|
'cmd': 'notify',
|
|
'notifier_id': NOTIFIER_ID,
|
|
'subject': subject,
|
|
'body': body}
|
|
|
|
try:
|
|
r = requests.post(TAUTULLI_URL.rstrip('/') + '/api/v2', params=payload)
|
|
response = r.json()
|
|
|
|
if response['response']['result'] == 'success':
|
|
sys.stdout.write("Successfully sent Tautulli notification.")
|
|
else:
|
|
raise Exception(response['response']['message'])
|
|
except Exception as e:
|
|
sys.stderr.write("Tautulli 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))
|
|
else:
|
|
sys.stdout.write('User: {} is ignored from this script.'.format(username))
|
|
else:
|
|
sys.stdout.write('IP: {} is in whitelist, ignoring.'.format(ip_address)) |