Mercurial > kallithea

comparison rhodecode/model/repo.py @ 629:7e536d1af60d beta

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Code refactoring,models renames cleaned up sqlalchemy sessions, added cache support to most queries in models fixed test.ini file
author Marcin Kuzminski <marcin@python-works.com>
date Thu, 21 Oct 2010 01:38:14 +0200
parents rhodecode/model/repo_model.py@1e757ac98988
children ae292c3b94ed
comparison
equal deleted inserted replaced
628:dd06bdf974c8 629:7e536d1af60d
1 #!/usr/bin/env python
2 # encoding: utf-8
3 # model for handling repositories actions
4 # Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>
5 # This program is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU General Public License
7 # as published by the Free Software Foundation; version 2
8 # of the License or (at your opinion) any later version of the license.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program; if not, write to the Free Software
17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
18 # MA 02110-1301, USA.
19 """
20 Created on Jun 5, 2010
21 model for handling repositories actions
22 :author: marcink
23 """
24
25 from datetime import datetime
26 from pylons import app_globals as g
27 from rhodecode.model.db import Repository, RepoToPerm, User, Permission
28 from rhodecode.model.meta import Session
29 from rhodecode.model.user import UserModel
30 from rhodecode.model.caching_query import FromCache
31 import logging
32 import os
33 import shutil
34 import traceback
35 log = logging.getLogger(__name__)
36
37 class RepoModel(object):
38
39 def __init__(self, sa=None):
40 if not sa:
41 self.sa = Session()
42 else:
43 self.sa = sa
44
45 def get(self, repo_id, cache=False):
46 repo = self.sa.query(Repository)\
47 .filter(Repository.repo_name == repo_id)
48
49 if cache:
50 repo = repo.options(FromCache("sql_cache_short",
51 "get_repo_%s" % repo))
52 return repo.scalar()
53
54 def get_users_js(self):
55
56 users = self.sa.query(User).filter(User.active == True).all()
57 u_tmpl = '''{id:%s, fname:"%s", lname:"%s", nname:"%s"},'''
58 users_array = '[%s];' % '\n'.join([u_tmpl % (u.user_id, u.name,
59 u.lastname, u.username)
60 for u in users])
61 return users_array
62
63
64 def update(self, repo_name, form_data):
65 try:
66
67 #update permissions
68 for username, perm in form_data['perms_updates']:
69 r2p = self.sa.query(RepoToPerm)\
70 .filter(RepoToPerm.user == UserModel(self.sa).get_by_username(username, cache=False))\
71 .filter(RepoToPerm.repository == self.get(repo_name))\
72 .one()
73
74 r2p.permission_id = self.sa.query(Permission).filter(
75 Permission.permission_name ==
76 perm).one().permission_id
77 self.sa.add(r2p)
78
79 #set new permissions
80 for username, perm in form_data['perms_new']:
81 r2p = RepoToPerm()
82 r2p.repository = self.get(repo_name)
83 r2p.user = UserModel(self.sa).get_by_username(username, cache=False)
84
85 r2p.permission_id = self.sa.query(Permission).filter(
86 Permission.permission_name == perm)\
87 .one().permission_id
88 self.sa.add(r2p)
89
90 #update current repo
91 cur_repo = self.get(repo_name, cache=False)
92
93 for k, v in form_data.items():
94 if k == 'user':
95 cur_repo.user_id = v
96 else:
97 setattr(cur_repo, k, v)
98
99 self.sa.add(cur_repo)
100
101 if repo_name != form_data['repo_name']:
102 #rename our data
103 self.__rename_repo(repo_name, form_data['repo_name'])
104
105 self.sa.commit()
106 except:
107 log.error(traceback.format_exc())
108 self.sa.rollback()
109 raise
110
111 def create(self, form_data, cur_user, just_db=False, fork=False):
112 try:
113 if fork:
114 repo_name = str(form_data['fork_name'])
115 org_name = str(form_data['repo_name'])
116
117 else:
118 org_name = repo_name = str(form_data['repo_name'])
119 new_repo = Repository()
120 for k, v in form_data.items():
121 if k == 'repo_name':
122 v = repo_name
123 setattr(new_repo, k, v)
124
125 if fork:
126 parent_repo = self.sa.query(Repository)\
127 .filter(Repository.repo_name == org_name).scalar()
128 new_repo.fork = parent_repo
129
130 new_repo.user_id = cur_user.user_id
131 self.sa.add(new_repo)
132
133 #create default permission
134 repo_to_perm = RepoToPerm()
135 default = 'repository.read'
136 for p in UserModel(self.sa).get_by_username('default', cache=False).user_perms:
137 if p.permission.permission_name.startswith('repository.'):
138 default = p.permission.permission_name
139 break
140
141 default_perm = 'repository.none' if form_data['private'] else default
142
143 repo_to_perm.permission_id = self.sa.query(Permission)\
144 .filter(Permission.permission_name == default_perm)\
145 .one().permission_id
146
147 repo_to_perm.repository_id = new_repo.repo_id
148 repo_to_perm.user_id = UserModel(self.sa).get_by_username('default', cache=False).user_id
149
150 self.sa.add(repo_to_perm)
151 self.sa.commit()
152 if not just_db:
153 self.__create_repo(repo_name)
154 except:
155 log.error(traceback.format_exc())
156 self.sa.rollback()
157 raise
158
159 def create_fork(self, form_data, cur_user):
160 from rhodecode.lib.celerylib import tasks, run_task
161 run_task(tasks.create_repo_fork, form_data, cur_user)
162
163 def delete(self, repo):
164 try:
165 self.sa.delete(repo)
166 self.sa.commit()
167 self.__delete_repo(repo.repo_name)
168 except:
169 log.error(traceback.format_exc())
170 self.sa.rollback()
171 raise
172
173 def delete_perm_user(self, form_data, repo_name):
174 try:
175 self.sa.query(RepoToPerm)\
176 .filter(RepoToPerm.repository == self.get(repo_name))\
177 .filter(RepoToPerm.user_id == form_data['user_id']).delete()
178 self.sa.commit()
179 except:
180 log.error(traceback.format_exc())
181 self.sa.rollback()
182 raise
183
184 def __create_repo(self, repo_name):
185 from rhodecode.lib.utils import check_repo
186 repo_path = os.path.join(g.base_path, repo_name)
187 if check_repo(repo_name, g.base_path):
188 log.info('creating repo %s in %s', repo_name, repo_path)
189 from vcs.backends.hg import MercurialRepository
190 MercurialRepository(repo_path, create=True)
191
192 def __rename_repo(self, old, new):
193 log.info('renaming repo from %s to %s', old, new)
194
195 old_path = os.path.join(g.base_path, old)
196 new_path = os.path.join(g.base_path, new)
197 if os.path.isdir(new_path):
198 raise Exception('Was trying to rename to already existing dir %s',
199 new_path)
200 shutil.move(old_path, new_path)
201
202 def __delete_repo(self, name):
203 rm_path = os.path.join(g.base_path, name)
204 log.info("Removing %s", rm_path)
205 #disable hg
206 shutil.move(os.path.join(rm_path, '.hg'), os.path.join(rm_path, 'rm__.hg'))
207 #disable repo
208 shutil.move(rm_path, os.path.join(g.base_path, 'rm__%s__%s' \
209 % (datetime.today(), name)))