108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
from config.config import Config
|
|
from tmdbv3api import TMDb, Movie, Collection
|
|
from plexapi.server import PlexServer
|
|
from fuzzywuzzy import fuzz
|
|
from models.movie_collection import MovieCollection
|
|
from datetime import datetime as dt
|
|
import yaml, os, urllib.request
|
|
|
|
# Instantiate Tmdb object
|
|
tmdb = TMDb()
|
|
tmdb.api_key = Config.tmdbApiKey
|
|
tmdb.language = Config.tmdbLanguage
|
|
|
|
# Instantiate TMDb movie & collection object
|
|
tmdbMovie = Movie()
|
|
tmdbCollection = Collection()
|
|
|
|
# Connect to Plex Server
|
|
baseurl = Config.plexBaseUrl
|
|
token = Config.plexToken
|
|
plex = PlexServer(baseurl, token)
|
|
|
|
# Create empty lists
|
|
PlexMovieList = []
|
|
MatchingList = []
|
|
TMDbCollection = []
|
|
MovieCollectionFinal = []
|
|
|
|
# Retrieve the library Movies from Plex
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Retrieving Movies section from Plex")
|
|
plexMoviesLibrary = plex.library.section('Movies')
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Movies section from Plex retrieved")
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Creating Movies list and Collection Available in Plex")
|
|
for movie in plexMoviesLibrary.search():
|
|
PlexMovieList.append(movie.title)
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Lists created, found {len(PlexMovieList)} movies")
|
|
|
|
i = 0
|
|
|
|
length = len(PlexMovieList)
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Comparing Movies in Plex to find similar movies...")
|
|
while i < length:
|
|
for movie in PlexMovieList:
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Checking matching between {PlexMovieList[i]} and {movie}, score is: {fuzz.ratio(PlexMovieList[i], movie)}")
|
|
if 85 < fuzz.ratio(PlexMovieList[i], movie) < 100:
|
|
if PlexMovieList[i] not in MatchingList:
|
|
MatchingList.append(PlexMovieList[i])
|
|
i += 1
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Comparison is finished...found {len(MatchingList)} similar movies")
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Searching TMDb for movies in the Matching List")
|
|
for movie in MatchingList:
|
|
movieSearch = tmdbMovie.search(movie)
|
|
movieId = movieSearch[0].id
|
|
movieDetails = tmdbMovie.details(movie_id=movieId)
|
|
|
|
if movieDetails.belongs_to_collection is not None:
|
|
collectionId = movieDetails.belongs_to_collection
|
|
|
|
if collectionId.id not in TMDbCollection:
|
|
TMDbCollection.append(collectionId.id)
|
|
|
|
for id in TMDbCollection:
|
|
MovieCollectionFinal.append(tmdbCollection.details(id))
|
|
|
|
MovieCollectionList = []
|
|
|
|
for movie in MovieCollectionFinal:
|
|
coll = MovieCollection()
|
|
nameLength = len(movie)
|
|
coll.collectionTitle = movie.name[:nameLength - 13]
|
|
coll.collectionId = movie.id
|
|
coll.collectionImgPath = movie.poster_path
|
|
MovieCollectionList.append(coll)
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Checking and Updating Yaml File...")
|
|
|
|
with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
|
|
try:
|
|
currentYaml = yaml.safe_load(f)
|
|
except yaml.YAMLError as exc:
|
|
print(exc)
|
|
|
|
currentYamlCollections = []
|
|
|
|
for movies in currentYaml['collections']:
|
|
currentYamlCollections.append(movies)
|
|
|
|
for collection in MovieCollectionList:
|
|
if collection.collectionTitle not in currentYamlCollections:
|
|
with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
|
|
new_coll = {f'{collection.collectionTitle}': {'sync_mode': 'sync', 'tmdb_collection': collection.collectionId, 'tmdb_summary': collection.collectionId}}
|
|
currentYaml['collections'].update(new_coll)
|
|
|
|
if currentYaml:
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Adding new collection: {collection.collectionTitle} to Yaml file...")
|
|
with open(Config.PlexAutoCollectionConfigFilePath, 'w') as f:
|
|
yaml.safe_dump(currentYaml, f)
|
|
|
|
if not os.path.exists(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.collectionTitle}"):
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: Download poster for new collection {collection.collectionTitle}...")
|
|
os.mkdir(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.collectionTitle}")
|
|
urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.collectionImgPath}", Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.collectionTitle}/poster.jpeg")
|
|
|
|
print(f":: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} :: CollectionUpdater :: All steps are completed...") |