Code layout change.

[wolnelektury.git] / src / reporting / utils.py

diff --git a/src/reporting/utils.py b/src/reporting/utils.py
new file mode 100755 (executable)
index 0000000..8ecb9b0
--- /dev/null
+++ b/src/reporting/utils.py
@@ -0,0 +1,123 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from errno import ENOENT
+import os
+import os.path
+from django.conf import settings
+import logging
+from django.http import HttpResponse
+
+logger = logging.getLogger(__name__)
+
+
+def render_to_pdf(output_path, template, context=None, add_files=None):
+    """Renders a TeXML document into a PDF file.
+
+    :param str output_path: is where the PDF file should go
+    :param str template: is a TeXML template path
+    :param context: is context for rendering the template
+    :param dict add_files: a dictionary of additional files XeTeX will need
+    """
+
+    from StringIO import StringIO
+    import shutil
+    from tempfile import mkdtemp
+    import subprocess
+    import Texml.processor
+    from django.template.loader import render_to_string
+
+    rendered = render_to_string(template, context)
+    texml = StringIO(rendered.encode('utf-8'))
+    tempdir = mkdtemp(prefix="render_to_pdf-")
+    tex_path = os.path.join(tempdir, "doc.tex")
+    with open(tex_path, 'w') as tex_file:
+        Texml.processor.process(texml, tex_file, encoding="utf-8")
+
+    if add_files:
+        for add_name, src_file in add_files.items():
+            add_path = os.path.join(tempdir, add_name)
+            if hasattr(src_file, "read"):
+                with open(add_path, 'w') as add_file:
+                    add_file.write(add_file.read())
+            else:
+                shutil.copy(src_file, add_path)
+
+    cwd = os.getcwd()
+    os.chdir(tempdir)
+    try:
+        subprocess.check_call(['xelatex', '-interaction=batchmode', tex_path],
+            stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        try:
+            os.makedirs(os.path.dirname(output_path))
+        except:
+            pass
+        shutil.move(os.path.join(tempdir, "doc.pdf"), output_path)
+    finally:
+        os.chdir(cwd)
+        shutil.rmtree(tempdir)
+
+
+def render_to_csv(output_path, template, context=None, add_files=None):
+    """Renders a TeXML document into a PDF file.
+
+    :param str output_path: is where the PDF file should go
+    :param str template: is a TeXML template path
+    :param context: is context for rendering the template
+    :param dict add_files: a dictionary of additional files XeTeX will need
+    """
+
+    from django.template.loader import render_to_string
+
+    try:
+        os.makedirs(os.path.dirname(output_path))
+    except:
+        pass
+
+    rendered = render_to_string(template, context)
+    with open(output_path, 'w') as csv_file:
+        csv_file.write(rendered.encode('utf-8'))
+
+
+def read_chunks(f, size=8192):
+    chunk = f.read(size)
+    while chunk:
+        yield chunk
+        chunk = f.read(size)
+
+
+def generated_file_view(file_name, mime_type, send_name=None, signals=None):
+    file_path = os.path.join(settings.MEDIA_ROOT, file_name)
+    if send_name is None:
+        send_name = os.path.basename(file_name)
+
+    def signal_handler(*args, **kwargs):
+        try:
+            os.unlink(file_path)
+        except OSError as oe:
+            if oe.errno != ENOENT:
+                raise oe
+
+    if signals:
+        for signal in signals:
+            signal.connect(signal_handler, weak=False)
+
+    def decorator(func):
+        def view(request, *args, **kwargs):
+            if not os.path.exists(file_path):
+                func(file_path, *args, **kwargs)
+
+            if hasattr(send_name, "__call__"):
+                name = send_name()
+            else:
+                name = send_name
+
+            response = HttpResponse(content_type=mime_type)
+            response['Content-Disposition'] = 'attachment; filename=%s' % name
+            with open(file_path) as f:
+                for chunk in read_chunks(f):
+                    response.write(chunk)
+            return response
+        return view
+    return decorator