class RegistrationForm(forms.Form):
username = forms.CharField()
email = forms.EmailField()
first_name = forms.CharField()
last_name = forms.CharField()
password = forms.CharField(widget=forms.PasswordInput(render_value=True))
password2 = forms.CharField(widget=forms.PasswordInput(render_value=True),
label='Confirm password')
def clean_username(self):
if User.objects.filter(
username=self.cleaned_data['username']).count() > 0:
raise forms.ValidationError(
'A user with such username is already registered')
return self.cleaned_data['username']
def clean_email(self):
if User.objects.filter(email=self.cleaned_data['email']).count() > 0:
raise forms.ValidationError(
'A user with such email is already registered')
return self.cleaned_data['email']
def clean(self, *args, **kwargs):
super(RegistrationForm, self).clean(*args, **kwargs)
if 'password' in self.cleaned_data and 'password2' in self.cleaned_data:
if self.cleaned_data['password'] != self.cleaned_data['password2']:
self._errors['password'] = [u'Passwords must match.']
self._errors['password2'] = [u'Passwords must match']
return self.cleaned_data
def save(self, obj):
user = User(username=self.cleaned_data['username'],
first_name=self.cleaned_data['first_name'],
last_name=self.cleaned_data['last_name'],
email=self.cleaned_data['email'],
education=[],
experience=[],
services=[])
user.set_password(self.cleaned_data['password'])
user.save()
user = authenticate(username=self.cleaned_data['username'],
password=self.cleaned_data['password'])
session_key = md5(self.cleaned_data['username'] +
self.cleaned_data['password'])
obj.set_secure_cookie("session_key", session_key.hexdigest())
session = Session.objects.get_or_create(user=user)[0]
session.session_key = session_key.hexdigest()
session.save()
return user
class NewPasswordForm(forms.Form):
password = forms.CharField(widget=forms.PasswordInput(render_value=True),
label='Please enter new password')
password2 = forms.CharField(widget=forms.PasswordInput(render_value=True),
label='Confirm password')
def clean(self, *args, **kwargs):
super(NewPasswordForm, self).clean(*args, **kwargs)
if 'password' in self.cleaned_data and 'password2' in self.cleaned_data:
if self.cleaned_data['password'] != self.cleaned_data['password2']:
self._errors['password'] = [u'Passwords must match.']
self._errors['password2'] = [u'Passwords must match']
return self.cleaned_data
def save(self, handler, user):
user.set_password(self.cleaned_data['password'])
user.save()
login(user, self.cleaned_data['password'], handler)
return user
class InterestsForm(forms.Form):
interest = forms.CharField(max_length=64, label='')
def save(self, handler):
user = handler.get_current_user()
interest = Interest(tag=self.cleaned_data['interest'])
interest.save()
user.interests.append(interest)
user.save()
return interest
class RecommendedBookForm(forms.Form):
text = forms.CharField(max_length=1024)
ref = forms.URLField()
def save(self, handler):
user = handler.get_current_user()
book = Book(text=self.cleaned_data['text'],
ref=self.cleaned_data['ref'])
book.save()
user.recommended_books.append(book)
user.save()
return book
class ExperienceForm(forms.Form):
employer = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
position = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
description = forms.CharField(
widget=forms.Textarea(attrs={'class': 'post-body'}), required=False)
start_date = forms.DateTimeField()
end_date = forms.DateTimeField(required=False)
def save(self, handler):
user = handler.get_current_user()
experience = Experience(employer=self.cleaned_data['employer'],
position=self.cleaned_data['position'],
description=self.cleaned_data['description'],
start_date=self.cleaned_data['start_date'],
end_date=self.cleaned_data['end_date'])
experience.save()
user.experience.append(experience)
user.save()
return experience
class ServicesForm(forms.Form):
name = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
def save(self, handler):
user = handler.get_current_user()
service = Service.objects.get_or_create(
name=self.cleaned_data['name'])[0]
service.providers_num += 1
service.save()
user.services.append(service)
user.save()
return service
class LoginForm(forms.Form):
username = forms.CharField(label="E-mail")
password = forms.CharField(widget=forms.PasswordInput(render_value=True))
def clean(self):
if 'username' in self.cleaned_data and 'password' in self.cleaned_data:
username = self.cleaned_data['username']
if not '@' in username:
user = authenticate(username=username,
password=self.cleaned_data['password'])
else:
user = authenticate(email=username,
password=self.cleaned_data['password'])
if not user:
raise forms.ValidationError('Email or password are incorrect')
else:
return None
self.user = user
return user
def save(self, handler):
login(self.user, self.cleaned_data['password'], handler)
return self.user
class EducationForm(forms.Form):
university = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
degree = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
major = forms.CharField(widget=forms.TextInput(
attrs={'class': 'large input-text'}))
description = forms.CharField(
widget=forms.Textarea(attrs={'class': 'post-body'}), required=False)
start_date = forms.DateTimeField()
end_date = forms.DateTimeField(required=False)
def save(self, handler):
user = handler.get_current_user()
education = Education(university=self.cleaned_data['university'],
degree=self.cleaned_data['degree'],
major=self.cleaned_data['major'],
description=self.cleaned_data['description'],
start_date=self.cleaned_data['start_date'],
end_date=self.cleaned_data['end_date'])
education.save()
user.education.append(education)
user.save()
return education
class PasswordRecoveryForm(forms.Form):
email_or_login = forms.CharField(
widget=forms.TextInput(attrs={'class': 'large input-text'}),
label='Please enter your email or password')
def clean_email_or_login(self):
data = self.cleaned_data['email_or_login']
try:
if '@' in data:
self.user = User.objects.get(email=data)
else:
self.user = User.objects.get(username=data)
except Exception as e:
raise forms.ValidationError(
'A user with such email or username does not exist')
return self.cleaned_data['email_or_login']
def save(self):
return self.user