diff --git a/autoProcessMedia.cfg.spec b/autoProcessMedia.cfg.spec index 02798251..f68cbd04 100644 --- a/autoProcessMedia.cfg.spec +++ b/autoProcessMedia.cfg.spec @@ -68,6 +68,8 @@ watch_dir = ##### Set the recursive directory permissions to the following (0 to disable) chmodDirectory = 0 + # api key for www.omdbapi.com (used as alternative to imdb) + omdbapikey = [Radarr] #### autoProcessing for Movies diff --git a/core/autoProcess/autoProcessMovie.py b/core/autoProcess/autoProcessMovie.py index 3665e670..5f10fcc2 100644 --- a/core/autoProcess/autoProcessMovie.py +++ b/core/autoProcess/autoProcessMovie.py @@ -154,6 +154,7 @@ class autoProcessMovie(object): web_root = cfg.get("web_root", "") remote_path = int(cfg.get("remote_path", 0)) protocol = "https://" if ssl else "http://" + omdbapikey = cfg.get("omdbapikey", "") status = int(status) if status > 0 and core.NOEXTRACTFAILED: extract = 0 diff --git a/core/nzbToMediaUtil.py b/core/nzbToMediaUtil.py index 9a2cab7e..79ae6580 100644 --- a/core/nzbToMediaUtil.py +++ b/core/nzbToMediaUtil.py @@ -1018,7 +1018,7 @@ def listMediaFiles(path, minSize=0, delete_ignored=0, media=True, audio=True, me return sorted(files, key=len) -def find_imdbid(dirName, inputName): +def find_imdbid(dirName, inputName, omdbApiKey): imdbid = None logger.info('Attemping imdbID lookup for {0}'.format(inputName)) @@ -1066,8 +1066,11 @@ def find_imdbid(dirName, inputName): logger.debug("Opening URL: {0}".format(url)) + if not omdbApiKey: + return try: - r = requests.get(url, params={'y': year, 't': title}, verify=False, timeout=(60, 300)) + r = requests.get(url, params={'apikey': omdbApiKey, 'y': year, 't': title}, + verify=False, timeout=(60, 300)) except requests.ConnectionError: logger.error("Unable to open URL {0}".format(url)) return