From 24385a8897258a4c582aabbbb70f2fa3551cf898 Mon Sep 17 00:00:00 2001
From: Donald Little <junior@jaj.com>
Date: Fri, 15 Sep 2023 17:05:03 +0000
Subject: [PATCH] Move a few things over to the config file

---
 apiconfig-default.py |  6 ++++++
 updatecoll.py        | 42 +++++++++++++++++-------------------------
 2 files changed, 23 insertions(+), 25 deletions(-)

diff --git a/apiconfig-default.py b/apiconfig-default.py
index 96878f2..72bc482 100644
--- a/apiconfig-default.py
+++ b/apiconfig-default.py
@@ -1,2 +1,8 @@
 url = 'http://radarr.host.ip:port'
 apikey = 'yourradarrapikey'
+mediafolder = '/videos/'
+libraryname = 'Movies'
+colmap = {
+    'Drama': 'Drama',
+    'SciFi': 'Sci-Fi'
+}
diff --git a/updatecoll.py b/updatecoll.py
index 106c80f..b45f7d1 100644
--- a/updatecoll.py
+++ b/updatecoll.py
@@ -4,41 +4,33 @@ from plexapi.media import Media
 from pprint import pprint
 import sys
 
-PLEX_URL = 'http://localhost:32400'
-PLEX_TOKEN = 'h3ffbZ4pDxgNtGuUDFys'
-PATH_FOLDER = '/videos/'
-LIBRARY = 'Junior Movies'
-
-colmap = {
-        "Action-Adventure": "Action/Adventure",
-        "Animated": "Animated",
-        "Christmas": "Christmas",
-        "Comedy-Romantic": "Comedy/Romance",
-        "Documentary": "Documentary",
-        "Drama": "Drama",
-        "Imports": "Imports",
-        "Scary": "Scary",
-        "Sci-Fi": "Sci-Fi"
-}
+try:
+    import apiconfig
+except:
+    print("Error!")
+    print("  \"apiconfig.py\" seems to be missing!")
+    print("  Have you copied the default config to \"apiconfig.py\" and made appropriate edits?")
+    print()
+    sys.exit()
 
 colcontents = {}
-for col in colmap.values():
+for col in apiconfig.colmap.values():
     colcontents[col] = []
 
-plex = PlexServer(PLEX_URL, PLEX_TOKEN)
+plex = PlexServer(apiconfig.url, apiconfig.apikey)
 
 print("Building collection contents lists...")
-for col in colmap.values():
-    videos = plex.library.section(LIBRARY).search(collection=col)
+for col in apiconfig.colmap.values():
+    videos = plex.library.section(apiconfig.libraryname).search(collection=col)
     for video in videos: colcontents[col].append(video.title)
 
 print("Checking collections for each movie...")
-videos = plex.library.section(LIBRARY).all()
+videos = plex.library.section(apiconfig.libraryname).all()
 for video in videos:
     file = None
     for part in video.iterParts():
         if file == None: file = part.file
-    [top_folder, junk] = file[len(PATH_FOLDER):].split("/", 1)
-    if video.title not in colcontents[colmap[top_folder]]:
-        print(f"Movie {video.title} is missing collection {colmap[top_folder]}")
-        video.addCollection(colmap[top_folder])
+    [top_folder, junk] = file[len(apiconfig.mediafolder):].split("/", 1)
+    if video.title not in colcontents[apiconfig.colmap[top_folder]]:
+        print(f"Movie {video.title} is missing collection {apiconfig.colmap[top_folder]}")
+        video.addCollection(apiconfig.colmap[top_folder])