user.py
230 lines
| 7.9 KiB
| text/x-python
|
PythonLexer
r761 | # -*- coding: utf-8 -*- | |||
""" | ||||
r956 | rhodecode.model.user | |||
~~~~~~~~~~~~~~~~~~~~ | ||||
r761 | ||||
users model for RhodeCode | ||||
r811 | ||||
r761 | :created_on: Apr 9, 2010 | |||
:author: marcink | ||||
r902 | :copyright: (C) 2009-2011 Marcin Kuzminski <marcin@python-works.com> | |||
r761 | :license: GPLv3, see COPYING for more details. | |||
""" | ||||
r629 | # 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. | ||||
r750 | ||||
r629 | import logging | |||
import traceback | ||||
r761 | from pylons.i18n.translation import _ | |||
from rhodecode.model import BaseModel | ||||
from rhodecode.model.caching_query import FromCache | ||||
from rhodecode.model.db import User | ||||
r629 | ||||
r761 | from rhodecode.lib.exceptions import DefaultUserException, UserOwnsReposException | |||
r713 | ||||
r761 | from sqlalchemy.exc import DatabaseError | |||
r1116 | from rhodecode.lib import generate_api_key | |||
r761 | ||||
log = logging.getLogger(__name__) | ||||
r629 | ||||
r752 | class UserModel(BaseModel): | |||
r629 | ||||
def get(self, user_id, cache=False): | ||||
user = self.sa.query(User) | ||||
if cache: | ||||
user = user.options(FromCache("sql_cache_short", | ||||
"get_user_%s" % user_id)) | ||||
return user.get(user_id) | ||||
r742 | def get_by_username(self, username, cache=False, case_insensitive=False): | |||
r750 | ||||
r742 | if case_insensitive: | |||
user = self.sa.query(User).filter(User.username.ilike(username)) | ||||
else: | ||||
user = self.sa.query(User)\ | ||||
.filter(User.username == username) | ||||
r629 | if cache: | |||
user = user.options(FromCache("sql_cache_short", | ||||
"get_user_%s" % username)) | ||||
return user.scalar() | ||||
def create(self, form_data): | ||||
try: | ||||
new_user = User() | ||||
for k, v in form_data.items(): | ||||
setattr(new_user, k, v) | ||||
r1116 | new_user.api_key = generate_api_key(form_data['username']) | |||
r629 | self.sa.add(new_user) | |||
self.sa.commit() | ||||
except: | ||||
log.error(traceback.format_exc()) | ||||
self.sa.rollback() | ||||
raise | ||||
Thayne Harbaugh
|
r991 | def create_ldap(self, username, password, user_dn, attrs): | ||
r705 | """ | |||
Checks if user is in database, if not creates this user marked | ||||
as ldap user | ||||
:param username: | ||||
:param password: | ||||
Thayne Harbaugh
|
r991 | :param user_dn: | ||
:param attrs: | ||||
r705 | """ | |||
r750 | from rhodecode.lib.auth import get_crypt_password | |||
r761 | log.debug('Checking for such ldap account in RhodeCode database') | |||
if self.get_by_username(username, case_insensitive=True) is None: | ||||
r705 | try: | |||
new_user = User() | ||||
Thayne Harbaugh
|
r991 | new_user.username = username.lower() # add ldap account always lowercase | ||
r750 | new_user.password = get_crypt_password(password) | |||
r1116 | new_user.api_key = generate_api_key(username) | |||
Thayne Harbaugh
|
r991 | new_user.email = attrs['email'] | ||
r705 | new_user.active = True | |||
Thayne Harbaugh
|
r991 | new_user.ldap_dn = user_dn | ||
new_user.name = attrs['name'] | ||||
new_user.lastname = attrs['lastname'] | ||||
r705 | ||||
self.sa.add(new_user) | ||||
self.sa.commit() | ||||
return True | ||||
r761 | except (DatabaseError,): | |||
r705 | log.error(traceback.format_exc()) | |||
self.sa.rollback() | ||||
raise | ||||
r761 | log.debug('this %s user exists skipping creation of ldap account', | |||
username) | ||||
r705 | return False | |||
r629 | def create_registration(self, form_data): | |||
r689 | from rhodecode.lib.celerylib import tasks, run_task | |||
r629 | 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() | ||||
r689 | body = ('New user registration\n' | |||
'username: %s\n' | ||||
'email: %s\n') | ||||
body = body % (form_data['username'], form_data['email']) | ||||
run_task(tasks.send_email, None, | ||||
_('[RhodeCode] New User registration'), | ||||
body) | ||||
r629 | except: | |||
log.error(traceback.format_exc()) | ||||
self.sa.rollback() | ||||
raise | ||||
def update(self, user_id, form_data): | ||||
try: | ||||
r1116 | user = self.get(user_id, cache=False) | |||
if user.username == 'default': | ||||
r629 | raise DefaultUserException( | |||
_("You can't Edit this user since it's" | ||||
" crucial for entire application")) | ||||
r713 | ||||
r629 | for k, v in form_data.items(): | |||
if k == 'new_password' and v != '': | ||||
r1116 | user.password = v | |||
user.api_key = generate_api_key(user.username) | ||||
r629 | else: | |||
r1116 | setattr(user, k, v) | |||
r629 | ||||
r1116 | self.sa.add(user) | |||
r629 | self.sa.commit() | |||
except: | ||||
log.error(traceback.format_exc()) | ||||
self.sa.rollback() | ||||
raise | ||||
def update_my_account(self, user_id, form_data): | ||||
try: | ||||
r1116 | user = self.get(user_id, cache=False) | |||
if user.username == 'default': | ||||
r629 | 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 != '': | ||||
r1116 | user.password = v | |||
user.api_key = generate_api_key(user.username) | ||||
r629 | else: | |||
if k not in ['admin', 'active']: | ||||
r1116 | setattr(user, k, v) | |||
r629 | ||||
r1116 | self.sa.add(user) | |||
r629 | self.sa.commit() | |||
except: | ||||
log.error(traceback.format_exc()) | ||||
self.sa.rollback() | ||||
raise | ||||
def delete(self, user_id): | ||||
try: | ||||
user = self.get(user_id, cache=False) | ||||
if user.username == 'default': | ||||
raise DefaultUserException( | ||||
_("You can't remove this user since it's" | ||||
" crucial for entire application")) | ||||
r713 | if user.repositories: | |||
raise UserOwnsReposException(_('This user still owns %s ' | ||||
'repositories and cannot be ' | ||||
'removed. Switch owners or ' | ||||
'remove those repositories') \ | ||||
% user.repositories) | ||||
r629 | self.sa.delete(user) | |||
self.sa.commit() | ||||
except: | ||||
log.error(traceback.format_exc()) | ||||
self.sa.rollback() | ||||
raise | ||||
def reset_password(self, data): | ||||
from rhodecode.lib.celerylib import tasks, run_task | ||||
run_task(tasks.reset_user_password, data['email']) | ||||
r673 | ||||
def fill_data(self, user): | ||||
""" | ||||
Fills user data with those from database and log out user if not | ||||
present in database | ||||
:param user: | ||||
""" | ||||
r686 | ||||
if not hasattr(user, 'user_id') or user.user_id is None: | ||||
raise Exception('passed in user has to have the user_id attribute') | ||||
r673 | log.debug('filling auth user data') | |||
try: | ||||
dbuser = self.get(user.user_id) | ||||
user.username = dbuser.username | ||||
user.is_admin = dbuser.admin | ||||
user.name = dbuser.name | ||||
user.lastname = dbuser.lastname | ||||
user.email = dbuser.email | ||||
except: | ||||
log.error(traceback.format_exc()) | ||||
user.is_authenticated = False | ||||
return user | ||||