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 18:06:46 +01:00
parent 261c00fc91
commit 936d2858c3
1 changed files with 53 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
PlexAutoCollectionUpdater.py
View File

@ -22,64 +22,66 @@ baseurl = Config.plexBaseUrl
token = Config.plexToken token = Config.plexToken
plex = PlexServer(baseurl, token) plex = PlexServer(baseurl, token)
plexMoviesLibrary = plex.library.section('Movies') def MovieCollectionList(library):
plexMoviesLibrary = plexMoviesLibrary.search() plexMoviesLibrary = plex.library.section(library)
plexMoviesLibrary = plexMoviesLibrary.search()
MoviesList = [] MoviesList = []
logger.info(f"Retrieving {len(plexMoviesLibrary)} Movies from Plex and Updating TMDb information...") logger.info(f"Retrieving {len(plexMoviesLibrary)} Movies from Plex and Updating TMDb information...")
for item in plexMoviesLibrary: for item in plexMoviesLibrary:
movie = UpdatedMovie() movie = UpdatedMovie()
tmdbMovie = Movie() tmdbMovie = Movie()
movie.PlexTitle = item.title movie.PlexTitle = item.title
for guid in item.guids: for guid in item.guids:
if "tmdb://" in guid.id: if "tmdb://" in guid.id:
movie.TMDbId = guid.id[7:] movie.TMDbId = guid.id[7:]
elif "imdb://" in guid.id: elif "imdb://" in guid.id:
movie.ImdbId = guid.id[7:] movie.ImdbId = guid.id[7:]
# Search for the movie on TMDb # Search for the movie on TMDb
try: try:
tmdbMovieDetails = tmdbMovie.details(movie_id=movie.TMDbId) tmdbMovieDetails = tmdbMovie.details(movie_id=movie.TMDbId)
logger.debug(f'Getting TMDb details for {movie.PlexTitle}') logger.debug(f'Getting TMDb details for {movie.PlexTitle}')
except: except:
logger.warning(f"CollectionUpdater :: Cannot get details of {movie.PlexTitle} from TMDb") logger.warning(f"CollectionUpdater :: Cannot get details of {movie.PlexTitle} from TMDb")
if tmdbMovieDetails['belongs_to_collection']: if tmdbMovieDetails['belongs_to_collection']:
movie.MovieTMDbCollectionName = tmdbMovieDetails['belongs_to_collection'].name[:len(tmdbMovieDetails['belongs_to_collection'].name) - 7] movie.MovieTMDbCollectionName = tmdbMovieDetails['belongs_to_collection'].name[:len(tmdbMovieDetails['belongs_to_collection'].name) - 7]
movie.MovieTMDbCollectionTMDbId = tmdbMovieDetails['belongs_to_collection'].id movie.MovieTMDbCollectionTMDbId = tmdbMovieDetails['belongs_to_collection'].id
movie.MovieTMDbCollectionPosterPath = tmdbMovieDetails['belongs_to_collection'].poster_path movie.MovieTMDbCollectionPosterPath = tmdbMovieDetails['belongs_to_collection'].poster_path
movie.TMDbTitle = tmdbMovieDetails.title movie.TMDbTitle = tmdbMovieDetails.title
# Add movie to the list of Movies # Add movie to the list of Movies
if movie.MovieTMDbCollectionTMDbId: if movie.MovieTMDbCollectionTMDbId:
logger.info(f'Movie {movie.PlexTitle} has been added to Collection List') logger.info(f'Movie {movie.PlexTitle} has been added to Collection List')
MoviesList.append(movie) MoviesList.append(movie)
logger.info(f"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 i = 0
count = 0
MovieListFinal = []
ExistingCollectionId = []
while i < len(MoviesList):
for movie in MoviesList:
if MoviesList[i].MovieTMDbCollectionTMDbId == movie.MovieTMDbCollectionTMDbId:
count += 1
if count > 2:
if not MoviesList[i].MovieTMDbCollectionTMDbId in ExistingCollectionId:
logger.info(f"Keeping collection {MoviesList[i].MovieTMDbCollectionName} which contains 3 or more movies")
ExistingCollectionId.append(MoviesList[i].MovieTMDbCollectionTMDbId)
MovieListFinal.append(MoviesList[i])
i += 1
count = 0 count = 0
MovieListFinal = []
ExistingCollectionId = []
logger.info(f"Found a total of {len(MovieListFinal)} collections, updating config file now !") while i < len(MoviesList):
for movie in MoviesList:
if MoviesList[i].MovieTMDbCollectionTMDbId == movie.MovieTMDbCollectionTMDbId:
count += 1
if count > 2:
if not MoviesList[i].MovieTMDbCollectionTMDbId in ExistingCollectionId:
logger.info(f"Keeping collection {MoviesList[i].MovieTMDbCollectionName} which contains 3 or more movies")
ExistingCollectionId.append(MoviesList[i].MovieTMDbCollectionTMDbId)
MovieListFinal.append(MoviesList[i])
i += 1
count = 0
def UpdateYAML(configFile): logger.info(f"Found a total of {len(MovieListFinal)} collections, updating config file now !")
return MovieListFinal
def UpdateYAML(configFile, MovieListFinal):
try: try:
with open(configFile, 'r') as f: with open(configFile, 'r') as f:
try: try:
@ -114,7 +116,10 @@ def UpdateYAML(configFile):
urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}", urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}",
Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg") Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg")
UpdateYAML(Config.PlexAutoCollectionConfigFilePath) MovieListFinal = MovieCollectionList('Movies')
#UpdateYAML(Config.PlexAutoCollectionConfigFilePath4K) UpdateYAML(Config.PlexAutoCollectionConfigFilePath, MovieListFinal)
MovieListFinal = MovieCollectionList('4K Movies')
UpdateYAML(Config.PlexAutoCollectionConfigFilePath4K)
logger.info(f"All collections have been updated in plex...") logger.info(f"All collections have been updated in plex...")