Extra check for imdb file. close #273

couchpotato/core/helpers/variable.py

@@ -77,9 +77,9 @@ def cleanHost(host):
 
     return host
 
-def getImdb(txt):
+def getImdb(txt, check_inside = True):
 
-    if os.path.isfile(txt):
+    if check_inside and os.path.isfile(txt):
         output = open(txt, 'r')
         txt = output.read()
         output.close()

couchpotato/core/plugins/scanner/main.py

@@ -443,7 +443,7 @@ class Scanner(Plugin):
                 log.debug('Found movie via CP tag: %s' % cur_file)
                 break
 
-        # Check and see if nfo contains the imdb-id
+        # Check and see if nfo or filenames contains the imdb-id
         if not imdb_id:
             try:
                 for nfo_file in files['nfo']:
@@ -454,6 +454,16 @@ class Scanner(Plugin):
             except:
                 pass
 
+            try:
+                for filetype in files:
+                    for filetype_file in files[filetype]:
+                        imdb_id = getImdb(filetype_file, check_inside = False)
+                        if imdb_id:
+                            log.debug('Found movie via imdb in filename: %s' % nfo_file)
+                            break
+            except:
+                pass
+
         # Check if path is already in db
         if not imdb_id:
             db = get_session()