Merge pull request #17 from burkasaurusrex/master

Add 'add_movie' option to Radarr config

Author: mza921

README.md

@@ -110,6 +110,14 @@ collections:
       - https://trakt.tv/users/2borno2b/lists/christmas-movies-extravanganza
   Marvel:
     trakt-list: https://trakt.tv/users/movistapp/lists/marvel
+  Pixar:
+    studio: Pixar
+    details:
+      summary: A collection of Pixar movies
+  1990s:
+    year: [1990,1991,1992,1993,1994,1995,1996,1997,1998,1999]
+    details:
+      summary: A collection of 1990s movies
 plex:
   library: Movies
   library_type: movie # or 'show'
@@ -120,7 +128,8 @@ radarr:
   token: ################ 
   quality_profile_id: 4
   root_folder_path: /mnt/user/PlexMedia/movies
-  search: true
+  add_movie: false
+  search_movie: false
 tmdb:
   apikey: ################ 
   language: en

config.yml.template

@@ -16,6 +16,14 @@ collections:
       tmdb-summary: 4169
   Marvel:
     trakt-list: https://trakt.tv/users/movistapp/lists/marvel
+  Pixar:
+    studio: Pixar
+    details:
+      summary: A collection of Pixar movies
+  1990s:
+    year: [1990,1991,1992,1993,1994,1995,1996,1997,1998,1999]
+    details:
+      summary: A collection of 1990s movies
 plex:
   library: Movies
   library_type: ### 'movie' or 'show' ###
@@ -26,7 +34,8 @@ radarr:
   token: ###########################
   quality_profile_id: 4
   root_folder_path: /mnt/user/PlexMedia/movies
-  search: false
+  add_movie: false
+  search_movie: false
 tmdb:
   apikey: ############################
   language: en

config_tools.py

@@ -56,8 +56,20 @@ def __init__(self, config_path):
         config = Config(config_path).radarr
         self.url = config['url']
         self.token = config['token']
-        self.quality = config['quality_profile_id']
-
+        self.quality_profile_id = config['quality_profile_id']
+        self.root_folder_path = config['root_folder_path']
+        # Set 'add_movie' to None if not set
+        if 'add_movie' in config: 
+            self.add_movie = config['add_movie']
+        else:
+            self.add_movie = None
+        # Support synonyms 'search_movie' and 'search'; add logical default to False
+        if 'search_movie' in config:
+            self.search_movie = config['search_movie']
+        elif 'search' in config:
+            self.search_movie = config['search']
+        else:
+            self.search_movie = False
 
 class TMDB:
     def __init__(self, config_path):
@@ -92,7 +104,7 @@ def __init__(self, config_path):
         except:
             a = 1
 
-def update_from_config(config_path, plex, skip_radarr=False):
+def update_from_config(config_path, plex):
     config = Config(config_path)
     collections = config.collections
     if isinstance(plex.Library, MovieSection):
@@ -137,7 +149,11 @@ def update_from_config(config_path, plex, skip_radarr=False):
                         else:
                             method_name = "TMDb"
                         print("{} missing movies from {} List: {}".format(len(missing), method_name, v))
-                        if not skip_radarr:
+                        if 'add_movie' in config.radarr:
+                            if config.radarr['add_movie'] is True:
+                                print("Adding missing movies to Radarr")
+                                add_to_radarr(config_path, missing)
+                        else:
                             if input("Add missing movies to Radarr? (y/n): ").upper() == "Y":
                                 add_to_radarr(config_path, missing)
                     elif libtype == "show":

radarr_tools.py

@@ -1,27 +1,15 @@
 import re, requests, os, yaml
+import config_tools
 from tmdbv3api import TMDb
 from tmdbv3api import Movie
 
 def add_to_radarr(config_path, missing):
-    # config_path = os.path.join(os.getcwd(), 'config.yml')
-    config = yaml.load(open(config_path), Loader=yaml.FullLoader)
+    config_tmdb = config_tools.TMDB(config_path)
+    config_radarr = config_tools.Radarr(config_path)
 
     tmdb = TMDb()
-    tmdb.api_key = config['tmdb']['apikey']
-    tmdb.language = "en"
-
-    url = config['radarr']['url'] + "/api/movie"
-    token = config['radarr']['token']
-    quality = config['radarr']['quality_profile_id']
-    rootfolder = config['radarr']['root_folder_path']
-    search = config['radarr']['search']
-    querystring = {"apikey": "{}".format(token)}
-
-    if "None" in (tmdb.api_key, url, token, quality, rootfolder):
-        print("All TMDB / Radarr details must be filled out in the configuration "
-              "file to import missing movies into Radarr")
-        print("\n")
-        return
+    tmdb.api_key = config_tmdb.apikey
+    tmdb.language = config_tmdb.language
 
     movie = Movie()
     for m in missing:
@@ -31,18 +19,18 @@ def add_to_radarr(config_path, missing):
             tmdb_title = tmdb_details['title']
             tmdb_id = tmdb_details['id']
         except IndexError:
-            print("Unable to fetch necessary external information")
+            print("Unable to fetch necessary external information, skipping")
             continue
 
         # Validate TMDb year (several TMDb entries don't yet have release dates)        
         try: 
             tmdb_year = tmdb_details['release_date'].split("-")[0]
         except KeyError:
-            print(tmdb_title + " does not have a release date yet")
+            print(tmdb_title + " does not have a release date yet, skipping")
             continue
 
         if tmdb_year.isdigit() == False:
-            print(tmdb_title + " does not have a release date yet")
+            print(tmdb_title + " does not have a release date yet, skipping")
             continue
 
         tmdb_poster = "https://image.tmdb.org/t/p/original{}".format(tmdb_details['poster_path'])
@@ -54,21 +42,24 @@ def add_to_radarr(config_path, missing):
 
         payload = {
             "title": tmdb_title,
-            "qualityProfileId": quality,
+            "qualityProfileId": config_radarr.quality_profile_id,
             "year": int(tmdb_year),
             "tmdbid": str(tmdb_id),
             "titleslug": titleslug,
             "monitored": "true",
-            "rootFolderPath": rootfolder,
+            "rootFolderPath": config_radarr.root_folder_path,
             "images": [{
                 "covertype": "poster",
                 "url": tmdb_poster
             }],
             "addOptions": {
-                "searchForMovie": search
+                "searchForMovie": config_radarr.search_movie
             }
         }
 
+        url = config_radarr.url + "/api/movie"
+        querystring = {"apikey": "{}".format(config_radarr.token)}
+
         response = requests.post(url, json=payload, params=querystring)
 
         try:

requirements.txt

@@ -1,7 +1,7 @@
 PyYAML==5.1.2
 tmdbv3api
 lxml
-requests
+requests>=2.4.2
 flask
 git+git://github.com/pkkid/python-plexapi.git#egg=plexapi
 trakt.py