diff options
Diffstat (limited to 'imdb-lookup/imdbinfo.py')
| -rwxr-xr-x | imdb-lookup/imdbinfo.py | 9 |
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) |