imokforms.py

"""Place to put field and form definitions."""

from google.appengine.api import users
from google.appengine.ext import webapp
from google.appengine.ext.webapp import template, util
from google.appengine.ext.webapp.util import login_required
from google.appengine.api import mail

# must import template before importing django stuff
from google.appengine.ext.db import djangoforms
try:
  from django import newforms as forms
except ImportError:
  from django import forms
import django.core.exceptions

from datastore import *
from imokutils import *

from twitter import validate_password

class PhoneField(forms.CharField):
  def __init__(self, *args, **kwargs):
    kwargs['max_length'] = 12
    super(PhoneField, self).__init__(*args, **kwargs)

  def clean(self, value):
    if not Phone.is_valid_number(value):
      raise forms.ValidationError('Please enter a valid 10-digit US phone number')
  
    user = users.get_current_user()

    clean_number = Phone.normalize_number(value)
    if Phone.all().filter('verified =', True).filter('number =', clean_number).filter('user !=', user).count() > 0:
      raise forms.ValidationError('Phone number taken, please enter a different one.')

    return clean_number

#    cleanNumber = re.sub(r'\D+', '', value) # ignore punctuation
#    if len(cleanNumber) == 10:
#      return "+1%s" % cleanNumber
#    elif len(cleanNumber) == 11 and cleanNumber.startswith('1'):
#      return "+%s" % cleanNumber
#    else:
#      raise forms.ValidationError('Please enter a valid 10-digit US phone number')
    

class UserProfileForm(djangoforms.ModelForm):
  phoneNumber = PhoneField(label="Mobile phone number*")
  twitterUsername = forms.CharField(required=False, label="Twitter Username", max_length=40)
  twitterPassword = forms.CharField(required=False, label="Twitter Password", widget=forms.PasswordInput(), max_length=140)

  def clean_twitterPassword(self):
    username = self.clean_data['twitterUsername']
    password = self.clean_data['twitterPassword']

    if not username and not password:
      # removing twitter info
      return password

    if validate_password(username, password) is False:
      raise forms.ValidationError("Your Twitter password is invalid!")

    # Always return the cleaned data, whether you have changed it or
    # not.
    return password


  def saveWithPhone(self):
    editedProfile = self.save(commit=False)
    editedProfile.twitter_password = self._cleaned_data()['twitterPassword']
    editedProfile.twitter_username = self._cleaned_data()['twitterUsername']
    editedProfile.put()

    phone = getPhone(createIfNeeded=True)
    newNumber = self._cleaned_data()['phoneNumber']
    phoneChanged = (newNumber != phone.number)
    if phoneChanged:
      phone.number = newNumber
      phone.code = ''
      phone.verified = False
      phone.put()
    return phoneChanged
  
  class Meta:
    model = ImokUser
    exclude = ['account', 'twitter_username', 'twitter_password']