From 99a86693a3a73b9009f593eceddf1ec1e97629f9 Mon Sep 17 00:00:00 2001
From: blacktwin <blacktwin@users.noreply.github.com>
Date: Wed, 14 Mar 2018 08:49:48 -0400
Subject: [PATCH] Update for killing streams. Adding config structure.

---
 killstream/kill_plex_streams.py | 37 ++++++++++++++++++++++-----------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/killstream/kill_plex_streams.py b/killstream/kill_plex_streams.py
index c2c43cb..2f8342e 100644
--- a/killstream/kill_plex_streams.py
+++ b/killstream/kill_plex_streams.py
@@ -1,28 +1,41 @@
 """
-Kill all streams
+Kill all Plex sessions
+
 """
 
+import sys
 import requests
 from plexapi.server import PlexServer
+import configparser
 
-## EDIT THESE SETTINGS ##
-PLEX_TOKEN = 'xxxxx'
-PLEX_URL = 'http://localhost:32400'
+# EDIT THESE SETTINGS #
+PLEX_URL = ''  # leave blank if using config.ini. Overrides config
+PLEX_TOKEN = ''  # leave blank if using config.ini. Overrides config
 
 MESSAGE = 'Because....'
-ignore_lst = ('')
-##/EDIT THESE SETTINGS ##
+# /EDIT THESE SETTINGS #
+
+config = configparser.ConfigParser()
+try:
+    config.read('../config.ini')
+    if not PLEX_URL:
+        PLEX_URL = config.get('plex', 'url')
+    if not PLEX_TOKEN:
+        PLEX_TOKEN = config.get('plex', 'token')
+except configparser.NoSectionError:
+    sys.exit('Error: No config and missing var values.')
 
 sess = requests.Session()
 sess.verify = False
 plex = PlexServer(PLEX_URL, PLEX_TOKEN, session=sess)
 
-def kill_session():
+
+def kill_sessions():
     for session in plex.sessions():
         user = session.usernames[0]
-        if user not in ignore_lst:
-            title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
-            print("Killing {}'s stream of {} for {}".format(user, title, MESSAGE))
-            session.stop(reason=MESSAGE)
+        title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
+        print("Killing {}'s stream of {} for {}".format(user, title, MESSAGE))
+        session.stop(reason=MESSAGE)
 
-kill_session()
+
+kill_sessions()