1 files changed, 141 insertions, 0 deletions
diff --git a/97suifangqa/apps/sfaccount/forms.py b/97suifangqa/apps/sfaccount/forms.py
new file mode 100644
index 0000000..d2a3bf1
--- /dev/null
+++ b/97suifangqa/apps/sfaccount/forms.py
@@ -0,0 +1,141 @@
+# -*- coding: utf-8 -*-
+
+from django import forms
+from django.template import loader
+from django.utils.http import int_to_base36
+from django.contrib.auth.models import User
+from django.contrib.auth.tokens import default_token_generator
+from django.contrib.sites.models import get_current_site
+
+from django.utils.translation import ugettext, ugettext_lazy as _
+
+from sfaccount.tasks import send_mail
+
+import re
+
+
+# AccountForm {{{
+class AccountForm(forms.Form):
+    """
+    form for signing up a new account
+    """
+    username = forms.RegexField(regex=r'^[A-Za-z0-9_-]+$',
+            max_length=30, label=u"用户名",
+            help_text=u"由字母、数字和下划线组成，长度6-30位",
+            error_messages={'invalid': u"用户名仅能包含字母、数字和下划线"},
+    )
+    email = forms.EmailField(max_length=75, label=u"邮箱")
+    password1 = forms.CharField(label=u"密码", max_length=30,
+            help_text=u"密码长度6-30位",
+            widget=forms.PasswordInput)
+    password2 = forms.CharField(label=u"确认密码", max_length=30,
+            widget=forms.PasswordInput)
+
+    def clean_username(self):
+        username = self.cleaned_data['username']
+        # check length
+        if len(username) < 6:
+            raise forms.ValidationError(u'用户名长度需大于6位')
+        # check first letter
+        p = re.compile('[a-zA-Z_]')
+        if p.match(username[0]):
+            pass
+        else:
+            raise forms.ValidationError(u'首字母必须是字母或下划线')
+        # check if exists
+        try:
+            User.objects.get(username=username)
+        except User.DoesNotExist:
+            return username
+        raise forms.ValidationError(u'用户名已经被占用')
+
+    def clean_email(self):
+        try:
+            User.objects.get(email__iexact=self.cleaned_data['email'])
+        except User.DoesNotExist:
+            return self.cleaned_data['email']
+        raise forms.ValidationError(u'邮箱地址已经被占用')
+
+    def clean_password1(self):
+        password1 = self.cleaned_data['password1']
+        if len(password1) < 6:
+            raise forms.ValidationError(u'密码长度需大于6位')
+        return password1
+
+    def clean(self):
+        cd = self.cleaned_data
+        if 'password1' in cd and 'password2' in cd:
+            if cd['password1'] != cd['password2']:
+                raise forms.ValidationError(u'两次输入的密码不一致')
+        #
+        return cd
+# }}}
+
+
+# SFPasswordResetForm {{{
+class SFPasswordResetForm(forms.Form):
+    """
+    to replace django's 'PasswordResetForm'
+    to use djcelery's async send mail
+    """
+    error_messages = {
+        'unknown': _("That e-mail address doesn't have an associated "
+                     "user account. Are you sure you've registered?"),
+        'unusable': _("The user account associated with this e-mail "
+                      "address cannot reset the password."),
+    }
+    email = forms.EmailField(label=_("E-mail"), max_length=75)
+
+    def save(self, domain_override=None,
+             subject_template_name='registration/password_reset_subject.txt',
+             email_template_name='registration/password_reset_email.txt',
+             use_https=False, token_generator=default_token_generator,
+             from_email=None, request=None,
+             html_email_template_name=None):
+        """
+        Generates a one-use only link for resetting password
+        and sends to the user.
+        """
+        # validate first
+        if not self.is_valid():
+            return self
+        # validated: has 'self.cleaned_data'
+        email = self.cleaned_data['email']
+        users = User.objects.filter(email__iexact=email)
+        if not len(users):
+            raise forms.ValidationError(self.error_messages['unknown'])
+        for user in users:
+            # make sure that no email is sent to a user that actually
+            # has a password marked as unusable
+            if not user.has_usable_password():
+                continue
+            if not domain_override:
+                current_site = get_current_site(request)
+                site_name = current_site.name
+                domain = current_site.domain
+            else:
+                site_name = domain = domain_override
+            c = {
+                'email': user.email,
+                'domain': domain,
+                'site_name': site_name,
+                'uid': int_to_base36(user.id),
+                'user': user,
+                'token': token_generator.make_token(user),
+                'protocol': 'https' if use_https else 'http',
+            }
+            subject = loader.render_to_string(subject_template_name, c)
+            # Email subject *must not* contain newlines
+            subject = ''.join(subject.splitlines())
+            body_text = loader.render_to_string(email_template_name, c)
+            # html email
+            if html_email_template_name:
+                body_html = loader.render_to_string(html_email_template_name, c)
+            else:
+                body_html = None
+            # send mail
+            to = user.email
+            send_mail(to, subject, body_text, body_html)
+# }}}
+
+