apps/cover/views.py

   1 # Create your views here.
   2 import os.path
   3 from django.conf import settings
   4 from django.http import HttpResponse, Http404
   5 from catalogue.models import Chunk
   6 from django.views.decorators.http import require_POST
   7 from django.views.decorators.csrf import csrf_exempt
   8 from django.shortcuts import render
   9
  10
  11 PREVIEW_SIZE = (216, 300)
  12
  13 def preview(request, book, chunk=None, rev=None):
  14     """Creates a cover image.
  15
  16     If chunk and rev number are given, use version from given revision.
  17     If rev is not given, use publishable version.
  18     """
  19     import Image
  20     from librarian.cover import WLCover
  21     from librarian.dcparser import BookInfo
  22
  23     chunk = Chunk.get(book, chunk)
  24     if rev is not None:
  25         try:
  26             revision = chunk.at_revision(rev)
  27         except Chunk.change_model.DoesNotExist:
  28             raise Http404
  29     else:
  30         revision = chunk.publishable()
  31         if revision is None:
  32             raise Http404
  33     xml = revision.materialize().encode('utf-8')
  34
  35     info = BookInfo.from_string(xml)
  36     cover = WLCover(info)
  37     response = HttpResponse(mimetype=cover.mime_type())
  38     image = cover.image().resize(PREVIEW_SIZE, Image.ANTIALIAS)
  39     image.save(response, cover.format)
  40     return response
  41
  42
  43 @csrf_exempt
  44 @require_POST
  45 def preview_from_xml(request):
  46     from hashlib import sha1
  47     import Image
  48     from os import makedirs
  49     from lxml import etree
  50     from librarian.cover import WLCover
  51     from librarian.dcparser import BookInfo
  52
  53     xml = request.POST['xml']
  54     info = BookInfo.from_string(xml.encode('utf-8'))
  55     coverid = sha1(etree.tostring(info.to_etree())).hexdigest()
  56     cover = WLCover(info)
  57
  58     cover_dir = 'cover/preview'
  59     try:
  60         makedirs(os.path.join(settings.MEDIA_ROOT, cover_dir))
  61     except OSError:
  62         pass
  63     fname = os.path.join(cover_dir, "%s.%s" % (coverid, cover.ext()))
  64     image = cover.image().resize(PREVIEW_SIZE, Image.ANTIALIAS)
  65     image.save(os.path.join(settings.MEDIA_ROOT, fname))
  66     return HttpResponse(os.path.join(settings.MEDIA_URL, fname))
  67
  68
  69 def flickr(request):
  70     url = request.POST.get('flickr_url')
  71     if url:
  72         import re
  73         from urllib2 import urlopen
  74
  75         html = urlopen(url).read()
  76         match = re.search(r'<a href="([^"]*)" rel="license cc:license">Some rights reserved</a>', html)
  77         if match:
  78             license_url = match.group(1)
  79
  80         re_license = re.compile(r'http://creativecommons.org/licenses/([^/]*)/([^/]*)/.*')
  81         m = re_license.match(license_url)
  82         if m:
  83             license_name = 'CC %s %s' % (m.group(1).upper(), m.group(2))
  84
  85         m = re.search(r'<strong class="username">By <a href="[^"]*">([^<]*)</a></strong>', html)
  86         if m:
  87             author = m.group(1)
  88
  89         url_size = url.rstrip('/') + '/sizes/o/'
  90         html = urlopen(url_size).read()
  91         m = re.search(r'<div id="allsizes-photo">\s*<img src="([^"]*)"', html)
  92         if m:
  93             image_url = m.group(1)
  94     else:
  95         url = ""
  96
  97     return render(request, 'cover/flickr.html', locals())