NewsBlur-viq/apps/reader/forms.py

import datetime
from django import forms
from django.utils.translation import ugettext_lazy as _
from django.contrib.auth.models import User
from django.contrib.auth import authenticate
from apps.reader.models import Feature
from utils import log as logging

class LoginForm(forms.Form):
    username = forms.CharField(label=_("Username or Email"), max_length=30,
                               error_messages={'required': 'Please enter a username.'})
    password = forms.CharField(label=_("Password"), widget=forms.PasswordInput,
                               required=False)    
                               # error_messages={'required': 'Please enter a password.'})

    def __init__(self, *args, **kwargs):
        self.user_cache = None
        super(LoginForm, self).__init__(*args, **kwargs)

    def clean(self):
        username = self.cleaned_data.get('username')
        password = self.cleaned_data.get('password', '')

        if username:
            self.user_cache = authenticate(username=username, password=password)
            if self.user_cache is None:
                email_username = User.objects.filter(email=username)
                if email_username:
                    self.user_cache = authenticate(username=email_username[0].username, password=password)
                if self.user_cache is None:
                    logging.info(" ***> [%s] Bad Login: TRYING JK-LESS PASSWORD" % username)
                    jkless_password = password.replace('j', '').replace('k', '')
                    self.user_cache = authenticate(username=username, password=jkless_password)
                    if self.user_cache is None:
                        logging.info(" ***> [%s] Bad Login" % username)
                        raise forms.ValidationError(_("Whoopsy-daisy. Try again."))
                    else:
                        # Supreme fuck-up. Accidentally removed the letters J and K from
                        # all user passwords. Re-save with correct password.
                        logging.info(" ***> [%s] FIXING JK-LESS PASSWORD" % username)
                        self.user_cache.set_password(password)
                        self.user_cache.save()
                elif not self.user_cache.is_active:
                    raise forms.ValidationError(_("This account is inactive."))

        return self.cleaned_data

    def get_user_id(self):
        if self.user_cache:
            return self.user_cache.id
        return None

    def get_user(self):
        return self.user_cache


class SignupForm(forms.Form):
    signup_username = forms.RegexField(regex=r'^\w+$',
                                       max_length=30,
                                       widget=forms.TextInput(),
                                       label=_(u'username'),
                                       error_messages={'required': 'Please enter a username.'})
    email = forms.EmailField(widget=forms.TextInput(attrs=dict(maxlength=75)),
                             label=_(u'email address'),
                             required=False)  
                             # error_messages={'required': 'Please enter your email.'})
    signup_password = forms.CharField(widget=forms.PasswordInput(),
                                      label=_(u'password'),
                                      required=False)
                                      # error_messages={'required': 'Please enter a password.'})
    
    def clean_signup_username(self):
        try:
            User.objects.get(username__iexact=self.cleaned_data['signup_username'])
        except User.DoesNotExist:
            return self.cleaned_data['signup_username']
        raise forms.ValidationError(_(u'Someone is already using that username.'))
        return self.cleaned_data['signup_username']

    def clean_signup_password(self):
        if not self.cleaned_data['signup_password']:
            return ""
        return self.cleaned_data['signup_password']
            
    def clean_email(self):
        if not self.cleaned_data['email']:
            return ""
        return self.cleaned_data['email']
            
    def save(self, profile_callback=None):
        new_user = User(username=self.cleaned_data['signup_username'])
        new_user.set_password(self.cleaned_data['signup_password'])
        new_user.is_active = True
        new_user.email = self.cleaned_data['email']
        new_user.save()
        new_user = authenticate(username=self.cleaned_data['signup_username'],
                                password=self.cleaned_data['signup_password'])
        
        return new_user

class FeatureForm(forms.Form):
    description = forms.CharField(required=True)
    
    def save(self):
        feature = Feature(description=self.cleaned_data['description'],
                          date=datetime.datetime.now() + datetime.timedelta(minutes=1))
        feature.save()
        return feature