rouggy/PlexAutoCollectionUpdater
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Gregory Salaun 2021-12-15 17:08:00 +01:00
parent 15f4540ec5
commit e8f06e0070
2 changed files with 36 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
PlexAutoCollectionUpdater.py
View File

@ -32,6 +32,7 @@ logger.info(f"Retrieving {len(plexMoviesLibrary)} Movies from Plex and Updating
for item in plexMoviesLibrary:
movie = UpdatedMovie()
tmdbMovie = Movie()
tmdbMovieDetails = object()
movie.PlexTitle = item.title
for guid in item.guids:
@ -58,7 +59,7 @@ for item in plexMoviesLibrary:
if movie.MovieTMDbCollectionTMDbId:
logger.info(f'Movie {movie.PlexTitle} has been added to Collection List')
MoviesList.append(movie)
logger.info("Retrieved Movies from Plex and Updated TMDb information for {len(MoviesList)} movies...")
logger.info(f"Retrieved Movies from Plex and Updated TMDb information for {len(MoviesList)} movies...")
i = 0
count = 0
@ -79,36 +80,44 @@ while i < len(MoviesList):
logger.info(f"Found a total of {len(MovieListFinal)} collections, updating config file now !")
currentYaml = object()
def UpdateYAML(configFile):
currentYaml = object()
try:
with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
try:
currentYaml = yaml.safe_load(f)
except yaml.YAMLError as exc:
print(exc)
except:
logger.warning(f"Could not find the config file with path: {Config.PlexAutoCollectionConfigFilePath}")
try:
with open(configFile, 'r') as f:
try:
currentYaml = yaml.safe_load(f)
except yaml.YAMLError as exc:
print(exc)
except:
logger.warning(f"Could not find the config file with path: {configFile}")
currentYamlCollections = []
currentYamlCollections = []
for movies in currentYaml['collections']:
currentYamlCollections.append(movies)
for movies in currentYaml['collections']:
currentYamlCollections.append(movies)
for collection in MovieListFinal:
if collection.MovieTMDbCollectionName not in currentYamlCollections:
with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
new_coll = {f'{collection.MovieTMDbCollectionName}': {'sync_mode': 'sync', 'tmdb_collection': collection.MovieTMDbCollectionTMDbId, 'tmdb_summary': collection.MovieTMDbCollectionTMDbId}}
currentYaml['collections'].update(new_coll)
for collection in MovieListFinal:
if collection.MovieTMDbCollectionName not in currentYamlCollections:
with open(configFile, 'r') as f:
new_coll = {f'{collection.MovieTMDbCollectionName}': {'sync_mode': 'sync',
'tmdb_collection': collection.MovieTMDbCollectionTMDbId,
'tmdb_summary': collection.MovieTMDbCollectionTMDbId}}
currentYaml['collections'].update(new_coll)
if currentYaml:
logger.info(f"Adding new collection: {collection.MovieTMDbCollectionName} to Yaml file...")
with open(Config.PlexAutoCollectionConfigFilePath, 'w') as f:
yaml.safe_dump(currentYaml, f)
if currentYaml:
logger.info(f"Adding new collection: {collection.MovieTMDbCollectionName} to {configFile} file...")
with open(configFile, 'w') as f:
yaml.safe_dump(currentYaml, f)
if not os.path.exists(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}"):
logger.info(f"Download poster for new collection {collection.MovieTMDbCollectionName}...")
os.mkdir(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}")
urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}", Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg")
if not os.path.exists(
Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}"):
logger.info(f"Download poster for new collection {collection.MovieTMDbCollectionName}...")
os.mkdir(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}")
urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}",
Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg")
UpdateYAML(Config.PlexAutoCollectionConfigFilePath)
UpdateYAML(Config.PlexAutoCollectionConfigFilePath4K)
logger.info(f"All collections have been updated in plex...")

1
config/config.py
View File

@ -6,4 +6,5 @@ class Config(object):
plexBaseUrl = 'https://plex.rouggy.ovh'
plexToken = 'VeQQwtf-sNtWWLwzCEih'
PlexAutoCollectionConfigFilePath = '/opt/Plex-Auto-Collections/config/config.yml'
PlexAutoCollectionConfigFilePath4K = '/opt/Plex-Auto-Collections/config/config-4k.yml'
PlexAutoCollectionConfigFileImagesPath = '/opt/Plex-Auto-Collections/config/images'