repo_permission.py
63 lines
| 2.3 KiB
| text/x-python
|
PythonLexer
|
r1586 | # -*- coding: utf-8 -*- | ||
""" | ||||
rhodecode.model.users_group | ||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~ | ||||
repository permission model for RhodeCode | ||||
:created_on: Oct 1, 2011 | ||||
:author: nvinot | ||||
:copyright: (C) 2011-2011 Nicolas Vinot <aeris@imirhil.fr> | ||||
: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 <http://www.gnu.org/licenses/>. | ||||
import logging | ||||
from rhodecode.model.db import BaseModel, RepoToPerm, Permission | ||||
from rhodecode.model.meta import Session | ||||
|
r1593 | log = logging.getLogger(__name__) | ||
|
r1586 | |||
|
r1593 | class RepositoryPermissionModel(BaseModel): | ||
def get_user_permission(self, repository, user): | ||||
|
r1586 | return RepoToPerm.query() \ | ||
|
r1593 | .filter(RepoToPerm.user == user) \ | ||
.filter(RepoToPerm.repository == repository) \ | ||||
|
r1586 | .scalar() | ||
|
r1593 | def update_user_permission(self, repository, user, permission): | ||
permission = Permission.get_by_key(permission) | ||||
current = self.get_user_permission(repository, user) | ||||
|
r1586 | if current: | ||
if not current.permission is permission: | ||||
current.permission = permission | ||||
else: | ||||
p = RepoToPerm() | ||||
p.user = user | ||||
p.repository = repository | ||||
p.permission = permission | ||||
|
r1593 | Session.add(p) | ||
|
r1586 | Session.commit() | ||
|
r1593 | def delete_user_permission(self, repository, user): | ||
current = self.get_user_permission(repository, user) | ||||
|
r1589 | if current: | ||
|
r1593 | Session.delete(current) | ||
|
r1589 | Session.commit() | ||
|
r1586 | |||
|
r1593 | def update_or_delete_user_permission(self, repository, user, permission): | ||
|
r1586 | if permission: | ||
|
r1593 | self.update_user_permission(repository, user, permission) | ||
|
r1586 | else: | ||
|
r1593 | self.delete_user_permission(repository, user) | ||