diff options
| -rwxr-xr-x | imdb-lookup/imdbrating.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/imdb-lookup/imdbrating.py b/imdb-lookup/imdbrating.py new file mode 100755 index 0000000..995324c --- /dev/null +++ b/imdb-lookup/imdbrating.py @@ -0,0 +1,41 @@ +#!/usr/bin/env python2 +# coding: utf-8 +# python3 works as well +import os +import sys +import shutil +import dbm + +try: + import tmdbsimple as tmdb +except ImportError as e: + print(u"Missing dependency: {0}".format(str(e))) + print(u"Install using system package manager or `pip install --user <module>`") + sys.exit(1) + +def read_key(): + if u"TMDB_KEY" in os.environ.keys(): + return os.environ[u"TMDB_KEY"] + if u"XDG_CONFIG_HOME" in os.environ.keys(): + cfg_home = os.environ[u"XDG_CONFIG_HOME"] + else: + cfg_home = os.path.join(os.path.expanduser(u"~"), ".config") + if os.path.exists(os.path.join(cfg_home, u"tmdbkey")): + return open(os.path.join(cfg_home, u"tmdbkey"), "r").read().strip() + if os.path.exists(os.path.join(os.path.expanduser(u"~"), ".tmdbkey")): + return open(os.path.join(os.path.expanduser(u"~"), ".tmdbkey")).read().strip() + raise Exception(u"No TheMovieDB Key defined. Set Env. var. TMDB_KEY or .tmdbkey file") + + +def get_rating(imdb_id): + return tmdb.Find(id=imdb_id).info(external_source="imdb_id")['movie_results'][0]['vote_average'] + +if __name__ == u"__main__": + tmdb.API_KEY = read_key() + imdb_id = sys.argv[1] + + db = dbm.open('cache', 'rw') + if imdb_id not in db.keys(): + db[imdb_id] = str(get_rating(imdb_id)) + print db[imdb_id] + db.close() |