82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""
|
|
|
|
Kill stream if bitrate is > BITRATE_LIMIT
|
|
|
|
PlexPy > Settings > Notification Agents > Scripts > Bell icon:
|
|
[X] Notify on playback start
|
|
|
|
PlexPy > Settings > Notification Agents > Scripts > Gear icon:
|
|
Playback Start: kill_session_bitrate.py
|
|
|
|
"""
|
|
import requests
|
|
import platform
|
|
from uuid import getnode
|
|
import unicodedata
|
|
|
|
## EDIT THESE SETTINGS ##
|
|
PLEX_HOST = ''
|
|
PLEX_PORT = 32400
|
|
PLEX_SSL = '' # s or ''
|
|
PLEX_TOKEN = 'xxxxxx'
|
|
|
|
BITRATE_LIMIT = 4000
|
|
|
|
ignore_lst = ('')
|
|
|
|
|
|
def fetch(path, t='GET'):
|
|
url = 'http{}://{}:{}/'.format(PLEX_SSL, PLEX_HOST, PLEX_PORT)
|
|
|
|
headers = {'X-Plex-Token': PLEX_TOKEN,
|
|
'Accept': 'application/json',
|
|
'X-Plex-Provides': 'controller',
|
|
'X-Plex-Platform': platform.uname()[0],
|
|
'X-Plex-Platform-Version': platform.uname()[2],
|
|
'X-Plex-Product': 'Plexpy script',
|
|
'X-Plex-Version': '0.9.5',
|
|
'X-Plex-Device': platform.platform(),
|
|
'X-Plex-Client-Identifier': str(hex(getnode()))
|
|
}
|
|
|
|
try:
|
|
if t == 'GET':
|
|
r = requests.get(url + path, headers=headers, verify=False)
|
|
elif t == 'POST':
|
|
r = requests.post(url + path, headers=headers, verify=False)
|
|
elif t == 'DELETE':
|
|
r = requests.delete(url + path, headers=headers, verify=False)
|
|
|
|
if r and len(r.content): # incase it dont return anything
|
|
|
|
return r.json()
|
|
else:
|
|
return r.content
|
|
|
|
except Exception as e:
|
|
print e
|
|
|
|
def kill_stream(sessionId, message):
|
|
headers = {'X-Plex-Token': PLEX_TOKEN}
|
|
params = {'sessionId': sessionId,
|
|
'reason': message}
|
|
requests.get('http{}://{}:{}/status/sessions/terminate'.format(PLEX_SSL, PLEX_HOST, PLEX_PORT),
|
|
headers=headers, params=params)
|
|
|
|
response = fetch('status/sessions')
|
|
|
|
sessions = []
|
|
for video in response['MediaContainer']['Video']:
|
|
sess_id = video['Session']['id']
|
|
user = video['User']['title']
|
|
title = (video['grandparentTitle'] + ' - ' if video['type'] == 'episode' else '') + video['title']
|
|
title = unicodedata.normalize('NFKD', title).encode('ascii','ignore')
|
|
bitrate = video['Media']['bitrate']
|
|
sessions.append((sess_id, user, title, bitrate))
|
|
|
|
for session in sessions:
|
|
if session[1] not in ignore_lst and int(session[3]) > BITRATE_LIMIT:
|
|
message = "You are not allowed to stream above 4 Mbps."
|
|
print("Killing {}'s stream of {} for {}".format(session[1], session[2], message))
|
|
kill_stream(session[0], message)
|