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'
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
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])