kallithea: rhodecode/model/repo.py comparison

comparison rhodecode/model/repo.py @ 2031:82a88013a3fd

merge 1.3 into stable

author	Marcin Kuzminski <marcin@python-works.com>
date	Sun, 26 Feb 2012 17:25:09 +0200
parents	24aeb43bbf51 14dffcfebb02
children	dc2584ba5fbc

comparison

equal deleted inserted replaced

-:ab0e122b38a7
+:82a88013a3fd
 Repository model for rhodecode
 :created_on: Jun 5, 2010
 :author: marcink
-:copyright: (C) 2009-2011 Marcin Kuzminski <marcin@python-works.com>
+:copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com>
 :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
 import shutil
 import logging
 import traceback
 from datetime import datetime
-from vcs.utils.lazy import LazyProperty
+from rhodecode.lib.vcs.backends import get_backend
-from vcs.backends import get_backend
+from rhodecode.lib import LazyProperty
 from rhodecode.lib import safe_str, safe_unicode
+from rhodecode.lib.caching_query import FromCache
+from rhodecode.lib.hooks import log_create_repository
 from rhodecode.model import BaseModel
-from rhodecode.model.caching_query import FromCache
+from rhodecode.model.db import Repository, UserRepoToPerm, User, Permission, \
-from rhodecode.model.db import Repository, RepoToPerm, User, Permission, \
+Statistics, UsersGroup, UsersGroupRepoToPerm, RhodeCodeUi, RepoGroup
-Statistics, UsersGroup, UsersGroupRepoToPerm, RhodeCodeUi, Group
 log = logging.getLogger(__name__)
 class RepoModel(BaseModel):
+def __get_user(self, user):
+return self._get_instance(User, user, callback=User.get_by_username)
+def __get_users_group(self, users_group):
+return self._get_instance(UsersGroup, users_group,
+callback=UsersGroup.get_by_group_name)
+def __get_repos_group(self, repos_group):
+return self._get_instance(RepoGroup, repos_group,
+callback=RepoGroup.get_by_group_name)
+def __get_repo(self, repository):
+return self._get_instance(Repository, repository,
+callback=Repository.get_by_repo_name)
+def __get_perm(self, permission):
+return self._get_instance(Permission, permission,
+callback=Permission.get_by_key)
 @LazyProperty
 def repos_path(self):
