Mirrored_Repos/JBOPS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update plex_api_parental_control.py

Browse Source
This commit is contained in:
blacktwin 2017-09-29 15:35:30 -04:00 committed by GitHub
parent a883b00dc6
commit 3418ca7cd7
1 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
utility/plex_api_parental_control.py
View File

@ -39,11 +39,12 @@ Usage:
import argparse import argparse
from time import sleep
from plexapi.server import PlexServer from plexapi.server import PlexServer
PLEX_URL = 'http://localhost:32400' PLEX_URL = 'http://localhost:32400'
PLEX_TOKEN = 'xxxxx' PLEX_TOKEN = 'xxxxxx'
plex = PlexServer(PLEX_URL, PLEX_TOKEN) plex = PlexServer(PLEX_URL, PLEX_TOKEN)
user_lst = [x.title for x in plex.myPlexAccount().users()] user_lst = [x.title for x in plex.myPlexAccount().users()]
@ -54,26 +55,19 @@ MESSAGE = "GET TO BED!"
def share(user, libraries): def share(user, libraries):
plex.myPlexAccount().updateFriend(user=user, server=plex, sections=libraries) plex.myPlexAccount().updateFriend(user=user, server=plex, sections=libraries)
if not libraries: print('Shared libraries: {libraries} with {user}.'.format(libraries=libraries, user=user))
print('Shared all libraries with {user}.'.format(user=user))
else:
print('Shared libraries: {libraries} with {user}.'.format(libraries=libraries, user=user))
def unshare(user, libraries): def unshare(user, libraries):
plex.myPlexAccount().updateFriend(user=user, server=plex, removeSections=True, sections=libraries) plex.myPlexAccount().updateFriend(user=user, server=plex, removeSections=True, sections=libraries)
print('Unshared libraries: {libraries} from {user}.'.format(libraries=libraries, user=user)) print('Unshared all libraries from {user}.'.format(libraries=libraries, user=user))
def kill_session(user, libraries): def kill_session(user, libraries):
for session in plex.sessions(): for session in plex.sessions():
# Check for users stream # Check for users stream
if session.usernames[0] in user: if session.usernames[0] in user:
# If to be unshared library is not being watched then stream is not killed.
title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
for library in libraries:
if session.librarySectionID == plex.library.section(library).key:
break
print('{user} is watching {title} and it\'s past their bedtime. Killing stream.'.format( print('{user} is watching {title} and it\'s past their bedtime. Killing stream.'.format(
user=user, title=title)) user=user, title=title))
session.stop(reason=MESSAGE) session.stop(reason=MESSAGE)
@ -84,7 +78,7 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Share or unshare libraries.", parser = argparse.ArgumentParser(description="Share or unshare libraries.",
formatter_class=argparse.RawTextHelpFormatter) formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument('-s', '--share', nargs='?', type=str, required=True, parser.add_argument('-s', '--share', nargs='?', type=str, required=True,
choices=['share', 'share_all', 'unshare', 'unshare_all'], metavar='', choices=['share', 'share_all', 'unshare'], metavar='',
help='To share or to unshare.: \n (choices: %(choices)s)') help='To share or to unshare.: \n (choices: %(choices)s)')
parser.add_argument('-u', '--user', nargs='?', type=str, required=True, choices=user_lst, metavar='', 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' help='Space separated list of case sensitive names to process. Allowed names are: \n'
@ -98,12 +92,10 @@ if __name__ == "__main__":
if opts.share == 'share': if opts.share == 'share':
share(opts.user, opts.libraries) share(opts.user, opts.libraries)
elif opts.share == 'share_all': elif opts.share == 'share_all':
unshare(opts.user, sections_lst) share(opts.user, sections_lst)
elif opts.share == 'unshare': elif opts.share == 'unshare':
kill_session(opts.user, opts.libraries)
unshare(opts.user, opts.libraries)
elif opts.share == 'unshare_all':
kill_session(opts.user, sections_lst) kill_session(opts.user, sections_lst)
unshare(opts.user, sections_lst) sleep(5)
unshare(opts.user, opts.libraries)
else: else:
print('I don\'t know what else you want.') print('I don\'t know what else you want.')