diff --git a/killstream/kill_device.py b/killstream/kill_device.py index 7c0df6e..a26eec1 100644 --- a/killstream/kill_device.py +++ b/killstream/kill_device.py @@ -13,20 +13,16 @@ from plexapi.server import PlexServer ## EDIT THESE SETTINGS ## -PLEX_TOKEN = 'xxxx' +PLEX_TOKEN = 'xxxxxx' PLEX_URL = 'http://localhost:32400' DEFAULT_REASON = 'This stream has ended due to your device type.' # Find platforms that have history in PlexPy in Play count by platform and stream type Graph -DEVICES = {'Android': - { 'message': 'Andriod message', 'kill': False}, - 'Chrome': - { 'message': 'Chrome message', 'kill': True}, - 'Plex Media Player': - { 'message': 'PMP message', 'kill': False}, - 'Chromecast': - { 'message': 'Chromecast message', 'kill': True}} +DEVICES = {'Android': 'Andriod message', + 'Chrome': 'Chrome message', + 'Plex Media Player': 'PMP message', + 'Chromecast': 'Chromecast message'} USER_IGNORE = ('') # ('Username','User2') ##/EDIT THESE SETTINGS ## @@ -35,20 +31,21 @@ sess = requests.Session() sess.verify = False plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess) - def kill_session(): + for session in plex.sessions(): user = session.usernames[0] if user in USER_IGNORE: - print('Ignoring {}\'s stream.'.format(user)) + print('Ignoring {}\'s stream from platform check.'.format(user)) exit() platform = session.players[0].platform - if DEVICES[platform]['kill'] is True: - MESSAGE = DEVICES[platform].get('message', DEFAULT_REASON) + if DEVICES[platform]: + MESSAGE = DEVICES.get(platform, DEFAULT_REASON) print('Killing {user}\'s stream on {plat}.'.format(user=user, plat=platform)) session.stop(reason=MESSAGE) - + else: + print('{user}\'s stream on {plat} is allowed to play.'.format(user=user, plat=platform)) if __name__ == '__main__': kill_session()