Add support for addition cover logos.

[librarian.git] / librarian / embeds / latex.py

diff --git a/librarian/embeds/latex.py b/librarian/embeds/latex.py
index e10d165..8425d03 100644 (file)
--- a/librarian/embeds/latex.py
+++ b/librarian/embeds/latex.py
@@ -1,21 +1,25 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+

 import os
 import shutil
 from subprocess import call, PIPE
 from tempfile import mkdtemp
 from librarian import get_resource
 import os
 import shutil
 from subprocess import call, PIPE
 from tempfile import mkdtemp
 from librarian import get_resource

-from . import DataEmbed, create_embed, downgrades_to, converts_to
+from . import DataEmbed, create_embed, downgrades_to
+

 
 class LaTeX(DataEmbed):
     @downgrades_to('image/png')
     def to_png(self):
 
 class LaTeX(DataEmbed):
     @downgrades_to('image/png')
     def to_png(self):

-        tmpl = open(get_resource('res/embeds/latex/template.tex')).read().decode('utf-8')
+        tmpl = open(get_resource('res/embeds/latex/template.tex'), 'rb').read().decode('utf-8')

         tempdir = mkdtemp('-librarian-embed-latex')
         fpath = os.path.join(tempdir, 'doc.tex')
         tempdir = mkdtemp('-librarian-embed-latex')
         fpath = os.path.join(tempdir, 'doc.tex')

-        with open(fpath, 'w') as f:
+        with open(fpath, 'wb') as f:

             f.write((tmpl % {'code': self.data}).encode('utf-8'))
         call(['xelatex', '-interaction=batchmode', '-output-directory', tempdir, fpath], stdout=PIPE, stderr=PIPE)
         call(['convert', '-density', '150', os.path.join(tempdir, 'doc.pdf'), '-trim',
             f.write((tmpl % {'code': self.data}).encode('utf-8'))
         call(['xelatex', '-interaction=batchmode', '-output-directory', tempdir, fpath], stdout=PIPE, stderr=PIPE)
         call(['convert', '-density', '150', os.path.join(tempdir, 'doc.pdf'), '-trim',

-                os.path.join(tempdir, 'doc.png')])
-        pngdata = open(os.path.join(tempdir, 'doc.png')).read()
+             os.path.join(tempdir, 'doc.png')])
+        pngdata = open(os.path.join(tempdir, 'doc.png'), 'rb').read()

         shutil.rmtree(tempdir)
         return create_embed('image/png', data=pngdata)
         shutil.rmtree(tempdir)
         return create_embed('image/png', data=pngdata)