diff options
Diffstat (limited to 'imdb-lookup/imdbinfo.py')
| -rwxr-xr-x | imdb-lookup/imdbinfo.py | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py index 4c903ec..40be75a 100755 --- a/imdb-lookup/imdbinfo.py +++ b/imdb-lookup/imdbinfo.py @@ -316,13 +316,17 @@ def do_index(args, imdb_ids): if curfile.endswith("avi"): yield "AVI" for curdir in dirs: - if "german" in curdir: - yield "GERMAN" - if "1080" in curdir: - yield "1080p" - if "720p" in curdir: - yield "720p" - return list(set(find())) + tags = { "german" : "German", + "1080" : "1080p", "720" : "720p", + "bdrip" : "BlueRay", "blueray" : "BlueRay", + "hevc" : "HEVC" } + for (key, tag) in tags.items(): + if key in curdir.lower(): + yield tag + + l = list(set(find())) + l.sort() + return l def poster(imdb_id): data = db.poster_low(imdb_id) |