diff --git a/stats.py b/stats.py index 794643e..12d691b 100644 --- a/stats.py +++ b/stats.py @@ -1,17 +1,17 @@ import json import sys -if len(sys.args) != 2: - print "usage: %s idsofcountry.json" % sys.args[0] +if len(sys.argv) != 2: + print "usage: %s idsofcountry.json" % sys.argv[0] sys.exit(1) idsofcountry = sys.argv[1] data = json.load(open(idsofcountry)) -mini_series = filter(lambda x: 'Mini-Series' in x[1], data) -tv_series = filter(lambda x: 'TV Series' in x[1], data) -tv_movies = filter(lambda x: 'TV Movie' in x[1], data) -tv_special = filter(lambda x: 'TV Special' in x[1], data) +mini_series = filter(lambda x: 'Mini-Series' in x['title'], data) +tv_series = filter(lambda x: 'TV Series' in x['title'], data) +tv_movies = filter(lambda x: 'TV Movie' in x['title'], data) +tv_special = filter(lambda x: 'TV Special' in x['title'], data) #cinema = set(data) - set(mini_series) - set(tv_series) - set(tv_movies)