mirror of https://github.com/nanoy42/coope
10 changed files with 155 additions and 1 deletions
@ -0,0 +1,40 @@ |
|||
|
|||
import os |
|||
from subprocess import PIPE, run |
|||
import tempfile |
|||
|
|||
from django.template.loader import get_template |
|||
|
|||
from django_tex.exceptions import TexError |
|||
from django.conf import settings |
|||
|
|||
DEFAULT_INTERPRETER = 'lualatex' |
|||
|
|||
def run_tex(source): |
|||
with tempfile.TemporaryDirectory() as tempdir: |
|||
filename = os.path.join(tempdir, 'texput.tex') |
|||
with open(filename, 'x', encoding='utf-8') as f: |
|||
f.write(source) |
|||
latex_interpreter = getattr(settings, 'LATEX_INTERPRETER', DEFAULT_INTERPRETER) |
|||
latex_command = f'cd "{tempdir}" && {latex_interpreter} -interaction=batchmode {os.path.basename(filename)}' |
|||
process = run(latex_command, shell=True, stdout=PIPE, stderr=PIPE) |
|||
try: |
|||
if process.returncode == 1: |
|||
with open(os.path.join(tempdir, 'texput.log'), encoding='utf8') as f: |
|||
log = f.read() |
|||
raise TexError(log=log, source=source) |
|||
with open(os.path.join(tempdir, 'texput.pdf'), 'rb') as pdf_file: |
|||
pdf = pdf_file.read() |
|||
except FileNotFoundError: |
|||
if process.stderr: |
|||
raise Exception(process.stderr.decode('utf-8')) |
|||
raise |
|||
return pdf |
|||
|
|||
def compile_template_to_pdf(template_name, context): |
|||
source = render_template_with_context(template_name, context) |
|||
return run_tex(source) |
|||
|
|||
def render_template_with_context(template_name, context): |
|||
template = get_template(template_name, using='tex') |
|||
return template.render(context) |
|||
@ -0,0 +1,11 @@ |
|||
|
|||
from django.template.backends.jinja2 import Jinja2 |
|||
|
|||
class TeXEngine(Jinja2): |
|||
app_dirname = 'templates' |
|||
|
|||
def __init__(self, params): |
|||
default_environment = 'django_tex.environment.environment' |
|||
if 'environment' not in params['OPTIONS'] or not params['OPTIONS']['environment']: |
|||
params['OPTIONS']['environment'] = default_environment |
|||
super().__init__(params) |
|||
@ -0,0 +1,16 @@ |
|||
|
|||
from jinja2 import Environment |
|||
|
|||
from django.template.defaultfilters import register |
|||
|
|||
from django_tex.filters import FILTERS as tex_specific_filters |
|||
|
|||
# Django's built-in filters ... |
|||
filters = register.filters |
|||
# ... updated with tex specific filters |
|||
filters.update(tex_specific_filters) |
|||
|
|||
def environment(**options): |
|||
env = Environment(**options) |
|||
env.filters = filters |
|||
return env |
|||
@ -0,0 +1,40 @@ |
|||
|
|||
import re |
|||
|
|||
def prettify_message(message): |
|||
''' |
|||
Helper methods that removes consecutive whitespaces and newline characters |
|||
''' |
|||
# Replace consecutive whitespaces with a single whitespace |
|||
message = re.sub(r'[ ]{2,}', ' ', message) |
|||
# Replace consecutive newline characters, optionally separated by whitespace, with a single newline |
|||
message = re.sub(r'([\r\n][ \t]*)+', '\n', message) |
|||
return message |
|||
|
|||
def tokenizer(code): |
|||
token_specification = [ |
|||
('ERROR', r'\! (?:.+[\r\n])+[\r\n]+'), |
|||
('WARNING', r'latex warning.*'), |
|||
('NOFILE', r'no file.*') |
|||
] |
|||
token_regex = '|'.join('(?P<{}>{})'.format(label, regex) for label, regex in token_specification) |
|||
for m in re.finditer(token_regex, code, re.IGNORECASE): |
|||
token_dict = dict(type=m.lastgroup, message=prettify_message(m.group())) |
|||
yield token_dict |
|||
|
|||
class TexError(Exception): |
|||
|
|||
def __init__(self, log, source): |
|||
self.log = log |
|||
self.source = source |
|||
self.tokens = list(tokenizer(self.log)) |
|||
self.message = self.get_message() |
|||
|
|||
def get_message(self): |
|||
for token in self.tokens: |
|||
if token['type'] == 'ERROR': |
|||
return token['message'] |
|||
return 'No error message found in log' |
|||
|
|||
def __str__(self): |
|||
return self.message |
|||
@ -0,0 +1,9 @@ |
|||
from django.utils.formats import localize_input |
|||
|
|||
def do_linebreaks(value): |
|||
return value.replace('\n', '\\\\\n') |
|||
|
|||
FILTERS = { |
|||
'localize': localize_input, |
|||
'linebreaks': do_linebreaks |
|||
} |
|||
@ -0,0 +1,19 @@ |
|||
from django.db import models |
|||
from django.core.exceptions import ValidationError |
|||
from django.template import TemplateDoesNotExist |
|||
from django.utils.translation import ugettext_lazy as _ |
|||
from django.template.loader import get_template |
|||
|
|||
def validate_template_path(name): |
|||
try: |
|||
get_template(name, using='tex') |
|||
except TemplateDoesNotExist: |
|||
raise ValidationError(_('Template not found.')) |
|||
|
|||
class TeXTemplateFile(models.Model): |
|||
|
|||
title = models.CharField(max_length=255) |
|||
name = models.CharField(max_length=255, validators=[validate_template_path,]) |
|||
|
|||
class Meta: |
|||
abstract = True |
|||
@ -0,0 +1,17 @@ |
|||
|
|||
from django.http import HttpResponse |
|||
|
|||
from django_tex.core import compile_template_to_pdf |
|||
|
|||
class PDFResponse(HttpResponse): |
|||
|
|||
def __init__(self, content, filename=None): |
|||
super(PDFResponse, self).__init__(content_type='application/pdf') |
|||
self['Content-Disposition'] = 'filename="{}"'.format(filename) |
|||
self.write(content) |
|||
|
|||
|
|||
def render_to_pdf(request, template_name, context=None, filename=None): |
|||
# Request is not needed and only included to make the signature conform to django's render function |
|||
pdf = compile_template_to_pdf(template_name, context) |
|||
return PDFResponse(pdf, filename=filename) |
|||
Loading…
Reference in new issue