diff --git a/rhodecode/model/users_group.py b/rhodecode/model/users_group.py new file mode 100644 --- /dev/null +++ b/rhodecode/model/users_group.py @@ -0,0 +1,89 @@ +# -*- coding: utf-8 -*- +""" + rhodecode.model.users_group + ~~~~~~~~~~~~~~~~~~~~~~~~~~~ + + users group model for RhodeCode + + :created_on: Oct 1, 2011 + :author: nvinot + :copyright: (C) 2011-2011 Nicolas Vinot + :license: GPLv3, see COPYING for more details. +""" +# 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, either version 3 of the License, or +# (at your option) any later version. +# +# 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, see . + +import logging +import traceback + +from rhodecode.model import BaseModel +from rhodecode.model.caching_query import FromCache +from rhodecode.model.db import UsersGroupMember, UsersGroup + +log = logging.getLogger(__name__) + +class UsersGroupModel(BaseModel): + + def get(self, users_group_id, cache = False): + users_group = UsersGroup.query() + if cache: + users_group = users_group.options(FromCache("sql_cache_short", + "get_users_group_%s" % users_group_id)) + return users_group.get(users_group_id) + + def get_by_name(self, name, cache = False, case_insensitive = False): + users_group = UsersGroup.query() + if case_insensitive: + users_group = users_group.filter(UsersGroup.users_group_name.ilike(name)) + else: + users_group = users_group.filter(UsersGroup.users_group_name == name) + if cache: + users_group = users_group.options(FromCache("sql_cache_short", + "get_users_group_%s" % name)) + return users_group.scalar() + + def create(self, form_data): + try: + new_users_group = UsersGroup() + for k, v in form_data.items(): + setattr(new_users_group, k, v) + + self.sa.add(new_users_group) + self.sa.commit() + return new_users_group + except: + log.error(traceback.format_exc()) + self.sa.rollback() + raise + + def add_user_to_group(self, users_group, user): + for m in users_group.members: + u = m.user + if u.user_id == user.user_id: + return m + + try: + users_group_member = UsersGroupMember() + users_group_member.user = user + users_group_member.users_group = users_group + + users_group.members.append(users_group_member) + user.group_member.append(users_group_member) + + self.sa.add(users_group_member) + self.sa.commit() + return users_group_member + except: + log.error(traceback.format_exc()) + self.sa.rollback() + raise