mirror of
https://github.com/mza921/Plex-Auto-Collections
synced 2024-11-14 00:07:18 +00:00
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
import re, requests, os, yaml
|
|
import config_tools
|
|
from tmdbv3api import TMDb
|
|
from tmdbv3api import Movie
|
|
|
|
def add_to_radarr(config_path, missing):
|
|
config_tmdb = config_tools.TMDB(config_path)
|
|
config_radarr = config_tools.Radarr(config_path)
|
|
|
|
tmdb = TMDb()
|
|
tmdb.api_key = config_tmdb.apikey
|
|
tmdb.language = config_tmdb.language
|
|
|
|
movie = Movie()
|
|
for tmdb_id in missing:
|
|
try:
|
|
tmovie = movie.details(tmdb_id)
|
|
tmdb_title = tmovie.title
|
|
except AttributeError:
|
|
print("| --- Unable to fetch necessary TMDb information for TMDb ID {}, skipping".format(tmdb_id))
|
|
continue
|
|
|
|
# Validate TMDb year (several TMDb entries don't yet have release dates)
|
|
try:
|
|
# This might be overly punitive
|
|
tmdb_year = tmovie.release_date.split("-")[0]
|
|
except AttributeError:
|
|
print("| --- {} does not have a release date on TMDb yet, skipping".format(tmdb_title))
|
|
continue
|
|
|
|
if tmdb_year.isdigit() == False:
|
|
print("| --- {} does not have a release date on TMDb yet, skipping".format(tmdb_title))
|
|
continue
|
|
|
|
tmdb_poster = "https://image.tmdb.org/t/p/original{}".format(tmovie.poster_path)
|
|
|
|
titleslug = "{} {}".format(tmdb_title, tmdb_year)
|
|
titleslug = re.sub(r'([^\s\w]|_)+', '', titleslug)
|
|
titleslug = titleslug.replace(" ", "-")
|
|
titleslug = titleslug.lower()
|
|
|
|
payload = {
|
|
"title": tmdb_title,
|
|
"qualityProfileId": config_radarr.quality_profile_id,
|
|
"year": int(tmdb_year),
|
|
"tmdbid": str(tmdb_id),
|
|
"titleslug": titleslug,
|
|
"monitored": "true",
|
|
"rootFolderPath": config_radarr.root_folder_path,
|
|
"images": [{
|
|
"covertype": "poster",
|
|
"url": tmdb_poster
|
|
}],
|
|
"addOptions": {
|
|
"searchForMovie": config_radarr.search_movie
|
|
}
|
|
}
|
|
|
|
if config_radarr.version == "v3":
|
|
slug = "/api/v3/movie"
|
|
else:
|
|
slug = "/api/movie"
|
|
url = config_radarr.url + slug
|
|
querystring = {"apikey": "{}".format(config_radarr.token)}
|
|
|
|
response = requests.post(url, json=payload, params=querystring)
|
|
|
|
if response.status_code < 400:
|
|
print("| +++ {}: Added to Radarr".format(tmdb_title))
|
|
else:
|
|
print("| --- {}: {} {}".format(tmdb_title, response.json()[0]['errorMessage'], response.status_code))
|