Skip to content

Instantly share code, notes, and snippets.

@blacktwin
Created May 19, 2017 18:22
Show Gist options
  • Save blacktwin/d47d3ada86d02a494f9dc33e50dd15b5 to your computer and use it in GitHub Desktop.
Save blacktwin/d47d3ada86d02a494f9dc33e50dd15b5 to your computer and use it in GitHub Desktop.
Kill concurrent transcode streams of other users if Admin user is experiencing buffering warnings from PlexPy.
'''
If server admin stream is experiencing buffering and there are concurrent transcode streams from
another user, kill concurrent transcode stream that has the lowest percent complete. Message in
kill stream will list why it was killed ('Server Admin's stream take priority and this user has X
concurrent streams'). Message will also include an approximation of when the other concurrent stream
will finish, stream that is closest to finish will be used.
PlexPy > Settings > Notification Agents > Scripts > Bell icon:
[X] Notify on buffer warning
PlexPy > Settings > Notification Agents > Scripts > Gear icon:
Buffer Warnings: kill_else_if_buffering.py
'''
import requests
import platform
from uuid import getnode
from operator import itemgetter
## EDIT THESE SETTINGS ##
PLEX_HOST = ''
PLEX_PORT = 32400
PLEX_SSL = '' # s or ''
PLEX_TOKEN = 'xxxxxx'
DEFAULT_REASON = 'Server Admin\'s stream takes priority and {user}(you) has {x} concurrent streams.' \
' {user}\'s stream of {video} is {time}% complete. Should be finished in {comp} minutes. ' \
'Try again then.'
ADMIN_USER = ('Admin') # additional usernames can be added ('Admin', 'user2')
##
def fetch(path, t='GET'):
url = 'http%s://%s:%s/' % (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_HOST, PLEX_PORT),
headers=headers, params=params)
def add_to_dictlist(d, key, val):
if key not in d:
d[key] = [val]
else:
d[key].append(val)
if __name__ == '__main__':
response = fetch('status/sessions')
user_dict = {}
sessions = []
if 'Video' in response['MediaContainer']:
for s in response['MediaContainer']['Video']:
try:
if s['TranscodeSession']['videoDecision'] == 'transcode' and s['User']['title'] not in ADMIN_USER:
id = s['Session']['id']
user = s['User']['title']
percent_comp = int((float(s['viewOffset']) / float(s['duration'])) * 100)
time_to_comp = int(int(s['duration']) - int(s['viewOffset'])) / 1000 / 60
title = s['title']
add_to_dictlist(user_dict, user, [id, percent_comp, title, user, time_to_comp])
except KeyError:
print('{} has a direct stream to ignore.'.format(s['User']['title']))
# Remove users with only 1 stream. Targeting users with multiple concurrent streams
filtered_dict = {key: value for key, value in user_dict.items()
if len(value) is not 1}
# Find who to kill and who will be finishing first.
for session in filtered_dict.values():
to_kill = min(session, key=itemgetter(1))
to_finish = max(session, key=itemgetter(1))
MESSAGE = DEFAULT_REASON.format(user=to_finish[3], x=len(filtered_dict.values()[0]),
video=to_finish[2], time=to_finish[1], comp=to_finish[4])
print(MESSAGE)
kill_stream(to_kill[0], MESSAGE)
@neuroryuu
Copy link

Hello friend, I am new to the plexpy script for a long time ago I am trying to make a script that allows me to have multiple users and some 2 to 3 transcodes, I want to close their transcodes if they are more than 2 and let them use only 1, it is possible???

@blacktwin
Copy link
Author

@neuroryuu
Copy link

@blacktwin thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment