Move a few things over to the config file
This commit is contained in:
		
							parent
							
								
									c8523d870b
								
							
						
					
					
						commit
						24385a8897
					
				| 
						 | 
				
			
			@ -1,2 +1,8 @@
 | 
			
		|||
url = 'http://radarr.host.ip:port'
 | 
			
		||||
apikey = 'yourradarrapikey'
 | 
			
		||||
mediafolder = '/videos/'
 | 
			
		||||
libraryname = 'Movies'
 | 
			
		||||
colmap = {
 | 
			
		||||
    'Drama': 'Drama',
 | 
			
		||||
    'SciFi': 'Sci-Fi'
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -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])
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue
	
	Block a user