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'
 | 
					url = 'http://radarr.host.ip:port'
 | 
				
			||||||
apikey = 'yourradarrapikey'
 | 
					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
 | 
					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])
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue
	
	Block a user