Mercurial > kallithea
diff rhodecode/model/repo_permission.py @ 1586:2ccb32ddcfd7 beta
Add API for repositories and groups (creation, permission)
author | Nicolas VINOT <aeris@imirhil.fr> |
---|---|
date | Sun, 02 Oct 2011 17:39:52 +0200 |
parents | |
children | 307ec693bdf2 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/rhodecode/model/repo_permission.py Sun Oct 02 17:39:52 2011 +0200 @@ -0,0 +1,62 @@ +# -*- 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 + +log = logging.getLogger(__name__) + +class RepositoryPermissionModel(BaseModel): + def getUserPermission(self, repository, user): + return RepoToPerm.query() \ + .filter(RepoToPerm.user == user) \ + .filter(RepoToPerm.repository == repository) \ + .scalar() + + def updateUserPermission(self, repository, user, permission): + permission = Permission.get_by_key(permission) + current = self.getUserPermission(repository, user) + if current: + if not current.permission is permission: + current.permission = permission + else: + p = RepoToPerm() + p.user = user + p.repository = repository + p.permission = permission + Session.add(p) + Session.commit() + + def deleteUserPermission(self, repository, user): + current = self.getUserPermission(repository, user) + Session.delete(current) + Session.commit() + + def updateOrDeleteUserPermission(self, repository, user, permission): + if permission: + self.updateUserPermission(repository, user, permission) + else: + self.deleteUserPermission(repository, user) \ No newline at end of file