comparison rhodecode/model/users_group.py @ 1713:54687aa00724 beta

Tests updates, Session refactoring
author Marcin Kuzminski <marcin@python-works.com>
date Wed, 23 Nov 2011 15:36:57 +0200
parents b76bb93db070
children 7d1fc253549e
comparison
equal deleted inserted replaced
1712:cac5109ac3b6 1713:54687aa00724
33 33
34 log = logging.getLogger(__name__) 34 log = logging.getLogger(__name__)
35 35
36 class UsersGroupModel(BaseModel): 36 class UsersGroupModel(BaseModel):
37 37
38 def __get_users_group(self, users_group):
39 return self.__get_instance(UsersGroup, users_group)
40
38 def get(self, users_group_id, cache = False): 41 def get(self, users_group_id, cache = False):
39 users_group = UsersGroup.query() 42 return UsersGroup.get(users_group_id)
40 if cache:
41 users_group = users_group.options(FromCache("sql_cache_short",
42 "get_users_group_%s" % users_group_id))
43 return users_group.get(users_group_id)
44 43
45 def get_by_name(self, name, cache = False, case_insensitive = False): 44 def get_by_name(self, name, cache = False, case_insensitive = False):
46 users_group = UsersGroup.query() 45 return UsersGroup.get_by_group_name(name, cache, case_insensitive)
47 if case_insensitive:
48 users_group = users_group.filter(UsersGroup.users_group_name.ilike(name))
49 else:
50 users_group = users_group.filter(UsersGroup.users_group_name == name)
51 if cache:
52 users_group = users_group.options(FromCache("sql_cache_short",
53 "get_users_group_%s" % name))
54 return users_group.scalar()
55 46
56 def create(self, form_data): 47 def create(self, form_data):
57 try: 48 try:
58 new_users_group = UsersGroup() 49 new_users_group = UsersGroup()
59 for k, v in form_data.items(): 50 for k, v in form_data.items():
64 return new_users_group 55 return new_users_group
65 except: 56 except:
66 log.error(traceback.format_exc()) 57 log.error(traceback.format_exc())
67 self.sa.rollback() 58 self.sa.rollback()
68 raise 59 raise
60
61
62 def create_(self, name, active=True):
63 new = UsersGroup()
64 new.users_group_name = name
65 new.users_group_active = active
66 self.sa.add(new)
67 return new
68
69 def delete(self, users_group):
70 obj = self.__get_users_group(users_group)
71 self.sa.delete(obj)
69 72
70 def add_user_to_group(self, users_group, user): 73 def add_user_to_group(self, users_group, user):
71 for m in users_group.members: 74 for m in users_group.members:
72 u = m.user 75 u = m.user
73 if u.user_id == user.user_id: 76 if u.user_id == user.user_id: