apps/cover/forms.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 re
   7 from urllib2 import urlopen
   8 from django import forms
   9 from django.utils.translation import ugettext_lazy as _
  10 from cover.models import Image
  11
  12 class ImageAddForm(forms.ModelForm):
  13     class Meta:
  14         model = Image
  15
  16 class ImageEditForm(forms.ModelForm):
  17     """Form used for editing a Book."""
  18     class Meta:
  19         model = Image
  20         exclude = ['download_url']
  21
  22
  23 class ReadonlyImageEditForm(ImageEditForm):
  24     """Form used for not editing a Book."""
  25
  26     def __init__(self, *args, **kwargs):
  27         ret = super(ReadonlyImageEditForm, self).__init__(*args, **kwargs)
  28         for field in self.fields.values():
  29             field.widget.attrs.update({"readonly": True})
  30         return ret
  31
  32     def save(self, *args, **kwargs):
  33         raise AssertionError, "ReadonlyImageEditForm should not be saved."
  34
  35
  36 class FlickrForm(forms.Form):
  37     source_url = forms.URLField(label=_('Flickr URL'))
  38
  39     def clean_source_url(self):
  40         url = self.cleaned_data['source_url']
  41         m = re.match(r'(https?://)?(www\.|secure\.)?flickr\.com/photos/(?P<author>[^/]+)/(?P<img>\d+)/?', url)
  42         if not m:
  43             raise forms.ValidationError("It doesn't look like Flickr URL.")
  44         author_slug, img_id = m.group('author'), m.group('img')
  45         base_url = "https://www.flickr.com/photos/%s/%s/" % (author_slug, img_id)
  46
  47         try:
  48             html = urlopen(url).read().decode('utf-8')
  49         except:
  50             raise forms.ValidationError('Error reading page.')
  51         match = re.search(r'<a href="([^"]*)" rel="license cc:license">Some rights reserved</a>', html)
  52         try:
  53             assert match
  54             license_url = match.group(1)
  55             self.cleaned_data['license_url'] = license_url
  56             re_license = re.compile(r'http://creativecommons.org/licenses/([^/]*)/([^/]*)/.*')
  57             m = re_license.match(license_url)
  58             assert m
  59             self.cleaned_data['license_name'] = 'CC %s %s' % (m.group(1).upper(), m.group(2))
  60         except AssertionError:
  61             raise forms.ValidationError('Error reading license name.')
  62
  63         m = re.search(r'<strong class="username">By <a href="[^"]*">([^<]*)</a></strong>', html)
  64         if m:
  65             self.cleaned_data['author'] = "%s@Flickr" % m.group(1)
  66         else:
  67             raise forms.ValidationError('Error reading author name.')
  68
  69         m = re.search(r'<h1[^>]*>(.*?)</h1>', html)
  70         if not m:
  71             raise forms.ValidationError('Error reading image title.')
  72         self.cleaned_data['title'] = m.group(1)
  73
  74         url_size = base_url + "sizes/o/"
  75         html = urlopen(url_size).read().decode('utf-8')
  76         m = re.search(r'<div id="allsizes-photo">\s*<img src="([^"]*)"', html)
  77         if m:
  78             self.cleaned_data['download_url'] = m.group(1)
  79         else:
  80             raise forms.ValidationError('Error reading image URL.')
  81         return base_url