From 111141631ef85e14d17b346c4d7cc3be55112546 Mon Sep 17 00:00:00 2001
From: l3uddz <l3uddz@gmail.com>
Date: Tue, 6 Mar 2018 22:43:05 +0000
Subject: [PATCH] added --no-search flag to disable automatic search when
 adding movies/series. fixed radarr being sent sonarr root_folder accidently.

---
 traktarr.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/traktarr.py b/traktarr.py
index 8418edd..95e702b 100644
--- a/traktarr.py
+++ b/traktarr.py
@@ -33,7 +33,8 @@ def app():
               help='Trakt list to process.', required=True)
 @click.option('--add-limit', '-l', default=0, help='Limit number of series added to Sonarr.', show_default=True)
 @click.option('--add-delay', '-d', default=2.5, help='Seconds between each add request to Sonarr.', show_default=True)
-def shows(list_type, add_limit=0, add_delay=2.5):
+@click.option('--no-search', is_flag=True, help='Disable search when adding series to Sonarr.')
+def shows(list_type, add_limit=0, add_delay=2.5, no_search=False):
     added_shows = 0
 
     # validate trakt api_key
@@ -107,7 +108,7 @@ def shows(list_type, add_limit=0, add_delay=2.5):
                 # add show to sonarr
                 if sonarr.add_series(series['show']['ids']['tvdb'], series['show']['title'], profile_id,
                                      cfg.sonarr.root_folder,
-                                     True):
+                                     not no_search):
                     log.info("ADDED %s (%d)", series['show']['title'], series['show']['year'])
                     added_shows += 1
                 else:
@@ -131,7 +132,8 @@ def shows(list_type, add_limit=0, add_delay=2.5):
               help='Trakt list to process.', required=True)
 @click.option('--add-limit', '-l', default=0, help='Limit number of movies added to Radarr.', show_default=True)
 @click.option('--add-delay', '-d', default=2.5, help='Seconds between each add request to Radarr.', show_default=True)
-def movies(list_type, add_limit=0, add_delay=2.5):
+@click.option('--no-search', is_flag=True, help='Disable search when adding movies to Radarr.')
+def movies(list_type, add_limit=0, add_delay=2.5, no_search=False):
     added_movies = 0
 
     # validate trakt api_key
@@ -202,7 +204,7 @@ def movies(list_type, add_limit=0, add_delay=2.5):
                          ', '.join(movie['movie']['genres']), movie['movie']['country'].upper())
                 # add movie to radarr
                 if radarr.add_movie(movie['movie']['ids']['tmdb'], movie['movie']['title'], movie['movie']['year'],
-                                    profile_id, cfg.sonarr.root_folder, True):
+                                    profile_id, cfg.radarr.root_folder, not no_search):
                     log.info("ADDED %s (%d)", movie['movie']['title'], movie['movie']['year'])
                     added_movies += 1
                 else: