summaryrefslogtreecommitdiff
path: root/imdb-lookup/imdbrating.py
diff options
context:
space:
mode:
authorYves Fischer <yvesf-git@xapek.org>2014-11-17 23:37:07 +0100
committerYves Fischer <yvesf-git@xapek.org>2014-11-17 23:37:07 +0100
commitebe3644acb28a51171fc0dad77ccd1bf59ed4e58 (patch)
treef73e2859e74463b7717fd6cca00c10483ab5471b /imdb-lookup/imdbrating.py
parent9dd1bd8f512f200091f3929b1eb846d601aa17ed (diff)
downloadscripts-ebe3644acb28a51171fc0dad77ccd1bf59ed4e58.tar.gz
scripts-ebe3644acb28a51171fc0dad77ccd1bf59ed4e58.zip
delete imdbrating.py use imdbinfo.py rating subcommand
Diffstat (limited to 'imdb-lookup/imdbrating.py')
-rwxr-xr-ximdb-lookup/imdbrating.py60
1 files changed, 0 insertions, 60 deletions
diff --git a/imdb-lookup/imdbrating.py b/imdb-lookup/imdbrating.py
deleted file mode 100755
index da40ed5..0000000
--- a/imdb-lookup/imdbrating.py
+++ /dev/null
@@ -1,60 +0,0 @@
-#!/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_db_filename():
- if u"XDG_CACHE_HOME" in os.environ.keys():
- cachedir = os.environ["XDG_CACHE_HOME"]
- else:
- cachedir = os.path.join(os.path.expanduser(u"~"), ".cache")
- return os.path.join(cachedir, "imdbrating.dbm")
-
-def get_rating(imdb_id):
- info = tmdb.Find(id=imdb_id).info(external_source="imdb_id")
- if 'movie_results' in info and len(info['movie_results']) == 1 and 'vote_average' in info['movie_results'][0]:
- return info['movie_results'][0]['vote_average']
- return 0
-
-def print_ratings(files, db):
- for filename in files:
- if os.path.basename(filename).startswith('.'):
- continue
- elif os.path.isdir(filename):
- print_ratings(map(lambda p: os.path.join(filename, p), os.listdir(filename)), db)
- imdb_id = os.path.basename(filename).split('#')[-1]
- if imdb_id.startswith('tt'):
- if imdb_id not in db.keys() or db[imdb_id] == 0:
- db[imdb_id] = str(get_rating(imdb_id))
- print "{rating} {filename}".format(rating=db[imdb_id], filename=filename)
-
-if __name__ == u"__main__":
- tmdb.API_KEY = read_key()
- db = dbm.open(get_db_filename(), 'rw')
- files = sum([os.listdir(x) if os.path.isdir(x) else [x] for x in sys.argv[1:]],[])
- print_ratings(files, db)
- db.close()
-