plexapi and ssl update

#12

killstream/kill_more_than.py

@@ -1,101 +1,61 @@
 """
-If user has 2* or more concurrent streams and the IP of the 2nd stream differs from 1st kill 2nd. 
+If user has 2* or more concurrent streams and the IP of the 2nd stream differs from 1st kill 2nd.
 If 2nd stream IP is the same as 1st stream don't kill.
-
-
 *PlexPy > Settings > Notification> User Concurrent Stream Threshold
     The number of concurrent streams by a single user for PlexPy to trigger a notification. Minimum 2.
-
 PlexPy > Settings > Notification Agents > Scripts > Bell icon:
         [X] Notify on user concurrent streams
-        
+
 PlexPy > Settings > Notification Agents > Scripts > Gear icon:
         Playback User Concurrent Streams: kill_more_than.py
-        
-PlexPy > Settings > Notifications > Script > Script Arguments      
-        {username} {ip_address}
+
+PlexPy > Settings > Notifications > Script > Script Arguments
+        {user} {ip_address}
 """
 
-
 import requests
-import platform
-from uuid import getnode
 import sys
-import unicodedata
+from plexapi.server import PlexServer
 
 ## EDIT THESE SETTINGS ##
-PLEX_HOST = ''
-PLEX_PORT = 32400
-PLEX_SSL = ''  # s or ''
-PLEX_TOKEN = 'xxxxxxx'
+PLEX_TOKEN = 'xxxx'
+PLEX_URL = 'http://localhost:32400'
 
-REASON = 'Because....too many streams'
+MESSAGE = 'Because....too many streams'
+##/EDIT THESE SETTINGS ##
 
 # 2nd stream information is passed
 USERNAME = sys.argv[1]
 ADDRESS = sys.argv[2]
 
-ignore_lst = ('')
-
-if USERNAME in ignore_lst:
-    print(u"{} ignored.".format(USERNAME))
-    exit()
+sess = requests.Session()
+sess.verify = False
+plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
 
 
-def fetch(path, t='GET'):
-    url = 'http{}://{}:{}/'.format(PLEX_SSL, PLEX_HOST, PLEX_PORT)
+def kill_session(session_key, reason):
+    for session in plex.sessions():
+        # Check for users session key
+        if session.sessionKey == session_key:
+            title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+            print('{user} is watching {title} and they might be asleep.'.format(user=user, title=title))
+            session.stop(reason=reason)
 
-    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)
+user_sessions = []
 
-        if r and len(r.content):  # incase it dont return anything
+for session in plex.sessions():
+    username = session.usernames[0]
+    ip_address = session.players[0].address
+    if username == USERNAME and ip_address == ADDRESS:
+        sess_key = session.sessionKey
+        title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+        user_sessions.append((sess_key, username, title))
 
-            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']:
-    if video['User']['title'] == USERNAME and video['Player']['address'].lstrip("::ffff:") == ADDRESS::
-        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')
-        sessions.append((sess_id, user, title))
-
-if len(sessions) == 1:
-    for session in sessions:
-        print(u"Killing {}'s second stream of {} for {}".format(session[1], session[2], REASON))
-        kill_stream(session[0], REASON)
+if len(user_sessions) == 1:
+    for session in user_sessions:
+        print(u"Killing {}'s second stream of {} for {}".format(session[1], session[2], MESSAGE))
+        kill_session(session[0], MESSAGE)
 else:
-    for session in sessions:
+    for session in user_sessions:
         print(u"Not killing {}'s second stream. Same IP".format(session[1]))
-        break