From 4ee5f5f479f3b8ce929029ff45aa9d48c83c93be Mon Sep 17 00:00:00 2001 From: Yves Fischer Date: Tue, 23 Dec 2014 17:50:40 +0100 Subject: improve tags --- imdb-lookup/imdbinfo.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) (limited to 'imdb-lookup/imdbinfo.py') 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) -- cgit v1.2.1