Preliminary math and tables support.

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

diff --git a/librarian/embeds/latex.py b/librarian/embeds/latex.py
new file mode 100644 (file)
index 0000000..e10d165
--- /dev/null
+++ b/librarian/embeds/latex.py
@@ -0,0 +1,21 @@
+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
+
+class LaTeX(DataEmbed):
+    @downgrades_to('image/png')
+    def to_png(self):
+        tmpl = open(get_resource('res/embeds/latex/template.tex')).read().decode('utf-8')
+        tempdir = mkdtemp('-librarian-embed-latex')
+        fpath = os.path.join(tempdir, 'doc.tex')
+        with open(fpath, 'w') 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',
+                os.path.join(tempdir, 'doc.png')])
+        pngdata = open(os.path.join(tempdir, 'doc.png')).read()
+        shutil.rmtree(tempdir)
+        return create_embed('image/png', data=pngdata)