135 lines
4.7 KiB
Python
135 lines
4.7 KiB
Python
import subprocess
|
|
import datetime
|
|
import hashlib
|
|
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.sites.models import Site
|
|
from django.utils.translation import ugettext, ugettext_lazy as _
|
|
|
|
from ietf.utils.mail import send_mail
|
|
from redesign.person.models import Person, Email
|
|
|
|
|
|
class RegistrationForm(forms.Form):
|
|
|
|
email = forms.EmailField(label="Your email")
|
|
realm = 'IETF'
|
|
expire = 3
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.send_email()
|
|
return True
|
|
|
|
def send_email(self):
|
|
domain = Site.objects.get_current().domain
|
|
subject = ugettext(u'Confirm registration at %s') % domain
|
|
from_email = settings.DEFAULT_FROM_EMAIL
|
|
to_email = self.cleaned_data['email']
|
|
today = datetime.date.today().strftime('%Y%m%d')
|
|
auth = hashlib.md5('%s%s%s%s' % (settings.SECRET_KEY, today, to_email, self.realm)).hexdigest()
|
|
context = {
|
|
'domain': domain,
|
|
'today': today,
|
|
'realm': self.realm,
|
|
'auth': auth,
|
|
'username': to_email,
|
|
'expire': settings.DAYS_TO_EXPIRE_REGISTRATION_LINK,
|
|
}
|
|
send_mail(None, to_email, from_email, subject, 'registration/creation_email.txt', context)
|
|
|
|
def clean_email(self):
|
|
email = self.cleaned_data.get('email', '')
|
|
if not email:
|
|
return email
|
|
if User.objects.filter(username=email).count():
|
|
raise forms.ValidationError(_('Email already in use'))
|
|
return email
|
|
|
|
|
|
class RecoverPasswordForm(RegistrationForm):
|
|
|
|
realm = 'IETF'
|
|
|
|
def send_email(self):
|
|
domain = Site.objects.get_current().domain
|
|
subject = 'Password recovery at %s' % domain
|
|
from_email = settings.DEFAULT_FROM_EMAIL
|
|
today = datetime.date.today().strftime('%Y%m%d')
|
|
to_email = self.cleaned_data["email"]
|
|
auth = hashlib.md5('%s%s%s%s' % (settings.SECRET_KEY, today, to_email, self.realm)).hexdigest()
|
|
context = {'domain': domain,
|
|
'today': today,
|
|
'realm': self.realm,
|
|
'auth': auth,
|
|
'username': to_email,
|
|
'expire': settings.DAYS_TO_EXPIRE_REGISTRATION_LINK,
|
|
}
|
|
send_mail(None, to_email, from_email, subject, 'registration/password_reset_email.txt', context)
|
|
|
|
def clean_email(self):
|
|
email = self.cleaned_data.get('email', '')
|
|
return email
|
|
|
|
|
|
class PasswordForm(forms.Form):
|
|
|
|
password1 = forms.CharField(label=_("Password"), widget=forms.PasswordInput)
|
|
password2 = forms.CharField(label=_("Password confirmation"), widget=forms.PasswordInput,
|
|
help_text=_("Enter the same password as above, for verification."))
|
|
realm = 'IETF'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.username = kwargs.pop('username')
|
|
self.update_user = kwargs.pop('update_user', False)
|
|
super(PasswordForm, self).__init__(*args, **kwargs)
|
|
|
|
def clean_password2(self):
|
|
password1 = self.cleaned_data.get("password1", "")
|
|
password2 = self.cleaned_data["password2"]
|
|
if password1 != password2:
|
|
raise forms.ValidationError(_("The two password fields didn't match."))
|
|
return password2
|
|
|
|
def get_password(self):
|
|
return self.cleaned_data.get('password1')
|
|
|
|
def create_user(self):
|
|
user = User.objects.create(username=self.username,
|
|
email=self.username)
|
|
person = Person.objects.create(user=user,
|
|
name=self.username,
|
|
ascii=self.username)
|
|
Email.objects.create(person=person,
|
|
address=self.username)
|
|
return user
|
|
|
|
def get_user(self):
|
|
return User.objects.get(username=self.username)
|
|
|
|
def save_password_file(self):
|
|
if settings.USE_PYTHON_HTDIGEST:
|
|
pass_file = settings.HTPASSWD_FILE
|
|
realm = settings.HTDIGEST_REALM
|
|
password = self.get_password()
|
|
username = self.username
|
|
prefix = '%s:%s:' % (username, realm)
|
|
key = hashlib.md5(prefix + password).hexdigest()
|
|
f = open(pass_file, 'a+')
|
|
f.write('%s%s\n' % (prefix, key))
|
|
f.close()
|
|
else:
|
|
p = subprocess.Popen([settings.HTPASSWD_COMMAND, "-b", settings.HTPASSWD_FILE, self.username, self.get_password()], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
stdout, stderr = p.communicate()
|
|
|
|
def save(self):
|
|
if self.update_user:
|
|
user = self.get_user()
|
|
else:
|
|
user = self.create_user()
|
|
user.set_password(self.get_password())
|
|
user.save()
|
|
self.save_password_file()
|
|
return user
|