list files

1 files changed, 8 insertions, 1 deletions

diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py
index caf00a4..259b2c6 100755
--- a/imdb-lookup/imdbinfo.py
+++ b/imdb-lookup/imdbinfo.py
@@ -140,7 +140,7 @@ class Protector(object):
         if callable(attr):
             return protected
         else:
-            return o
+            return attr
 
 def do_aka(args, imdb_ids):
     with TMDBCache() as tmdbcache:
@@ -178,6 +178,12 @@ def do_rating(args, imdb_ids):
 
 def do_index(args, imdb_ids):
     """creates a index website"""
+    def listMovieFiles(path):
+        for root, dirs, files in os.walk(path):
+            for curfile in files:
+                if sum(map(curfile.endswith, [".mkv", ".avi", ".mov", ".mp4"])):
+                    yield os.path.join(root, curfile)
+
     def asBase64(poster):
         contentType, data = poster
         data64 = filter(lambda c: c!='\n', base64.encodestring(data))
@@ -197,6 +203,7 @@ def do_index(args, imdb_ids):
             'title' : 'Movie overview',
             'urlencode' : urlencode,
             'int' : int,
+            'listMovieFiles' : listMovieFiles,
         }
         assert not os.path.exists("index.html"), "index.html already exists"
         stream = template.generate(mapping)