#!/usr/bin/python2.5

#

# Copyright 2008 the Melange authors.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#   http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"""Generic cleaning methods.

"""

__authors__ = [

    '"Todd Larsen" <tlarsen@google.com>',

    '"Sverre Rabbelier" <sverre@rabbelier.nl>',

    '"Lennard de Rijk" <ljvderijk@gmail.com>',

    ]

from google.appengine.api import users

from django import forms

from soc.logic import validate

from soc.logic.models import user as user_logic

def clean_link_id(self):

  # convert to lowercase for user comfort

  link_id = self.cleaned_data.get('link_id').lower()

  if not validate.isLinkIdFormatValid(link_id):

    raise forms.ValidationError("This link ID is in wrong format.")

  return link_id

def clean_existing_user(field_name):

  """Check if the field_name field is a valid user.

  """

  def wrapped(self):

    link_id = self.cleaned_data.get(field_name).lower()

    if not validate.isLinkIdFormatValid(link_id):

      raise forms.ValidationError("This link ID is in wrong format.")

    user_entity = user_logic.logic.getForFields({'link_id': link_id}, 

        unique=True)

    if not user_entity:

      # user does not exist

      raise forms.ValidationError("This user does not exist.")

    return user_entity

  return wrapped

def clean_user_not_exist(field_name):

  """Check if the field_name value is a valid link_id and a user with the

     link id does not exist.

  """ 

  def wrapped(self):

    link_id = self.cleaned_data.get(field_name).lower()

    if not validate.isLinkIdFormatValid(link_id):

      raise forms.ValidationError("This link ID is in wrong format.")

    user_entity = user_logic.logic.getForFields({'link_id': link_id}, 

        unique=True)

    if user_entity:

      # user exists already

      raise forms.ValidationError("There is already a user with this link id.")

    return link_id

  return wrapped

def clean_users_not_same(field_name):

  """Check if the field_name field is a valid user and is not 

     equal to the current user.

  """

  def wrapped(self):

    clean_user_field = clean_existing_user(field_name)

    user_entity = clean_user_field(self)

    current_user_entity = user_logic.logic.getForCurrentAccount()

    if user_entity.key() == current_user_entity.key():

      # users are equal

      raise forms.ValidationError("You cannot enter yourself here.")

    return user_entity

  return wrapped

def clean_user_account(field_name):

  """Returns the User with the given field_name value.

  """

  def wrapped(self):

    email_adress = self.cleaned_data.get(field_name).lower()

    # get the user account for this email

    user_account = users.User(email_adress)

    return user_account

  return wrapped

def clean_user_account_not_in_use(field_name):

  """Check if the field_name value contains an email 

     address that hasn't been used for an existing account.

  """ 

  def wrapped(self):

    email_adress = self.cleaned_data.get(field_name).lower()

    # get the user account for this email and check if it's in use

    user_account = users.User(email_adress)

    fields = {'account': user_account}

    user_entity = user_logic.logic.getForFields(fields, unique=True)

    if user_entity or user_logic.logic.isFormerAccount(user_account):

      raise forms.ValidationError("There is already a user with this email adress.")

    return user_account

  return wrapped

def clean_feed_url(self):

  feed_url = self.cleaned_data.get('feed_url')

  if feed_url == '':

    # feed url not supplied (which is OK), so do not try to validate it

    return None

  if not validate.isFeedURLValid(feed_url):

    raise forms.ValidationError('This URL is not a valid ATOM or RSS feed.')

  return feed_url

def clean_url(field_name):

  """Clean method for cleaning a field belonging to a LinkProperty.

  """

  def wrapped(self):

    value = self.cleaned_data.get(field_name)

    # LinkProperty does not accept the empty string so we must return None

    if not value or value == u'':

      return None

    # call the Django URLField cleaning method to properly clean/validate this field

    return forms.URLField.clean(self.fields[field_name], value)

  return wrapped

def validate_user_edit(link_id_field, account_field):

  """Clean method for cleaning user edit form.

  Raises ValidationError if:

    -Another User has the given email address as account

    -Another User has the given email address in it's FormerAccounts list

  """

  def wrapper(self):

    cleaned_data = self.cleaned_data

    link_id = cleaned_data.get(link_id_field)

    user_account = cleaned_data.get(account_field)

    # if both fields were valid do this check

    if link_id and user_account:

      # get the user from the link_id in the form

      fields = {'link_id': link_id}

      user_entity = user_logic.logic.getForFields(fields, unique=True)

      former_accounts = user_entity.former_accounts

      # if it's not the user's current account or one of his former accounts

      if (user_entity.account != user_account  and 

          user_account not in former_accounts):

        # get the user having the given account

        fields = {'account': user_account}

        user_from_account_entity = user_logic.logic.getForFields(fields, 

            unique=True)

        # if there is a user with the given account or it's a former account

        if user_from_account_entity or user_logic.logic.isFormerAccount(user_account):

          # raise an error because this email address can't be used

          raise forms.ValidationError("There is already a user with this email adress.")

    return cleaned_data

  return wrapper