diff --git a/rhodecode/model/user_group.py b/rhodecode/model/users_group.py copy from rhodecode/model/user_group.py copy to rhodecode/model/users_group.py --- a/rhodecode/model/user_group.py +++ b/rhodecode/model/users_group.py @@ -24,3 +24,56 @@ # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, # MA 02110-1301, USA. + +import logging +import traceback + +from pylons.i18n.translation import _ + +from rhodecode.model import BaseModel +from rhodecode.model.caching_query import FromCache +from rhodecode.model.db import UsersGroup + +from rhodecode.lib.exceptions import DefaultUserException, UserOwnsReposException + +from sqlalchemy.exc import DatabaseError + +log = logging.getLogger(__name__) + + +class UsersGroupModel(BaseModel): + + def get(self, users_group_id, cache=False): + users_group = self.sa.query(UsersGroup) + 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_groupname(self, users_group_name, cache=False, case_insensitive=False): + + if case_insensitive: + user = self.sa.query(UsersGroup)\ + .filter(UsersGroup.users_group_name.ilike(users_group_name)) + else: + user = self.sa.query(UsersGroup)\ + .filter(UsersGroup.users_group_name == users_group_name) + if cache: + user = user.options(FromCache("sql_cache_short", + "get_user_%s" % users_group_name)) + return user.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() + except: + log.error(traceback.format_exc()) + self.sa.rollback() + raise +