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 = {
|
bgneal@6
|
43 'all': settings.GPP_THIRD_PARTY_CSS['markitup'] + \
|
bgneal@6
|
44 settings.GPP_THIRD_PARTY_CSS['jquery-ui']
|
gremmie@1
|
45 }
|
bgneal@6
|
46 js = settings.GPP_THIRD_PARTY_JS['jquery'] + \
|
bgneal@6
|
47 settings.GPP_THIRD_PARTY_JS['markitup'] + \
|
bgneal@6
|
48 settings.GPP_THIRD_PARTY_JS['jquery-ui'] + \
|
bgneal@6
|
49 ('js/bio.js', )
|
gremmie@1
|
50
|
gremmie@1
|
51
|
gremmie@1
|
52 def get_image(file):
|
gremmie@1
|
53 """
|
gremmie@1
|
54 Returns a PIL Image from the supplied file.
|
gremmie@1
|
55 Throws ValidationError if the file does not parse as an image file.
|
gremmie@1
|
56 """
|
gremmie@1
|
57 parser = ImageFile.Parser()
|
gremmie@1
|
58 for chunk in file.chunks():
|
gremmie@1
|
59 parser.feed(chunk)
|
gremmie@1
|
60 try:
|
gremmie@1
|
61 image = parser.close()
|
gremmie@1
|
62 return image
|
gremmie@1
|
63 except IOError:
|
gremmie@1
|
64 pass
|
gremmie@1
|
65 raise forms.ValidationError("Upload a valid image. " +
|
gremmie@1
|
66 "The file you uploaded was either not an image or a corrupted image.")
|
gremmie@1
|
67
|
gremmie@1
|
68
|
gremmie@1
|
69 def scale_image(image, size):
|
gremmie@1
|
70 """Scales an image file if necessary."""
|
gremmie@1
|
71
|
gremmie@1
|
72 # don't upscale
|
gremmie@1
|
73 if (size, size) >= image.size:
|
gremmie@1
|
74 return image
|
gremmie@1
|
75
|
gremmie@1
|
76 (w, h) = image.size
|
gremmie@1
|
77 if w > h:
|
gremmie@1
|
78 diff = (w - h) / 2
|
gremmie@1
|
79 image = image.crop((diff, 0, w - diff, h))
|
gremmie@1
|
80 elif h > w:
|
gremmie@1
|
81 diff = (h - w) / 2
|
gremmie@1
|
82 image = image.crop((0, diff, w, h - diff))
|
gremmie@1
|
83 image = image.resize((size, size), Image.ANTIALIAS)
|
gremmie@1
|
84 return image
|
gremmie@1
|
85
|
gremmie@1
|
86
|
gremmie@1
|
87 class UploadAvatarForm(forms.Form):
|
gremmie@1
|
88 """Form used to change a user's avatar"""
|
gremmie@1
|
89 avatar_file = forms.ImageField(required=False)
|
gremmie@1
|
90 image = None
|
gremmie@1
|
91
|
gremmie@1
|
92 def clean_avatar_file(self):
|
gremmie@1
|
93 file = self.cleaned_data['avatar_file']
|
gremmie@1
|
94 if file is not None:
|
gremmie@1
|
95 if file.size > settings.MAX_AVATAR_SIZE_BYTES:
|
gremmie@1
|
96 raise forms.ValidationError("Please upload a file smaller than %s bytes." % \
|
gremmie@1
|
97 settings.MAX_AVATAR_SIZE)
|
gremmie@1
|
98 self.image = get_image(file)
|
gremmie@1
|
99 self.format = self.image.format
|
gremmie@1
|
100 return file
|
gremmie@1
|
101
|
gremmie@1
|
102 def get_file(self):
|
gremmie@1
|
103 if self.image is not None:
|
gremmie@1
|
104 self.image = scale_image(self.image, settings.MAX_AVATAR_SIZE_PIXELS)
|
gremmie@1
|
105 s = StringIO()
|
gremmie@1
|
106 self.image.save(s, self.format)
|
gremmie@1
|
107 return ContentFile(s.getvalue())
|
gremmie@1
|
108 return None
|
gremmie@1
|
109
|
gremmie@1
|
110 def get_filename(self):
|
gremmie@1
|
111 return self.cleaned_data['avatar_file'].name
|
gremmie@1
|
112
|
gremmie@1
|
113 # vim: ts=4 sw=4
|