620ad1aa2beef5a854177771149f48581dd5c1d9
[redakcja.git] / src / cover / views.py
1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 import os.path
5 from django.conf import settings
6 from django.contrib.auth.decorators import permission_required
7 from django.http import HttpResponse, HttpResponseRedirect, Http404
8 from django.shortcuts import get_object_or_404, render
9 from django.views.decorators.csrf import csrf_exempt
10 from django.views.decorators.http import require_POST
11 from lxml import etree
12 from librarian import RDFNS, DCNS
13 from documents.helpers import active_tab
14 from documents.models import Book, Chunk
15 from cover.models import Image
16 from cover import forms
17 from cover.utils import get_import_data
18
19
20 PREVIEW_SIZE = (216, 300)
21
22
23 def preview(request, book, chunk=None, rev=None):
24     """Creates a cover image.
25
26     If chunk and rev number are given, use version from given revision.
27     If rev is not given, use publishable version.
28     """
29     from PIL import Image
30     from librarian.cover import make_cover
31     from librarian.dcparser import BookInfo
32
33     chunk = Chunk.get(book, chunk)
34     if rev is not None:
35         try:
36             revision = chunk.at_revision(rev)
37         except Chunk.change_model.DoesNotExist:
38             raise Http404
39     else:
40         revision = chunk.publishable()
41         if revision is None:
42             revision = chunk.head
43     xml = revision.materialize().encode('utf-8')
44
45     try:
46         info = BookInfo.from_bytes(xml)
47     except:
48         return HttpResponseRedirect(os.path.join(settings.STATIC_URL, "img/sample_cover.png"))
49     width = request.GET.get('width')
50     width = int(width) if width else None
51     height=request.GET.get('height')
52     height = int(height) if height else None
53     cover = make_cover(info, width=width, height=height)
54     #cover = make_cover(info)
55     response = HttpResponse(content_type=cover.mime_type())
56     if height or width:
57         size = (width, height)
58     else:
59         size = PREVIEW_SIZE
60     img = cover.image().resize(size, Image.ANTIALIAS)
61     img.save(response, cover.format)
62
63     if 'download' in request.GET:
64         response['Content-Disposition'] = 'attachment; filename=%s.jpg' % chunk.book.slug
65
66     return response
67
68
69 @csrf_exempt
70 @require_POST
71 def preview_from_xml(request):
72     from hashlib import sha1
73     from PIL import Image
74     from os import makedirs
75     from lxml import etree
76     from librarian.cover import make_cover
77     from librarian.dcparser import BookInfo
78
79     xml = request.POST['xml']
80     try:
81         info = BookInfo.from_bytes(xml.encode('utf-8'))
82     except:
83         return HttpResponse(os.path.join(settings.STATIC_URL, "img/sample_cover.png"))
84     coverid = sha1(etree.tostring(info.to_etree())).hexdigest()
85     cover = make_cover(info)
86
87     cover_dir = 'cover/preview'
88     try:
89         makedirs(os.path.join(settings.MEDIA_ROOT, cover_dir))
90     except OSError:
91         pass
92     fname = os.path.join(cover_dir, "%s.%s" % (coverid, cover.ext()))
93     img = cover.image().resize(PREVIEW_SIZE, Image.ANTIALIAS)
94     img.save(os.path.join(settings.MEDIA_ROOT, fname))
95     return HttpResponse(os.path.join(settings.MEDIA_URL, fname))
96
97
98 @active_tab('cover')
99 def image(request, pk):
100     img = get_object_or_404(Image, pk=pk)
101
102     if request.user.has_perm('cover.change_image'):
103         if request.method == "POST":
104             form = forms.ImageEditForm(request.POST, request.FILES, instance=img)
105             if form.is_valid():
106                 form.save()
107                 return HttpResponseRedirect(img.get_absolute_url())
108         else:
109             form = forms.ImageEditForm(instance=img)
110         editable = True
111     else:
112         form = forms.ReadonlyImageEditForm(instance=img)
113         editable = False
114
115     return render(request, "cover/image_detail.html", {
116         "object": Image.objects.get(id=img.id),
117         "form": form,
118         "editable": editable,
119     })
120
121
122 def image_file(request, pk):
123     img = get_object_or_404(Image, pk=pk)
124     return HttpResponseRedirect(img.file.url)
125
126
127 @active_tab('cover')
128 def image_list(request):
129     return render(request, "cover/image_list.html", {
130         'object_list': Image.objects.all().order_by('-id'),
131         'can_add': request.user.has_perm('cover.add_image'),
132     })
133
134
135 @permission_required('cover.add_image')
136 @active_tab('cover')
137 def add_image(request):
138     form = ff = None
139     if request.method == 'POST':
140         if request.POST.get('form_id') == 'import':
141             ff = forms.ImportForm(request.POST)
142             if ff.is_valid():
143                 form = forms.ImageAddForm(ff.cleaned_data)
144         else:
145             form = forms.ImageAddForm(request.POST, request.FILES)
146             if form.is_valid():
147                 obj = form.save()
148                 return HttpResponseRedirect(obj.get_absolute_url())
149     if form is None:
150         form = forms.ImageAddForm()
151     if ff is None:
152         ff = forms.ImportForm()
153     return render(request, 'cover/add_image.html', {
154             'form': form,
155             'ff': ff,
156         })
157
158 @permission_required('cover.add_image')
159 def quick_import(request, pk):
160     url = request.POST.get('url')
161     if url.startswith('%s://%s/' % (
162             request.scheme,
163             request.get_host())):
164         cover_id = url.rsplit('/', 1)[-1]
165         cover = Image.objects.get(pk=cover_id)
166     else:
167         data = get_import_data(url)
168         same = Image.objects.filter(source_url=data['source_url'])
169         if not same.exists():
170             same = Image.objects.filter(download_url=data['download_url'])
171         if same.exists():
172             cover = same.first()
173         else:
174             form = forms.ImageAddForm(data)
175             if form.is_valid():
176                 cover = form.save()
177
178     # We have a cover. Now let's commit.
179     book = Book.objects.get(pk=pk)
180     chunk = book[0]
181     text = chunk.head.materialize()
182
183     root = etree.fromstring(text)
184     rdf = root.find('.//' + RDFNS('Description'))
185     for tag in 'url', 'attribution', 'source':
186         for elem in rdf.findall('.//' + DCNS('relation.coverImage.%s' % tag)):
187             rdf.remove(elem)
188     e = etree.Element(DCNS('relation.coverImage.url'))
189     e.text = request.build_absolute_uri(cover.use_file.url)
190     rdf.append(e)
191     e.tail = '\n    '
192     e = etree.Element(DCNS('relation.coverImage.attribution'))
193     e.text = ''
194     if cover.title:
195         e.text += cover.title + ', '
196     if cover.author:
197         e.text += cover.author + ', '
198     e.text += cover.license_name
199     e.tail = '\n    '
200     rdf.append(e)
201     e = etree.Element(DCNS('relation.coverImage.source'))
202     e.text = cover.get_full_url()
203     e.tail = '\n    '
204     rdf.append(e)
205
206     xml = etree.tostring(root, encoding='unicode')
207     chunk.commit(
208         xml,
209         author=request.user,
210         comment='Cover',
211         publishable=chunk.head.publishable,
212     )
213     return HttpResponseRedirect(book.get_absolute_url())
214