Create check_play.py

https://gist.github.com/blacktwin/f070dff29ddbeb87973be9c0a94a1df7
2017-07-12 14:20:43 -04:00 · 2017-07-12 14:20:43 -04:00 · cbdf582b35
commit cbdf582b35
parent 56e38cf9c3
1 changed files with 87 additions and 0 deletions
--- a/check_play.py
+++ b/check_play.py
@ -0,0 +1,87 @@
+#   1. Install the requests module for python.
+#       pip install requests
+#   2. Add script arguments in PlexPy.
+#       {user} {title}
+#   Add to Playback Resume
+
+import requests
+import sys
+
+user = sys.argv[1]
+title = sys.argv[2]
+
+## EDIT THESE SETTINGS ##
+PLEXPY_APIKEY = 'XXXXXXXXXX'  # Your PlexPy API key
+PLEXPY_URL = 'http://localhost:8181/'  # Your PlexPy URL
+AGENT_ID = 10  # The notification agent ID for PlexPy
+
+SUBJECT_TEXT = "PlexPy Notification"
+BODY_TEXT = """\
+<html>
+  <head></head>
+  <body>
+    <p>Hi!<br>
+    <br>User %s has attempted to watch %s more than 3 times unsuccessfully.<br>
+    </p>
+  </body>
+</html>
+""" %(user, title)
+
+
+class UserHIS(object):
+    def __init__(self, data=None):
+        data = data or {}
+        self.watched = [d['watched_status'] for d in data]
+
+def get_get_history():
+    # Get the user IP list from PlexPy
+    payload = {'apikey': PLEXPY_APIKEY,
+               'cmd': 'get_history',
+               'user': user,
+               'search': title}
+
+    try:
+        r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload)
+        response = r.json()
+        if response['response']['data']['recordsFiltered'] > 2:
+            res_data = response['response']['data']['data']
+            return UserHIS(data=res_data)
+
+    except Exception as e:
+        sys.stderr.write("PlexPy API 'get_get_history' request failed: {0}.".format(e))
+
+
+def send_notification():
+    # Format notification text
+    try:
+        subject = SUBJECT_TEXT
+        body = BODY_TEXT
+    except LookupError as e:
+        sys.stderr.write("Unable to substitute '{0}' in the notification subject or body".format(e))
+        return None
+    # Send the notification through PlexPy
+    payload = {'apikey': PLEXPY_APIKEY,
+               'cmd': 'notify',
+               'agent_id': AGENT_ID,
+               'subject': subject,
+               'body': body}
+
+    try:
+        r = requests.post(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload)
+        response = r.json()
+
+        if response['response']['result'] == 'success':
+            sys.stdout.write("Successfully sent PlexPy notification.")
+        else:
+            raise Exception(response['response']['message'])
+    except Exception as e:
+        sys.stderr.write("PlexPy API 'notify' request failed: {0}.".format(e))
+        return None
+
+
+if __name__ == '__main__':
+    hisy = get_get_history()
+
+    if sum(hisy.watched) == 0:
+        sys.stdout.write(user + ' has attempted to watch ' + title + ' more than 3 times unsuccessfully.')
+        send_notification()