Files
@ 6b934c9607e7
Branch filter:
Location: kallithea/pylons_app/model/user_model.py
6b934c9607e7
4.2 KiB
text/x-python
Improved testing scenarios. Made test env creator
Fixed hg_model error message
some other tweeks and fixes
Models fixe for uniq email, and removed some extra not needed imports from model main module
Fixed hg_model error message
some other tweeks and fixes
Models fixe for uniq email, and removed some extra not needed imports from model main module
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 | #!/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_default(self):
return self.sa.query(User).filter(User.username == 'default').scalar()
def get_user(self, id):
return self.sa.query(User).get(id)
def get_user_by_name(self,name):
return self.sa.query(User).filter(User.username == name).scalar()
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':
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, uid, form_data):
try:
new_user = self.sa.query(User).get(uid)
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 update_my_account(self, uid, form_data):
try:
new_user = self.sa.query(User).get(uid)
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:
if k not in ['admin', 'active']:
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
|