links to movies sites

1 files changed, 18 insertions, 12 deletions

diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py
index 7b23356..caf00a4 100755
--- a/imdb-lookup/imdbinfo.py
+++ b/imdb-lookup/imdbinfo.py
@@ -6,15 +6,16 @@ import sys
 import re
 import dbm
 import json
+import time
 import base64
 import argparse
 import math
 import logging
-from StringIO import StringIO
-try:
-    from urllib.parse import quote as urlencode
-except:
-    from urllib import quote as urlencode
+try: from StringIO import StringIO
+except: from io import StringIO
+
+try: from urllib.parse import quote as urlencode
+except: from urllib import quote as urlencode
 
 try:
     from PIL import Image
@@ -112,7 +113,16 @@ class TMDBCache(object):
         data64 = "".join(map(lambda c: isinstance(c,int) and chr(c) or c, 
                     filter(lambda c: c!='\n', base64.encodestring(buf.getvalue()))))
         return "data:{};base64,{}".format(contentType, data64)
- 
+
+
+def weight_rating(infos):
+    """ add 'rating' to all infos"""
+    maxvotes = max(map(lambda i: i["vote_count"], infos))
+    for info in infos:
+        f = math.sin(math.pi * ( info["vote_average"]/10.0 ) )
+        d = (float(info["vote_count"]) / maxvotes) - 0.5
+        info['rating'] = info["vote_average"] + 2 * d * f
+    return infos
 
 class Protector(object):
     def __init__(self, child):
@@ -161,11 +171,8 @@ def do_rating(args, imdb_ids):
     with TMDBCache() as tmdbcache:
         infos = list(filter(lambda i: "vote_average" in i and "vote_count" in i, 
                         map(lambda fid: tmdbcache.infos(fid[1]), imdb_ids)))
-        maxvotes = max(map(lambda i: i["vote_count"], infos))
+        weight_rating(infos)
         for info in infos:
-            f = math.sin(math.pi * ( info["vote_average"]/10.0 ) )
-            d = (float(info["vote_count"]) / maxvotes) - 0.5
-            info['rating'] = info["vote_average"] + 2 * d * f
             print(u"{rating:.02f} {imdb_id} {title:30s} avg={vote_average:.1f} count={vote_count:.0f}".format(**info))
 
 
@@ -183,9 +190,8 @@ def do_index(args, imdb_ids):
     template_file = os.path.join(os.path.dirname(__file__), "index.jinja2.html")
     template = Template(open(template_file, "rb").read().decode('utf-8'))
     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 = {
+            'gmtime' : time.gmtime(),
             'input' : imdb_ids,
             'tmdbcache' : Protector(tmdbcache),
             'title' : 'Movie overview',