more compact

1 files changed, 9 insertions, 11 deletions

diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py
index eeea947..4290dac 100755
--- a/imdb-lookup/imdbinfo.py
+++ b/imdb-lookup/imdbinfo.py
@@ -17,7 +17,7 @@ from io import BytesIO
 from urllib.parse import quote as urlencode
 
 try:
-    from PIL import Image
+    from PIL import Image, ImageFilter
     import tmdbsimple as tmdb
     import requests
 except ImportError as e:
@@ -126,12 +126,11 @@ class TMDBCache(object):
             return None
         contentType, data = p
         image = Image.open(BytesIO(data))
+        image = image.filter(ImageFilter.GaussianBlur(radius=1))
         buf = BytesIO()
-        image.save(buf, "JPEG", quality=10, optimize=True)
-        data64 = base64.encodestring(buf.getvalue())
-        data64 = "".join(map(lambda c: isinstance(c, int) and chr(c) or c,
-                             filter(lambda c: c != '\n', data64)))
-        return "data:{};base64,{}".format(contentType, data64)
+        image.save(buf, "JPEG", quality=15, optimize=True)
+        data64 = base64.encodestring(buf.getvalue()).translate(None, b'\n')
+        return "data:{};base64,{}".format("image/jpeg", data64.decode())
 
     def _imdb_request(self, path, query):
         # see also https://github.com/richardasaurus/imdb-pie
@@ -287,10 +286,6 @@ def do_index(args, imdb_ids):
                 if sum(map(curfile.endswith, valid_extensions)):
                     yield os.path.join(root, curfile)
 
-    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:
@@ -309,12 +304,15 @@ def do_index(args, imdb_ids):
             'int': int,
             'listMovieFiles': listMovieFiles,
             'math': math,
+            'pathjoin': os.path.join,
         }
         assert not os.path.exists("index.html"), "index.html already exists"
         stream = template.generate(mapping)
         outfile = open("index.html", "wb")
         for output in stream:
-            outfile.write(output.encode('utf-8'))
+            out = output.strip()
+            out = re.sub(" {2,}", " ", out)
+            outfile.write(out.encode('utf-8'))
 
 
 class HelpAction(argparse._HelpAction):