130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
import datetime
|
|
import hashlib
|
|
import subprocess
|
|
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth.forms import PasswordResetForm
|
|
from django.contrib.sites.models import Site
|
|
from django.utils.translation import ugettext, ugettext_lazy as _
|
|
|
|
from ietf.utils.mail import send_mail
|
|
from ietf.utils import debug
|
|
#from ietf.person.models import Person, Email
|
|
|
|
|
|
class RegistrationForm(forms.Form):
|
|
|
|
email = forms.EmailField(label="Your email")
|
|
realm = 'IETF'
|
|
expire = 3
|
|
|
|
def save(self, *args, **kwargs):
|
|
# why is there a save when it doesn't save?
|
|
self.send_email()
|
|
return True
|
|
|
|
def send_email(self):
|
|
domain = Site.objects.get_current().domain
|
|
subject = '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(self.request, 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 reset at %s' % domain
|
|
from_email = settings.DEFAULT_FROM_EMAIL
|
|
today = datetime.date.today().strftime('%Y%m%d')
|
|
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(self.request, to_email, from_email, subject, 'registration/password_reset_email.txt', context)
|
|
|
|
|
|
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."))
|
|
|
|
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
|
|
return None
|
|
|
|
def get_user(self):
|
|
return User.objects.get(username=self.username)
|
|
|
|
@debug.trace
|
|
def save(self):
|
|
# if self.update_user:
|
|
# user = self.get_user()
|
|
# else:
|
|
# user = self.create_user()
|
|
# user.set_password(self.get_password())
|
|
# user.save()
|
|
# return user
|
|
debug.show("[settings.HTPASSWD_COMMAND, settings.HTPASSWD_FILE, self.username, self.get_password()]")
|
|
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()
|
|
debug.show('stdout')
|
|
debug.show('stderr')
|
|
return p.returncode
|
|
|
|
class TestEmailForm(forms.Form):
|
|
email = forms.EmailField(required=False)
|