1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
#!/usr/bin/python2.5
import os
import PIL
from pytask.taskapp.utilities.helper import get_key
from django import forms
from pytask.taskapp.models import GENDER_CHOICES, Profile
from registration.forms import RegistrationFormUniqueEmail
from registration.models import RegistrationProfile
from pytask.taskapp.utilities.notification import create_notification
class UserProfileEditForm(forms.ModelForm):
"""Form used to edit the profile of a user"""
class Meta:
model = Profile
exclude = ('user','rights','dob','credits')
def clean_photo(self):
uploaded_photo = self.data.get('photo', None)
prev_photo = self.instance.photo
if uploaded_photo:
if uploaded_photo.size > 1048576:
raise forms.ValidationError('Images only smaller than 1MB allowed')
tmp_im_path = '/tmp/'+get_key()
tmp_file = open(tmp_im_path, 'w')
tmp_file.write(uploaded_photo.read())
tmp_file.close()
try:
PIL.Image.open(tmp_im_path)
except IOError:
raise forms.ValidationError('Image format unknown')
os.remove(tmp_im_path)
if prev_photo: os.remove(prev_photo.path)
return uploaded_photo
else:
return prev_photo
class RegistrationFormCustom(RegistrationFormUniqueEmail):
"""Used instead of RegistrationForm used by default django-registration backend, this adds date of birth and gender to the default django-registration RegistrationForm"""
dob = forms.DateField(help_text = "YYYY-MM-DD", required=True, label=u'date of birth')
gender = forms.ChoiceField(choices = GENDER_CHOICES, required=True, label=u'gender')
def save(self,profile_callback=None):
new_user = RegistrationProfile.objects.create_inactive_user(username=self.cleaned_data['username'],password=self.cleaned_data['password1'],email=self.cleaned_data['email'])
new_profile = Profile(user=new_user,dob=self.cleaned_data['dob'],gender=self.cleaned_data['gender'])
new_profile.save()
create_notification('NU',new_user)
return new_user
def UserChoiceForm(choices, instance=None):
""" take a list of users and return a choice form.
"""
class myForm(forms.Form):
user = forms.ChoiceField(choices, required=True)
return myForm(instance) if instance else myForm()
|
