Plex-Auto-Collections/app/radarr_tools.py

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))