Move a few things over to the config file

This commit is contained in:
Junior 2023-09-15 17:05:03 +00:00
parent c8523d870b
commit 24385a8897
2 changed files with 23 additions and 25 deletions

View File

@ -1,2 +1,8 @@
url = 'http://radarr.host.ip:port' url = 'http://radarr.host.ip:port'
apikey = 'yourradarrapikey' apikey = 'yourradarrapikey'
mediafolder = '/videos/'
libraryname = 'Movies'
colmap = {
'Drama': 'Drama',
'SciFi': 'Sci-Fi'
}

View File

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