From 88e962a3a0f87b46a6716f50c7fd99c16dd229f7 Mon Sep 17 00:00:00 2001 From: clinton-hall Date: Sat, 9 Aug 2014 10:07:23 +0930 Subject: [PATCH] check for imdb id in file names. Helps #526 --- nzbtomedia/nzbToMediaUtil.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nzbtomedia/nzbToMediaUtil.py b/nzbtomedia/nzbToMediaUtil.py index 254c9cee..51aa8395 100644 --- a/nzbtomedia/nzbToMediaUtil.py +++ b/nzbtomedia/nzbToMediaUtil.py @@ -836,7 +836,12 @@ def find_imdbid(dirName, inputName): imdbid = m.group(1) logger.info("Found imdbID [%s]" % imdbid) return imdbid - + for file in os.listdir(dirName): + m = re.search('(tt\d{7})', file) + if m: + imdbid = m.group(1) + logger.info("Found imdbID [%s]" % imdbid) + return imdbid logger.info('Searching IMDB for imdbID ...') guess = guessit.guess_movie_info(inputName) if guess: