generate shiny html stuff with pictures

1 files changed, 84 insertions, 30 deletions

diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py
index 2835eb8..7d00a72 100755
--- a/imdb-lookup/imdbinfo.py
+++ b/imdb-lookup/imdbinfo.py
@@ -6,11 +6,13 @@ import sys
 import re
 import dbm
 import json
+import base64
 import argparse
 import math
 
 try:
     import tmdbsimple as tmdb
+    import requests
 except ImportError as e:
     print(u"Missing dependency: {0}".format(str(e)))
     print(u"Install using system package manager or `pip install --user <module>`")
@@ -31,18 +33,20 @@ def read_key():
         
 class TMDBCache(object):
     def __enter__(self):
-        self.db = dbm.open(self._get_db_filename(),"c")
+        self.db = dbm.open(self._get_db_filename("tmdbmovie.dbm"),"c")
+        self.db_images = dbm.open(self._get_db_filename("tmdbposter.dbm"), "c")
         return self
 
     def __exit__(self, type, value, traceback):
         self.db.close()
+        self.db_images.close()
 
-    def _get_db_filename(self):
+    def _get_db_filename(self, name):
         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, "tmdbmovie.dbm")
+        return os.path.join(cachedir, name)
 
     def _cache(self, key, callable_func):
         if key not in self.db:
@@ -75,15 +79,31 @@ class TMDBCache(object):
             if key in self.db:
                 print("Remove {}".format(key))
                 del self.db[key]
-        
+
+    def poster(self, poster_path, format="w185"):
+        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:
+            r = requests.get(url)
+            self.db_images[key] = r.content
+            self.db_images[keyContentType] = r.headers['content-type']
+        return (self.db_images[keyContentType], self.db_images[key])
+
+    def poster_base64(self, poster_path, format="w185"):
+        contentType, data = self.poster(poster_path, format)
+        data64 = "".join(map(lambda c: isinstance(c,int) and chr(c) or c, 
+                    filter(lambda c: c!='\n', base64.encodestring(data))))
+        return "data:{};base64,{}".format(contentType, data64)
+ 
 def do_aka(args, imdb_ids):
     with TMDBCache() as tmdbcache:
-        for imdb_id in imdb_ids:
+        for (filename, imdb_id) in imdb_ids:
             print(tmdbcache.alternative_title(imdb_id, locale=args.lang))
 
 def do_data(args, imdb_ids):
     with TMDBCache() as tmdbcache:
-        for imdb_id in imdb_ids:
+        for (filename, imdb_id) in imdb_ids:
             selected_properties = ["imdb_id", "revenue", "vote_average", "vote_count", "runtime", "budget", "vote_avarage", "release_date", "popularity", ]
             kv = map(lambda kv: u"{}={}".format(*kv), 
                     filter(lambda kv: kv[0] in selected_properties,
@@ -92,19 +112,19 @@ def do_data(args, imdb_ids):
 
 def do_year(args, imdb_ids):
     with TMDBCache() as tmdbcache:
-        for imdb_id in imdb_ids:
+        for (filename, imdb_id) in imdb_ids:
             print(tmdbcache.infos(imdb_id)["release_date"].split("-")[0])
 
 def do_prune(args, imdb_ids):
     with TMDBCache() as tmdbcache:
-        for imdb_id in imdb_ids:
+        for (filename, imdb_id) in imdb_ids:
             tmdbcache.prune(imdb_id)
 
 def do_rating(args, imdb_ids):
     """Calculates a rating based on vote_average and vote_count"""
     with TMDBCache() as tmdbcache:
         infos = list(filter(lambda i: "vote_average" in i and "vote_count" in i, 
-                        map(lambda imdb_id: tmdbcache.infos(imdb_id), imdb_ids)))
+                        map(lambda filename,imdb_id: tmdbcache.infos(imdb_id), imdb_ids)))
         maxvotes = max(map(lambda i: i["vote_count"], infos))
         for info in infos:
             f = math.sin(math.pi * ( info["vote_average"]/10.0 ) )
@@ -113,29 +133,59 @@ def do_rating(args, imdb_ids):
             print("{rating:.02f} {imdb_id} {title:30s} avg={vote_average:.1f} count={vote_count:.0f}".format(**info))
 
 
