Epub fixes.

[librarian.git] / src / librarian / elements / figures / ilustr.py

diff --git a/src/librarian/elements/figures/ilustr.py b/src/librarian/elements/figures/ilustr.py
index 143132e..c6105f4 100644 (file)
--- a/src/librarian/elements/figures/ilustr.py
+++ b/src/librarian/elements/figures/ilustr.py
@@ -1,12 +1,50 @@
+import six.moves
+from PIL import Image
 from ..base import WLElement
 
 
 class Ilustr(WLElement):
-    HTML_TAG = 'img'
+    EPUB_TAG = HTML_TAG = 'img'
 
     def get_html_attr(self, builder):
+        ## TODO: thumbnail.
+
+        url = six.moves.urllib.parse.urljoin(
+            builder.base_url,
+            self.get('src')
+        )
+        
+        imgfile = six.moves.urllib.request.urlopen(url)
+        img = Image.open(imgfile)
+        th_format, ext, media_type = {
+            'GIF': ('GIF', 'gif', 'image/gif'),
+            'PNG': ('PNG', 'png', 'image/png'),
+        }.get(img.format, ('JPEG', 'jpg', 'image/jpeg'))
+
+        width = 1200
+        if img.size[0] < width:
+            th = img
+        else:
+            th = img.resize((width, round(width * img.size[1] / img.size[0])))
+
+        buffer = six.BytesIO()
+        th.save(buffer, format=th_format)
+        imgfile.close()
+        file_name = 'image%d.%s' % (
+            builder.assign_image_number(),
+            ext
+        )
+
+        builder.add_file(
+            content=buffer.getvalue(),
+            file_name=file_name,
+            media_type=media_type,
+        )
+        
         return {
-            'src': builder.image_location + self.attrib['src'],
-            'alt': self.attr['alt'],
-            'title': self.attr['alt'],
+            'src': file_name,
+            'alt': self.attrib['alt'],
+            'title': self.attrib['alt'],
         }
+
+    get_epub_attr = get_html_attr