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