Changeset - 0d26d46bd370
[Not reviewed]
default
0 1 0
Marcin Kuzminski - 15 years ago 2010-06-29 20:36:09
marcin@python-works.com
protected againts changing default user.
1 file changed with 15 insertions and 1 deletions:
0 comments (0 inline, 0 general)
pylons_app/model/user_model.py
Show inline comments
 
@@ -26,9 +26,12 @@ Model for users
 

	
 
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):
 
@@ -53,6 +56,10 @@ class UserModel(object):
 
    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 != '':
 
                    
 
@@ -68,8 +75,15 @@ class UserModel(object):
 
            raise      
 

	
 
    def delete(self, id):
 
        
 
        try:
 
            self.sa.delete(self.sa.query(User).get(id))
 
            
 
            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)
0 comments (0 inline, 0 general)