From 12275e32539210db062f577435c731ee7ab7cc1b Mon Sep 17 00:00:00 2001 From: clinton-hall Date: Sat, 9 Aug 2014 21:58:15 +0930 Subject: [PATCH] get imdb from dognzb info where available. Fixes #526 --- nzbtomedia/nzbToMediaUtil.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/nzbtomedia/nzbToMediaUtil.py b/nzbtomedia/nzbToMediaUtil.py index 51aa8395..f8915cbf 100644 --- a/nzbtomedia/nzbToMediaUtil.py +++ b/nzbtomedia/nzbToMediaUtil.py @@ -840,8 +840,17 @@ def find_imdbid(dirName, inputName): m = re.search('(tt\d{7})', file) if m: imdbid = m.group(1) - logger.info("Found imdbID [%s]" % imdbid) - return imdbid + logger.info("Found imdbID [%s] via file name" % imdbid) + return imdbid + if os.environ.has_key('NZBPR__DNZB_MOREINFO'): + dnzb_more_info=os.environ.get('NZBPR__DNZB_MOREINFO', '') + if dnzb_more_info != '': + regex = re.compile(r'^http://www.imdb.com/title/(tt[0-9]+)/$', re.IGNORECASE) + m = regex.match(dnzb_more_info) + if m: + imdbid = m.group(1) + logger.info("Found imdbID [%s] from DNZB-MoreInfo" % imdbid) + return imdbid logger.info('Searching IMDB for imdbID ...') guess = guessit.guess_movie_info(inputName) if guess: