diff --git a/killstream/play_limit.py b/killstream/play_limit.py new file mode 100644 index 0000000..65bb981 --- /dev/null +++ b/killstream/play_limit.py @@ -0,0 +1,74 @@ +""" +Kill streams if user has watched too much Plex Today. + +PlexPy > Settings > Notification Agents > Scripts > Bell icon: + [X] Notify on playback start + +PlexPy > Settings > Notification Agents > Scripts > Gear icon: + Playback Start: play_limit.py + +PlexPy > Settings > Notifications > Script > Script Arguments + {username} + +""" + +import requests +import sys +import datetime +from plexapi.server import PlexServer + + +## EDIT THESE SETTINGS ## +PLEXPY_APIKEY = 'xxxxx' # Your PlexPy API key +PLEXPY_URL = 'http://localhost:8182/' # Your PlexPy URL + +PLEX_TOKEN = 'xxxxx' +PLEX_URL = 'http://localhost:32400' + +PLAY_LIMIT = {'user1': 2, + 'user2': 3, + 'user3': 4} + +MESSAGE = 'You have reached your play limit for today.' +##/EDIT THESE SETTINGS ## + +username = str(sys.argv[1]) + +TODAY = datetime.datetime.today().strftime('%Y-%m-%d') + +sess = requests.Session() +sess.verify = False +plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess) + + +def get_get_history(username): + # Get the PlexPy history. + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_history', + 'user': username, + 'start_date': TODAY} + + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + res_data = response['response']['data']['recordsFiltered'] + return res_data + + except Exception as e: + sys.stderr.write("PlexPy API 'get_history' request failed: {0}.".format(e)) + + +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 played too much today. Killing stream.' + .format(user=user, title=title)) + session.stop(reason=MESSAGE) + + +if get_get_history(username) > PLAY_LIMIT[username]: + print('User has reached play limit for today.') + kill_session(username)