Mercurial > public > sg101

annotate gpp/bio/forms.py @ 1:dbd703f7d63a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Initial import of sg101 stuff from private repository.
author gremmie
date Mon, 06 Apr 2009 02:43:12 +0000
parents
children b6263ac72052
rev   line source
gremmie@1 1 """
gremmie@1 2 This file contains the forms used by the bio application.
gremmie@1 3 """
gremmie@1 4 from PIL import ImageFile
gremmie@1 5 from PIL import Image
gremmie@1 6
gremmie@1 7 try:
gremmie@1 8 from cStringIO import StringIO
gremmie@1 9 except:
gremmie@1 10 from StringIO import StringIO
gremmie@1 11
gremmie@1 12 from django import forms
gremmie@1 13 from django.conf import settings
gremmie@1 14 from django.core.files.base import ContentFile
gremmie@1 15 from django.contrib.auth.models import User
gremmie@1 16
gremmie@1 17 from bio.models import UserProfile
gremmie@1 18
gremmie@1 19
gremmie@1 20 class EditUserForm(forms.ModelForm):
gremmie@1 21 """Form for editing the fields of the User model."""
gremmie@1 22 email = forms.EmailField(label='Email', required=True)
gremmie@1 23 class Meta:
gremmie@1 24 model = User
gremmie@1 25 fields = ('first_name', 'last_name', 'email')
gremmie@1 26
gremmie@1 27
gremmie@1 28 class EditUserProfileForm(forms.ModelForm):
gremmie@1 29 """Form for editing the fields of the UserProfile model."""
gremmie@1 30 location = forms.CharField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 31 occupation = forms.CharField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 32 interests = forms.CharField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 33 website_1 = forms.URLField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 34 website_2 = forms.URLField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 35 website_3 = forms.URLField(required=False, widget=forms.TextInput(attrs={'size' : 64 }))
gremmie@1 36
gremmie@1 37 class Meta:
gremmie@1 38 model = UserProfile
gremmie@1 39 exclude = ('user', 'avatar', 'profile_html', 'signature_html')
gremmie@1 40
gremmie@1 41 class Media:
gremmie@1 42 css = {
gremmie@1 43 'all': ('js/markitup/skins/markitup/style.css',
gremmie@1 44 'js/markitup/sets/markdown/style.css',
gremmie@1 45 'http://ajax.googleapis.com/ajax/libs/jqueryui/1.7.0/themes/redmond/jquery-ui.css',
gremmie@1 46 ),
gremmie@1 47 }
gremmie@1 48 js = (
gremmie@1 49 'http://ajax.googleapis.com/ajax/libs/jquery/1.3.2/jquery.min.js',
gremmie@1 50 'js/markitup/jquery.markitup.pack.js',
gremmie@1 51 'js/markitup/sets/markdown/set.js',
gremmie@1 52 'http://ajax.googleapis.com/ajax/libs/jqueryui/1.7.0/jquery-ui.js',
gremmie@1 53 'js/bio.js',
gremmie@1 54 )
gremmie@1 55
gremmie@1 56
gremmie@1 57 def get_image(file):
gremmie@1 58 """
gremmie@1 59 Returns a PIL Image from the supplied file.
gremmie@1 60 Throws ValidationError if the file does not parse as an image file.
gremmie@1 61 """
gremmie@1 62 parser = ImageFile.Parser()
gremmie@1 63 for chunk in file.chunks():
gremmie@1 64 parser.feed(chunk)
gremmie@1 65 try:
gremmie@1 66 image = parser.close()
gremmie@1 67 return image
gremmie@1 68 except IOError:
gremmie@1 69 pass
gremmie@1 70 raise forms.ValidationError("Upload a valid image. " +
gremmie@1 71 "The file you uploaded was either not an image or a corrupted image.")
gremmie@1 72
gremmie@1 73
gremmie@1 74 def scale_image(image, size):
gremmie@1 75 """Scales an image file if necessary."""
gremmie@1 76
gremmie@1 77 # don't upscale
gremmie@1 78 if (size, size) >= image.size:
gremmie@1 79 return image
gremmie@1 80
gremmie@1 81 (w, h) = image.size
gremmie@1 82 if w > h:
gremmie@1 83 diff = (w - h) / 2
gremmie@1 84 image = image.crop((diff, 0, w - diff, h))
gremmie@1 85 elif h > w:
gremmie@1 86 diff = (h - w) / 2
gremmie@1 87 image = image.crop((0, diff, w, h - diff))
gremmie@1 88 image = image.resize((size, size), Image.ANTIALIAS)
gremmie@1 89 return image
gremmie@1 90
gremmie@1 91
gremmie@1 92 class UploadAvatarForm(forms.Form):
gremmie@1 93 """Form used to change a user's avatar"""
gremmie@1 94 avatar_file = forms.ImageField(required=False)
gremmie@1 95 image = None
gremmie@1 96
gremmie@1 97 def clean_avatar_file(self):
gremmie@1 98 file = self.cleaned_data['avatar_file']
gremmie@1 99 if file is not None:
gremmie@1 100 if file.size > settings.MAX_AVATAR_SIZE_BYTES:
gremmie@1 101 raise forms.ValidationError("Please upload a file smaller than %s bytes." % \
gremmie@1 102 settings.MAX_AVATAR_SIZE)
gremmie@1 103 self.image = get_image(file)
gremmie@1 104 self.format = self.image.format
gremmie@1 105 return file
gremmie@1 106
gremmie@1 107 def get_file(self):
gremmie@1 108 if self.image is not None:
gremmie@1 109 self.image = scale_image(self.image, settings.MAX_AVATAR_SIZE_PIXELS)
gremmie@1 110 s = StringIO()
gremmie@1 111 self.image.save(s, self.format)
gremmie@1 112 return ContentFile(s.getvalue())
gremmie@1 113 return None
gremmie@1 114
gremmie@1 115 def get_filename(self):
gremmie@1 116 return self.cleaned_data['avatar_file'].name
gremmie@1 117
gremmie@1 118 # vim: ts=4 sw=4