<name> = formencode.validators.<name of validator>

<name> must equal form name

list=[1,2,3,4,5]

for SELECT use formencode.All(OneOf(list), Int())

"""

from formencode import All

from formencode.validators import UnicodeString, OneOf, Int, Number, Regex, \

    Email, Bool, StringBoolean

from pylons import session

from pylons.i18n.translation import _

from pylons_app.lib.auth import get_crypt_password

import pylons_app.lib.helpers as h

from pylons_app.model import meta

from pylons_app.model.db import User, Repository

from sqlalchemy.exc import OperationalError

from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound

from webhelpers.pylonslib.secure_form import authentication_token

import datetime

import formencode

import logging

log = logging.getLogger(__name__)

#this is needed to translate the messages using _() in validators

class State_obj(object):

    _ = staticmethod(_)

#===============================================================================

# VALIDATORS

#===============================================================================

class ValidAuthToken(formencode.validators.FancyValidator):

    messages = {'invalid_token':_('Token mismatch')}

    def validate_python(self, value, state):

        if value != authentication_token():

            raise formencode.Invalid(self.message('invalid_token', state,

                                            search_number=value), value, state)

class ValidUsername(formencode.validators.FancyValidator):

    def validate_python(self, value, state):

        if value in ['default', 'new_user']:

            raise formencode.Invalid(_('Invalid username'), value, state)

class ValidPassword(formencode.validators.FancyValidator):

    def to_python(self, value, state):

class ValidAuth(formencode.validators.FancyValidator):

    messages = {

            'invalid_password':_('invalid password'),

            'invalid_login':_('invalid user name'),

            'disabled_account':_('Your acccount is disabled')

            }

    #error mapping

    e_dict = {'username':messages['invalid_login'],

              'password':messages['invalid_password']}

    e_dict_disable = {'username':messages['disabled_account']}

    def validate_python(self, value, state):

        sa = meta.Session

        crypted_passwd = get_crypt_password(value['password'])

        username = value['username']

        try:

            user = sa.query(User).filter(User.username == username).one()

        except (NoResultFound, MultipleResultsFound, OperationalError) as e:

            log.error(e)

            user = None

            raise formencode.Invalid(self.message('invalid_password',

                                     state=State_obj), value, state,

                                     error_dict=self.e_dict)            

        if user:

            if user.active:

                if user.username == username and user.password == crypted_passwd:

                    from pylons_app.lib.auth import AuthUser

                    auth_user = AuthUser()

                    auth_user.username = username

                    auth_user.is_authenticated = True

                    auth_user.is_admin = user.admin

                    auth_user.user_id = user.user_id

                    session['hg_app_user'] = auth_user

                    session.save()

                    log.info('user %s is now authenticated', username)

                    try:

                        user.last_login = datetime.datetime.now()

                        sa.add(user)

                        sa.commit()                        

                    except (OperationalError) as e:

                        log.error(e)

                        sa.rollback()

                    return value

                else:

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

# MA  02110-1301, USA.

"""

Created on April 9, 2010

Model for users

@author: marcink

"""

from pylons_app.model.db import User

from pylons_app.model.meta import Session

from pylons.i18n.translation import _

import logging

log = logging.getLogger(__name__)

class DefaultUserException(Exception):pass

class UserModel(object):

    def __init__(self):

        self.sa = Session() 

    def get_user(self, id):

        return self.sa.query(User).get(id)

    def create(self, form_data):

        try:

            new_user = User()

            for k, v in form_data.items():

                setattr(new_user, k, v)

            self.sa.add(new_user)

            self.sa.commit()

        except Exception as e:

            log.error(e)

            self.sa.rollback()

            raise      

    def update(self, id, form_data):

        try:

            new_user = self.sa.query(User).get(id)

            if new_user.username == 'default':

                raise DefaultUserException(

                                _("You can't Edit this user since it's" 

                                  " crucial for entire application"))

            for k, v in form_data.items():

                if k == 'new_password' and v != '':

                    new_user.password = v

                else:

                    setattr(new_user, k, v)

            self.sa.add(new_user)

            self.sa.commit()

        except Exception as e:

            log.error(e)

            self.sa.rollback()

            raise      

    def delete(self, id):

        try:

            user = self.sa.query(User).get(id)

            if user.username == 'default':

                raise DefaultUserException(

                                _("You can't remove this user since it's" 

                                  " crucial for entire application"))

            self.sa.delete(user)

            self.sa.commit()            

        except Exception as e:

            log.error(e)

            self.sa.rollback()

            raise