8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-12-23 15:33:45 +00:00
re2o/cotisations/tex.py

87 lines
3 KiB
Python
Raw Normal View History

2017-01-15 23:01:18 +00:00
# Re2o est un logiciel d'administration développé initiallement au rezometz. Il
# se veut agnostique au réseau considéré, de manière à être installable en
# quelques clics.
#
# Copyright © 2017 Gabriel Détraz
# Copyright © 2017 Goulven Kermarec
# Copyright © 2017 Augustin Lemesle
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
"""tex.py
Module in charge of rendering some LaTex templates.
Used to generated PDF invoice.
"""
2017-01-15 23:01:18 +00:00
2018-04-14 13:39:51 +00:00
import tempfile
from subprocess import Popen, PIPE
import os
from datetime import datetime
2016-07-09 15:16:44 +00:00
from django.template.loader import get_template
2017-11-14 19:00:24 +00:00
from django.template import Context
from django.http import HttpResponse
2016-07-09 15:16:44 +00:00
from django.conf import settings
2017-11-14 19:00:24 +00:00
from django.utils.text import slugify
2016-07-09 15:16:44 +00:00
TEMP_PREFIX = getattr(settings, 'TEX_TEMP_PREFIX', 'render_tex-')
CACHE_PREFIX = getattr(settings, 'TEX_CACHE_PREFIX', 'render-tex')
CACHE_TIMEOUT = getattr(settings, 'TEX_CACHE_TIMEOUT', 86400) # 1 day
2018-04-14 13:39:51 +00:00
def render_invoice(_request, ctx={}):
"""
Render an invoice using some available information such as the current
date, the user, the articles, the prices, ...
"""
2017-11-14 19:00:24 +00:00
filename = '_'.join([
2018-04-13 18:58:29 +00:00
'invoice',
2018-04-14 13:39:51 +00:00
slugify(ctx.get('asso_name', "")),
slugify(ctx.get('recipient_name', "")),
str(ctx.get('DATE', datetime.now()).year),
str(ctx.get('DATE', datetime.now()).month),
str(ctx.get('DATE', datetime.now()).day),
2017-11-14 19:00:24 +00:00
])
2018-04-14 13:39:51 +00:00
r = render_tex(_request, 'cotisations/factures.tex', ctx)
2018-04-13 18:58:29 +00:00
r['Content-Disposition'] = 'attachment; filename="{name}.pdf"'.format(
name=filename
)
2017-11-14 19:00:24 +00:00
return r
2018-04-13 18:58:29 +00:00
2018-04-14 13:39:51 +00:00
def render_tex(_request, template, ctx={}):
"""
Creates a PDF from a LaTex templates using pdflatex.
Writes it in a temporary directory and send back an HTTP response for
accessing this file.
"""
2016-07-09 21:26:17 +00:00
context = Context(ctx)
2017-11-14 19:00:24 +00:00
template = get_template(template)
2016-07-09 21:26:17 +00:00
rendered_tpl = template.render(context).encode('utf-8')
2018-04-13 18:58:29 +00:00
2016-07-09 21:26:17 +00:00
with tempfile.TemporaryDirectory() as tempdir:
for i in range(2):
process = Popen(
['pdflatex', '-output-directory', tempdir],
2018-04-13 18:58:29 +00:00
stdin=PIPE,
stdout=PIPE,
2016-07-09 21:26:17 +00:00
)
process.communicate(rendered_tpl)
with open(os.path.join(tempdir, 'texput.pdf'), 'rb') as f:
pdf = f.read()
r = HttpResponse(content_type='application/pdf')
r.write(pdf)
return r