diff --git a/killstream/kill_else_if_buffering.py b/killstream/kill_else_if_buffering.py index db9198d..11f514d 100644 --- a/killstream/kill_else_if_buffering.py +++ b/killstream/kill_else_if_buffering.py @@ -40,9 +40,11 @@ def kill_session(sess_key, message): # Check for users stream username = session.usernames[0] if session.sessionKey == sess_key: - title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title - print('{user} is watching {title} and they might be asleep.'.format(user=username, title=title)) + title = unicode(session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title + title = unicodedata.normalize('NFKD', title).encode('ascii', 'ignore').translate(None, "'") session.stop(reason=message) + print('Terminated {user}\'s stream of {title} to prioritize admin stream.'.format(user=username, + title=title)) def add_to_dictlist(d, key, val): @@ -56,30 +58,32 @@ def main(): user_dict = {} for session in plex.sessions(): - trans_dec = session.transcodeSessions[0].videoDecision - username = session.usernames[0] - if trans_dec == 'transcode' and username not in ADMIN_USER: - sess_key = session.sessionKey - percent_comp = int((float(session.viewOffset) / float(session.duration)) * 100) - time_to_comp = int(int(session.duration) - int(session.viewOffset)) / 1000 / 60 - title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title - title = unicodedata.normalize('NFKD', title).encode('ascii', 'ignore').translate(None, "'") - add_to_dictlist(user_dict, username, [sess_key, percent_comp, title, username, time_to_comp]) + if session.transcodeSessions: + trans_dec = session.transcodeSessions[0].videoDecision + username = session.usernames[0] + if trans_dec == 'transcode' and username not in ADMIN_USER: + sess_key = session.sessionKey + percent_comp = int((float(session.viewOffset) / float(session.duration)) * 100) + time_to_comp = int(int(session.duration) - int(session.viewOffset)) / 1000 / 60 + title = unicode(session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title + title = unicodedata.normalize('NFKD', title).encode('ascii', 'ignore').translate(None, "'") + add_to_dictlist(user_dict, username, [sess_key, percent_comp, title, username, time_to_comp]) # 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 users in filtered_dict.values(): - to_kill = min(users, key=itemgetter(1)) - to_finish = max(users, 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_session(to_kill[0], MESSAGE) + if filtered_dict: + for users in filtered_dict.values(): + to_kill = min(users, key=itemgetter(1)) + to_finish = max(users, 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_session(to_kill[0], MESSAGE) if __name__ == '__main__':