Jonas Havelka
3 months ago
3 changed files with 87 additions and 78 deletions
@ -0,0 +1,77 @@ |
|||||
|
""" |
||||
|
Stejně jako je `django.views.generic` jsou zde generické Views |
||||
|
a pár mixinů, které upravují chování Views. |
||||
|
""" |
||||
|
|
||||
|
import http |
||||
|
import shutil |
||||
|
import subprocess |
||||
|
import tempfile |
||||
|
|
||||
|
from pathlib import Path |
||||
|
|
||||
|
from django.http import HttpResponse |
||||
|
from django.shortcuts import render |
||||
|
from django.template.loader import render_to_string |
||||
|
from django.views import generic |
||||
|
|
||||
|
|
||||
|
class NeprazdnyListView(generic.ListView): |
||||
|
""" |
||||
|
Použití jako generic.ListView, jen při prázdném listu vyhodí M&M stránku |
||||
|
s titlem `self.if_prazdny_title` a textem `self.if_prazdny_text` |
||||
|
a způsob renderování (např. CSV) lze změnit přepsáním metody render. |
||||
|
""" |
||||
|
allow_empty = False |
||||
|
if_prazdny_title = "V seznamu nic není" |
||||
|
if_prazdny_text = "V seznamu nic není. Zkus to napravit v adminu, nebo se zeptej webařů." |
||||
|
|
||||
|
# Skoro copy-paste generic.list.ListView.get, |
||||
|
# protože nemůžu chytat 404, neboť může nastat i v get_context_data |
||||
|
def get(self, request, *args, **kwargs): |
||||
|
self.object_list = self.get_queryset() |
||||
|
|
||||
|
if self.get_paginate_by(self.object_list) is not None and hasattr( |
||||
|
self.object_list, "exists" |
||||
|
): |
||||
|
is_empty = not self.object_list.exists() |
||||
|
else: |
||||
|
is_empty = not self.object_list |
||||
|
if is_empty: |
||||
|
return render(request, 'universal.html', { |
||||
|
'title': self.if_prazdny_title, |
||||
|
'text': self.if_prazdny_text, |
||||
|
}, status=http.HTTPStatus.NOT_FOUND) |
||||
|
|
||||
|
return self.render(request, *args, **kwargs) |
||||
|
|
||||
|
# Tohle jsem vyčlenil, aby šlo generovat i něco jiného než template |
||||
|
def render(self, request, *args, **kwargs): |
||||
|
context = self.get_context_data() |
||||
|
return self.render_to_response(context) |
||||
|
|
||||
|
|
||||
|
class TeXResponseMixin: |
||||
|
""" |
||||
|
Mixin pro TemplateView, aby výsledek projel TeXem a vrátil rovnou PDF. |
||||
|
Obrázky a jiné soubory lze přidat nastavením `dalsi_potrebne_soubory` |
||||
|
(např. na `[django.contrib.staticfiles.finders.find('bla')]`, |
||||
|
nebo jiný seznam absolutních cest). |
||||
|
""" |
||||
|
dalsi_potrebne_soubory = [] |
||||
|
tex = "pdflatex" |
||||
|
|
||||
|
def render_to_response(self, context, **response_kwargs): |
||||
|
zdrojak = render_to_string(self.get_template_names(), context) |
||||
|
|
||||
|
with tempfile.TemporaryDirectory() as tempdirfn: |
||||
|
tempdir = Path(tempdirfn) |
||||
|
with open(tempdir / "main.tex", "w") as texfile: |
||||
|
texfile.write(zdrojak) |
||||
|
for file in self.dalsi_potrebne_soubory: |
||||
|
shutil.copy(file, tempdir) |
||||
|
subprocess.call([self.tex, "main.tex"], cwd=tempdir, stdout=subprocess.DEVNULL) |
||||
|
|
||||
|
with open(tempdir / "main.pdf", "rb") as pdffile: |
||||
|
response = HttpResponse(pdffile.read(), content_type='application/pdf', **response_kwargs) |
||||
|
return response |
Loading…
Reference in new issue