logging

1 files changed, 9 insertions, 0 deletions

diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py
index 54c4816..70d2790 100755
--- a/imdb-lookup/imdbinfo.py
+++ b/imdb-lookup/imdbinfo.py
@@ -9,6 +9,7 @@ import json
 import base64
 import argparse
 import math
+import logging
 
 try:
     import tmdbsimple as tmdb
@@ -50,12 +51,14 @@ class TMDBCache(object):
 
     def _cache(self, key, callable_func):
         if key not in self.db:
+            logging.info("Load value for key %s", key)
             self.db[key] = json.dumps(callable_func())
         d = self.db[key].decode('utf-8')
         return json.loads(d)
 
     def infos(self, movie_id):
         try:
+            logging.debug("movie %s", movie_id)
             return self._cache(movie_id + "movies.info", tmdb.Movies(movie_id).info)
         except Exception as e:
             raise Exception("Failed to query movie with id {id}: {reason}".format(id=movie_id, reason=str(e)))
@@ -81,10 +84,12 @@ class TMDBCache(object):
                 del self.db[key]
 
     def poster(self, poster_path, format="w185"):
+        logging.debug("poster %s", poster_path)
         key = "poster_{}_{}".format(format, poster_path)
         keyContentType = "poster_{}_{}_content_type".format(format, poster_path)
         url = "http://image.tmdb.org/t/p/{}/{}".format(format, poster_path)
         if key not in self.db_images:
+            logging.info("Load poster %s", url)
             r = requests.get(url)
             self.db_images[key] = r.content
             self.db_images[keyContentType] = r.headers['content-type']
@@ -211,6 +216,8 @@ if __name__ == u"__main__":
 
     parser = argparse.ArgumentParser(description="get movie data", add_help=False)
     parser.add_argument("--help", action=HelpAction, help="Display full help")
+    parser.add_argument("--log-level", action='store', type=int, 
+            help="Set log level (CRITICAL=50,ERROR=40,WARNING=30,INFO=20,DEBUG=10,NOTSET=0")
     parser.add_argument("-h", action=argparse._HelpAction, help="Display short help")
 
     subparsers = parser.add_subparsers()
@@ -240,6 +247,8 @@ if __name__ == u"__main__":
     parser_index.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
     args = parser.parse_args(sys.argv[1:])
+    if "log_level" in args:
+        logging.basicConfig(level=args.log_level)
 
     ids = map(lambda filename: (lambda x: (filename, x.groups()[0]) if x else None)(re.match(".*#(tt[0-9]{7}).*", filename)),
                 args.files[0])