This commit is contained in:
Gregory Salaun 2021-12-08 12:18:19 +01:00
parent 4701035529
commit b642a8451f

View File

@ -3,6 +3,7 @@ from tmdbv3api import TMDb, Movie, Collection
from plexapi.server import PlexServer from plexapi.server import PlexServer
from fuzzywuzzy import fuzz from fuzzywuzzy import fuzz
from models.movie_collection import MovieCollection from models.movie_collection import MovieCollection
from datetime import datetime as dt
# Instantiate Tmdb object # Instantiate Tmdb object
tmdb = TMDb() tmdb = TMDb()
@ -24,20 +25,26 @@ PlexMovieList = []
MatchingList = [] MatchingList = []
TMDbCollection = [] TMDbCollection = []
MovieCollectionFinal = [] MovieCollectionFinal = []
print(":: CollectionUpdater :: Creating Empty lists")
# Retrieve the library Movies from Plex # Retrieve the library Movies from Plex
print(f":: {dt.now()} :: CollectionUpdater :: Retrieving Movies section from Plex")
plexMoviesLibrary = plex.library.section('Movies') plexMoviesLibrary = plex.library.section('Movies')
print(f":: {dt.now()} :: CollectionUpdater :: Movies section from Plex retrieved")
print(f":: {dt.now()} :: CollectionUpdater :: Creating Movies list and Collection Available in Plex")
for movie in plexMoviesLibrary.search(): for movie in plexMoviesLibrary.search():
for collection in movie.collections: for collection in movie.collections:
if collection.tag and collection.tag not in AvailablePlexCollectionsMovies: if collection.tag and collection.tag not in AvailablePlexCollectionsMovies:
AvailablePlexCollectionsMovies.append(collection.tag) AvailablePlexCollectionsMovies.append(collection.tag)
PlexMovieList.append(movie.title) PlexMovieList.append(movie.title)
print(f":: {dt.now()} :: CollectionUpdater :: Lists created")
i = 0 i = 0
length = len(PlexMovieList) length = len(PlexMovieList)
print(f":: {dt.now()} :: CollectionUpdater :: Comparing Movies in Plex to find similar movies...")
while i < length: while i < length:
for movie in PlexMovieList: for movie in PlexMovieList:
if 85 < fuzz.ratio(PlexMovieList[i], movie) < 100: if 85 < fuzz.ratio(PlexMovieList[i], movie) < 100:
@ -45,6 +52,8 @@ while i < length:
MatchingList.append(PlexMovieList[i]) MatchingList.append(PlexMovieList[i])
i += 1 i += 1
print(f":: {dt.now()} :: CollectionUpdater :: Comparison is finished...")
for movie in MatchingList: for movie in MatchingList:
movieSearch = tmdbMovie.search(movie) movieSearch = tmdbMovie.search(movie)
movieId = movieSearch[0].id movieId = movieSearch[0].id
@ -60,4 +69,4 @@ for id in TMDbCollection:
MovieCollectionFinal.append(tmdbCollection.details(id)) MovieCollectionFinal.append(tmdbCollection.details(id))
for movie in MovieCollectionFinal: for movie in MovieCollectionFinal:
print(movie) print(movie.name)