+def do_index(args, imdb_ids):
+    """creates a index website"""
+    def asBase64(poster):
+        contentType, data = poster
+        data64 = filter(lambda c: c!='\n', base64.encodestring(data))
+        return "data:{};base64,{}".format(contentType, data64)
+    try:
+        from jinja2 import Template
+    except ImportError:
+        print("Failed to import jinja2 library for html-templating")
+        sys.exit(1)
+    template_file = os.path.join(os.path.dirname(__file__), "index.jinja2.html")
+    template = Template(open(template_file).read())
+    with TMDBCache() as tmdbcache:
+        #infos = list(map(tmdbcache.infos, imdb_ids))
+        #posters = list(map(asBase64, map(tmdbcache.poster, map(lambda info: info['poster_path'], infos))))
+        mapping = {
+            'input' : imdb_ids,
+            'tmdbcache' : tmdbcache,
+            'title' : 'Movie overview',
+        }
+        assert not os.path.exists("index.html"), "index.html already exists"
+        open("index.html", "w").write(template.render(mapping))
 
 class HelpAction(argparse._HelpAction):
-
     def __call__(self, parser, namespace, values, option_string=None):
-        parser.print_help()
-        print("")
-
-        # retrieve subparsers from parser
+        formatter = parser._get_formatter()
+        formatter.add_usage(parser.usage, parser._actions, parser._mutually_exclusive_groups)
+        formatter.add_text(parser.description)
+
+        formatter.start_section(parser._optionals.title)
+        formatter.add_text(parser._optionals.description)
+        formatter.add_arguments(parser._optionals._group_actions)
+        formatter.end_section()
+    
         subparsers_actions = [
             action for action in parser._actions
             if isinstance(action, argparse._SubParsersAction)]
-        # there will probably only be one subparser_action,
-        # but better save than sorry
+
         for subparsers_action in subparsers_actions:
             # get all subparsers and print help
-            for choice, subparser in subparsers_action.choices.items():
-                formatter = argparse.HelpFormatter(prog=choice)
-                print("subcommand {}:".format(choice))
-                for action_group in subparser._action_groups:
-                    formatter.add_arguments(action_group._group_actions)
-                print(formatter.format_help())
-
-        parser.exit()
+            subparsers = subparsers_action.choices
+            for subaction in subparsers_action._get_subactions():
+                subparser = subparsers[subaction.dest]
+                formatter.start_section("{} {} {}".format(formatter._prog, subaction.dest, 
+                            formatter._format_actions_usage(subparser._actions, [])))
+                formatter.add_text(subaction.help)
+                formatter.add_arguments(subparser._positionals._group_actions)
+                formatter.add_arguments(subparser._optionals._group_actions)
+                formatter.end_section()
+
+        print(formatter.format_help())
+        parser.exit(0)
 
 if __name__ == u"__main__":
     tmdb.API_KEY = read_key()
@@ -145,29 +195,33 @@ if __name__ == u"__main__":
     parser.add_argument("-h", action=argparse._HelpAction, help="Display short help")
 
     subparsers = parser.add_subparsers()
-    parser_aka = subparsers.add_parser("aka", add_help=False)
+    parser_aka = subparsers.add_parser("aka", add_help=False, help="Print alternative title in other languages")
     parser_aka.add_argument("--lang", help="Language code (default 'DE')")
     parser_aka.set_defaults(func=do_aka)
     parser_aka.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
-    parser_data = subparsers.add_parser("data", add_help=False)
+    parser_data = subparsers.add_parser("data", add_help=False, help="Print all available data")
     parser_data.set_defaults(func=do_data)
     parser_data.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
-    parser_year = subparsers.add_parser("year", add_help=False)
+    parser_year = subparsers.add_parser("year", add_help=False, help="Print only the release year")
     parser_year.set_defaults(func=do_year)
     parser_year.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
-    parser_prune = subparsers.add_parser("prune", add_help=False)
+    parser_prune = subparsers.add_parser("prune", add_help=False, help="Delete cache entries")
     parser_prune.set_defaults(func=do_prune)
     parser_prune.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
-    parser_rating = subparsers.add_parser("rating", add_help=False)
+    parser_rating = subparsers.add_parser("rating", add_help=False, help="Print movie ratings")
     parser_rating.set_defaults(func=do_rating)
     parser_rating.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
 
+    parser_index = subparsers.add_parser("index", add_help=False, help="Generate index.html file")
+    parser_index.set_defaults(func=do_index)
+    parser_index.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids")
+
     args = parser.parse_args(sys.argv[1:])
 
-    ids = map(lambda filename: (lambda x: x.groups()[0] if x else None)(re.match(".*#(tt[0-9]{7}).*", filename)),
+    ids = map(lambda filename: (lambda x: (filename, x.groups()[0]) if x else None)(re.match(".*#(tt[0-9]{7}).*", filename)),
                 args.files[0])
     args.func(args, filter(lambda i: i is not None, ids))