6b3344b441cbadeaf23608a9898ac79383d209a8
[redakcja.git] / apps / cover / views.py
1 # -*- coding: utf-8 -*-
2 #
3 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 import os.path
7 from django.conf import settings
8 from django.contrib.auth.decorators import permission_required
9 from django.http import HttpResponse, HttpResponseRedirect, Http404
10 from django.shortcuts import get_object_or_404, render
11 from django.views.decorators.csrf import csrf_exempt
12 from django.views.decorators.http import require_POST
13 from catalogue.helpers import active_tab
14 from catalogue.models import Chunk
15 from cover.models import Image
16 from cover import forms
17
18
19 PREVIEW_SIZE = (216, 300)
20
21
22 def preview(request, book, chunk=None, rev=None):
23     """Creates a cover image.
24
25     If chunk and rev number are given, use version from given revision.
26     If rev is not given, use publishable version.
27     """
28     from PIL import Image
29     from librarian.cover import DefaultEbookCover
30     from librarian.dcparser import BookInfo
31
32     chunk = Chunk.get(book, chunk)
33     if rev is not None:
34         try:
35             revision = chunk.at_revision(rev)
36         except Chunk.change_model.DoesNotExist:
37             raise Http404
38     else:
39         revision = chunk.publishable()
40         if revision is None:
41             raise Http404
42     xml = revision.materialize().encode('utf-8')
43
44     try:
45         info = BookInfo.from_string(xml)
46     except:
47         return HttpResponseRedirect(os.path.join(settings.STATIC_URL, "img/sample_cover.png"))
48     cover = DefaultEbookCover(info)
49     response = HttpResponse(mimetype=cover.mime_type())
50     image = cover.image().resize(PREVIEW_SIZE, Image.ANTIALIAS)
51     image.save(response, cover.format)
52     return response
53
54
55 @csrf_exempt
56 @require_POST
57 def preview_from_xml(request):
58     from hashlib import sha1
59     from PIL import Image
60     from os import makedirs
61     from lxml import etree
62     from librarian.cover import DefaultEbookCover
63     from librarian.dcparser import BookInfo
64
65     xml = request.POST['xml']
66     try:
67         info = BookInfo.from_string(xml.encode('utf-8'))
68     except:
69         return HttpResponse(os.path.join(settings.STATIC_URL, "img/sample_cover.png"))
70     coverid = sha1(etree.tostring(info.to_etree())).hexdigest()
71     cover = DefaultEbookCover(info)
72
73     cover_dir = 'cover/preview'
74     try:
75         makedirs(os.path.join(settings.MEDIA_ROOT, cover_dir))
76     except OSError:
77         pass
78     fname = os.path.join(cover_dir, "%s.%s" % (coverid, cover.ext()))
79     image = cover.image().resize(PREVIEW_SIZE, Image.ANTIALIAS)
80     image.save(os.path.join(settings.MEDIA_ROOT, fname))
81     return HttpResponse(os.path.join(settings.MEDIA_URL, fname))
82
83
84 @active_tab('cover')
85 def image(request, pk):
86     image = get_object_or_404(Image, pk=pk)
87
88     if request.user.has_perm('cover.change_image'):
89         if request.method == "POST":
90             form = forms.ImageEditForm(request.POST, request.FILES, instance=image)
91             if form.is_valid():
92                 form.save()
93                 return HttpResponseRedirect(image.get_absolute_url())
94         else:
95             form = forms.ImageEditForm(instance=image)
96         editable = True
97     else:
98         form = forms.ReadonlyImageEditForm(instance=image)
99         editable = False
100
101     return render(request, "cover/image_detail.html", {
102         "object": image,
103         "form": form,
104         "editable": editable,
105     })
106
107
108 @active_tab('cover')
109 def image_list(request):
110     objects = Image.objects.all()
111     enable_add = request.user.has_perm('cover.add_image')
112     return render(request, "cover/image_list.html", {
113         'object_list': Image.objects.all(),
114         'can_add': request.user.has_perm('cover.add_image'),
115     })
116
117
118 @permission_required('cover.add_image')
119 @active_tab('cover')
120 def add_image(request):
121     form = ff = None
122     if request.method == 'POST':
123         if request.POST.get('form_id') == 'flickr':
124             ff = forms.FlickrForm(request.POST)
125             if ff.is_valid():
126                 form = forms.ImageAddForm(ff.cleaned_data)
127         else:
128             form = forms.ImageAddForm(request.POST, request.FILES)
129             if form.is_valid():
130                 obj = form.save()
131                 return HttpResponseRedirect(obj.get_absolute_url())
132     if form is None:
133         form = forms.ImageAddForm()
134     if ff is None:
135         ff = forms.FlickrForm()
136     return render(request, 'cover/add_image.html', {
137             'form': form,
138             'ff': ff,
139         })