Files @ a26f48ad7a8a
Branch filter:

Location: kallithea/pylons_app/model/user_model.py

Marcin Kuzminski
fixes issue #16 reimplementation of database repository, for using generic pk instead of repo naming as pk. Which caused to many problems.
Fixed issue when redirecting after repo name change to 404.
Removed update hook from basic app setup
#!/usr/bin/env python
# encoding: utf-8
# Model for users
# Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>
 
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; version 2
# of the License or (at your opinion) any later version of the license.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# 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 create_registration(self, form_data):
        try:
            new_user = User()
            for k, v in form_data.items():
                if k != 'admin' or k != 'active':
                    setattr(new_user, k, v)
                setattr(new_user, 'active', True)
                
            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