Mercurial > kallithea
comparison rhodecode/model/permission_model.py @ 547:1e757ac98988
renamed project to rhodecode
author | Marcin Kuzminski <marcin@python-works.com> |
---|---|
date | Wed, 06 Oct 2010 03:18:16 +0200 |
parents | pylons_app/model/permission_model.py@63c697d1a631 |
children | 7e536d1af60d |
comparison
equal
deleted
inserted
replaced
546:7c2f5e4d7bbf | 547:1e757ac98988 |
---|---|
1 #!/usr/bin/env python | |
2 # encoding: utf-8 | |
3 # Model for permissions | |
4 # Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com> | |
5 | |
6 # This program is free software; you can redistribute it and/or | |
7 # modify it under the terms of the GNU General Public License | |
8 # as published by the Free Software Foundation; version 2 | |
9 # of the License or (at your opinion) any later version of the license. | |
10 # | |
11 # This program is distributed in the hope that it will be useful, | |
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 # GNU General Public License for more details. | |
15 # | |
16 # You should have received a copy of the GNU General Public License | |
17 # along with this program; if not, write to the Free Software | |
18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, | |
19 # MA 02110-1301, USA. | |
20 """ | |
21 Created on Aug 20, 2010 | |
22 Model for permissions | |
23 @author: marcink | |
24 """ | |
25 | |
26 from pylons.i18n.translation import _ | |
27 from rhodecode.model.db import User, Permission, UserToPerm, RepoToPerm | |
28 from rhodecode.model.meta import Session | |
29 import logging | |
30 import traceback | |
31 log = logging.getLogger(__name__) | |
32 | |
33 | |
34 class PermissionModel(object): | |
35 | |
36 def __init__(self): | |
37 self.sa = Session() | |
38 | |
39 def get_default(self): | |
40 return self.sa.query(User).filter(User.username == 'default').scalar() | |
41 | |
42 def get_permission(self, id): | |
43 return self.sa.query(Permission).get(id) | |
44 | |
45 def get_permission_by_name(self, name): | |
46 return self.sa.query(Permission)\ | |
47 .filter(Permission.permission_name == name).scalar() | |
48 | |
49 | |
50 def update(self, form_result): | |
51 perm_user = self.sa.query(User)\ | |
52 .filter(User.username == form_result['perm_user_name']).scalar() | |
53 u2p = self.sa.query(UserToPerm).filter(UserToPerm.user == perm_user).all() | |
54 if len(u2p) != 3: | |
55 raise Exception('There is more than 3 defined \ | |
56 permissions for defualt user. This should not happen please verify\ | |
57 your database') | |
58 | |
59 try: | |
60 #stage 1 change defaults | |
61 for p in u2p: | |
62 if p.permission.permission_name.startswith('repository.'): | |
63 p.permission = self.get_permission_by_name(form_result['default_perm']) | |
64 self.sa.add(p) | |
65 | |
66 if p.permission.permission_name.startswith('hg.register.'): | |
67 p.permission = self.get_permission_by_name(form_result['default_register']) | |
68 self.sa.add(p) | |
69 | |
70 if p.permission.permission_name.startswith('hg.create.'): | |
71 p.permission = self.get_permission_by_name(form_result['default_create']) | |
72 self.sa.add(p) | |
73 #stage 2 update all default permissions for repos if checked | |
74 if form_result['overwrite_default'] == 'true': | |
75 for r2p in self.sa.query(RepoToPerm).filter(RepoToPerm.user == perm_user).all(): | |
76 r2p.permission = self.get_permission_by_name(form_result['default_perm']) | |
77 self.sa.add(r2p) | |
78 | |
79 self.sa.commit() | |
80 except: | |
81 log.error(traceback.format_exc()) | |
82 self.sa.rollback() | |
83 raise | |
84 | |
85 | |
86 | |
87 | |
88 |