Create plex_api_parental_control.py

2017-09-29 14:33:18 -04:00 · 2017-09-29 14:33:18 -04:00 · 1d09a07d0b
commit 1d09a07d0b
parent cd734effd5
1 changed files with 93 additions and 0 deletions
--- a/utility/plex_api_parental_control.py
+++ b/utility/plex_api_parental_control.py
@ -0,0 +1,93 @@
+'''
+Set as cron or task for times of allowing and not allowing user access to server.
+Unsharing will kill any current stream from user before unsharing.
+
+Share or unshare libraries.
+
+optional arguments:
+  -h, --help            show this help message and exit
+  -s [], --share []     To share or to unshare.:
+                        (choices: share, unshare)
+  -u [], --user []      Space separated list of case sensitive names to process. Allowed names are:
+                        (choices: All users names)
+  -l  [ ...], --libraries  [ ...]
+                        Space separated list of case sensitive names to process. Allowed names are:
+                        (choices: All library names)
+                        (default: All Libraries)
+
+Usage:
+
+   plex_api_share.py -s unshare -u USER
+       - Kill users current stream.
+       - Unshared all libraries with USER.
+       - USER still exists as a Friend or Home User
+
+   plex_api_share.py -s share -u USER -l Movies
+       - Shared libraries: ['Movies'] with USER
+
+   plex_api_share.py -s share -u USER -l Movies "TV Shows"
+       - Shared libraries: ['Movies', 'TV Shows'] with USER
+          * Double Quote libraries with spaces
+
+'''
+
+
+import argparse
+from plexapi.server import PlexServer
+
+
+PLEX_URL = 'http://localhost:32400'
+PLEX_TOKEN = 'xxxxx'
+plex = PlexServer(PLEX_URL, PLEX_TOKEN)
+
+user_lst = [x.title for x in plex.myPlexAccount().users()]
+sections_lst = [x.title for x in plex.library.sections()]
+
+MESSAGE = "GET TO BED!"
+
+
+def share(user, libraries):
+    plex.myPlexAccount().updateFriend(user=user, server=plex, sections=libraries)
+    if not libraries:
+        print('Shared all libraries with {user}.'.format(user=user))
+    else:
+        print('Shared libraries: {libraries} with {user}.'.format(libraries=libraries, user=user))
+
+
+def unshare(user):
+    plex.myPlexAccount().updateFriend(user=user, server=plex, removeSections=True)
+    print('Unshared all libraries with {user}.'.format(user=user))
+
+
+def kill_session(user):
+    for session in plex.sessions():
+        if session.usernames[0] in user:
+            title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+            print('{user} is watching {title} and it\'s past their bedtime. Killing stream.'.format(
+                user=user, title=title))
+            session.stop(reason=MESSAGE)
+
+
+if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser(description="Share or unshare libraries.",
+                                     formatter_class=argparse.RawTextHelpFormatter)
+    parser.add_argument('-s', '--share', nargs='?', type=str, required=True, choices=['share', 'unshare'], metavar='',
+                        help='To share or to unshare.: \n'
+                             '(choices: %(choices)s)')
+    parser.add_argument('-u', '--user', nargs='?', type=str, required=True, choices=user_lst, metavar='',
+                        help='Space separated list of case sensitive names to process. Allowed names are: \n'
+                             '(choices: %(choices)s)')
+    parser.add_argument('-l', '--libraries', nargs='+', default='', choices=sections_lst, metavar='',
+                        help='Space separated list of case sensitive names to process. Allowed names are: \n'
+                             '(choices: %(choices)s \n(default: All Libraries)')
+
+    opts = parser.parse_args()
+
+    if opts.share == 'share':
+        share(opts.user, opts.libraries)
+    elif opts.share == 'unshare':
+        kill_session(opts.user)
+        unshare(opts.user)
+    else:
+        print('I don\'t know what else you want.')