user_model.py
105 lines
| 3.3 KiB
| text/x-python
|
PythonLexer
r238 | #!/usr/bin/env python | |||
# encoding: utf-8 | ||||
r252 | # 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 | ||||
""" | ||||
r238 | from pylons_app.model.db import User | |||
from pylons_app.model.meta import Session | ||||
r314 | from pylons.i18n.translation import _ | |||
r265 | import logging | |||
log = logging.getLogger(__name__) | ||||
r238 | ||||
r314 | class DefaultUserException(Exception):pass | |||
r238 | 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() | ||||
r265 | except Exception as e: | |||
log.error(e) | ||||
r238 | self.sa.rollback() | |||
raise | ||||
r363 | 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 | ||||
r238 | def update(self, id, form_data): | |||
try: | ||||
new_user = self.sa.query(User).get(id) | ||||
r314 | if new_user.username == 'default': | |||
raise DefaultUserException( | ||||
_("You can't Edit this user since it's" | ||||
" crucial for entire application")) | ||||
r238 | 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() | ||||
r265 | except Exception as e: | |||
log.error(e) | ||||
r238 | self.sa.rollback() | |||
raise | ||||
r265 | ||||
def delete(self, id): | ||||
r314 | ||||
r265 | try: | |||
r314 | ||||
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) | ||||
r265 | self.sa.commit() | |||
except Exception as e: | ||||
log.error(e) | ||||
self.sa.rollback() | ||||
raise | ||||