From c4abdfadc12ca2f43b3173c5d8fde0adcf49a747 Mon Sep 17 00:00:00 2001 From: clinton-hall Date: Wed, 29 Oct 2014 12:07:02 +1030 Subject: [PATCH] don't treat single file as dir. Fixes #596 --- nzbtomedia/nzbToMediaUtil.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/nzbtomedia/nzbToMediaUtil.py b/nzbtomedia/nzbToMediaUtil.py index a5fedb93..4871fbb2 100644 --- a/nzbtomedia/nzbToMediaUtil.py +++ b/nzbtomedia/nzbToMediaUtil.py @@ -899,12 +899,13 @@ 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] via file name" % imdbid) - return imdbid + if os.path.isdir(dirName): + for file in os.listdir(dirName): + m = re.search('(tt\d{7})', file) + if m: + imdbid = m.group(1) + 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 != '':