From 155e6a129fde5d3756f76e913cf62a021d9103d0 Mon Sep 17 00:00:00 2001
From: blacktwin <blacktwin@users.noreply.github.com>
Date: Thu, 26 Oct 2017 12:37:48 -0400
Subject: [PATCH] Create time_limit.py

So many more feature to add to this. notify user of removal, record user invitedAt and removedAt, the libraries shared with user (in case you invite them back you could provide the same libraries).
---
 killstream/time_limit.py | 98 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100644 killstream/time_limit.py

diff --git a/killstream/time_limit.py b/killstream/time_limit.py
new file mode 100644
index 0000000..6e027ab
--- /dev/null
+++ b/killstream/time_limit.py
@@ -0,0 +1,98 @@
+"""
+Kill streams if user has exceeded time limit on Plex server. Choose to unshare or remove user.
+
+PlexPy > Settings > Notification Agents > Scripts > Bell icon:
+        [X] Notify on playback start
+
+PlexPy > Settings > Notification Agents > Scripts > Gear icon:
+        Playback Start: time_limit.py
+
+PlexPy > Settings > Notifications > Script > Script Arguments
+        {username}
+
+"""
+
+import requests
+import sys
+from plexapi.server import PlexServer
+
+
+## EDIT THESE SETTINGS ##
+PLEXPY_APIKEY = 'xxxx'  # Your PlexPy API key
+PLEXPY_URL = 'http://localhost:8182/'  # Your PlexPy URL
+
+PLEX_TOKEN = 'xxxx'
+PLEX_URL = 'http://localhost:32400'
+
+TIME_LIMIT = {'user1': {'d': 1, 'h': 2, 'm': 30, 'remove': True, 'unshare': True},
+              'user2': {'d': 0, 'h': 2, 'm': 30, 'remove': False, 'unshare': True},
+              'user3': {'d': 0, 'h': 20, 'm': 30, 'remove': True, 'unshare': False}}
+
+MESSAGE = 'You have reached your time limit on my server.'
+##/EDIT THESE SETTINGS ##
+
+username = str(sys.argv[1])
+
+total_time = 0
+
+sess = requests.Session()
+sess.verify = False
+plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
+
+sections_lst = [x.title for x in plex.library.sections()]
+
+def get_get_history(username):
+    # Get the PlexPy history.
+    payload = {'apikey': PLEXPY_APIKEY,
+               'cmd': 'get_history',
+               'user': username}
+
+    try:
+        r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload)
+        response = r.json()
+
+        res_data = response['response']['data']['data']
+        return sum([data['duration'] for data in res_data])
+
+    except Exception as e:
+        sys.stderr.write("PlexPy API 'get_history' request failed: {0}.".format(e))
+
+
+def unshare(user, libraries):
+    print('{user} has reached their time limit. Unsharing.'.format(user=user))
+    plex.myPlexAccount().updateFriend(user=user, server=plex, removeSections=True, sections=libraries)
+    print('Unshared all libraries from {user}.'.format(libraries=libraries, user=user))
+
+
+def remove_friend(user):
+    print('{user} has reached their time limit. Removing.'.format(user=user))
+    plex.myPlexAccount().removeFriend(user)
+    print('Removed {user}.'.format(user=user))
+
+
+def kill_session(user):
+    for session in plex.sessions():
+        # Check for users stream
+        if session.usernames[0] in user:
+            title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+            print('{user} is watching {title} and they\'ve reached their time limit. Killing stream.'
+                .format(user=user, title=title))
+            session.stop(reason=MESSAGE)
+
+
+if TIME_LIMIT[username]['d']:
+    total_time += TIME_LIMIT[username]['d'] * 24 * 60 * 60
+if TIME_LIMIT[username]['m']:
+    total_time += TIME_LIMIT[username]['m'] * 60 * 60
+if TIME_LIMIT[username]['h']:
+    total_time += TIME_LIMIT[username]['h'] * 60
+
+
+if get_get_history(username) > total_time:
+    print('User has reached time limit.')
+    kill_session(username)
+    if TIME_LIMIT[username]['remove']:
+        remove_friend(username)
+    if TIME_LIMIT[username]['unshare']:
+        unshare(username, sections_lst)
+