Revert "Merge branch 'develop' into feature-rating"

This reverts commit 3ba90aacc3, reversing
changes made to ed78edae57.

misc/config.py

@@ -55,6 +55,9 @@ class Config(object, metaclass=Singleton):
             'profile': 'HD-1080p',
             'root_folder': '/movies/'
         },
+        'omdb': {
+            'api_key': ''
+        },
         'filters': {
             'shows': {
                 'disabled_for': [],
@@ -85,7 +88,8 @@ class Config(object, metaclass=Singleton):
                 'anticipated': 3,
                 'trending': 3,
                 'popular': 3,
-                'boxoffice': 10
+                'boxoffice': 10,
+                'rating_limit':0
             },
             'shows': {
                 'interval': 48,

traktarr.py

@@ -394,6 +394,7 @@ def movie(movie_id, folder=None, no_search=False):
 @click.option('--add-delay', '-d', default=2.5, help='Seconds between each add request to Radarr.', show_default=True)
 @click.option('--sort', '-s', default='votes', type=click.Choice(['votes', 'rating', 'release']),
               help='Sort list to process.')
+@click.option('--rating','-r',default=None,type=(int),help='Set a minimum rating threshold (according to Rotten Tomatoes)')
 @click.option('--genre', '-g', default=None, help='Only add movies from this genre to Radarr.')
 @click.option('--folder', '-f', default=None, help='Add movies with this root folder to Radarr.')
 @click.option('--actor', '-a', default=None, help='Only add movies from this actor to Radarr.')
@@ -404,13 +405,14 @@ def movie(movie_id, folder=None, no_search=False):
 @click.option('--ignore-blacklist', is_flag=True, help='Ignores the blacklist when running the command.')
 @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', genre=None, folder=None, actor=None, no_search=False,
+def movies(list_type, add_limit=0, add_delay=2.5, sort='votes', rating=None, genre=None, folder=None, actor=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
     from helpers import misc as misc_helper
     from helpers import radarr as radarr_helper
     from helpers import trakt as trakt_helper
+    from helpers import rating as rating_helper
 
     added_movies = 0
 
@@ -518,6 +520,15 @@ def movies(list_type, add_limit=0, add_delay=2.5, sort='votes', genre=None, fold
             if not trakt_helper.is_movie_blacklisted(movie, cfg.filters.movies, ignore_blacklist,
                                                      callback_remove_recommended if remove_rejected_from_recommended
                                                      else None):
+                # Assuming the movie is not blacklisted, proceed to pull RT score if the user wishes to restrict
+                movieRating = None
+                if (rating != None and cfg['omdb']['api_key'] != ''):
+                    movieRating = rating_helper.get_rating(cfg['omdb']['api_key'],movie)
+                    if (movieRating == -1):
+                        log.debug("Skipping: %s because it did not have a rating/lacked imdbID",
+                                  movie['movie']['title'])
+                        continue
+                if (rating == None or movieRating >= rating):
                     log.info("Adding: %s (%d) | Genres: %s | Country: %s", movie['movie']['title'], movie['movie']['year'],
                              ', '.join(movie['movie']['genres']), movie['movie']['country'].upper())
                     # add movie to radarr
@@ -529,7 +540,10 @@ def movies(list_type, add_limit=0, add_delay=2.5, sort='votes', genre=None, fold
                         added_movies += 1
                     else:
                         log.error("FAILED adding %s (%d)", movie['movie']['title'], movie['movie']['year'])
-
+                else:
+                    log.info("SKIPPING: %s (%d) | Genres: %s | Country: %s", movie['movie']['title'],
+                             movie['movie']['year'],
+                             ', '.join(movie['movie']['genres']), movie['movie']['country'].upper())
                 # stop adding movies, if added_movies >= add_limit
                 if add_limit and added_movies >= add_limit:
                     break
@@ -691,7 +705,7 @@ def automatic_shows(add_delay=2.5, sort='votes', no_search=False, notifications=
     return
 
 
-def automatic_movies(add_delay=2.5, sort='votes', no_search=False, notifications=False, ignore_blacklist=False):
+def automatic_movies(add_delay=2.5, sort='votes', no_search=False, notifications=False, ignore_blacklist=False,rating_limit=None):
     from media.trakt import Trakt
 
     total_movies_added = 0
@@ -722,7 +736,7 @@ def automatic_movies(add_delay=2.5, sort='votes', no_search=False, notifications
                 # run movies
                 added_movies = movies.callback(list_type=list_type, add_limit=limit,
                                                add_delay=add_delay, sort=sort, no_search=no_search,
-                                               notifications=notifications)
+                                               notifications=notifications,rating=rating_limit)
             elif list_type.lower() == 'watchlist':
                 for authenticate_user, limit in value.items():
                     if limit <= 0:
@@ -740,7 +754,7 @@ def automatic_movies(add_delay=2.5, sort='votes', no_search=False, notifications
                     added_movies = movies.callback(list_type=list_type, add_limit=limit,
                                                    add_delay=add_delay, sort=sort, no_search=no_search,
                                                    notifications=notifications, authenticate_user=authenticate_user,
-                                                   ignore_blacklist=local_ignore_blacklist)
+                                                   ignore_blacklist=local_ignore_blacklist,rating=rating_limit)
             elif list_type.lower() == 'lists':
                 for list, v in value.items():
                     if isinstance(v, dict):
@@ -759,7 +773,7 @@ def automatic_movies(add_delay=2.5, sort='votes', no_search=False, notifications
                     added_movies = movies.callback(list_type=list, add_limit=limit,
                                                    add_delay=add_delay, sort=sort, no_search=no_search,
                                                    notifications=notifications, authenticate_user=authenticate_user,
-                                                   ignore_blacklist=local_ignore_blacklist)
+                                                   ignore_blacklist=local_ignore_blacklist,rating=rating_limit)
 
             if added_movies is None:
                 log.error("Failed adding movies from Trakt's %s list", list_type)
@@ -800,7 +814,8 @@ def run(add_delay=2.5, sort='votes', no_search=False, run_now=False, no_notifica
             sort,
             no_search,
             not no_notifications,
-            ignore_blacklist
+            ignore_blacklist,
+            cfg.automatic.movies.rating_limit
         )
         if run_now:
             movie_schedule.run()