-"""Get's the repositories root path from database
+"""
+Get's the repositories root path from database
 """
 q = self.sa.query(RhodeCodeUi).filter(RhodeCodeUi.ui_key == '/').one()
 return q.ui_value
 if cache:
 repo = repo.options(FromCache("sql_cache_short",
 "get_repo_%s" % repo_id))
 return repo.scalar()
+def get_repo(self, repository):
+return self.__get_repo(repository)
 def get_by_repo_name(self, repo_name, cache=False):
 repo = self.sa.query(Repository)\
 .filter(Repository.repo_name == repo_name)
 if cache:
 repo = repo.options(FromCache("sql_cache_short",
 "get_repo_%s" % repo_name))
 return repo.scalar()
 def get_users_js(self):
 users = self.sa.query(User).filter(User.active == True).all()
 u_tmpl = '''{id:%s, fname:"%s", lname:"%s", nname:"%s"},'''
 for gr in users_groups])
 return users_groups_array
 def _get_defaults(self, repo_name):
 """
 Get's information about repository, and returns a dict for
 usage in forms
 :param repo_name:
 """
 repo_info = Repository.get_by_repo_name(repo_name)
 defaults.update({'g_perm_%s' % p.users_group.users_group_name:
 p.permission.permission_name})
 return defaults
 def update(self, repo_name, form_data):
 try:
 cur_repo = self.get_by_repo_name(repo_name, cache=False)
 # update permissions
 for member, perm, member_type in form_data['perms_updates']:
 if member_type == 'user':
-r2p = self.sa.query(RepoToPerm)\
+# this updates existing one
-.filter(RepoToPerm.user == User.get_by_username(member))\
+RepoModel().grant_user_permission(
-.filter(RepoToPerm.repository == cur_repo)\
+repo=cur_repo, user=member, perm=perm
-.one()
+)
-r2p.permission = self.sa.query(Permission)\
-.filter(Permission.permission_name ==
-perm).scalar()
-self.sa.add(r2p)
 else:
-g2p = self.sa.query(UsersGroupRepoToPerm)\
+RepoModel().grant_users_group_permission(
-.filter(UsersGroupRepoToPerm.users_group ==
+repo=cur_repo, group_name=member, perm=perm
-UsersGroup.get_by_group_name(member))\
+)
-.filter(UsersGroupRepoToPerm.repository ==
-cur_repo).one()
-g2p.permission = self.sa.query(Permission)\
-.filter(Permission.permission_name ==
-perm).scalar()
-self.sa.add(g2p)
 # set new permissions
 for member, perm, member_type in form_data['perms_new']:
 if member_type == 'user':
-r2p = RepoToPerm()
+RepoModel().grant_user_permission(
-r2p.repository = cur_repo
+repo=cur_repo, user=member, perm=perm
-r2p.user = User.get_by_username(member)
+)
-r2p.permission = self.sa.query(Permission)\
-.filter(Permission.
-permission_name == perm)\
-.scalar()
-self.sa.add(r2p)
 else:
-g2p = UsersGroupRepoToPerm()
+RepoModel().grant_users_group_permission(
-g2p.repository = cur_repo
+repo=cur_repo, group_name=member, perm=perm
-g2p.users_group = UsersGroup.get_by_group_name(member)
+)
-g2p.permission = self.sa.query(Permission)\
-.filter(Permission.
-permission_name == perm)\
-.scalar()
-self.sa.add(g2p)
 # update current repo
 for k, v in form_data.items():
 if k == 'user':
 cur_repo.user = User.get_by_username(v)
 elif k == 'repo_name':
 pass
 elif k == 'repo_group':
-cur_repo.group = Group.get(v)
+cur_repo.group = RepoGroup.get(v)
 else:
 setattr(cur_repo, k, v)
 new_name = cur_repo.get_new_name(form_data['repo_name'])
 if repo_name != new_name:
 # rename repository
 self.__rename_repo(old=repo_name, new=new_name)
-self.sa.commit()
 return cur_repo
 except:
 log.error(traceback.format_exc())
-self.sa.rollback()
 raise
 def create(self, form_data, cur_user, just_db=False, fork=False):
+from rhodecode.model.scm import ScmModel
 try:
 if fork:
-repo_name = form_data['fork_name']
+fork_parent_id = form_data['fork_parent_id']
-org_name = form_data['repo_name']
-org_full_name = org_name
+# repo name is just a name of repository
+# while repo_name_full is a full qualified name that is combined
-else:
+# with name and path of group
-org_name = repo_name = form_data['repo_name']
+repo_name = form_data['repo_name']
 repo_name_full = form_data['repo_name_full']
 new_repo = Repository()
 new_repo.enable_statistics = False
 for k, v in form_data.items():
 if k == 'repo_name':
-if fork:
+v = repo_name_full
-v = repo_name
-else:
-v = repo_name_full
 if k == 'repo_group':
 k = 'group_id'
 if k == 'description':
-v = safe_unicode(v) or repo_name
+v = v or repo_name
 setattr(new_repo, k, v)
 if fork:
-parent_repo = self.sa.query(Repository)\
+parent_repo = Repository.get(fork_parent_id)
-.filter(Repository.repo_name == org_full_name).one()
 new_repo.fork = parent_repo
 new_repo.user_id = cur_user.user_id
 self.sa.add(new_repo)
-#create default permission
+def _create_default_perms():
-repo_to_perm = RepoToPerm()
+# create default permission
-default = 'repository.read'
+repo_to_perm = UserRepoToPerm()
-for p in User.get_by_username('default').user_perms:
+default = 'repository.read'
-if p.permission.permission_name.startswith('repository.'):
+for p in User.get_by_username('default').user_perms:
-default = p.permission.permission_name
+if p.permission.permission_name.startswith('repository.'):
-break
+default = p.permission.permission_name
+break
-default_perm = 'repository.none' if form_data['private'] else default
+default_perm = 'repository.none' if form_data['private'] else default
-repo_to_perm.permission_id = self.sa.query(Permission)\
-.filter(Permission.permission_name == default_perm)\
+repo_to_perm.permission_id = self.sa.query(Permission)\
-.one().permission_id
+.filter(Permission.permission_name == default_perm)\
+.one().permission_id
-repo_to_perm.repository = new_repo
-repo_to_perm.user_id = User.get_by_username('default').user_id
+repo_to_perm.repository = new_repo
+repo_to_perm.user_id = User.get_by_username('default').user_id
-self.sa.add(repo_to_perm)
+self.sa.add(repo_to_perm)
+if fork:
+if form_data.get('copy_permissions'):
+repo = Repository.get(fork_parent_id)
+user_perms = UserRepoToPerm.query()\
+.filter(UserRepoToPerm.repository == repo).all()
+group_perms = UsersGroupRepoToPerm.query()\
+.filter(UsersGroupRepoToPerm.repository == repo).all()
+for perm in user_perms:
+UserRepoToPerm.create(perm.user, new_repo,
+perm.permission)
+for perm in group_perms:
+UsersGroupRepoToPerm.create(perm.users_group, new_repo,
+perm.permission)
+else:
+_create_default_perms()
+else:
+_create_default_perms()
 if not just_db:
 self.__create_repo(repo_name, form_data['repo_type'],
 form_data['repo_group'],
 form_data['clone_uri'])
-self.sa.commit()
+# now automatically start following this repository as owner
-#now automatically start following this repository as owner
-from rhodecode.model.scm import ScmModel
 ScmModel(self.sa).toggle_following_repo(new_repo.repo_id,
 cur_user.user_id)
+log_create_repository(new_repo.get_dict(),
+created_by=cur_user.username)
 return new_repo
 except:
 log.error(traceback.format_exc())
-self.sa.rollback()
 raise
 def create_fork(self, form_data, cur_user):
+"""
+Simple wrapper into executing celery task for fork creation
+:param form_data:
+:param cur_user:
+"""
 from rhodecode.lib.celerylib import tasks, run_task
 run_task(tasks.create_repo_fork, form_data, cur_user)
 def delete(self, repo):
+repo = self.__get_repo(repo)
 try:
 self.sa.delete(repo)
 self.__delete_repo(repo)
-self.sa.commit()
 except:
 log.error(traceback.format_exc())
-self.sa.rollback()
 raise
-def delete_perm_user(self, form_data, repo_name):
+def grant_user_permission(self, repo, user, perm):
+"""
+Grant permission for user on given repository, or update existing one
+if found
+:param repo: Instance of Repository, repository_id, or repository name
+:param user: Instance of User, user_id or username
+:param perm: Instance of Permission, or permission_name
+"""
+user = self.__get_user(user)
+repo = self.__get_repo(repo)
+permission = self.__get_perm(perm)
+# check if we have that permission already
+obj = self.sa.query(UserRepoToPerm)\
+.filter(UserRepoToPerm.user == user)\
+.filter(UserRepoToPerm.repository == repo)\
+.scalar()
+if obj is None:
+# create new !
+obj = UserRepoToPerm()
+obj.repository = repo
+obj.user = user
+obj.permission = permission
+self.sa.add(obj)
+def revoke_user_permission(self, repo, user):
+"""
+Revoke permission for user on given repository
+:param repo: Instance of Repository, repository_id, or repository name
+:param user: Instance of User, user_id or username
+"""
+user = self.__get_user(user)
+repo = self.__get_repo(repo)
+obj = self.sa.query(UserRepoToPerm)\
+.filter(UserRepoToPerm.repository == repo)\
+.filter(UserRepoToPerm.user == user)\
+.one()
+self.sa.delete(obj)
+def grant_users_group_permission(self, repo, group_name, perm):
+"""
+Grant permission for users group on given repository, or update
+existing one if found
+:param repo: Instance of Repository, repository_id, or repository name
+:param group_name: Instance of UserGroup, users_group_id,
+or users group name
+:param perm: Instance of Permission, or permission_name
+"""
+repo = self.__get_repo(repo)
+group_name = self.__get_users_group(group_name)
+permission = self.__get_perm(perm)
+# check if we have that permission already
+obj = self.sa.query(UsersGroupRepoToPerm)\
+.filter(UsersGroupRepoToPerm.users_group == group_name)\
+.filter(UsersGroupRepoToPerm.repository == repo)\
+.scalar()
+if obj is None:
+# create new
+obj = UsersGroupRepoToPerm()
+obj.repository = repo
+obj.users_group = group_name
+obj.permission = permission
+self.sa.add(obj)
+def revoke_users_group_permission(self, repo, group_name):
+"""
+Revoke permission for users group on given repository
+:param repo: Instance of Repository, repository_id, or repository name
+:param group_name: Instance of UserGroup, users_group_id,
+or users group name
+"""
+repo = self.__get_repo(repo)
+group_name = self.__get_users_group(group_name)
+obj = self.sa.query(UsersGroupRepoToPerm)\
+.filter(UsersGroupRepoToPerm.repository == repo)\
+.filter(UsersGroupRepoToPerm.users_group == group_name)\
+.one()
+self.sa.delete(obj)
+def delete_stats(self, repo_name):
+"""
+removes stats for given repo
+:param repo_name:
+"""
 try:
-self.sa.query(RepoToPerm)\
+obj = self.sa.query(Statistics)\
-.filter(RepoToPerm.repository \
+.filter(Statistics.repository ==
-== self.get_by_repo_name(repo_name))\
+self.get_by_repo_name(repo_name))\
-.filter(RepoToPerm.user_id == form_data['user_id']).delete()
+.one()
-self.sa.commit()
+self.sa.delete(obj)
 except:
 log.error(traceback.format_exc())
-self.sa.rollback()
-raise
-def delete_perm_users_group(self, form_data, repo_name):
-try:
-self.sa.query(UsersGroupRepoToPerm)\
-.filter(UsersGroupRepoToPerm.repository \
-== self.get_by_repo_name(repo_name))\
-.filter(UsersGroupRepoToPerm.users_group_id \
-== form_data['users_group_id']).delete()
-self.sa.commit()
-except:
-log.error(traceback.format_exc())
-self.sa.rollback()
-raise
-def delete_stats(self, repo_name):
-try:
-self.sa.query(Statistics)\
-.filter(Statistics.repository == \
-self.get_by_repo_name(repo_name)).delete()
-self.sa.commit()
-except:
-log.error(traceback.format_exc())
-self.sa.rollback()
 raise
 def __create_repo(self, repo_name, alias, new_parent_id, clone_uri=False):
 """
 makes repository on filesystem. It's group aware means it'll create
 :param clone_uri:
 """
 from rhodecode.lib.utils import is_valid_repo, is_valid_repos_group
 if new_parent_id:
-paths = Group.get(new_parent_id).full_path.split(Group.url_sep())
+paths = RepoGroup.get(new_parent_id)\
+.full_path.split(RepoGroup.url_sep())
 new_parent_path = os.sep.join(paths)
 else:
 new_parent_path = ''
-repo_path = os.path.join(*map(lambda x:safe_str(x),
+# we need to make it str for mercurial
+repo_path = os.path.join(*map(lambda x: safe_str(x),
 [self.repos_path, new_parent_path, repo_name]))
 # check if this path is not a repository
 if is_valid_repo(repo_path, self.repos_path):
 raise Exception('This path %s is a valid repository' % repo_path)
 # check if this path is a group
 if is_valid_repos_group(repo_path, self.repos_path):
 raise Exception('This path %s is a valid group' % repo_path)
-log.info('creating repo %s in %s @ %s', repo_name, repo_path,
+log.info('creating repo %s in %s @ %s' % (
-clone_uri)
+repo_name, safe_unicode(repo_path), clone_uri
+)
+)
 backend = get_backend(alias)
 backend(repo_path, create=True, src_url=clone_uri)
 def __rename_repo(self, old, new):
 """
 renames repository on filesystem
 :param old: old name
 :param new: new name
 """
-log.info('renaming repo from %s to %s', old, new)
+log.info('renaming repo from %s to %s' % (old, new))
 old_path = os.path.join(self.repos_path, old)
 new_path = os.path.join(self.repos_path, new)
 if os.path.isdir(new_path):
-raise Exception('Was trying to rename to already existing dir %s' \
+raise Exception(
-		     % new_path)
+'Was trying to rename to already existing dir %s' % new_path
+)
 shutil.move(old_path, new_path)
 def __delete_repo(self, repo):
 """
 removes repo from filesystem, the removal is acctually made by
 by reverting the renames on this repository
 :param repo: repo object
 """
 rm_path = os.path.join(self.repos_path, repo.repo_name)
-log.info("Removing %s", rm_path)
+log.info("Removing %s" % (rm_path))
-#disable hg/git
+# disable hg/git
 alias = repo.repo_type
 shutil.move(os.path.join(rm_path, '.%s' % alias),
 os.path.join(rm_path, 'rm__.%s' % alias))
-#disable repo
+# disable repo
-shutil.move(rm_path, os.path.join(self.repos_path, 'rm__%s__%s' \
+_d = 'rm__%s__%s' % (datetime.now().strftime('%Y%m%d_%H%M%S_%f'),
-% (datetime.today()\
+repo.repo_name)
-.strftime('%Y%m%d_%H%M%S_%f'),
+shutil.move(rm_path, os.path.join(self.repos_path, _d))
-repo.repo_name)))

Mercurial > kallithea

comparison rhodecode/model/repo.py @ 2031:82a88013a3fd