-# -*- 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