import re
from urllib2 import urlopen
from django import forms
-from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import ugettext_lazy as _, ugettext
from cover.models import Image
+from django.utils.text import mark_safe
class ImageAddForm(forms.ModelForm):
class Meta:
model = Image
+ def __init__(self, *args, **kwargs):
+ super(ImageAddForm, self).__init__(*args, **kwargs)
+ self.fields['file'].required = False
+
+ def clean_download_url(self):
+ cl = self.cleaned_data['download_url'] or None
+ if cl is not None:
+ try:
+ img = Image.objects.get(download_url=cl)
+ except Image.DoesNotExist:
+ pass
+ else:
+ raise forms.ValidationError(mark_safe(
+ ugettext('Image <a href="%(url)s">already in repository</a>.'
+ ) % {'url': img.get_absolute_url()}))
+ return cl
+
+ def clean(self):
+ cleaned_data = super(ImageAddForm, self).clean()
+ if not cleaned_data.get('download_url', None) and not cleaned_data.get('file', None):
+ raise forms.ValidationError('No image specified')
+ return cleaned_data
+
class ImageEditForm(forms.ModelForm):
"""Form used for editing a Book."""
class Meta:
class ReadonlyImageEditForm(ImageEditForm):
- """Form used for not editing a Book."""
+ """Form used for not editing an Image."""
def __init__(self, *args, **kwargs):
ret = super(ReadonlyImageEditForm, self).__init__(*args, **kwargs)
except AssertionError:
raise forms.ValidationError('Error reading license name.')
- m = re.search(r'<span class="photo-name-line-2">([^<]*)</span>', html)
+ m = re.search(r'"ownername":"([^"]*)', html)
if m:
self.cleaned_data['author'] = "%s@Flickr" % m.group(1)
else: