From 53d2f2ef0e0dffc14020b12b4426a075056cb22f Mon Sep 17 00:00:00 2001 From: desimaniac <5501908+desimaniac@users.noreply.github.com> Date: Thu, 6 Jun 2019 10:19:53 -0500 Subject: [PATCH] CLI: Fixed argument for minimum availability --- traktarr.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/traktarr.py b/traktarr.py index 9b56752..967b6fb 100755 --- a/traktarr.py +++ b/traktarr.py @@ -352,7 +352,7 @@ def shows(list_type, add_limit=0, add_delay=2.5, sort='votes', genre=None, folde type=click.Choice(['announced', 'in_cinemas', 'released', 'predb']), help='Add movies with this minimum availability to Radarr.') @click.option('--no-search', is_flag=True, help='Disable search when adding movie to Radarr.') -def movie(movie_id, folder=None, minimum_availability=None, no_search=False): +def movie(movie_id, folder=None, min_avail=None, no_search=False): from media.radarr import Radarr from media.trakt import Trakt @@ -360,6 +360,10 @@ def movie(movie_id, folder=None, minimum_availability=None, no_search=False): if folder: cfg['radarr']['root_folder'] = folder + # replace radarr minimum_availability if min_avail is supplied + if min_avail: + cfg['radarr']['minimum_availability'] = min_avail + # validate trakt api_key trakt = Trakt(cfg) radarr = Radarr(cfg.radarr.url, cfg.radarr.api_key) @@ -415,7 +419,7 @@ def movie(movie_id, folder=None, minimum_availability=None, no_search=False): @click.option('--remove-rejected-from-recommended', is_flag=True, help='Removes rejected/existing movies from recommended.') def movies(list_type, add_limit=0, add_delay=2.5, sort='votes', rating=None, genre=None, folder=None, actor=None, - minimum_availability=None, no_search=False, notifications=False, authenticate_user=None, + min_avail=None, no_search=False, notifications=False, authenticate_user=None, ignore_blacklist=False, remove_rejected_from_recommended=False): from media.radarr import Radarr from media.trakt import Trakt @@ -434,6 +438,10 @@ def movies(list_type, add_limit=0, add_delay=2.5, sort='votes', rating=None, gen if folder: cfg['radarr']['root_folder'] = folder + # replace radarr minimum_availability if min_avail is supplied + if min_avail: + cfg['radarr']['minimum_availability'] = min_avail + # validate trakt api_key trakt = Trakt(cfg) radarr = Radarr(cfg.radarr.url, cfg.radarr.api_key)