##// END OF EJS Templates
user-bookmarks: make it easier to re-organize entries.
ergo -
r3995:daa7294b default
parent child Browse files
Show More
@@ -1,743 +1,760 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2016-2019 RhodeCode GmbH
3 # Copyright (C) 2016-2019 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import logging
21 import logging
22 import datetime
22 import datetime
23 import string
23 import string
24
24
25 import formencode
25 import formencode
26 import formencode.htmlfill
26 import formencode.htmlfill
27 import peppercorn
27 import peppercorn
28 from pyramid.httpexceptions import HTTPFound
28 from pyramid.httpexceptions import HTTPFound
29 from pyramid.view import view_config
29 from pyramid.view import view_config
30
30
31 from rhodecode.apps._base import BaseAppView, DataGridAppView
31 from rhodecode.apps._base import BaseAppView, DataGridAppView
32 from rhodecode import forms
32 from rhodecode import forms
33 from rhodecode.lib import helpers as h
33 from rhodecode.lib import helpers as h
34 from rhodecode.lib import audit_logger
34 from rhodecode.lib import audit_logger
35 from rhodecode.lib.ext_json import json
35 from rhodecode.lib.ext_json import json
36 from rhodecode.lib.auth import LoginRequired, NotAnonymous, CSRFRequired, \
36 from rhodecode.lib.auth import LoginRequired, NotAnonymous, CSRFRequired, \
37 HasRepoPermissionAny, HasRepoGroupPermissionAny
37 HasRepoPermissionAny, HasRepoGroupPermissionAny
38 from rhodecode.lib.channelstream import (
38 from rhodecode.lib.channelstream import (
39 channelstream_request, ChannelstreamException)
39 channelstream_request, ChannelstreamException)
40 from rhodecode.lib.utils2 import safe_int, md5, str2bool
40 from rhodecode.lib.utils2 import safe_int, md5, str2bool
41 from rhodecode.model.auth_token import AuthTokenModel
41 from rhodecode.model.auth_token import AuthTokenModel
42 from rhodecode.model.comment import CommentsModel
42 from rhodecode.model.comment import CommentsModel
43 from rhodecode.model.db import (
43 from rhodecode.model.db import (
44 IntegrityError, joinedload,
44 IntegrityError, joinedload,
45 Repository, UserEmailMap, UserApiKeys, UserFollowing,
45 Repository, UserEmailMap, UserApiKeys, UserFollowing,
46 PullRequest, UserBookmark, RepoGroup)
46 PullRequest, UserBookmark, RepoGroup)
47 from rhodecode.model.meta import Session
47 from rhodecode.model.meta import Session
48 from rhodecode.model.pull_request import PullRequestModel
48 from rhodecode.model.pull_request import PullRequestModel
49 from rhodecode.model.scm import RepoList
49 from rhodecode.model.scm import RepoList
50 from rhodecode.model.user import UserModel
50 from rhodecode.model.user import UserModel
51 from rhodecode.model.repo import RepoModel
51 from rhodecode.model.repo import RepoModel
52 from rhodecode.model.user_group import UserGroupModel
52 from rhodecode.model.user_group import UserGroupModel
53 from rhodecode.model.validation_schema.schemas import user_schema
53 from rhodecode.model.validation_schema.schemas import user_schema
54
54
55 log = logging.getLogger(__name__)
55 log = logging.getLogger(__name__)
56
56
57
57
58 class MyAccountView(BaseAppView, DataGridAppView):
58 class MyAccountView(BaseAppView, DataGridAppView):
59 ALLOW_SCOPED_TOKENS = False
59 ALLOW_SCOPED_TOKENS = False
60 """
60 """
61 This view has alternative version inside EE, if modified please take a look
61 This view has alternative version inside EE, if modified please take a look
62 in there as well.
62 in there as well.
63 """
63 """
64
64
65 def load_default_context(self):
65 def load_default_context(self):
66 c = self._get_local_tmpl_context()
66 c = self._get_local_tmpl_context()
67 c.user = c.auth_user.get_instance()
67 c.user = c.auth_user.get_instance()
68 c.allow_scoped_tokens = self.ALLOW_SCOPED_TOKENS
68 c.allow_scoped_tokens = self.ALLOW_SCOPED_TOKENS
69
69
70 return c
70 return c
71
71
72 @LoginRequired()
72 @LoginRequired()
73 @NotAnonymous()
73 @NotAnonymous()
74 @view_config(
74 @view_config(
75 route_name='my_account_profile', request_method='GET',
75 route_name='my_account_profile', request_method='GET',
76 renderer='rhodecode:templates/admin/my_account/my_account.mako')
76 renderer='rhodecode:templates/admin/my_account/my_account.mako')
77 def my_account_profile(self):
77 def my_account_profile(self):
78 c = self.load_default_context()
78 c = self.load_default_context()
79 c.active = 'profile'
79 c.active = 'profile'
80 return self._get_template_context(c)
80 return self._get_template_context(c)
81
81
82 @LoginRequired()
82 @LoginRequired()
83 @NotAnonymous()
83 @NotAnonymous()
84 @view_config(
84 @view_config(
85 route_name='my_account_password', request_method='GET',
85 route_name='my_account_password', request_method='GET',
86 renderer='rhodecode:templates/admin/my_account/my_account.mako')
86 renderer='rhodecode:templates/admin/my_account/my_account.mako')
87 def my_account_password(self):
87 def my_account_password(self):
88 c = self.load_default_context()
88 c = self.load_default_context()
89 c.active = 'password'
89 c.active = 'password'
90 c.extern_type = c.user.extern_type
90 c.extern_type = c.user.extern_type
91
91
92 schema = user_schema.ChangePasswordSchema().bind(
92 schema = user_schema.ChangePasswordSchema().bind(
93 username=c.user.username)
93 username=c.user.username)
94
94
95 form = forms.Form(
95 form = forms.Form(
96 schema,
96 schema,
97 action=h.route_path('my_account_password_update'),
97 action=h.route_path('my_account_password_update'),
98 buttons=(forms.buttons.save, forms.buttons.reset))
98 buttons=(forms.buttons.save, forms.buttons.reset))
99
99
100 c.form = form
100 c.form = form
101 return self._get_template_context(c)
101 return self._get_template_context(c)
102
102
103 @LoginRequired()
103 @LoginRequired()
104 @NotAnonymous()
104 @NotAnonymous()
105 @CSRFRequired()
105 @CSRFRequired()
106 @view_config(
106 @view_config(
107 route_name='my_account_password_update', request_method='POST',
107 route_name='my_account_password_update', request_method='POST',
108 renderer='rhodecode:templates/admin/my_account/my_account.mako')
108 renderer='rhodecode:templates/admin/my_account/my_account.mako')
109 def my_account_password_update(self):
109 def my_account_password_update(self):
110 _ = self.request.translate
110 _ = self.request.translate
111 c = self.load_default_context()
111 c = self.load_default_context()
112 c.active = 'password'
112 c.active = 'password'
113 c.extern_type = c.user.extern_type
113 c.extern_type = c.user.extern_type
114
114
115 schema = user_schema.ChangePasswordSchema().bind(
115 schema = user_schema.ChangePasswordSchema().bind(
116 username=c.user.username)
116 username=c.user.username)
117
117
118 form = forms.Form(
118 form = forms.Form(
119 schema, buttons=(forms.buttons.save, forms.buttons.reset))
119 schema, buttons=(forms.buttons.save, forms.buttons.reset))
120
120
121 if c.extern_type != 'rhodecode':
121 if c.extern_type != 'rhodecode':
122 raise HTTPFound(self.request.route_path('my_account_password'))
122 raise HTTPFound(self.request.route_path('my_account_password'))
123
123
124 controls = self.request.POST.items()
124 controls = self.request.POST.items()
125 try:
125 try:
126 valid_data = form.validate(controls)
126 valid_data = form.validate(controls)
127 UserModel().update_user(c.user.user_id, **valid_data)
127 UserModel().update_user(c.user.user_id, **valid_data)
128 c.user.update_userdata(force_password_change=False)
128 c.user.update_userdata(force_password_change=False)
129 Session().commit()
129 Session().commit()
130 except forms.ValidationFailure as e:
130 except forms.ValidationFailure as e:
131 c.form = e
131 c.form = e
132 return self._get_template_context(c)
132 return self._get_template_context(c)
133
133
134 except Exception:
134 except Exception:
135 log.exception("Exception updating password")
135 log.exception("Exception updating password")
136 h.flash(_('Error occurred during update of user password'),
136 h.flash(_('Error occurred during update of user password'),
137 category='error')
137 category='error')
138 else:
138 else:
139 instance = c.auth_user.get_instance()
139 instance = c.auth_user.get_instance()
140 self.session.setdefault('rhodecode_user', {}).update(
140 self.session.setdefault('rhodecode_user', {}).update(
141 {'password': md5(instance.password)})
141 {'password': md5(instance.password)})
142 self.session.save()
142 self.session.save()
143 h.flash(_("Successfully updated password"), category='success')
143 h.flash(_("Successfully updated password"), category='success')
144
144
145 raise HTTPFound(self.request.route_path('my_account_password'))
145 raise HTTPFound(self.request.route_path('my_account_password'))
146
146
147 @LoginRequired()
147 @LoginRequired()
148 @NotAnonymous()
148 @NotAnonymous()
149 @view_config(
149 @view_config(
150 route_name='my_account_auth_tokens', request_method='GET',
150 route_name='my_account_auth_tokens', request_method='GET',
151 renderer='rhodecode:templates/admin/my_account/my_account.mako')
151 renderer='rhodecode:templates/admin/my_account/my_account.mako')
152 def my_account_auth_tokens(self):
152 def my_account_auth_tokens(self):
153 _ = self.request.translate
153 _ = self.request.translate
154
154
155 c = self.load_default_context()
155 c = self.load_default_context()
156 c.active = 'auth_tokens'
156 c.active = 'auth_tokens'
157 c.lifetime_values = AuthTokenModel.get_lifetime_values(translator=_)
157 c.lifetime_values = AuthTokenModel.get_lifetime_values(translator=_)
158 c.role_values = [
158 c.role_values = [
159 (x, AuthTokenModel.cls._get_role_name(x))
159 (x, AuthTokenModel.cls._get_role_name(x))
160 for x in AuthTokenModel.cls.ROLES]
160 for x in AuthTokenModel.cls.ROLES]
161 c.role_options = [(c.role_values, _("Role"))]
161 c.role_options = [(c.role_values, _("Role"))]
162 c.user_auth_tokens = AuthTokenModel().get_auth_tokens(
162 c.user_auth_tokens = AuthTokenModel().get_auth_tokens(
163 c.user.user_id, show_expired=True)
163 c.user.user_id, show_expired=True)
164 c.role_vcs = AuthTokenModel.cls.ROLE_VCS
164 c.role_vcs = AuthTokenModel.cls.ROLE_VCS
165 return self._get_template_context(c)
165 return self._get_template_context(c)
166
166
167 def maybe_attach_token_scope(self, token):
167 def maybe_attach_token_scope(self, token):
168 # implemented in EE edition
168 # implemented in EE edition
169 pass
169 pass
170
170
171 @LoginRequired()
171 @LoginRequired()
172 @NotAnonymous()
172 @NotAnonymous()
173 @CSRFRequired()
173 @CSRFRequired()
174 @view_config(
174 @view_config(
175 route_name='my_account_auth_tokens_add', request_method='POST',)
175 route_name='my_account_auth_tokens_add', request_method='POST',)
176 def my_account_auth_tokens_add(self):
176 def my_account_auth_tokens_add(self):
177 _ = self.request.translate
177 _ = self.request.translate
178 c = self.load_default_context()
178 c = self.load_default_context()
179
179
180 lifetime = safe_int(self.request.POST.get('lifetime'), -1)
180 lifetime = safe_int(self.request.POST.get('lifetime'), -1)
181 description = self.request.POST.get('description')
181 description = self.request.POST.get('description')
182 role = self.request.POST.get('role')
182 role = self.request.POST.get('role')
183
183
184 token = UserModel().add_auth_token(
184 token = UserModel().add_auth_token(
185 user=c.user.user_id,
185 user=c.user.user_id,
186 lifetime_minutes=lifetime, role=role, description=description,
186 lifetime_minutes=lifetime, role=role, description=description,
187 scope_callback=self.maybe_attach_token_scope)
187 scope_callback=self.maybe_attach_token_scope)
188 token_data = token.get_api_data()
188 token_data = token.get_api_data()
189
189
190 audit_logger.store_web(
190 audit_logger.store_web(
191 'user.edit.token.add', action_data={
191 'user.edit.token.add', action_data={
192 'data': {'token': token_data, 'user': 'self'}},
192 'data': {'token': token_data, 'user': 'self'}},
193 user=self._rhodecode_user, )
193 user=self._rhodecode_user, )
194 Session().commit()
194 Session().commit()
195
195
196 h.flash(_("Auth token successfully created"), category='success')
196 h.flash(_("Auth token successfully created"), category='success')
197 return HTTPFound(h.route_path('my_account_auth_tokens'))
197 return HTTPFound(h.route_path('my_account_auth_tokens'))
198
198
199 @LoginRequired()
199 @LoginRequired()
200 @NotAnonymous()
200 @NotAnonymous()
201 @CSRFRequired()
201 @CSRFRequired()
202 @view_config(
202 @view_config(
203 route_name='my_account_auth_tokens_delete', request_method='POST')
203 route_name='my_account_auth_tokens_delete', request_method='POST')
204 def my_account_auth_tokens_delete(self):
204 def my_account_auth_tokens_delete(self):
205 _ = self.request.translate
205 _ = self.request.translate
206 c = self.load_default_context()
206 c = self.load_default_context()
207
207
208 del_auth_token = self.request.POST.get('del_auth_token')
208 del_auth_token = self.request.POST.get('del_auth_token')
209
209
210 if del_auth_token:
210 if del_auth_token:
211 token = UserApiKeys.get_or_404(del_auth_token)
211 token = UserApiKeys.get_or_404(del_auth_token)
212 token_data = token.get_api_data()
212 token_data = token.get_api_data()
213
213
214 AuthTokenModel().delete(del_auth_token, c.user.user_id)
214 AuthTokenModel().delete(del_auth_token, c.user.user_id)
215 audit_logger.store_web(
215 audit_logger.store_web(
216 'user.edit.token.delete', action_data={
216 'user.edit.token.delete', action_data={
217 'data': {'token': token_data, 'user': 'self'}},
217 'data': {'token': token_data, 'user': 'self'}},
218 user=self._rhodecode_user,)
218 user=self._rhodecode_user,)
219 Session().commit()
219 Session().commit()
220 h.flash(_("Auth token successfully deleted"), category='success')
220 h.flash(_("Auth token successfully deleted"), category='success')
221
221
222 return HTTPFound(h.route_path('my_account_auth_tokens'))
222 return HTTPFound(h.route_path('my_account_auth_tokens'))
223
223
224 @LoginRequired()
224 @LoginRequired()
225 @NotAnonymous()
225 @NotAnonymous()
226 @view_config(
226 @view_config(
227 route_name='my_account_emails', request_method='GET',
227 route_name='my_account_emails', request_method='GET',
228 renderer='rhodecode:templates/admin/my_account/my_account.mako')
228 renderer='rhodecode:templates/admin/my_account/my_account.mako')
229 def my_account_emails(self):
229 def my_account_emails(self):
230 _ = self.request.translate
230 _ = self.request.translate
231
231
232 c = self.load_default_context()
232 c = self.load_default_context()
233 c.active = 'emails'
233 c.active = 'emails'
234
234
235 c.user_email_map = UserEmailMap.query()\
235 c.user_email_map = UserEmailMap.query()\
236 .filter(UserEmailMap.user == c.user).all()
236 .filter(UserEmailMap.user == c.user).all()
237
237
238 schema = user_schema.AddEmailSchema().bind(
238 schema = user_schema.AddEmailSchema().bind(
239 username=c.user.username, user_emails=c.user.emails)
239 username=c.user.username, user_emails=c.user.emails)
240
240
241 form = forms.RcForm(schema,
241 form = forms.RcForm(schema,
242 action=h.route_path('my_account_emails_add'),
242 action=h.route_path('my_account_emails_add'),
243 buttons=(forms.buttons.save, forms.buttons.reset))
243 buttons=(forms.buttons.save, forms.buttons.reset))
244
244
245 c.form = form
245 c.form = form
246 return self._get_template_context(c)
246 return self._get_template_context(c)
247
247
248 @LoginRequired()
248 @LoginRequired()
249 @NotAnonymous()
249 @NotAnonymous()
250 @CSRFRequired()
250 @CSRFRequired()
251 @view_config(
251 @view_config(
252 route_name='my_account_emails_add', request_method='POST',
252 route_name='my_account_emails_add', request_method='POST',
253 renderer='rhodecode:templates/admin/my_account/my_account.mako')
253 renderer='rhodecode:templates/admin/my_account/my_account.mako')
254 def my_account_emails_add(self):
254 def my_account_emails_add(self):
255 _ = self.request.translate
255 _ = self.request.translate
256 c = self.load_default_context()
256 c = self.load_default_context()
257 c.active = 'emails'
257 c.active = 'emails'
258
258
259 schema = user_schema.AddEmailSchema().bind(
259 schema = user_schema.AddEmailSchema().bind(
260 username=c.user.username, user_emails=c.user.emails)
260 username=c.user.username, user_emails=c.user.emails)
261
261
262 form = forms.RcForm(
262 form = forms.RcForm(
263 schema, action=h.route_path('my_account_emails_add'),
263 schema, action=h.route_path('my_account_emails_add'),
264 buttons=(forms.buttons.save, forms.buttons.reset))
264 buttons=(forms.buttons.save, forms.buttons.reset))
265
265
266 controls = self.request.POST.items()
266 controls = self.request.POST.items()
267 try:
267 try:
268 valid_data = form.validate(controls)
268 valid_data = form.validate(controls)
269 UserModel().add_extra_email(c.user.user_id, valid_data['email'])
269 UserModel().add_extra_email(c.user.user_id, valid_data['email'])
270 audit_logger.store_web(
270 audit_logger.store_web(
271 'user.edit.email.add', action_data={
271 'user.edit.email.add', action_data={
272 'data': {'email': valid_data['email'], 'user': 'self'}},
272 'data': {'email': valid_data['email'], 'user': 'self'}},
273 user=self._rhodecode_user,)
273 user=self._rhodecode_user,)
274 Session().commit()
274 Session().commit()
275 except formencode.Invalid as error:
275 except formencode.Invalid as error:
276 h.flash(h.escape(error.error_dict['email']), category='error')
276 h.flash(h.escape(error.error_dict['email']), category='error')
277 except forms.ValidationFailure as e:
277 except forms.ValidationFailure as e:
278 c.user_email_map = UserEmailMap.query() \
278 c.user_email_map = UserEmailMap.query() \
279 .filter(UserEmailMap.user == c.user).all()
279 .filter(UserEmailMap.user == c.user).all()
280 c.form = e
280 c.form = e
281 return self._get_template_context(c)
281 return self._get_template_context(c)
282 except Exception:
282 except Exception:
283 log.exception("Exception adding email")
283 log.exception("Exception adding email")
284 h.flash(_('Error occurred during adding email'),
284 h.flash(_('Error occurred during adding email'),
285 category='error')
285 category='error')
286 else:
286 else:
287 h.flash(_("Successfully added email"), category='success')
287 h.flash(_("Successfully added email"), category='success')
288
288
289 raise HTTPFound(self.request.route_path('my_account_emails'))
289 raise HTTPFound(self.request.route_path('my_account_emails'))
290
290
291 @LoginRequired()
291 @LoginRequired()
292 @NotAnonymous()
292 @NotAnonymous()
293 @CSRFRequired()
293 @CSRFRequired()
294 @view_config(
294 @view_config(
295 route_name='my_account_emails_delete', request_method='POST')
295 route_name='my_account_emails_delete', request_method='POST')
296 def my_account_emails_delete(self):
296 def my_account_emails_delete(self):
297 _ = self.request.translate
297 _ = self.request.translate
298 c = self.load_default_context()
298 c = self.load_default_context()
299
299
300 del_email_id = self.request.POST.get('del_email_id')
300 del_email_id = self.request.POST.get('del_email_id')
301 if del_email_id:
301 if del_email_id:
302 email = UserEmailMap.get_or_404(del_email_id).email
302 email = UserEmailMap.get_or_404(del_email_id).email
303 UserModel().delete_extra_email(c.user.user_id, del_email_id)
303 UserModel().delete_extra_email(c.user.user_id, del_email_id)
304 audit_logger.store_web(
304 audit_logger.store_web(
305 'user.edit.email.delete', action_data={
305 'user.edit.email.delete', action_data={
306 'data': {'email': email, 'user': 'self'}},
306 'data': {'email': email, 'user': 'self'}},
307 user=self._rhodecode_user,)
307 user=self._rhodecode_user,)
308 Session().commit()
308 Session().commit()
309 h.flash(_("Email successfully deleted"),
309 h.flash(_("Email successfully deleted"),
310 category='success')
310 category='success')
311 return HTTPFound(h.route_path('my_account_emails'))
311 return HTTPFound(h.route_path('my_account_emails'))
312
312
313 @LoginRequired()
313 @LoginRequired()
314 @NotAnonymous()
314 @NotAnonymous()
315 @CSRFRequired()
315 @CSRFRequired()
316 @view_config(
316 @view_config(
317 route_name='my_account_notifications_test_channelstream',
317 route_name='my_account_notifications_test_channelstream',
318 request_method='POST', renderer='json_ext')
318 request_method='POST', renderer='json_ext')
319 def my_account_notifications_test_channelstream(self):
319 def my_account_notifications_test_channelstream(self):
320 message = 'Test message sent via Channelstream by user: {}, on {}'.format(
320 message = 'Test message sent via Channelstream by user: {}, on {}'.format(
321 self._rhodecode_user.username, datetime.datetime.now())
321 self._rhodecode_user.username, datetime.datetime.now())
322 payload = {
322 payload = {
323 # 'channel': 'broadcast',
323 # 'channel': 'broadcast',
324 'type': 'message',
324 'type': 'message',
325 'timestamp': datetime.datetime.utcnow(),
325 'timestamp': datetime.datetime.utcnow(),
326 'user': 'system',
326 'user': 'system',
327 'pm_users': [self._rhodecode_user.username],
327 'pm_users': [self._rhodecode_user.username],
328 'message': {
328 'message': {
329 'message': message,
329 'message': message,
330 'level': 'info',
330 'level': 'info',
331 'topic': '/notifications'
331 'topic': '/notifications'
332 }
332 }
333 }
333 }
334
334
335 registry = self.request.registry
335 registry = self.request.registry
336 rhodecode_plugins = getattr(registry, 'rhodecode_plugins', {})
336 rhodecode_plugins = getattr(registry, 'rhodecode_plugins', {})
337 channelstream_config = rhodecode_plugins.get('channelstream', {})
337 channelstream_config = rhodecode_plugins.get('channelstream', {})
338
338
339 try:
339 try:
340 channelstream_request(channelstream_config, [payload], '/message')
340 channelstream_request(channelstream_config, [payload], '/message')
341 except ChannelstreamException as e:
341 except ChannelstreamException as e:
342 log.exception('Failed to send channelstream data')
342 log.exception('Failed to send channelstream data')
343 return {"response": 'ERROR: {}'.format(e.__class__.__name__)}
343 return {"response": 'ERROR: {}'.format(e.__class__.__name__)}
344 return {"response": 'Channelstream data sent. '
344 return {"response": 'Channelstream data sent. '
345 'You should see a new live message now.'}
345 'You should see a new live message now.'}
346
346
347 def _load_my_repos_data(self, watched=False):
347 def _load_my_repos_data(self, watched=False):
348 if watched:
348 if watched:
349 admin = False
349 admin = False
350 follows_repos = Session().query(UserFollowing)\
350 follows_repos = Session().query(UserFollowing)\
351 .filter(UserFollowing.user_id == self._rhodecode_user.user_id)\
351 .filter(UserFollowing.user_id == self._rhodecode_user.user_id)\
352 .options(joinedload(UserFollowing.follows_repository))\
352 .options(joinedload(UserFollowing.follows_repository))\
353 .all()
353 .all()
354 repo_list = [x.follows_repository for x in follows_repos]
354 repo_list = [x.follows_repository for x in follows_repos]
355 else:
355 else:
356 admin = True
356 admin = True
357 repo_list = Repository.get_all_repos(
357 repo_list = Repository.get_all_repos(
358 user_id=self._rhodecode_user.user_id)
358 user_id=self._rhodecode_user.user_id)
359 repo_list = RepoList(repo_list, perm_set=[
359 repo_list = RepoList(repo_list, perm_set=[
360 'repository.read', 'repository.write', 'repository.admin'])
360 'repository.read', 'repository.write', 'repository.admin'])
361
361
362 repos_data = RepoModel().get_repos_as_dict(
362 repos_data = RepoModel().get_repos_as_dict(
363 repo_list=repo_list, admin=admin, short_name=False)
363 repo_list=repo_list, admin=admin, short_name=False)
364 # json used to render the grid
364 # json used to render the grid
365 return json.dumps(repos_data)
365 return json.dumps(repos_data)
366
366
367 @LoginRequired()
367 @LoginRequired()
368 @NotAnonymous()
368 @NotAnonymous()
369 @view_config(
369 @view_config(
370 route_name='my_account_repos', request_method='GET',
370 route_name='my_account_repos', request_method='GET',
371 renderer='rhodecode:templates/admin/my_account/my_account.mako')
371 renderer='rhodecode:templates/admin/my_account/my_account.mako')
372 def my_account_repos(self):
372 def my_account_repos(self):
373 c = self.load_default_context()
373 c = self.load_default_context()
374 c.active = 'repos'
374 c.active = 'repos'
375
375
376 # json used to render the grid
376 # json used to render the grid
377 c.data = self._load_my_repos_data()
377 c.data = self._load_my_repos_data()
378 return self._get_template_context(c)
378 return self._get_template_context(c)
379
379
380 @LoginRequired()
380 @LoginRequired()
381 @NotAnonymous()
381 @NotAnonymous()
382 @view_config(
382 @view_config(
383 route_name='my_account_watched', request_method='GET',
383 route_name='my_account_watched', request_method='GET',
384 renderer='rhodecode:templates/admin/my_account/my_account.mako')
384 renderer='rhodecode:templates/admin/my_account/my_account.mako')
385 def my_account_watched(self):
385 def my_account_watched(self):
386 c = self.load_default_context()
386 c = self.load_default_context()
387 c.active = 'watched'
387 c.active = 'watched'
388
388
389 # json used to render the grid
389 # json used to render the grid
390 c.data = self._load_my_repos_data(watched=True)
390 c.data = self._load_my_repos_data(watched=True)
391 return self._get_template_context(c)
391 return self._get_template_context(c)
392
392
393 @LoginRequired()
393 @LoginRequired()
394 @NotAnonymous()
394 @NotAnonymous()
395 @view_config(
395 @view_config(
396 route_name='my_account_bookmarks', request_method='GET',
396 route_name='my_account_bookmarks', request_method='GET',
397 renderer='rhodecode:templates/admin/my_account/my_account.mako')
397 renderer='rhodecode:templates/admin/my_account/my_account.mako')
398 def my_account_bookmarks(self):
398 def my_account_bookmarks(self):
399 c = self.load_default_context()
399 c = self.load_default_context()
400 c.active = 'bookmarks'
400 c.active = 'bookmarks'
401 return self._get_template_context(c)
401 return self._get_template_context(c)
402
402
403 def _process_entry(self, entry, user_id):
403 def _process_bookmark_entry(self, entry, user_id):
404 position = safe_int(entry.get('position'))
404 position = safe_int(entry.get('position'))
405 if position is None:
405 cur_position = safe_int(entry.get('cur_position'))
406 if position is None or cur_position is None:
406 return
407 return
407
408
408 # check if this is an existing entry
409 # check if this is an existing entry
409 is_new = False
410 is_new = False
410 db_entry = UserBookmark().get_by_position_for_user(position, user_id)
411 db_entry = UserBookmark().get_by_position_for_user(cur_position, user_id)
411
412
412 if db_entry and str2bool(entry.get('remove')):
413 if db_entry and str2bool(entry.get('remove')):
413 log.debug('Marked bookmark %s for deletion', db_entry)
414 log.debug('Marked bookmark %s for deletion', db_entry)
414 Session().delete(db_entry)
415 Session().delete(db_entry)
415 return
416 return
416
417
417 if not db_entry:
418 if not db_entry:
418 # new
419 # new
419 db_entry = UserBookmark()
420 db_entry = UserBookmark()
420 is_new = True
421 is_new = True
421
422
422 should_save = False
423 should_save = False
423 default_redirect_url = ''
424 default_redirect_url = ''
424
425
425 # save repo
426 # save repo
426 if entry.get('bookmark_repo') and safe_int(entry.get('bookmark_repo')):
427 if entry.get('bookmark_repo') and safe_int(entry.get('bookmark_repo')):
427 repo = Repository.get(entry['bookmark_repo'])
428 repo = Repository.get(entry['bookmark_repo'])
428 perm_check = HasRepoPermissionAny(
429 perm_check = HasRepoPermissionAny(
429 'repository.read', 'repository.write', 'repository.admin')
430 'repository.read', 'repository.write', 'repository.admin')
430 if repo and perm_check(repo_name=repo.repo_name):
431 if repo and perm_check(repo_name=repo.repo_name):
431 db_entry.repository = repo
432 db_entry.repository = repo
432 should_save = True
433 should_save = True
433 default_redirect_url = '${repo_url}'
434 default_redirect_url = '${repo_url}'
434 # save repo group
435 # save repo group
435 elif entry.get('bookmark_repo_group') and safe_int(entry.get('bookmark_repo_group')):
436 elif entry.get('bookmark_repo_group') and safe_int(entry.get('bookmark_repo_group')):
436 repo_group = RepoGroup.get(entry['bookmark_repo_group'])
437 repo_group = RepoGroup.get(entry['bookmark_repo_group'])
437 perm_check = HasRepoGroupPermissionAny(
438 perm_check = HasRepoGroupPermissionAny(
438 'group.read', 'group.write', 'group.admin')
439 'group.read', 'group.write', 'group.admin')
439
440
440 if repo_group and perm_check(group_name=repo_group.group_name):
441 if repo_group and perm_check(group_name=repo_group.group_name):
441 db_entry.repository_group = repo_group
442 db_entry.repository_group = repo_group
442 should_save = True
443 should_save = True
443 default_redirect_url = '${repo_group_url}'
444 default_redirect_url = '${repo_group_url}'
444 # save generic info
445 # save generic info
445 elif entry.get('title') and entry.get('redirect_url'):
446 elif entry.get('title') and entry.get('redirect_url'):
446 should_save = True
447 should_save = True
447
448
448 if should_save:
449 if should_save:
449 log.debug('Saving bookmark %s, new:%s', db_entry, is_new)
450 # mark user and position
450 # mark user and position
451 db_entry.user_id = user_id
451 db_entry.user_id = user_id
452 db_entry.position = position
452 db_entry.position = position
453 db_entry.title = entry.get('title')
453 db_entry.title = entry.get('title')
454 db_entry.redirect_url = entry.get('redirect_url') or default_redirect_url
454 db_entry.redirect_url = entry.get('redirect_url') or default_redirect_url
455 log.debug('Saving bookmark %s, new:%s', db_entry, is_new)
455
456
456 Session().add(db_entry)
457 Session().add(db_entry)
457
458
458 @LoginRequired()
459 @LoginRequired()
459 @NotAnonymous()
460 @NotAnonymous()
460 @CSRFRequired()
461 @CSRFRequired()
461 @view_config(
462 @view_config(
462 route_name='my_account_bookmarks_update', request_method='POST')
463 route_name='my_account_bookmarks_update', request_method='POST')
463 def my_account_bookmarks_update(self):
464 def my_account_bookmarks_update(self):
464 _ = self.request.translate
465 _ = self.request.translate
465 c = self.load_default_context()
466 c = self.load_default_context()
466 c.active = 'bookmarks'
467 c.active = 'bookmarks'
467
468
468 controls = peppercorn.parse(self.request.POST.items())
469 controls = peppercorn.parse(self.request.POST.items())
469 user_id = c.user.user_id
470 user_id = c.user.user_id
470
471
471 try:
472 # validate positions
473 positions = {}
472 for entry in controls.get('bookmarks', []):
474 for entry in controls.get('bookmarks', []):
473 self._process_entry(entry, user_id)
475 position = safe_int(entry['position'])
476 if position is None:
477 continue
478
479 if position in positions:
480 h.flash(_("Position {} is defined twice. "
481 "Please correct this error.").format(position), category='error')
482 return HTTPFound(h.route_path('my_account_bookmarks'))
483
484 entry['position'] = position
485 entry['cur_position'] = safe_int(entry.get('cur_position'))
486 positions[position] = entry
487
488 try:
489 for entry in positions.values():
490 self._process_bookmark_entry(entry, user_id)
474
491
475 Session().commit()
492 Session().commit()
476 h.flash(_("Update Bookmarks"), category='success')
493 h.flash(_("Update Bookmarks"), category='success')
477 except IntegrityError:
494 except IntegrityError:
478 h.flash(_("Failed to update bookmarks. "
495 h.flash(_("Failed to update bookmarks. "
479 "Make sure an unique position is used"), category='error')
496 "Make sure an unique position is used."), category='error')
480
497
481 return HTTPFound(h.route_path('my_account_bookmarks'))
498 return HTTPFound(h.route_path('my_account_bookmarks'))
482
499
483 @LoginRequired()
500 @LoginRequired()
484 @NotAnonymous()
501 @NotAnonymous()
485 @view_config(
502 @view_config(
486 route_name='my_account_goto_bookmark', request_method='GET',
503 route_name='my_account_goto_bookmark', request_method='GET',
487 renderer='rhodecode:templates/admin/my_account/my_account.mako')
504 renderer='rhodecode:templates/admin/my_account/my_account.mako')
488 def my_account_goto_bookmark(self):
505 def my_account_goto_bookmark(self):
489
506
490 bookmark_id = self.request.matchdict['bookmark_id']
507 bookmark_id = self.request.matchdict['bookmark_id']
491 user_bookmark = UserBookmark().query()\
508 user_bookmark = UserBookmark().query()\
492 .filter(UserBookmark.user_id == self.request.user.user_id) \
509 .filter(UserBookmark.user_id == self.request.user.user_id) \
493 .filter(UserBookmark.position == bookmark_id).scalar()
510 .filter(UserBookmark.position == bookmark_id).scalar()
494
511
495 redirect_url = h.route_path('my_account_bookmarks')
512 redirect_url = h.route_path('my_account_bookmarks')
496 if not user_bookmark:
513 if not user_bookmark:
497 raise HTTPFound(redirect_url)
514 raise HTTPFound(redirect_url)
498
515
499 # repository set
516 # repository set
500 if user_bookmark.repository:
517 if user_bookmark.repository:
501 repo_name = user_bookmark.repository.repo_name
518 repo_name = user_bookmark.repository.repo_name
502 base_redirect_url = h.route_path(
519 base_redirect_url = h.route_path(
503 'repo_summary', repo_name=repo_name)
520 'repo_summary', repo_name=repo_name)
504 if user_bookmark.redirect_url and \
521 if user_bookmark.redirect_url and \
505 '${repo_url}' in user_bookmark.redirect_url:
522 '${repo_url}' in user_bookmark.redirect_url:
506 redirect_url = string.Template(user_bookmark.redirect_url)\
523 redirect_url = string.Template(user_bookmark.redirect_url)\
507 .safe_substitute({'repo_url': base_redirect_url})
524 .safe_substitute({'repo_url': base_redirect_url})
508 else:
525 else:
509 redirect_url = base_redirect_url
526 redirect_url = base_redirect_url
510 # repository group set
527 # repository group set
511 elif user_bookmark.repository_group:
528 elif user_bookmark.repository_group:
512 repo_group_name = user_bookmark.repository_group.group_name
529 repo_group_name = user_bookmark.repository_group.group_name
513 base_redirect_url = h.route_path(
530 base_redirect_url = h.route_path(
514 'repo_group_home', repo_group_name=repo_group_name)
531 'repo_group_home', repo_group_name=repo_group_name)
515 if user_bookmark.redirect_url and \
532 if user_bookmark.redirect_url and \
516 '${repo_group_url}' in user_bookmark.redirect_url:
533 '${repo_group_url}' in user_bookmark.redirect_url:
517 redirect_url = string.Template(user_bookmark.redirect_url)\
534 redirect_url = string.Template(user_bookmark.redirect_url)\
518 .safe_substitute({'repo_group_url': base_redirect_url})
535 .safe_substitute({'repo_group_url': base_redirect_url})
519 else:
536 else:
520 redirect_url = base_redirect_url
537 redirect_url = base_redirect_url
521 # custom URL set
538 # custom URL set
522 elif user_bookmark.redirect_url:
539 elif user_bookmark.redirect_url:
523 server_url = h.route_url('home').rstrip('/')
540 server_url = h.route_url('home').rstrip('/')
524 redirect_url = string.Template(user_bookmark.redirect_url) \
541 redirect_url = string.Template(user_bookmark.redirect_url) \
525 .safe_substitute({'server_url': server_url})
542 .safe_substitute({'server_url': server_url})
526
543
527 log.debug('Redirecting bookmark %s to %s', user_bookmark, redirect_url)
544 log.debug('Redirecting bookmark %s to %s', user_bookmark, redirect_url)
528 raise HTTPFound(redirect_url)
545 raise HTTPFound(redirect_url)
529
546
530 @LoginRequired()
547 @LoginRequired()
531 @NotAnonymous()
548 @NotAnonymous()
532 @view_config(
549 @view_config(
533 route_name='my_account_perms', request_method='GET',
550 route_name='my_account_perms', request_method='GET',
534 renderer='rhodecode:templates/admin/my_account/my_account.mako')
551 renderer='rhodecode:templates/admin/my_account/my_account.mako')
535 def my_account_perms(self):
552 def my_account_perms(self):
536 c = self.load_default_context()
553 c = self.load_default_context()
537 c.active = 'perms'
554 c.active = 'perms'
538
555
539 c.perm_user = c.auth_user
556 c.perm_user = c.auth_user
540 return self._get_template_context(c)
557 return self._get_template_context(c)
541
558
542 @LoginRequired()
559 @LoginRequired()
543 @NotAnonymous()
560 @NotAnonymous()
544 @view_config(
561 @view_config(
545 route_name='my_account_notifications', request_method='GET',
562 route_name='my_account_notifications', request_method='GET',
546 renderer='rhodecode:templates/admin/my_account/my_account.mako')
563 renderer='rhodecode:templates/admin/my_account/my_account.mako')
547 def my_notifications(self):
564 def my_notifications(self):
548 c = self.load_default_context()
565 c = self.load_default_context()
549 c.active = 'notifications'
566 c.active = 'notifications'
550
567
551 return self._get_template_context(c)
568 return self._get_template_context(c)
552
569
553 @LoginRequired()
570 @LoginRequired()
554 @NotAnonymous()
571 @NotAnonymous()
555 @CSRFRequired()
572 @CSRFRequired()
556 @view_config(
573 @view_config(
557 route_name='my_account_notifications_toggle_visibility',
574 route_name='my_account_notifications_toggle_visibility',
558 request_method='POST', renderer='json_ext')
575 request_method='POST', renderer='json_ext')
559 def my_notifications_toggle_visibility(self):
576 def my_notifications_toggle_visibility(self):
560 user = self._rhodecode_db_user
577 user = self._rhodecode_db_user
561 new_status = not user.user_data.get('notification_status', True)
578 new_status = not user.user_data.get('notification_status', True)
562 user.update_userdata(notification_status=new_status)
579 user.update_userdata(notification_status=new_status)
563 Session().commit()
580 Session().commit()
564 return user.user_data['notification_status']
581 return user.user_data['notification_status']
565
582
566 @LoginRequired()
583 @LoginRequired()
567 @NotAnonymous()
584 @NotAnonymous()
568 @view_config(
585 @view_config(
569 route_name='my_account_edit',
586 route_name='my_account_edit',
570 request_method='GET',
587 request_method='GET',
571 renderer='rhodecode:templates/admin/my_account/my_account.mako')
588 renderer='rhodecode:templates/admin/my_account/my_account.mako')
572 def my_account_edit(self):
589 def my_account_edit(self):
573 c = self.load_default_context()
590 c = self.load_default_context()
574 c.active = 'profile_edit'
591 c.active = 'profile_edit'
575 c.extern_type = c.user.extern_type
592 c.extern_type = c.user.extern_type
576 c.extern_name = c.user.extern_name
593 c.extern_name = c.user.extern_name
577
594
578 schema = user_schema.UserProfileSchema().bind(
595 schema = user_schema.UserProfileSchema().bind(
579 username=c.user.username, user_emails=c.user.emails)
596 username=c.user.username, user_emails=c.user.emails)
580 appstruct = {
597 appstruct = {
581 'username': c.user.username,
598 'username': c.user.username,
582 'email': c.user.email,
599 'email': c.user.email,
583 'firstname': c.user.firstname,
600 'firstname': c.user.firstname,
584 'lastname': c.user.lastname,
601 'lastname': c.user.lastname,
585 }
602 }
586 c.form = forms.RcForm(
603 c.form = forms.RcForm(
587 schema, appstruct=appstruct,
604 schema, appstruct=appstruct,
588 action=h.route_path('my_account_update'),
605 action=h.route_path('my_account_update'),
589 buttons=(forms.buttons.save, forms.buttons.reset))
606 buttons=(forms.buttons.save, forms.buttons.reset))
590
607
591 return self._get_template_context(c)
608 return self._get_template_context(c)
592
609
593 @LoginRequired()
610 @LoginRequired()
594 @NotAnonymous()
611 @NotAnonymous()
595 @CSRFRequired()
612 @CSRFRequired()
596 @view_config(
613 @view_config(
597 route_name='my_account_update',
614 route_name='my_account_update',
598 request_method='POST',
615 request_method='POST',
599 renderer='rhodecode:templates/admin/my_account/my_account.mako')
616 renderer='rhodecode:templates/admin/my_account/my_account.mako')
600 def my_account_update(self):
617 def my_account_update(self):
601 _ = self.request.translate
618 _ = self.request.translate
602 c = self.load_default_context()
619 c = self.load_default_context()
603 c.active = 'profile_edit'
620 c.active = 'profile_edit'
604 c.perm_user = c.auth_user
621 c.perm_user = c.auth_user
605 c.extern_type = c.user.extern_type
622 c.extern_type = c.user.extern_type
606 c.extern_name = c.user.extern_name
623 c.extern_name = c.user.extern_name
607
624
608 schema = user_schema.UserProfileSchema().bind(
625 schema = user_schema.UserProfileSchema().bind(
609 username=c.user.username, user_emails=c.user.emails)
626 username=c.user.username, user_emails=c.user.emails)
610 form = forms.RcForm(
627 form = forms.RcForm(
611 schema, buttons=(forms.buttons.save, forms.buttons.reset))
628 schema, buttons=(forms.buttons.save, forms.buttons.reset))
612
629
613 controls = self.request.POST.items()
630 controls = self.request.POST.items()
614 try:
631 try:
615 valid_data = form.validate(controls)
632 valid_data = form.validate(controls)
616 skip_attrs = ['admin', 'active', 'extern_type', 'extern_name',
633 skip_attrs = ['admin', 'active', 'extern_type', 'extern_name',
617 'new_password', 'password_confirmation']
634 'new_password', 'password_confirmation']
618 if c.extern_type != "rhodecode":
635 if c.extern_type != "rhodecode":
619 # forbid updating username for external accounts
636 # forbid updating username for external accounts
620 skip_attrs.append('username')
637 skip_attrs.append('username')
621 old_email = c.user.email
638 old_email = c.user.email
622 UserModel().update_user(
639 UserModel().update_user(
623 self._rhodecode_user.user_id, skip_attrs=skip_attrs,
640 self._rhodecode_user.user_id, skip_attrs=skip_attrs,
624 **valid_data)
641 **valid_data)
625 if old_email != valid_data['email']:
642 if old_email != valid_data['email']:
626 old = UserEmailMap.query() \
643 old = UserEmailMap.query() \
627 .filter(UserEmailMap.user == c.user).filter(UserEmailMap.email == valid_data['email']).first()
644 .filter(UserEmailMap.user == c.user).filter(UserEmailMap.email == valid_data['email']).first()
628 old.email = old_email
645 old.email = old_email
629 h.flash(_('Your account was updated successfully'), category='success')
646 h.flash(_('Your account was updated successfully'), category='success')
630 Session().commit()
647 Session().commit()
631 except forms.ValidationFailure as e:
648 except forms.ValidationFailure as e:
632 c.form = e
649 c.form = e
633 return self._get_template_context(c)
650 return self._get_template_context(c)
634 except Exception:
651 except Exception:
635 log.exception("Exception updating user")
652 log.exception("Exception updating user")
636 h.flash(_('Error occurred during update of user'),
653 h.flash(_('Error occurred during update of user'),
637 category='error')
654 category='error')
638 raise HTTPFound(h.route_path('my_account_profile'))
655 raise HTTPFound(h.route_path('my_account_profile'))
639
656
640 def _get_pull_requests_list(self, statuses):
657 def _get_pull_requests_list(self, statuses):
641 draw, start, limit = self._extract_chunk(self.request)
658 draw, start, limit = self._extract_chunk(self.request)
642 search_q, order_by, order_dir = self._extract_ordering(self.request)
659 search_q, order_by, order_dir = self._extract_ordering(self.request)
643 _render = self.request.get_partial_renderer(
660 _render = self.request.get_partial_renderer(
644 'rhodecode:templates/data_table/_dt_elements.mako')
661 'rhodecode:templates/data_table/_dt_elements.mako')
645
662
646 pull_requests = PullRequestModel().get_im_participating_in(
663 pull_requests = PullRequestModel().get_im_participating_in(
647 user_id=self._rhodecode_user.user_id,
664 user_id=self._rhodecode_user.user_id,
648 statuses=statuses,
665 statuses=statuses,
649 offset=start, length=limit, order_by=order_by,
666 offset=start, length=limit, order_by=order_by,
650 order_dir=order_dir)
667 order_dir=order_dir)
651
668
652 pull_requests_total_count = PullRequestModel().count_im_participating_in(
669 pull_requests_total_count = PullRequestModel().count_im_participating_in(
653 user_id=self._rhodecode_user.user_id, statuses=statuses)
670 user_id=self._rhodecode_user.user_id, statuses=statuses)
654
671
655 data = []
672 data = []
656 comments_model = CommentsModel()
673 comments_model = CommentsModel()
657 for pr in pull_requests:
674 for pr in pull_requests:
658 repo_id = pr.target_repo_id
675 repo_id = pr.target_repo_id
659 comments = comments_model.get_all_comments(
676 comments = comments_model.get_all_comments(
660 repo_id, pull_request=pr)
677 repo_id, pull_request=pr)
661 owned = pr.user_id == self._rhodecode_user.user_id
678 owned = pr.user_id == self._rhodecode_user.user_id
662
679
663 data.append({
680 data.append({
664 'target_repo': _render('pullrequest_target_repo',
681 'target_repo': _render('pullrequest_target_repo',
665 pr.target_repo.repo_name),
682 pr.target_repo.repo_name),
666 'name': _render('pullrequest_name',
683 'name': _render('pullrequest_name',
667 pr.pull_request_id, pr.target_repo.repo_name,
684 pr.pull_request_id, pr.target_repo.repo_name,
668 short=True),
685 short=True),
669 'name_raw': pr.pull_request_id,
686 'name_raw': pr.pull_request_id,
670 'status': _render('pullrequest_status',
687 'status': _render('pullrequest_status',
671 pr.calculated_review_status()),
688 pr.calculated_review_status()),
672 'title': _render('pullrequest_title', pr.title, pr.description),
689 'title': _render('pullrequest_title', pr.title, pr.description),
673 'description': h.escape(pr.description),
690 'description': h.escape(pr.description),
674 'updated_on': _render('pullrequest_updated_on',
691 'updated_on': _render('pullrequest_updated_on',
675 h.datetime_to_time(pr.updated_on)),
692 h.datetime_to_time(pr.updated_on)),
676 'updated_on_raw': h.datetime_to_time(pr.updated_on),
693 'updated_on_raw': h.datetime_to_time(pr.updated_on),
677 'created_on': _render('pullrequest_updated_on',
694 'created_on': _render('pullrequest_updated_on',
678 h.datetime_to_time(pr.created_on)),
695 h.datetime_to_time(pr.created_on)),
679 'created_on_raw': h.datetime_to_time(pr.created_on),
696 'created_on_raw': h.datetime_to_time(pr.created_on),
680 'state': pr.pull_request_state,
697 'state': pr.pull_request_state,
681 'author': _render('pullrequest_author',
698 'author': _render('pullrequest_author',
682 pr.author.full_contact, ),
699 pr.author.full_contact, ),
683 'author_raw': pr.author.full_name,
700 'author_raw': pr.author.full_name,
684 'comments': _render('pullrequest_comments', len(comments)),
701 'comments': _render('pullrequest_comments', len(comments)),
685 'comments_raw': len(comments),
702 'comments_raw': len(comments),
686 'closed': pr.is_closed(),
703 'closed': pr.is_closed(),
687 'owned': owned
704 'owned': owned
688 })
705 })
689
706
690 # json used to render the grid
707 # json used to render the grid
691 data = ({
708 data = ({
692 'draw': draw,
709 'draw': draw,
693 'data': data,
710 'data': data,
694 'recordsTotal': pull_requests_total_count,
711 'recordsTotal': pull_requests_total_count,
695 'recordsFiltered': pull_requests_total_count,
712 'recordsFiltered': pull_requests_total_count,
696 })
713 })
697 return data
714 return data
698
715
699 @LoginRequired()
716 @LoginRequired()
700 @NotAnonymous()
717 @NotAnonymous()
701 @view_config(
718 @view_config(
702 route_name='my_account_pullrequests',
719 route_name='my_account_pullrequests',
703 request_method='GET',
720 request_method='GET',
704 renderer='rhodecode:templates/admin/my_account/my_account.mako')
721 renderer='rhodecode:templates/admin/my_account/my_account.mako')
705 def my_account_pullrequests(self):
722 def my_account_pullrequests(self):
706 c = self.load_default_context()
723 c = self.load_default_context()
707 c.active = 'pullrequests'
724 c.active = 'pullrequests'
708 req_get = self.request.GET
725 req_get = self.request.GET
709
726
710 c.closed = str2bool(req_get.get('pr_show_closed'))
727 c.closed = str2bool(req_get.get('pr_show_closed'))
711
728
712 return self._get_template_context(c)
729 return self._get_template_context(c)
713
730
714 @LoginRequired()
731 @LoginRequired()
715 @NotAnonymous()
732 @NotAnonymous()
716 @view_config(
733 @view_config(
717 route_name='my_account_pullrequests_data',
734 route_name='my_account_pullrequests_data',
718 request_method='GET', renderer='json_ext')
735 request_method='GET', renderer='json_ext')
719 def my_account_pullrequests_data(self):
736 def my_account_pullrequests_data(self):
720 self.load_default_context()
737 self.load_default_context()
721 req_get = self.request.GET
738 req_get = self.request.GET
722 closed = str2bool(req_get.get('closed'))
739 closed = str2bool(req_get.get('closed'))
723
740
724 statuses = [PullRequest.STATUS_NEW, PullRequest.STATUS_OPEN]
741 statuses = [PullRequest.STATUS_NEW, PullRequest.STATUS_OPEN]
725 if closed:
742 if closed:
726 statuses += [PullRequest.STATUS_CLOSED]
743 statuses += [PullRequest.STATUS_CLOSED]
727
744
728 data = self._get_pull_requests_list(statuses=statuses)
745 data = self._get_pull_requests_list(statuses=statuses)
729 return data
746 return data
730
747
731 @LoginRequired()
748 @LoginRequired()
732 @NotAnonymous()
749 @NotAnonymous()
733 @view_config(
750 @view_config(
734 route_name='my_account_user_group_membership',
751 route_name='my_account_user_group_membership',
735 request_method='GET',
752 request_method='GET',
736 renderer='rhodecode:templates/admin/my_account/my_account.mako')
753 renderer='rhodecode:templates/admin/my_account/my_account.mako')
737 def my_account_user_group_membership(self):
754 def my_account_user_group_membership(self):
738 c = self.load_default_context()
755 c = self.load_default_context()
739 c.active = 'user_group_membership'
756 c.active = 'user_group_membership'
740 groups = [UserGroupModel.get_user_groups_as_dict(group.users_group)
757 groups = [UserGroupModel.get_user_groups_as_dict(group.users_group)
741 for group in self._rhodecode_db_user.group_member]
758 for group in self._rhodecode_db_user.group_member]
742 c.user_groups = json.dumps(groups)
759 c.user_groups = json.dumps(groups)
743 return self._get_template_context(c)
760 return self._get_template_context(c)
@@ -1,5316 +1,5316 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2019 RhodeCode GmbH
3 # Copyright (C) 2010-2019 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 Database Models for RhodeCode Enterprise
22 Database Models for RhodeCode Enterprise
23 """
23 """
24
24
25 import re
25 import re
26 import os
26 import os
27 import time
27 import time
28 import string
28 import string
29 import hashlib
29 import hashlib
30 import logging
30 import logging
31 import datetime
31 import datetime
32 import uuid
32 import uuid
33 import warnings
33 import warnings
34 import ipaddress
34 import ipaddress
35 import functools
35 import functools
36 import traceback
36 import traceback
37 import collections
37 import collections
38
38
39 from sqlalchemy import (
39 from sqlalchemy import (
40 or_, and_, not_, func, TypeDecorator, event,
40 or_, and_, not_, func, TypeDecorator, event,
41 Index, Sequence, UniqueConstraint, ForeignKey, CheckConstraint, Column,
41 Index, Sequence, UniqueConstraint, ForeignKey, CheckConstraint, Column,
42 Boolean, String, Unicode, UnicodeText, DateTime, Integer, LargeBinary,
42 Boolean, String, Unicode, UnicodeText, DateTime, Integer, LargeBinary,
43 Text, Float, PickleType)
43 Text, Float, PickleType)
44 from sqlalchemy.sql.expression import true, false, case
44 from sqlalchemy.sql.expression import true, false, case
45 from sqlalchemy.sql.functions import coalesce, count # pragma: no cover
45 from sqlalchemy.sql.functions import coalesce, count # pragma: no cover
46 from sqlalchemy.orm import (
46 from sqlalchemy.orm import (
47 relationship, joinedload, class_mapper, validates, aliased)
47 relationship, joinedload, class_mapper, validates, aliased)
48 from sqlalchemy.ext.declarative import declared_attr
48 from sqlalchemy.ext.declarative import declared_attr
49 from sqlalchemy.ext.hybrid import hybrid_property
49 from sqlalchemy.ext.hybrid import hybrid_property
50 from sqlalchemy.exc import IntegrityError # pragma: no cover
50 from sqlalchemy.exc import IntegrityError # pragma: no cover
51 from sqlalchemy.dialects.mysql import LONGTEXT
51 from sqlalchemy.dialects.mysql import LONGTEXT
52 from zope.cachedescriptors.property import Lazy as LazyProperty
52 from zope.cachedescriptors.property import Lazy as LazyProperty
53 from pyramid import compat
53 from pyramid import compat
54 from pyramid.threadlocal import get_current_request
54 from pyramid.threadlocal import get_current_request
55 from webhelpers.text import collapse, remove_formatting
55 from webhelpers.text import collapse, remove_formatting
56
56
57 from rhodecode.translation import _
57 from rhodecode.translation import _
58 from rhodecode.lib.vcs import get_vcs_instance
58 from rhodecode.lib.vcs import get_vcs_instance
59 from rhodecode.lib.vcs.backends.base import EmptyCommit, Reference
59 from rhodecode.lib.vcs.backends.base import EmptyCommit, Reference
60 from rhodecode.lib.utils2 import (
60 from rhodecode.lib.utils2 import (
61 str2bool, safe_str, get_commit_safe, safe_unicode, sha1_safe,
61 str2bool, safe_str, get_commit_safe, safe_unicode, sha1_safe,
62 time_to_datetime, aslist, Optional, safe_int, get_clone_url, AttributeDict,
62 time_to_datetime, aslist, Optional, safe_int, get_clone_url, AttributeDict,
63 glob2re, StrictAttributeDict, cleaned_uri, datetime_to_time, OrderedDefaultDict)
63 glob2re, StrictAttributeDict, cleaned_uri, datetime_to_time, OrderedDefaultDict)
64 from rhodecode.lib.jsonalchemy import MutationObj, MutationList, JsonType, \
64 from rhodecode.lib.jsonalchemy import MutationObj, MutationList, JsonType, \
65 JsonRaw
65 JsonRaw
66 from rhodecode.lib.ext_json import json
66 from rhodecode.lib.ext_json import json
67 from rhodecode.lib.caching_query import FromCache
67 from rhodecode.lib.caching_query import FromCache
68 from rhodecode.lib.encrypt import AESCipher, validate_and_get_enc_data
68 from rhodecode.lib.encrypt import AESCipher, validate_and_get_enc_data
69 from rhodecode.lib.encrypt2 import Encryptor
69 from rhodecode.lib.encrypt2 import Encryptor
70 from rhodecode.model.meta import Base, Session
70 from rhodecode.model.meta import Base, Session
71
71
72 URL_SEP = '/'
72 URL_SEP = '/'
73 log = logging.getLogger(__name__)
73 log = logging.getLogger(__name__)
74
74
75 # =============================================================================
75 # =============================================================================
76 # BASE CLASSES
76 # BASE CLASSES
77 # =============================================================================
77 # =============================================================================
78
78
79 # this is propagated from .ini file rhodecode.encrypted_values.secret or
79 # this is propagated from .ini file rhodecode.encrypted_values.secret or
80 # beaker.session.secret if first is not set.
80 # beaker.session.secret if first is not set.
81 # and initialized at environment.py
81 # and initialized at environment.py
82 ENCRYPTION_KEY = None
82 ENCRYPTION_KEY = None
83
83
84 # used to sort permissions by types, '#' used here is not allowed to be in
84 # used to sort permissions by types, '#' used here is not allowed to be in
85 # usernames, and it's very early in sorted string.printable table.
85 # usernames, and it's very early in sorted string.printable table.
86 PERMISSION_TYPE_SORT = {
86 PERMISSION_TYPE_SORT = {
87 'admin': '####',
87 'admin': '####',
88 'write': '###',
88 'write': '###',
89 'read': '##',
89 'read': '##',
90 'none': '#',
90 'none': '#',
91 }
91 }
92
92
93
93
94 def display_user_sort(obj):
94 def display_user_sort(obj):
95 """
95 """
96 Sort function used to sort permissions in .permissions() function of
96 Sort function used to sort permissions in .permissions() function of
97 Repository, RepoGroup, UserGroup. Also it put the default user in front
97 Repository, RepoGroup, UserGroup. Also it put the default user in front
98 of all other resources
98 of all other resources
99 """
99 """
100
100
101 if obj.username == User.DEFAULT_USER:
101 if obj.username == User.DEFAULT_USER:
102 return '#####'
102 return '#####'
103 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
103 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
104 return prefix + obj.username
104 return prefix + obj.username
105
105
106
106
107 def display_user_group_sort(obj):
107 def display_user_group_sort(obj):
108 """
108 """
109 Sort function used to sort permissions in .permissions() function of
109 Sort function used to sort permissions in .permissions() function of
110 Repository, RepoGroup, UserGroup. Also it put the default user in front
110 Repository, RepoGroup, UserGroup. Also it put the default user in front
111 of all other resources
111 of all other resources
112 """
112 """
113
113
114 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
114 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
115 return prefix + obj.users_group_name
115 return prefix + obj.users_group_name
116
116
117
117
118 def _hash_key(k):
118 def _hash_key(k):
119 return sha1_safe(k)
119 return sha1_safe(k)
120
120
121
121
122 def in_filter_generator(qry, items, limit=500):
122 def in_filter_generator(qry, items, limit=500):
123 """
123 """
124 Splits IN() into multiple with OR
124 Splits IN() into multiple with OR
125 e.g.::
125 e.g.::
126 cnt = Repository.query().filter(
126 cnt = Repository.query().filter(
127 or_(
127 or_(
128 *in_filter_generator(Repository.repo_id, range(100000))
128 *in_filter_generator(Repository.repo_id, range(100000))
129 )).count()
129 )).count()
130 """
130 """
131 if not items:
131 if not items:
132 # empty list will cause empty query which might cause security issues
132 # empty list will cause empty query which might cause security issues
133 # this can lead to hidden unpleasant results
133 # this can lead to hidden unpleasant results
134 items = [-1]
134 items = [-1]
135
135
136 parts = []
136 parts = []
137 for chunk in xrange(0, len(items), limit):
137 for chunk in xrange(0, len(items), limit):
138 parts.append(
138 parts.append(
139 qry.in_(items[chunk: chunk + limit])
139 qry.in_(items[chunk: chunk + limit])
140 )
140 )
141
141
142 return parts
142 return parts
143
143
144
144
145 base_table_args = {
145 base_table_args = {
146 'extend_existing': True,
146 'extend_existing': True,
147 'mysql_engine': 'InnoDB',
147 'mysql_engine': 'InnoDB',
148 'mysql_charset': 'utf8',
148 'mysql_charset': 'utf8',
149 'sqlite_autoincrement': True
149 'sqlite_autoincrement': True
150 }
150 }
151
151
152
152
153 class EncryptedTextValue(TypeDecorator):
153 class EncryptedTextValue(TypeDecorator):
154 """
154 """
155 Special column for encrypted long text data, use like::
155 Special column for encrypted long text data, use like::
156
156
157 value = Column("encrypted_value", EncryptedValue(), nullable=False)
157 value = Column("encrypted_value", EncryptedValue(), nullable=False)
158
158
159 This column is intelligent so if value is in unencrypted form it return
159 This column is intelligent so if value is in unencrypted form it return
160 unencrypted form, but on save it always encrypts
160 unencrypted form, but on save it always encrypts
161 """
161 """
162 impl = Text
162 impl = Text
163
163
164 def process_bind_param(self, value, dialect):
164 def process_bind_param(self, value, dialect):
165 """
165 """
166 Setter for storing value
166 Setter for storing value
167 """
167 """
168 import rhodecode
168 import rhodecode
169 if not value:
169 if not value:
170 return value
170 return value
171
171
172 # protect against double encrypting if values is already encrypted
172 # protect against double encrypting if values is already encrypted
173 if value.startswith('enc$aes$') \
173 if value.startswith('enc$aes$') \
174 or value.startswith('enc$aes_hmac$') \
174 or value.startswith('enc$aes_hmac$') \
175 or value.startswith('enc2$'):
175 or value.startswith('enc2$'):
176 raise ValueError('value needs to be in unencrypted format, '
176 raise ValueError('value needs to be in unencrypted format, '
177 'ie. not starting with enc$ or enc2$')
177 'ie. not starting with enc$ or enc2$')
178
178
179 algo = rhodecode.CONFIG.get('rhodecode.encrypted_values.algorithm') or 'aes'
179 algo = rhodecode.CONFIG.get('rhodecode.encrypted_values.algorithm') or 'aes'
180 if algo == 'aes':
180 if algo == 'aes':
181 return 'enc$aes_hmac$%s' % AESCipher(ENCRYPTION_KEY, hmac=True).encrypt(value)
181 return 'enc$aes_hmac$%s' % AESCipher(ENCRYPTION_KEY, hmac=True).encrypt(value)
182 elif algo == 'fernet':
182 elif algo == 'fernet':
183 return Encryptor(ENCRYPTION_KEY).encrypt(value)
183 return Encryptor(ENCRYPTION_KEY).encrypt(value)
184 else:
184 else:
185 ValueError('Bad encryption algorithm, should be fernet or aes, got: {}'.format(algo))
185 ValueError('Bad encryption algorithm, should be fernet or aes, got: {}'.format(algo))
186
186
187 def process_result_value(self, value, dialect):
187 def process_result_value(self, value, dialect):
188 """
188 """
189 Getter for retrieving value
189 Getter for retrieving value
190 """
190 """
191
191
192 import rhodecode
192 import rhodecode
193 if not value:
193 if not value:
194 return value
194 return value
195
195
196 algo = rhodecode.CONFIG.get('rhodecode.encrypted_values.algorithm') or 'aes'
196 algo = rhodecode.CONFIG.get('rhodecode.encrypted_values.algorithm') or 'aes'
197 enc_strict_mode = str2bool(rhodecode.CONFIG.get('rhodecode.encrypted_values.strict') or True)
197 enc_strict_mode = str2bool(rhodecode.CONFIG.get('rhodecode.encrypted_values.strict') or True)
198 if algo == 'aes':
198 if algo == 'aes':
199 decrypted_data = validate_and_get_enc_data(value, ENCRYPTION_KEY, enc_strict_mode)
199 decrypted_data = validate_and_get_enc_data(value, ENCRYPTION_KEY, enc_strict_mode)
200 elif algo == 'fernet':
200 elif algo == 'fernet':
201 return Encryptor(ENCRYPTION_KEY).decrypt(value)
201 return Encryptor(ENCRYPTION_KEY).decrypt(value)
202 else:
202 else:
203 ValueError('Bad encryption algorithm, should be fernet or aes, got: {}'.format(algo))
203 ValueError('Bad encryption algorithm, should be fernet or aes, got: {}'.format(algo))
204 return decrypted_data
204 return decrypted_data
205
205
206
206
207 class BaseModel(object):
207 class BaseModel(object):
208 """
208 """
209 Base Model for all classes
209 Base Model for all classes
210 """
210 """
211
211
212 @classmethod
212 @classmethod
213 def _get_keys(cls):
213 def _get_keys(cls):
214 """return column names for this model """
214 """return column names for this model """
215 return class_mapper(cls).c.keys()
215 return class_mapper(cls).c.keys()
216
216
217 def get_dict(self):
217 def get_dict(self):
218 """
218 """
219 return dict with keys and values corresponding
219 return dict with keys and values corresponding
220 to this model data """
220 to this model data """
221
221
222 d = {}
222 d = {}
223 for k in self._get_keys():
223 for k in self._get_keys():
224 d[k] = getattr(self, k)
224 d[k] = getattr(self, k)
225
225
226 # also use __json__() if present to get additional fields
226 # also use __json__() if present to get additional fields
227 _json_attr = getattr(self, '__json__', None)
227 _json_attr = getattr(self, '__json__', None)
228 if _json_attr:
228 if _json_attr:
229 # update with attributes from __json__
229 # update with attributes from __json__
230 if callable(_json_attr):
230 if callable(_json_attr):
231 _json_attr = _json_attr()
231 _json_attr = _json_attr()
232 for k, val in _json_attr.iteritems():
232 for k, val in _json_attr.iteritems():
233 d[k] = val
233 d[k] = val
234 return d
234 return d
235
235
236 def get_appstruct(self):
236 def get_appstruct(self):
237 """return list with keys and values tuples corresponding
237 """return list with keys and values tuples corresponding
238 to this model data """
238 to this model data """
239
239
240 lst = []
240 lst = []
241 for k in self._get_keys():
241 for k in self._get_keys():
242 lst.append((k, getattr(self, k),))
242 lst.append((k, getattr(self, k),))
243 return lst
243 return lst
244
244
245 def populate_obj(self, populate_dict):
245 def populate_obj(self, populate_dict):
246 """populate model with data from given populate_dict"""
246 """populate model with data from given populate_dict"""
247
247
248 for k in self._get_keys():
248 for k in self._get_keys():
249 if k in populate_dict:
249 if k in populate_dict:
250 setattr(self, k, populate_dict[k])
250 setattr(self, k, populate_dict[k])
251
251
252 @classmethod
252 @classmethod
253 def query(cls):
253 def query(cls):
254 return Session().query(cls)
254 return Session().query(cls)
255
255
256 @classmethod
256 @classmethod
257 def get(cls, id_):
257 def get(cls, id_):
258 if id_:
258 if id_:
259 return cls.query().get(id_)
259 return cls.query().get(id_)
260
260
261 @classmethod
261 @classmethod
262 def get_or_404(cls, id_):
262 def get_or_404(cls, id_):
263 from pyramid.httpexceptions import HTTPNotFound
263 from pyramid.httpexceptions import HTTPNotFound
264
264
265 try:
265 try:
266 id_ = int(id_)
266 id_ = int(id_)
267 except (TypeError, ValueError):
267 except (TypeError, ValueError):
268 raise HTTPNotFound()
268 raise HTTPNotFound()
269
269
270 res = cls.query().get(id_)
270 res = cls.query().get(id_)
271 if not res:
271 if not res:
272 raise HTTPNotFound()
272 raise HTTPNotFound()
273 return res
273 return res
274
274
275 @classmethod
275 @classmethod
276 def getAll(cls):
276 def getAll(cls):
277 # deprecated and left for backward compatibility
277 # deprecated and left for backward compatibility
278 return cls.get_all()
278 return cls.get_all()
279
279
280 @classmethod
280 @classmethod
281 def get_all(cls):
281 def get_all(cls):
282 return cls.query().all()
282 return cls.query().all()
283
283
284 @classmethod
284 @classmethod
285 def delete(cls, id_):
285 def delete(cls, id_):
286 obj = cls.query().get(id_)
286 obj = cls.query().get(id_)
287 Session().delete(obj)
287 Session().delete(obj)
288
288
289 @classmethod
289 @classmethod
290 def identity_cache(cls, session, attr_name, value):
290 def identity_cache(cls, session, attr_name, value):
291 exist_in_session = []
291 exist_in_session = []
292 for (item_cls, pkey), instance in session.identity_map.items():
292 for (item_cls, pkey), instance in session.identity_map.items():
293 if cls == item_cls and getattr(instance, attr_name) == value:
293 if cls == item_cls and getattr(instance, attr_name) == value:
294 exist_in_session.append(instance)
294 exist_in_session.append(instance)
295 if exist_in_session:
295 if exist_in_session:
296 if len(exist_in_session) == 1:
296 if len(exist_in_session) == 1:
297 return exist_in_session[0]
297 return exist_in_session[0]
298 log.exception(
298 log.exception(
299 'multiple objects with attr %s and '
299 'multiple objects with attr %s and '
300 'value %s found with same name: %r',
300 'value %s found with same name: %r',
301 attr_name, value, exist_in_session)
301 attr_name, value, exist_in_session)
302
302
303 def __repr__(self):
303 def __repr__(self):
304 if hasattr(self, '__unicode__'):
304 if hasattr(self, '__unicode__'):
305 # python repr needs to return str
305 # python repr needs to return str
306 try:
306 try:
307 return safe_str(self.__unicode__())
307 return safe_str(self.__unicode__())
308 except UnicodeDecodeError:
308 except UnicodeDecodeError:
309 pass
309 pass
310 return '<DB:%s>' % (self.__class__.__name__)
310 return '<DB:%s>' % (self.__class__.__name__)
311
311
312
312
313 class RhodeCodeSetting(Base, BaseModel):
313 class RhodeCodeSetting(Base, BaseModel):
314 __tablename__ = 'rhodecode_settings'
314 __tablename__ = 'rhodecode_settings'
315 __table_args__ = (
315 __table_args__ = (
316 UniqueConstraint('app_settings_name'),
316 UniqueConstraint('app_settings_name'),
317 base_table_args
317 base_table_args
318 )
318 )
319
319
320 SETTINGS_TYPES = {
320 SETTINGS_TYPES = {
321 'str': safe_str,
321 'str': safe_str,
322 'int': safe_int,
322 'int': safe_int,
323 'unicode': safe_unicode,
323 'unicode': safe_unicode,
324 'bool': str2bool,
324 'bool': str2bool,
325 'list': functools.partial(aslist, sep=',')
325 'list': functools.partial(aslist, sep=',')
326 }
326 }
327 DEFAULT_UPDATE_URL = 'https://rhodecode.com/api/v1/info/versions'
327 DEFAULT_UPDATE_URL = 'https://rhodecode.com/api/v1/info/versions'
328 GLOBAL_CONF_KEY = 'app_settings'
328 GLOBAL_CONF_KEY = 'app_settings'
329
329
330 app_settings_id = Column("app_settings_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
330 app_settings_id = Column("app_settings_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
331 app_settings_name = Column("app_settings_name", String(255), nullable=True, unique=None, default=None)
331 app_settings_name = Column("app_settings_name", String(255), nullable=True, unique=None, default=None)
332 _app_settings_value = Column("app_settings_value", String(4096), nullable=True, unique=None, default=None)
332 _app_settings_value = Column("app_settings_value", String(4096), nullable=True, unique=None, default=None)
333 _app_settings_type = Column("app_settings_type", String(255), nullable=True, unique=None, default=None)
333 _app_settings_type = Column("app_settings_type", String(255), nullable=True, unique=None, default=None)
334
334
335 def __init__(self, key='', val='', type='unicode'):
335 def __init__(self, key='', val='', type='unicode'):
336 self.app_settings_name = key
336 self.app_settings_name = key
337 self.app_settings_type = type
337 self.app_settings_type = type
338 self.app_settings_value = val
338 self.app_settings_value = val
339
339
340 @validates('_app_settings_value')
340 @validates('_app_settings_value')
341 def validate_settings_value(self, key, val):
341 def validate_settings_value(self, key, val):
342 assert type(val) == unicode
342 assert type(val) == unicode
343 return val
343 return val
344
344
345 @hybrid_property
345 @hybrid_property
346 def app_settings_value(self):
346 def app_settings_value(self):
347 v = self._app_settings_value
347 v = self._app_settings_value
348 _type = self.app_settings_type
348 _type = self.app_settings_type
349 if _type:
349 if _type:
350 _type = self.app_settings_type.split('.')[0]
350 _type = self.app_settings_type.split('.')[0]
351 # decode the encrypted value
351 # decode the encrypted value
352 if 'encrypted' in self.app_settings_type:
352 if 'encrypted' in self.app_settings_type:
353 cipher = EncryptedTextValue()
353 cipher = EncryptedTextValue()
354 v = safe_unicode(cipher.process_result_value(v, None))
354 v = safe_unicode(cipher.process_result_value(v, None))
355
355
356 converter = self.SETTINGS_TYPES.get(_type) or \
356 converter = self.SETTINGS_TYPES.get(_type) or \
357 self.SETTINGS_TYPES['unicode']
357 self.SETTINGS_TYPES['unicode']
358 return converter(v)
358 return converter(v)
359
359
360 @app_settings_value.setter
360 @app_settings_value.setter
361 def app_settings_value(self, val):
361 def app_settings_value(self, val):
362 """
362 """
363 Setter that will always make sure we use unicode in app_settings_value
363 Setter that will always make sure we use unicode in app_settings_value
364
364
365 :param val:
365 :param val:
366 """
366 """
367 val = safe_unicode(val)
367 val = safe_unicode(val)
368 # encode the encrypted value
368 # encode the encrypted value
369 if 'encrypted' in self.app_settings_type:
369 if 'encrypted' in self.app_settings_type:
370 cipher = EncryptedTextValue()
370 cipher = EncryptedTextValue()
371 val = safe_unicode(cipher.process_bind_param(val, None))
371 val = safe_unicode(cipher.process_bind_param(val, None))
372 self._app_settings_value = val
372 self._app_settings_value = val
373
373
374 @hybrid_property
374 @hybrid_property
375 def app_settings_type(self):
375 def app_settings_type(self):
376 return self._app_settings_type
376 return self._app_settings_type
377
377
378 @app_settings_type.setter
378 @app_settings_type.setter
379 def app_settings_type(self, val):
379 def app_settings_type(self, val):
380 if val.split('.')[0] not in self.SETTINGS_TYPES:
380 if val.split('.')[0] not in self.SETTINGS_TYPES:
381 raise Exception('type must be one of %s got %s'
381 raise Exception('type must be one of %s got %s'
382 % (self.SETTINGS_TYPES.keys(), val))
382 % (self.SETTINGS_TYPES.keys(), val))
383 self._app_settings_type = val
383 self._app_settings_type = val
384
384
385 @classmethod
385 @classmethod
386 def get_by_prefix(cls, prefix):
386 def get_by_prefix(cls, prefix):
387 return RhodeCodeSetting.query()\
387 return RhodeCodeSetting.query()\
388 .filter(RhodeCodeSetting.app_settings_name.startswith(prefix))\
388 .filter(RhodeCodeSetting.app_settings_name.startswith(prefix))\
389 .all()
389 .all()
390
390
391 def __unicode__(self):
391 def __unicode__(self):
392 return u"<%s('%s:%s[%s]')>" % (
392 return u"<%s('%s:%s[%s]')>" % (
393 self.__class__.__name__,
393 self.__class__.__name__,
394 self.app_settings_name, self.app_settings_value,
394 self.app_settings_name, self.app_settings_value,
395 self.app_settings_type
395 self.app_settings_type
396 )
396 )
397
397
398
398
399 class RhodeCodeUi(Base, BaseModel):
399 class RhodeCodeUi(Base, BaseModel):
400 __tablename__ = 'rhodecode_ui'
400 __tablename__ = 'rhodecode_ui'
401 __table_args__ = (
401 __table_args__ = (
402 UniqueConstraint('ui_key'),
402 UniqueConstraint('ui_key'),
403 base_table_args
403 base_table_args
404 )
404 )
405
405
406 HOOK_REPO_SIZE = 'changegroup.repo_size'
406 HOOK_REPO_SIZE = 'changegroup.repo_size'
407 # HG
407 # HG
408 HOOK_PRE_PULL = 'preoutgoing.pre_pull'
408 HOOK_PRE_PULL = 'preoutgoing.pre_pull'
409 HOOK_PULL = 'outgoing.pull_logger'
409 HOOK_PULL = 'outgoing.pull_logger'
410 HOOK_PRE_PUSH = 'prechangegroup.pre_push'
410 HOOK_PRE_PUSH = 'prechangegroup.pre_push'
411 HOOK_PRETX_PUSH = 'pretxnchangegroup.pre_push'
411 HOOK_PRETX_PUSH = 'pretxnchangegroup.pre_push'
412 HOOK_PUSH = 'changegroup.push_logger'
412 HOOK_PUSH = 'changegroup.push_logger'
413 HOOK_PUSH_KEY = 'pushkey.key_push'
413 HOOK_PUSH_KEY = 'pushkey.key_push'
414
414
415 HOOKS_BUILTIN = [
415 HOOKS_BUILTIN = [
416 HOOK_PRE_PULL,
416 HOOK_PRE_PULL,
417 HOOK_PULL,
417 HOOK_PULL,
418 HOOK_PRE_PUSH,
418 HOOK_PRE_PUSH,
419 HOOK_PRETX_PUSH,
419 HOOK_PRETX_PUSH,
420 HOOK_PUSH,
420 HOOK_PUSH,
421 HOOK_PUSH_KEY,
421 HOOK_PUSH_KEY,
422 ]
422 ]
423
423
424 # TODO: johbo: Unify way how hooks are configured for git and hg,
424 # TODO: johbo: Unify way how hooks are configured for git and hg,
425 # git part is currently hardcoded.
425 # git part is currently hardcoded.
426
426
427 # SVN PATTERNS
427 # SVN PATTERNS
428 SVN_BRANCH_ID = 'vcs_svn_branch'
428 SVN_BRANCH_ID = 'vcs_svn_branch'
429 SVN_TAG_ID = 'vcs_svn_tag'
429 SVN_TAG_ID = 'vcs_svn_tag'
430
430
431 ui_id = Column(
431 ui_id = Column(
432 "ui_id", Integer(), nullable=False, unique=True, default=None,
432 "ui_id", Integer(), nullable=False, unique=True, default=None,
433 primary_key=True)
433 primary_key=True)
434 ui_section = Column(
434 ui_section = Column(
435 "ui_section", String(255), nullable=True, unique=None, default=None)
435 "ui_section", String(255), nullable=True, unique=None, default=None)
436 ui_key = Column(
436 ui_key = Column(
437 "ui_key", String(255), nullable=True, unique=None, default=None)
437 "ui_key", String(255), nullable=True, unique=None, default=None)
438 ui_value = Column(
438 ui_value = Column(
439 "ui_value", String(255), nullable=True, unique=None, default=None)
439 "ui_value", String(255), nullable=True, unique=None, default=None)
440 ui_active = Column(
440 ui_active = Column(
441 "ui_active", Boolean(), nullable=True, unique=None, default=True)
441 "ui_active", Boolean(), nullable=True, unique=None, default=True)
442
442
443 def __repr__(self):
443 def __repr__(self):
444 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.ui_section,
444 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.ui_section,
445 self.ui_key, self.ui_value)
445 self.ui_key, self.ui_value)
446
446
447
447
448 class RepoRhodeCodeSetting(Base, BaseModel):
448 class RepoRhodeCodeSetting(Base, BaseModel):
449 __tablename__ = 'repo_rhodecode_settings'
449 __tablename__ = 'repo_rhodecode_settings'
450 __table_args__ = (
450 __table_args__ = (
451 UniqueConstraint(
451 UniqueConstraint(
452 'app_settings_name', 'repository_id',
452 'app_settings_name', 'repository_id',
453 name='uq_repo_rhodecode_setting_name_repo_id'),
453 name='uq_repo_rhodecode_setting_name_repo_id'),
454 base_table_args
454 base_table_args
455 )
455 )
456
456
457 repository_id = Column(
457 repository_id = Column(
458 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
458 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
459 nullable=False)
459 nullable=False)
460 app_settings_id = Column(
460 app_settings_id = Column(
461 "app_settings_id", Integer(), nullable=False, unique=True,
461 "app_settings_id", Integer(), nullable=False, unique=True,
462 default=None, primary_key=True)
462 default=None, primary_key=True)
463 app_settings_name = Column(
463 app_settings_name = Column(
464 "app_settings_name", String(255), nullable=True, unique=None,
464 "app_settings_name", String(255), nullable=True, unique=None,
465 default=None)
465 default=None)
466 _app_settings_value = Column(
466 _app_settings_value = Column(
467 "app_settings_value", String(4096), nullable=True, unique=None,
467 "app_settings_value", String(4096), nullable=True, unique=None,
468 default=None)
468 default=None)
469 _app_settings_type = Column(
469 _app_settings_type = Column(
470 "app_settings_type", String(255), nullable=True, unique=None,
470 "app_settings_type", String(255), nullable=True, unique=None,
471 default=None)
471 default=None)
472
472
473 repository = relationship('Repository')
473 repository = relationship('Repository')
474
474
475 def __init__(self, repository_id, key='', val='', type='unicode'):
475 def __init__(self, repository_id, key='', val='', type='unicode'):
476 self.repository_id = repository_id
476 self.repository_id = repository_id
477 self.app_settings_name = key
477 self.app_settings_name = key
478 self.app_settings_type = type
478 self.app_settings_type = type
479 self.app_settings_value = val
479 self.app_settings_value = val
480
480
481 @validates('_app_settings_value')
481 @validates('_app_settings_value')
482 def validate_settings_value(self, key, val):
482 def validate_settings_value(self, key, val):
483 assert type(val) == unicode
483 assert type(val) == unicode
484 return val
484 return val
485
485
486 @hybrid_property
486 @hybrid_property
487 def app_settings_value(self):
487 def app_settings_value(self):
488 v = self._app_settings_value
488 v = self._app_settings_value
489 type_ = self.app_settings_type
489 type_ = self.app_settings_type
490 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
490 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
491 converter = SETTINGS_TYPES.get(type_) or SETTINGS_TYPES['unicode']
491 converter = SETTINGS_TYPES.get(type_) or SETTINGS_TYPES['unicode']
492 return converter(v)
492 return converter(v)
493
493
494 @app_settings_value.setter
494 @app_settings_value.setter
495 def app_settings_value(self, val):
495 def app_settings_value(self, val):
496 """
496 """
497 Setter that will always make sure we use unicode in app_settings_value
497 Setter that will always make sure we use unicode in app_settings_value
498
498
499 :param val:
499 :param val:
500 """
500 """
501 self._app_settings_value = safe_unicode(val)
501 self._app_settings_value = safe_unicode(val)
502
502
503 @hybrid_property
503 @hybrid_property
504 def app_settings_type(self):
504 def app_settings_type(self):
505 return self._app_settings_type
505 return self._app_settings_type
506
506
507 @app_settings_type.setter
507 @app_settings_type.setter
508 def app_settings_type(self, val):
508 def app_settings_type(self, val):
509 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
509 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
510 if val not in SETTINGS_TYPES:
510 if val not in SETTINGS_TYPES:
511 raise Exception('type must be one of %s got %s'
511 raise Exception('type must be one of %s got %s'
512 % (SETTINGS_TYPES.keys(), val))
512 % (SETTINGS_TYPES.keys(), val))
513 self._app_settings_type = val
513 self._app_settings_type = val
514
514
515 def __unicode__(self):
515 def __unicode__(self):
516 return u"<%s('%s:%s:%s[%s]')>" % (
516 return u"<%s('%s:%s:%s[%s]')>" % (
517 self.__class__.__name__, self.repository.repo_name,
517 self.__class__.__name__, self.repository.repo_name,
518 self.app_settings_name, self.app_settings_value,
518 self.app_settings_name, self.app_settings_value,
519 self.app_settings_type
519 self.app_settings_type
520 )
520 )
521
521
522
522
523 class RepoRhodeCodeUi(Base, BaseModel):
523 class RepoRhodeCodeUi(Base, BaseModel):
524 __tablename__ = 'repo_rhodecode_ui'
524 __tablename__ = 'repo_rhodecode_ui'
525 __table_args__ = (
525 __table_args__ = (
526 UniqueConstraint(
526 UniqueConstraint(
527 'repository_id', 'ui_section', 'ui_key',
527 'repository_id', 'ui_section', 'ui_key',
528 name='uq_repo_rhodecode_ui_repository_id_section_key'),
528 name='uq_repo_rhodecode_ui_repository_id_section_key'),
529 base_table_args
529 base_table_args
530 )
530 )
531
531
532 repository_id = Column(
532 repository_id = Column(
533 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
533 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
534 nullable=False)
534 nullable=False)
535 ui_id = Column(
535 ui_id = Column(
536 "ui_id", Integer(), nullable=False, unique=True, default=None,
536 "ui_id", Integer(), nullable=False, unique=True, default=None,
537 primary_key=True)
537 primary_key=True)
538 ui_section = Column(
538 ui_section = Column(
539 "ui_section", String(255), nullable=True, unique=None, default=None)
539 "ui_section", String(255), nullable=True, unique=None, default=None)
540 ui_key = Column(
540 ui_key = Column(
541 "ui_key", String(255), nullable=True, unique=None, default=None)
541 "ui_key", String(255), nullable=True, unique=None, default=None)
542 ui_value = Column(
542 ui_value = Column(
543 "ui_value", String(255), nullable=True, unique=None, default=None)
543 "ui_value", String(255), nullable=True, unique=None, default=None)
544 ui_active = Column(
544 ui_active = Column(
545 "ui_active", Boolean(), nullable=True, unique=None, default=True)
545 "ui_active", Boolean(), nullable=True, unique=None, default=True)
546
546
547 repository = relationship('Repository')
547 repository = relationship('Repository')
548
548
549 def __repr__(self):
549 def __repr__(self):
550 return '<%s[%s:%s]%s=>%s]>' % (
550 return '<%s[%s:%s]%s=>%s]>' % (
551 self.__class__.__name__, self.repository.repo_name,
551 self.__class__.__name__, self.repository.repo_name,
552 self.ui_section, self.ui_key, self.ui_value)
552 self.ui_section, self.ui_key, self.ui_value)
553
553
554
554
555 class User(Base, BaseModel):
555 class User(Base, BaseModel):
556 __tablename__ = 'users'
556 __tablename__ = 'users'
557 __table_args__ = (
557 __table_args__ = (
558 UniqueConstraint('username'), UniqueConstraint('email'),
558 UniqueConstraint('username'), UniqueConstraint('email'),
559 Index('u_username_idx', 'username'),
559 Index('u_username_idx', 'username'),
560 Index('u_email_idx', 'email'),
560 Index('u_email_idx', 'email'),
561 base_table_args
561 base_table_args
562 )
562 )
563
563
564 DEFAULT_USER = 'default'
564 DEFAULT_USER = 'default'
565 DEFAULT_USER_EMAIL = 'anonymous@rhodecode.org'
565 DEFAULT_USER_EMAIL = 'anonymous@rhodecode.org'
566 DEFAULT_GRAVATAR_URL = 'https://secure.gravatar.com/avatar/{md5email}?d=identicon&s={size}'
566 DEFAULT_GRAVATAR_URL = 'https://secure.gravatar.com/avatar/{md5email}?d=identicon&s={size}'
567
567
568 user_id = Column("user_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
568 user_id = Column("user_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
569 username = Column("username", String(255), nullable=True, unique=None, default=None)
569 username = Column("username", String(255), nullable=True, unique=None, default=None)
570 password = Column("password", String(255), nullable=True, unique=None, default=None)
570 password = Column("password", String(255), nullable=True, unique=None, default=None)
571 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
571 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
572 admin = Column("admin", Boolean(), nullable=True, unique=None, default=False)
572 admin = Column("admin", Boolean(), nullable=True, unique=None, default=False)
573 name = Column("firstname", String(255), nullable=True, unique=None, default=None)
573 name = Column("firstname", String(255), nullable=True, unique=None, default=None)
574 lastname = Column("lastname", String(255), nullable=True, unique=None, default=None)
574 lastname = Column("lastname", String(255), nullable=True, unique=None, default=None)
575 _email = Column("email", String(255), nullable=True, unique=None, default=None)
575 _email = Column("email", String(255), nullable=True, unique=None, default=None)
576 last_login = Column("last_login", DateTime(timezone=False), nullable=True, unique=None, default=None)
576 last_login = Column("last_login", DateTime(timezone=False), nullable=True, unique=None, default=None)
577 last_activity = Column('last_activity', DateTime(timezone=False), nullable=True, unique=None, default=None)
577 last_activity = Column('last_activity', DateTime(timezone=False), nullable=True, unique=None, default=None)
578
578
579 extern_type = Column("extern_type", String(255), nullable=True, unique=None, default=None)
579 extern_type = Column("extern_type", String(255), nullable=True, unique=None, default=None)
580 extern_name = Column("extern_name", String(255), nullable=True, unique=None, default=None)
580 extern_name = Column("extern_name", String(255), nullable=True, unique=None, default=None)
581 _api_key = Column("api_key", String(255), nullable=True, unique=None, default=None)
581 _api_key = Column("api_key", String(255), nullable=True, unique=None, default=None)
582 inherit_default_permissions = Column("inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
582 inherit_default_permissions = Column("inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
583 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
583 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
584 _user_data = Column("user_data", LargeBinary(), nullable=True) # JSON data
584 _user_data = Column("user_data", LargeBinary(), nullable=True) # JSON data
585
585
586 user_log = relationship('UserLog')
586 user_log = relationship('UserLog')
587 user_perms = relationship('UserToPerm', primaryjoin="User.user_id==UserToPerm.user_id", cascade='all, delete-orphan')
587 user_perms = relationship('UserToPerm', primaryjoin="User.user_id==UserToPerm.user_id", cascade='all, delete-orphan')
588
588
589 repositories = relationship('Repository')
589 repositories = relationship('Repository')
590 repository_groups = relationship('RepoGroup')
590 repository_groups = relationship('RepoGroup')
591 user_groups = relationship('UserGroup')
591 user_groups = relationship('UserGroup')
592
592
593 user_followers = relationship('UserFollowing', primaryjoin='UserFollowing.follows_user_id==User.user_id', cascade='all')
593 user_followers = relationship('UserFollowing', primaryjoin='UserFollowing.follows_user_id==User.user_id', cascade='all')
594 followings = relationship('UserFollowing', primaryjoin='UserFollowing.user_id==User.user_id', cascade='all')
594 followings = relationship('UserFollowing', primaryjoin='UserFollowing.user_id==User.user_id', cascade='all')
595
595
596 repo_to_perm = relationship('UserRepoToPerm', primaryjoin='UserRepoToPerm.user_id==User.user_id', cascade='all, delete-orphan')
596 repo_to_perm = relationship('UserRepoToPerm', primaryjoin='UserRepoToPerm.user_id==User.user_id', cascade='all, delete-orphan')
597 repo_group_to_perm = relationship('UserRepoGroupToPerm', primaryjoin='UserRepoGroupToPerm.user_id==User.user_id', cascade='all, delete-orphan')
597 repo_group_to_perm = relationship('UserRepoGroupToPerm', primaryjoin='UserRepoGroupToPerm.user_id==User.user_id', cascade='all, delete-orphan')
598 user_group_to_perm = relationship('UserUserGroupToPerm', primaryjoin='UserUserGroupToPerm.user_id==User.user_id', cascade='all, delete-orphan')
598 user_group_to_perm = relationship('UserUserGroupToPerm', primaryjoin='UserUserGroupToPerm.user_id==User.user_id', cascade='all, delete-orphan')
599
599
600 group_member = relationship('UserGroupMember', cascade='all')
600 group_member = relationship('UserGroupMember', cascade='all')
601
601
602 notifications = relationship('UserNotification', cascade='all')
602 notifications = relationship('UserNotification', cascade='all')
603 # notifications assigned to this user
603 # notifications assigned to this user
604 user_created_notifications = relationship('Notification', cascade='all')
604 user_created_notifications = relationship('Notification', cascade='all')
605 # comments created by this user
605 # comments created by this user
606 user_comments = relationship('ChangesetComment', cascade='all')
606 user_comments = relationship('ChangesetComment', cascade='all')
607 # user profile extra info
607 # user profile extra info
608 user_emails = relationship('UserEmailMap', cascade='all')
608 user_emails = relationship('UserEmailMap', cascade='all')
609 user_ip_map = relationship('UserIpMap', cascade='all')
609 user_ip_map = relationship('UserIpMap', cascade='all')
610 user_auth_tokens = relationship('UserApiKeys', cascade='all')
610 user_auth_tokens = relationship('UserApiKeys', cascade='all')
611 user_ssh_keys = relationship('UserSshKeys', cascade='all')
611 user_ssh_keys = relationship('UserSshKeys', cascade='all')
612
612
613 # gists
613 # gists
614 user_gists = relationship('Gist', cascade='all')
614 user_gists = relationship('Gist', cascade='all')
615 # user pull requests
615 # user pull requests
616 user_pull_requests = relationship('PullRequest', cascade='all')
616 user_pull_requests = relationship('PullRequest', cascade='all')
617 # external identities
617 # external identities
618 extenal_identities = relationship(
618 extenal_identities = relationship(
619 'ExternalIdentity',
619 'ExternalIdentity',
620 primaryjoin="User.user_id==ExternalIdentity.local_user_id",
620 primaryjoin="User.user_id==ExternalIdentity.local_user_id",
621 cascade='all')
621 cascade='all')
622 # review rules
622 # review rules
623 user_review_rules = relationship('RepoReviewRuleUser', cascade='all')
623 user_review_rules = relationship('RepoReviewRuleUser', cascade='all')
624
624
625 def __unicode__(self):
625 def __unicode__(self):
626 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
626 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
627 self.user_id, self.username)
627 self.user_id, self.username)
628
628
629 @hybrid_property
629 @hybrid_property
630 def email(self):
630 def email(self):
631 return self._email
631 return self._email
632
632
633 @email.setter
633 @email.setter
634 def email(self, val):
634 def email(self, val):
635 self._email = val.lower() if val else None
635 self._email = val.lower() if val else None
636
636
637 @hybrid_property
637 @hybrid_property
638 def first_name(self):
638 def first_name(self):
639 from rhodecode.lib import helpers as h
639 from rhodecode.lib import helpers as h
640 if self.name:
640 if self.name:
641 return h.escape(self.name)
641 return h.escape(self.name)
642 return self.name
642 return self.name
643
643
644 @hybrid_property
644 @hybrid_property
645 def last_name(self):
645 def last_name(self):
646 from rhodecode.lib import helpers as h
646 from rhodecode.lib import helpers as h
647 if self.lastname:
647 if self.lastname:
648 return h.escape(self.lastname)
648 return h.escape(self.lastname)
649 return self.lastname
649 return self.lastname
650
650
651 @hybrid_property
651 @hybrid_property
652 def api_key(self):
652 def api_key(self):
653 """
653 """
654 Fetch if exist an auth-token with role ALL connected to this user
654 Fetch if exist an auth-token with role ALL connected to this user
655 """
655 """
656 user_auth_token = UserApiKeys.query()\
656 user_auth_token = UserApiKeys.query()\
657 .filter(UserApiKeys.user_id == self.user_id)\
657 .filter(UserApiKeys.user_id == self.user_id)\
658 .filter(or_(UserApiKeys.expires == -1,
658 .filter(or_(UserApiKeys.expires == -1,
659 UserApiKeys.expires >= time.time()))\
659 UserApiKeys.expires >= time.time()))\
660 .filter(UserApiKeys.role == UserApiKeys.ROLE_ALL).first()
660 .filter(UserApiKeys.role == UserApiKeys.ROLE_ALL).first()
661 if user_auth_token:
661 if user_auth_token:
662 user_auth_token = user_auth_token.api_key
662 user_auth_token = user_auth_token.api_key
663
663
664 return user_auth_token
664 return user_auth_token
665
665
666 @api_key.setter
666 @api_key.setter
667 def api_key(self, val):
667 def api_key(self, val):
668 # don't allow to set API key this is deprecated for now
668 # don't allow to set API key this is deprecated for now
669 self._api_key = None
669 self._api_key = None
670
670
671 @property
671 @property
672 def reviewer_pull_requests(self):
672 def reviewer_pull_requests(self):
673 return PullRequestReviewers.query() \
673 return PullRequestReviewers.query() \
674 .options(joinedload(PullRequestReviewers.pull_request)) \
674 .options(joinedload(PullRequestReviewers.pull_request)) \
675 .filter(PullRequestReviewers.user_id == self.user_id) \
675 .filter(PullRequestReviewers.user_id == self.user_id) \
676 .all()
676 .all()
677
677
678 @property
678 @property
679 def firstname(self):
679 def firstname(self):
680 # alias for future
680 # alias for future
681 return self.name
681 return self.name
682
682
683 @property
683 @property
684 def emails(self):
684 def emails(self):
685 other = UserEmailMap.query()\
685 other = UserEmailMap.query()\
686 .filter(UserEmailMap.user == self) \
686 .filter(UserEmailMap.user == self) \
687 .order_by(UserEmailMap.email_id.asc()) \
687 .order_by(UserEmailMap.email_id.asc()) \
688 .all()
688 .all()
689 return [self.email] + [x.email for x in other]
689 return [self.email] + [x.email for x in other]
690
690
691 @property
691 @property
692 def auth_tokens(self):
692 def auth_tokens(self):
693 auth_tokens = self.get_auth_tokens()
693 auth_tokens = self.get_auth_tokens()
694 return [x.api_key for x in auth_tokens]
694 return [x.api_key for x in auth_tokens]
695
695
696 def get_auth_tokens(self):
696 def get_auth_tokens(self):
697 return UserApiKeys.query()\
697 return UserApiKeys.query()\
698 .filter(UserApiKeys.user == self)\
698 .filter(UserApiKeys.user == self)\
699 .order_by(UserApiKeys.user_api_key_id.asc())\
699 .order_by(UserApiKeys.user_api_key_id.asc())\
700 .all()
700 .all()
701
701
702 @LazyProperty
702 @LazyProperty
703 def feed_token(self):
703 def feed_token(self):
704 return self.get_feed_token()
704 return self.get_feed_token()
705
705
706 def get_feed_token(self, cache=True):
706 def get_feed_token(self, cache=True):
707 feed_tokens = UserApiKeys.query()\
707 feed_tokens = UserApiKeys.query()\
708 .filter(UserApiKeys.user == self)\
708 .filter(UserApiKeys.user == self)\
709 .filter(UserApiKeys.role == UserApiKeys.ROLE_FEED)
709 .filter(UserApiKeys.role == UserApiKeys.ROLE_FEED)
710 if cache:
710 if cache:
711 feed_tokens = feed_tokens.options(
711 feed_tokens = feed_tokens.options(
712 FromCache("sql_cache_short", "get_user_feed_token_%s" % self.user_id))
712 FromCache("sql_cache_short", "get_user_feed_token_%s" % self.user_id))
713
713
714 feed_tokens = feed_tokens.all()
714 feed_tokens = feed_tokens.all()
715 if feed_tokens:
715 if feed_tokens:
716 return feed_tokens[0].api_key
716 return feed_tokens[0].api_key
717 return 'NO_FEED_TOKEN_AVAILABLE'
717 return 'NO_FEED_TOKEN_AVAILABLE'
718
718
719 @classmethod
719 @classmethod
720 def get(cls, user_id, cache=False):
720 def get(cls, user_id, cache=False):
721 if not user_id:
721 if not user_id:
722 return
722 return
723
723
724 user = cls.query()
724 user = cls.query()
725 if cache:
725 if cache:
726 user = user.options(
726 user = user.options(
727 FromCache("sql_cache_short", "get_users_%s" % user_id))
727 FromCache("sql_cache_short", "get_users_%s" % user_id))
728 return user.get(user_id)
728 return user.get(user_id)
729
729
730 @classmethod
730 @classmethod
731 def extra_valid_auth_tokens(cls, user, role=None):
731 def extra_valid_auth_tokens(cls, user, role=None):
732 tokens = UserApiKeys.query().filter(UserApiKeys.user == user)\
732 tokens = UserApiKeys.query().filter(UserApiKeys.user == user)\
733 .filter(or_(UserApiKeys.expires == -1,
733 .filter(or_(UserApiKeys.expires == -1,
734 UserApiKeys.expires >= time.time()))
734 UserApiKeys.expires >= time.time()))
735 if role:
735 if role:
736 tokens = tokens.filter(or_(UserApiKeys.role == role,
736 tokens = tokens.filter(or_(UserApiKeys.role == role,
737 UserApiKeys.role == UserApiKeys.ROLE_ALL))
737 UserApiKeys.role == UserApiKeys.ROLE_ALL))
738 return tokens.all()
738 return tokens.all()
739
739
740 def authenticate_by_token(self, auth_token, roles=None, scope_repo_id=None):
740 def authenticate_by_token(self, auth_token, roles=None, scope_repo_id=None):
741 from rhodecode.lib import auth
741 from rhodecode.lib import auth
742
742
743 log.debug('Trying to authenticate user: %s via auth-token, '
743 log.debug('Trying to authenticate user: %s via auth-token, '
744 'and roles: %s', self, roles)
744 'and roles: %s', self, roles)
745
745
746 if not auth_token:
746 if not auth_token:
747 return False
747 return False
748
748
749 roles = (roles or []) + [UserApiKeys.ROLE_ALL]
749 roles = (roles or []) + [UserApiKeys.ROLE_ALL]
750 tokens_q = UserApiKeys.query()\
750 tokens_q = UserApiKeys.query()\
751 .filter(UserApiKeys.user_id == self.user_id)\
751 .filter(UserApiKeys.user_id == self.user_id)\
752 .filter(or_(UserApiKeys.expires == -1,
752 .filter(or_(UserApiKeys.expires == -1,
753 UserApiKeys.expires >= time.time()))
753 UserApiKeys.expires >= time.time()))
754
754
755 tokens_q = tokens_q.filter(UserApiKeys.role.in_(roles))
755 tokens_q = tokens_q.filter(UserApiKeys.role.in_(roles))
756
756
757 crypto_backend = auth.crypto_backend()
757 crypto_backend = auth.crypto_backend()
758 enc_token_map = {}
758 enc_token_map = {}
759 plain_token_map = {}
759 plain_token_map = {}
760 for token in tokens_q:
760 for token in tokens_q:
761 if token.api_key.startswith(crypto_backend.ENC_PREF):
761 if token.api_key.startswith(crypto_backend.ENC_PREF):
762 enc_token_map[token.api_key] = token
762 enc_token_map[token.api_key] = token
763 else:
763 else:
764 plain_token_map[token.api_key] = token
764 plain_token_map[token.api_key] = token
765 log.debug(
765 log.debug(
766 'Found %s plain and %s encrypted user tokens to check for authentication',
766 'Found %s plain and %s encrypted user tokens to check for authentication',
767 len(plain_token_map), len(enc_token_map))
767 len(plain_token_map), len(enc_token_map))
768
768
769 # plain token match comes first
769 # plain token match comes first
770 match = plain_token_map.get(auth_token)
770 match = plain_token_map.get(auth_token)
771
771
772 # check encrypted tokens now
772 # check encrypted tokens now
773 if not match:
773 if not match:
774 for token_hash, token in enc_token_map.items():
774 for token_hash, token in enc_token_map.items():
775 # NOTE(marcink): this is expensive to calculate, but most secure
775 # NOTE(marcink): this is expensive to calculate, but most secure
776 if crypto_backend.hash_check(auth_token, token_hash):
776 if crypto_backend.hash_check(auth_token, token_hash):
777 match = token
777 match = token
778 break
778 break
779
779
780 if match:
780 if match:
781 log.debug('Found matching token %s', match)
781 log.debug('Found matching token %s', match)
782 if match.repo_id:
782 if match.repo_id:
783 log.debug('Found scope, checking for scope match of token %s', match)
783 log.debug('Found scope, checking for scope match of token %s', match)
784 if match.repo_id == scope_repo_id:
784 if match.repo_id == scope_repo_id:
785 return True
785 return True
786 else:
786 else:
787 log.debug(
787 log.debug(
788 'AUTH_TOKEN: scope mismatch, token has a set repo scope: %s, '
788 'AUTH_TOKEN: scope mismatch, token has a set repo scope: %s, '
789 'and calling scope is:%s, skipping further checks',
789 'and calling scope is:%s, skipping further checks',
790 match.repo, scope_repo_id)
790 match.repo, scope_repo_id)
791 return False
791 return False
792 else:
792 else:
793 return True
793 return True
794
794
795 return False
795 return False
796
796
797 @property
797 @property
798 def ip_addresses(self):
798 def ip_addresses(self):
799 ret = UserIpMap.query().filter(UserIpMap.user == self).all()
799 ret = UserIpMap.query().filter(UserIpMap.user == self).all()
800 return [x.ip_addr for x in ret]
800 return [x.ip_addr for x in ret]
801
801
802 @property
802 @property
803 def username_and_name(self):
803 def username_and_name(self):
804 return '%s (%s %s)' % (self.username, self.first_name, self.last_name)
804 return '%s (%s %s)' % (self.username, self.first_name, self.last_name)
805
805
806 @property
806 @property
807 def username_or_name_or_email(self):
807 def username_or_name_or_email(self):
808 full_name = self.full_name if self.full_name is not ' ' else None
808 full_name = self.full_name if self.full_name is not ' ' else None
809 return self.username or full_name or self.email
809 return self.username or full_name or self.email
810
810
811 @property
811 @property
812 def full_name(self):
812 def full_name(self):
813 return '%s %s' % (self.first_name, self.last_name)
813 return '%s %s' % (self.first_name, self.last_name)
814
814
815 @property
815 @property
816 def full_name_or_username(self):
816 def full_name_or_username(self):
817 return ('%s %s' % (self.first_name, self.last_name)
817 return ('%s %s' % (self.first_name, self.last_name)
818 if (self.first_name and self.last_name) else self.username)
818 if (self.first_name and self.last_name) else self.username)
819
819
820 @property
820 @property
821 def full_contact(self):
821 def full_contact(self):
822 return '%s %s <%s>' % (self.first_name, self.last_name, self.email)
822 return '%s %s <%s>' % (self.first_name, self.last_name, self.email)
823
823
824 @property
824 @property
825 def short_contact(self):
825 def short_contact(self):
826 return '%s %s' % (self.first_name, self.last_name)
826 return '%s %s' % (self.first_name, self.last_name)
827
827
828 @property
828 @property
829 def is_admin(self):
829 def is_admin(self):
830 return self.admin
830 return self.admin
831
831
832 def AuthUser(self, **kwargs):
832 def AuthUser(self, **kwargs):
833 """
833 """
834 Returns instance of AuthUser for this user
834 Returns instance of AuthUser for this user
835 """
835 """
836 from rhodecode.lib.auth import AuthUser
836 from rhodecode.lib.auth import AuthUser
837 return AuthUser(user_id=self.user_id, username=self.username, **kwargs)
837 return AuthUser(user_id=self.user_id, username=self.username, **kwargs)
838
838
839 @hybrid_property
839 @hybrid_property
840 def user_data(self):
840 def user_data(self):
841 if not self._user_data:
841 if not self._user_data:
842 return {}
842 return {}
843
843
844 try:
844 try:
845 return json.loads(self._user_data)
845 return json.loads(self._user_data)
846 except TypeError:
846 except TypeError:
847 return {}
847 return {}
848
848
849 @user_data.setter
849 @user_data.setter
850 def user_data(self, val):
850 def user_data(self, val):
851 if not isinstance(val, dict):
851 if not isinstance(val, dict):
852 raise Exception('user_data must be dict, got %s' % type(val))
852 raise Exception('user_data must be dict, got %s' % type(val))
853 try:
853 try:
854 self._user_data = json.dumps(val)
854 self._user_data = json.dumps(val)
855 except Exception:
855 except Exception:
856 log.error(traceback.format_exc())
856 log.error(traceback.format_exc())
857
857
858 @classmethod
858 @classmethod
859 def get_by_username(cls, username, case_insensitive=False,
859 def get_by_username(cls, username, case_insensitive=False,
860 cache=False, identity_cache=False):
860 cache=False, identity_cache=False):
861 session = Session()
861 session = Session()
862
862
863 if case_insensitive:
863 if case_insensitive:
864 q = cls.query().filter(
864 q = cls.query().filter(
865 func.lower(cls.username) == func.lower(username))
865 func.lower(cls.username) == func.lower(username))
866 else:
866 else:
867 q = cls.query().filter(cls.username == username)
867 q = cls.query().filter(cls.username == username)
868
868
869 if cache:
869 if cache:
870 if identity_cache:
870 if identity_cache:
871 val = cls.identity_cache(session, 'username', username)
871 val = cls.identity_cache(session, 'username', username)
872 if val:
872 if val:
873 return val
873 return val
874 else:
874 else:
875 cache_key = "get_user_by_name_%s" % _hash_key(username)
875 cache_key = "get_user_by_name_%s" % _hash_key(username)
876 q = q.options(
876 q = q.options(
877 FromCache("sql_cache_short", cache_key))
877 FromCache("sql_cache_short", cache_key))
878
878
879 return q.scalar()
879 return q.scalar()
880
880
881 @classmethod
881 @classmethod
882 def get_by_auth_token(cls, auth_token, cache=False):
882 def get_by_auth_token(cls, auth_token, cache=False):
883 q = UserApiKeys.query()\
883 q = UserApiKeys.query()\
884 .filter(UserApiKeys.api_key == auth_token)\
884 .filter(UserApiKeys.api_key == auth_token)\
885 .filter(or_(UserApiKeys.expires == -1,
885 .filter(or_(UserApiKeys.expires == -1,
886 UserApiKeys.expires >= time.time()))
886 UserApiKeys.expires >= time.time()))
887 if cache:
887 if cache:
888 q = q.options(
888 q = q.options(
889 FromCache("sql_cache_short", "get_auth_token_%s" % auth_token))
889 FromCache("sql_cache_short", "get_auth_token_%s" % auth_token))
890
890
891 match = q.first()
891 match = q.first()
892 if match:
892 if match:
893 return match.user
893 return match.user
894
894
895 @classmethod
895 @classmethod
896 def get_by_email(cls, email, case_insensitive=False, cache=False):
896 def get_by_email(cls, email, case_insensitive=False, cache=False):
897
897
898 if case_insensitive:
898 if case_insensitive:
899 q = cls.query().filter(func.lower(cls.email) == func.lower(email))
899 q = cls.query().filter(func.lower(cls.email) == func.lower(email))
900
900
901 else:
901 else:
902 q = cls.query().filter(cls.email == email)
902 q = cls.query().filter(cls.email == email)
903
903
904 email_key = _hash_key(email)
904 email_key = _hash_key(email)
905 if cache:
905 if cache:
906 q = q.options(
906 q = q.options(
907 FromCache("sql_cache_short", "get_email_key_%s" % email_key))
907 FromCache("sql_cache_short", "get_email_key_%s" % email_key))
908
908
909 ret = q.scalar()
909 ret = q.scalar()
910 if ret is None:
910 if ret is None:
911 q = UserEmailMap.query()
911 q = UserEmailMap.query()
912 # try fetching in alternate email map
912 # try fetching in alternate email map
913 if case_insensitive:
913 if case_insensitive:
914 q = q.filter(func.lower(UserEmailMap.email) == func.lower(email))
914 q = q.filter(func.lower(UserEmailMap.email) == func.lower(email))
915 else:
915 else:
916 q = q.filter(UserEmailMap.email == email)
916 q = q.filter(UserEmailMap.email == email)
917 q = q.options(joinedload(UserEmailMap.user))
917 q = q.options(joinedload(UserEmailMap.user))
918 if cache:
918 if cache:
919 q = q.options(
919 q = q.options(
920 FromCache("sql_cache_short", "get_email_map_key_%s" % email_key))
920 FromCache("sql_cache_short", "get_email_map_key_%s" % email_key))
921 ret = getattr(q.scalar(), 'user', None)
921 ret = getattr(q.scalar(), 'user', None)
922
922
923 return ret
923 return ret
924
924
925 @classmethod
925 @classmethod
926 def get_from_cs_author(cls, author):
926 def get_from_cs_author(cls, author):
927 """
927 """
928 Tries to get User objects out of commit author string
928 Tries to get User objects out of commit author string
929
929
930 :param author:
930 :param author:
931 """
931 """
932 from rhodecode.lib.helpers import email, author_name
932 from rhodecode.lib.helpers import email, author_name
933 # Valid email in the attribute passed, see if they're in the system
933 # Valid email in the attribute passed, see if they're in the system
934 _email = email(author)
934 _email = email(author)
935 if _email:
935 if _email:
936 user = cls.get_by_email(_email, case_insensitive=True)
936 user = cls.get_by_email(_email, case_insensitive=True)
937 if user:
937 if user:
938 return user
938 return user
939 # Maybe we can match by username?
939 # Maybe we can match by username?
940 _author = author_name(author)
940 _author = author_name(author)
941 user = cls.get_by_username(_author, case_insensitive=True)
941 user = cls.get_by_username(_author, case_insensitive=True)
942 if user:
942 if user:
943 return user
943 return user
944
944
945 def update_userdata(self, **kwargs):
945 def update_userdata(self, **kwargs):
946 usr = self
946 usr = self
947 old = usr.user_data
947 old = usr.user_data
948 old.update(**kwargs)
948 old.update(**kwargs)
949 usr.user_data = old
949 usr.user_data = old
950 Session().add(usr)
950 Session().add(usr)
951 log.debug('updated userdata with %s', kwargs)
951 log.debug('updated userdata with %s', kwargs)
952
952
953 def update_lastlogin(self):
953 def update_lastlogin(self):
954 """Update user lastlogin"""
954 """Update user lastlogin"""
955 self.last_login = datetime.datetime.now()
955 self.last_login = datetime.datetime.now()
956 Session().add(self)
956 Session().add(self)
957 log.debug('updated user %s lastlogin', self.username)
957 log.debug('updated user %s lastlogin', self.username)
958
958
959 def update_password(self, new_password):
959 def update_password(self, new_password):
960 from rhodecode.lib.auth import get_crypt_password
960 from rhodecode.lib.auth import get_crypt_password
961
961
962 self.password = get_crypt_password(new_password)
962 self.password = get_crypt_password(new_password)
963 Session().add(self)
963 Session().add(self)
964
964
965 @classmethod
965 @classmethod
966 def get_first_super_admin(cls):
966 def get_first_super_admin(cls):
967 user = User.query()\
967 user = User.query()\
968 .filter(User.admin == true()) \
968 .filter(User.admin == true()) \
969 .order_by(User.user_id.asc()) \
969 .order_by(User.user_id.asc()) \
970 .first()
970 .first()
971
971
972 if user is None:
972 if user is None:
973 raise Exception('FATAL: Missing administrative account!')
973 raise Exception('FATAL: Missing administrative account!')
974 return user
974 return user
975
975
976 @classmethod
976 @classmethod
977 def get_all_super_admins(cls, only_active=False):
977 def get_all_super_admins(cls, only_active=False):
978 """
978 """
979 Returns all admin accounts sorted by username
979 Returns all admin accounts sorted by username
980 """
980 """
981 qry = User.query().filter(User.admin == true()).order_by(User.username.asc())
981 qry = User.query().filter(User.admin == true()).order_by(User.username.asc())
982 if only_active:
982 if only_active:
983 qry = qry.filter(User.active == true())
983 qry = qry.filter(User.active == true())
984 return qry.all()
984 return qry.all()
985
985
986 @classmethod
986 @classmethod
987 def get_default_user(cls, cache=False, refresh=False):
987 def get_default_user(cls, cache=False, refresh=False):
988 user = User.get_by_username(User.DEFAULT_USER, cache=cache)
988 user = User.get_by_username(User.DEFAULT_USER, cache=cache)
989 if user is None:
989 if user is None:
990 raise Exception('FATAL: Missing default account!')
990 raise Exception('FATAL: Missing default account!')
991 if refresh:
991 if refresh:
992 # The default user might be based on outdated state which
992 # The default user might be based on outdated state which
993 # has been loaded from the cache.
993 # has been loaded from the cache.
994 # A call to refresh() ensures that the
994 # A call to refresh() ensures that the
995 # latest state from the database is used.
995 # latest state from the database is used.
996 Session().refresh(user)
996 Session().refresh(user)
997 return user
997 return user
998
998
999 def _get_default_perms(self, user, suffix=''):
999 def _get_default_perms(self, user, suffix=''):
1000 from rhodecode.model.permission import PermissionModel
1000 from rhodecode.model.permission import PermissionModel
1001 return PermissionModel().get_default_perms(user.user_perms, suffix)
1001 return PermissionModel().get_default_perms(user.user_perms, suffix)
1002
1002
1003 def get_default_perms(self, suffix=''):
1003 def get_default_perms(self, suffix=''):
1004 return self._get_default_perms(self, suffix)
1004 return self._get_default_perms(self, suffix)
1005
1005
1006 def get_api_data(self, include_secrets=False, details='full'):
1006 def get_api_data(self, include_secrets=False, details='full'):
1007 """
1007 """
1008 Common function for generating user related data for API
1008 Common function for generating user related data for API
1009
1009
1010 :param include_secrets: By default secrets in the API data will be replaced
1010 :param include_secrets: By default secrets in the API data will be replaced
1011 by a placeholder value to prevent exposing this data by accident. In case
1011 by a placeholder value to prevent exposing this data by accident. In case
1012 this data shall be exposed, set this flag to ``True``.
1012 this data shall be exposed, set this flag to ``True``.
1013
1013
1014 :param details: details can be 'basic|full' basic gives only a subset of
1014 :param details: details can be 'basic|full' basic gives only a subset of
1015 the available user information that includes user_id, name and emails.
1015 the available user information that includes user_id, name and emails.
1016 """
1016 """
1017 user = self
1017 user = self
1018 user_data = self.user_data
1018 user_data = self.user_data
1019 data = {
1019 data = {
1020 'user_id': user.user_id,
1020 'user_id': user.user_id,
1021 'username': user.username,
1021 'username': user.username,
1022 'firstname': user.name,
1022 'firstname': user.name,
1023 'lastname': user.lastname,
1023 'lastname': user.lastname,
1024 'email': user.email,
1024 'email': user.email,
1025 'emails': user.emails,
1025 'emails': user.emails,
1026 }
1026 }
1027 if details == 'basic':
1027 if details == 'basic':
1028 return data
1028 return data
1029
1029
1030 auth_token_length = 40
1030 auth_token_length = 40
1031 auth_token_replacement = '*' * auth_token_length
1031 auth_token_replacement = '*' * auth_token_length
1032
1032
1033 extras = {
1033 extras = {
1034 'auth_tokens': [auth_token_replacement],
1034 'auth_tokens': [auth_token_replacement],
1035 'active': user.active,
1035 'active': user.active,
1036 'admin': user.admin,
1036 'admin': user.admin,
1037 'extern_type': user.extern_type,
1037 'extern_type': user.extern_type,
1038 'extern_name': user.extern_name,
1038 'extern_name': user.extern_name,
1039 'last_login': user.last_login,
1039 'last_login': user.last_login,
1040 'last_activity': user.last_activity,
1040 'last_activity': user.last_activity,
1041 'ip_addresses': user.ip_addresses,
1041 'ip_addresses': user.ip_addresses,
1042 'language': user_data.get('language')
1042 'language': user_data.get('language')
1043 }
1043 }
1044 data.update(extras)
1044 data.update(extras)
1045
1045
1046 if include_secrets:
1046 if include_secrets:
1047 data['auth_tokens'] = user.auth_tokens
1047 data['auth_tokens'] = user.auth_tokens
1048 return data
1048 return data
1049
1049
1050 def __json__(self):
1050 def __json__(self):
1051 data = {
1051 data = {
1052 'full_name': self.full_name,
1052 'full_name': self.full_name,
1053 'full_name_or_username': self.full_name_or_username,
1053 'full_name_or_username': self.full_name_or_username,
1054 'short_contact': self.short_contact,
1054 'short_contact': self.short_contact,
1055 'full_contact': self.full_contact,
1055 'full_contact': self.full_contact,
1056 }
1056 }
1057 data.update(self.get_api_data())
1057 data.update(self.get_api_data())
1058 return data
1058 return data
1059
1059
1060
1060
1061 class UserApiKeys(Base, BaseModel):
1061 class UserApiKeys(Base, BaseModel):
1062 __tablename__ = 'user_api_keys'
1062 __tablename__ = 'user_api_keys'
1063 __table_args__ = (
1063 __table_args__ = (
1064 Index('uak_api_key_idx', 'api_key'),
1064 Index('uak_api_key_idx', 'api_key'),
1065 Index('uak_api_key_expires_idx', 'api_key', 'expires'),
1065 Index('uak_api_key_expires_idx', 'api_key', 'expires'),
1066 base_table_args
1066 base_table_args
1067 )
1067 )
1068 __mapper_args__ = {}
1068 __mapper_args__ = {}
1069
1069
1070 # ApiKey role
1070 # ApiKey role
1071 ROLE_ALL = 'token_role_all'
1071 ROLE_ALL = 'token_role_all'
1072 ROLE_HTTP = 'token_role_http'
1072 ROLE_HTTP = 'token_role_http'
1073 ROLE_VCS = 'token_role_vcs'
1073 ROLE_VCS = 'token_role_vcs'
1074 ROLE_API = 'token_role_api'
1074 ROLE_API = 'token_role_api'
1075 ROLE_FEED = 'token_role_feed'
1075 ROLE_FEED = 'token_role_feed'
1076 ROLE_PASSWORD_RESET = 'token_password_reset'
1076 ROLE_PASSWORD_RESET = 'token_password_reset'
1077
1077
1078 ROLES = [ROLE_ALL, ROLE_HTTP, ROLE_VCS, ROLE_API, ROLE_FEED]
1078 ROLES = [ROLE_ALL, ROLE_HTTP, ROLE_VCS, ROLE_API, ROLE_FEED]
1079
1079
1080 user_api_key_id = Column("user_api_key_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1080 user_api_key_id = Column("user_api_key_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1081 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1081 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1082 api_key = Column("api_key", String(255), nullable=False, unique=True)
1082 api_key = Column("api_key", String(255), nullable=False, unique=True)
1083 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
1083 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
1084 expires = Column('expires', Float(53), nullable=False)
1084 expires = Column('expires', Float(53), nullable=False)
1085 role = Column('role', String(255), nullable=True)
1085 role = Column('role', String(255), nullable=True)
1086 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1086 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1087
1087
1088 # scope columns
1088 # scope columns
1089 repo_id = Column(
1089 repo_id = Column(
1090 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
1090 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
1091 nullable=True, unique=None, default=None)
1091 nullable=True, unique=None, default=None)
1092 repo = relationship('Repository', lazy='joined')
1092 repo = relationship('Repository', lazy='joined')
1093
1093
1094 repo_group_id = Column(
1094 repo_group_id = Column(
1095 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
1095 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
1096 nullable=True, unique=None, default=None)
1096 nullable=True, unique=None, default=None)
1097 repo_group = relationship('RepoGroup', lazy='joined')
1097 repo_group = relationship('RepoGroup', lazy='joined')
1098
1098
1099 user = relationship('User', lazy='joined')
1099 user = relationship('User', lazy='joined')
1100
1100
1101 def __unicode__(self):
1101 def __unicode__(self):
1102 return u"<%s('%s')>" % (self.__class__.__name__, self.role)
1102 return u"<%s('%s')>" % (self.__class__.__name__, self.role)
1103
1103
1104 def __json__(self):
1104 def __json__(self):
1105 data = {
1105 data = {
1106 'auth_token': self.api_key,
1106 'auth_token': self.api_key,
1107 'role': self.role,
1107 'role': self.role,
1108 'scope': self.scope_humanized,
1108 'scope': self.scope_humanized,
1109 'expired': self.expired
1109 'expired': self.expired
1110 }
1110 }
1111 return data
1111 return data
1112
1112
1113 def get_api_data(self, include_secrets=False):
1113 def get_api_data(self, include_secrets=False):
1114 data = self.__json__()
1114 data = self.__json__()
1115 if include_secrets:
1115 if include_secrets:
1116 return data
1116 return data
1117 else:
1117 else:
1118 data['auth_token'] = self.token_obfuscated
1118 data['auth_token'] = self.token_obfuscated
1119 return data
1119 return data
1120
1120
1121 @hybrid_property
1121 @hybrid_property
1122 def description_safe(self):
1122 def description_safe(self):
1123 from rhodecode.lib import helpers as h
1123 from rhodecode.lib import helpers as h
1124 return h.escape(self.description)
1124 return h.escape(self.description)
1125
1125
1126 @property
1126 @property
1127 def expired(self):
1127 def expired(self):
1128 if self.expires == -1:
1128 if self.expires == -1:
1129 return False
1129 return False
1130 return time.time() > self.expires
1130 return time.time() > self.expires
1131
1131
1132 @classmethod
1132 @classmethod
1133 def _get_role_name(cls, role):
1133 def _get_role_name(cls, role):
1134 return {
1134 return {
1135 cls.ROLE_ALL: _('all'),
1135 cls.ROLE_ALL: _('all'),
1136 cls.ROLE_HTTP: _('http/web interface'),
1136 cls.ROLE_HTTP: _('http/web interface'),
1137 cls.ROLE_VCS: _('vcs (git/hg/svn protocol)'),
1137 cls.ROLE_VCS: _('vcs (git/hg/svn protocol)'),
1138 cls.ROLE_API: _('api calls'),
1138 cls.ROLE_API: _('api calls'),
1139 cls.ROLE_FEED: _('feed access'),
1139 cls.ROLE_FEED: _('feed access'),
1140 }.get(role, role)
1140 }.get(role, role)
1141
1141
1142 @property
1142 @property
1143 def role_humanized(self):
1143 def role_humanized(self):
1144 return self._get_role_name(self.role)
1144 return self._get_role_name(self.role)
1145
1145
1146 def _get_scope(self):
1146 def _get_scope(self):
1147 if self.repo:
1147 if self.repo:
1148 return 'Repository: {}'.format(self.repo.repo_name)
1148 return 'Repository: {}'.format(self.repo.repo_name)
1149 if self.repo_group:
1149 if self.repo_group:
1150 return 'RepositoryGroup: {} (recursive)'.format(self.repo_group.group_name)
1150 return 'RepositoryGroup: {} (recursive)'.format(self.repo_group.group_name)
1151 return 'Global'
1151 return 'Global'
1152
1152
1153 @property
1153 @property
1154 def scope_humanized(self):
1154 def scope_humanized(self):
1155 return self._get_scope()
1155 return self._get_scope()
1156
1156
1157 @property
1157 @property
1158 def token_obfuscated(self):
1158 def token_obfuscated(self):
1159 if self.api_key:
1159 if self.api_key:
1160 return self.api_key[:4] + "****"
1160 return self.api_key[:4] + "****"
1161
1161
1162
1162
1163 class UserEmailMap(Base, BaseModel):
1163 class UserEmailMap(Base, BaseModel):
1164 __tablename__ = 'user_email_map'
1164 __tablename__ = 'user_email_map'
1165 __table_args__ = (
1165 __table_args__ = (
1166 Index('uem_email_idx', 'email'),
1166 Index('uem_email_idx', 'email'),
1167 UniqueConstraint('email'),
1167 UniqueConstraint('email'),
1168 base_table_args
1168 base_table_args
1169 )
1169 )
1170 __mapper_args__ = {}
1170 __mapper_args__ = {}
1171
1171
1172 email_id = Column("email_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1172 email_id = Column("email_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1173 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1173 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1174 _email = Column("email", String(255), nullable=True, unique=False, default=None)
1174 _email = Column("email", String(255), nullable=True, unique=False, default=None)
1175 user = relationship('User', lazy='joined')
1175 user = relationship('User', lazy='joined')
1176
1176
1177 @validates('_email')
1177 @validates('_email')
1178 def validate_email(self, key, email):
1178 def validate_email(self, key, email):
1179 # check if this email is not main one
1179 # check if this email is not main one
1180 main_email = Session().query(User).filter(User.email == email).scalar()
1180 main_email = Session().query(User).filter(User.email == email).scalar()
1181 if main_email is not None:
1181 if main_email is not None:
1182 raise AttributeError('email %s is present is user table' % email)
1182 raise AttributeError('email %s is present is user table' % email)
1183 return email
1183 return email
1184
1184
1185 @hybrid_property
1185 @hybrid_property
1186 def email(self):
1186 def email(self):
1187 return self._email
1187 return self._email
1188
1188
1189 @email.setter
1189 @email.setter
1190 def email(self, val):
1190 def email(self, val):
1191 self._email = val.lower() if val else None
1191 self._email = val.lower() if val else None
1192
1192
1193
1193
1194 class UserIpMap(Base, BaseModel):
1194 class UserIpMap(Base, BaseModel):
1195 __tablename__ = 'user_ip_map'
1195 __tablename__ = 'user_ip_map'
1196 __table_args__ = (
1196 __table_args__ = (
1197 UniqueConstraint('user_id', 'ip_addr'),
1197 UniqueConstraint('user_id', 'ip_addr'),
1198 base_table_args
1198 base_table_args
1199 )
1199 )
1200 __mapper_args__ = {}
1200 __mapper_args__ = {}
1201
1201
1202 ip_id = Column("ip_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1202 ip_id = Column("ip_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1203 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1203 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1204 ip_addr = Column("ip_addr", String(255), nullable=True, unique=False, default=None)
1204 ip_addr = Column("ip_addr", String(255), nullable=True, unique=False, default=None)
1205 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
1205 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
1206 description = Column("description", String(10000), nullable=True, unique=None, default=None)
1206 description = Column("description", String(10000), nullable=True, unique=None, default=None)
1207 user = relationship('User', lazy='joined')
1207 user = relationship('User', lazy='joined')
1208
1208
1209 @hybrid_property
1209 @hybrid_property
1210 def description_safe(self):
1210 def description_safe(self):
1211 from rhodecode.lib import helpers as h
1211 from rhodecode.lib import helpers as h
1212 return h.escape(self.description)
1212 return h.escape(self.description)
1213
1213
1214 @classmethod
1214 @classmethod
1215 def _get_ip_range(cls, ip_addr):
1215 def _get_ip_range(cls, ip_addr):
1216 net = ipaddress.ip_network(safe_unicode(ip_addr), strict=False)
1216 net = ipaddress.ip_network(safe_unicode(ip_addr), strict=False)
1217 return [str(net.network_address), str(net.broadcast_address)]
1217 return [str(net.network_address), str(net.broadcast_address)]
1218
1218
1219 def __json__(self):
1219 def __json__(self):
1220 return {
1220 return {
1221 'ip_addr': self.ip_addr,
1221 'ip_addr': self.ip_addr,
1222 'ip_range': self._get_ip_range(self.ip_addr),
1222 'ip_range': self._get_ip_range(self.ip_addr),
1223 }
1223 }
1224
1224
1225 def __unicode__(self):
1225 def __unicode__(self):
1226 return u"<%s('user_id:%s=>%s')>" % (self.__class__.__name__,
1226 return u"<%s('user_id:%s=>%s')>" % (self.__class__.__name__,
1227 self.user_id, self.ip_addr)
1227 self.user_id, self.ip_addr)
1228
1228
1229
1229
1230 class UserSshKeys(Base, BaseModel):
1230 class UserSshKeys(Base, BaseModel):
1231 __tablename__ = 'user_ssh_keys'
1231 __tablename__ = 'user_ssh_keys'
1232 __table_args__ = (
1232 __table_args__ = (
1233 Index('usk_ssh_key_fingerprint_idx', 'ssh_key_fingerprint'),
1233 Index('usk_ssh_key_fingerprint_idx', 'ssh_key_fingerprint'),
1234
1234
1235 UniqueConstraint('ssh_key_fingerprint'),
1235 UniqueConstraint('ssh_key_fingerprint'),
1236
1236
1237 base_table_args
1237 base_table_args
1238 )
1238 )
1239 __mapper_args__ = {}
1239 __mapper_args__ = {}
1240
1240
1241 ssh_key_id = Column('ssh_key_id', Integer(), nullable=False, unique=True, default=None, primary_key=True)
1241 ssh_key_id = Column('ssh_key_id', Integer(), nullable=False, unique=True, default=None, primary_key=True)
1242 ssh_key_data = Column('ssh_key_data', String(10240), nullable=False, unique=None, default=None)
1242 ssh_key_data = Column('ssh_key_data', String(10240), nullable=False, unique=None, default=None)
1243 ssh_key_fingerprint = Column('ssh_key_fingerprint', String(255), nullable=False, unique=None, default=None)
1243 ssh_key_fingerprint = Column('ssh_key_fingerprint', String(255), nullable=False, unique=None, default=None)
1244
1244
1245 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
1245 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
1246
1246
1247 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1247 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1248 accessed_on = Column('accessed_on', DateTime(timezone=False), nullable=True, default=None)
1248 accessed_on = Column('accessed_on', DateTime(timezone=False), nullable=True, default=None)
1249 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1249 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1250
1250
1251 user = relationship('User', lazy='joined')
1251 user = relationship('User', lazy='joined')
1252
1252
1253 def __json__(self):
1253 def __json__(self):
1254 data = {
1254 data = {
1255 'ssh_fingerprint': self.ssh_key_fingerprint,
1255 'ssh_fingerprint': self.ssh_key_fingerprint,
1256 'description': self.description,
1256 'description': self.description,
1257 'created_on': self.created_on
1257 'created_on': self.created_on
1258 }
1258 }
1259 return data
1259 return data
1260
1260
1261 def get_api_data(self):
1261 def get_api_data(self):
1262 data = self.__json__()
1262 data = self.__json__()
1263 return data
1263 return data
1264
1264
1265
1265
1266 class UserLog(Base, BaseModel):
1266 class UserLog(Base, BaseModel):
1267 __tablename__ = 'user_logs'
1267 __tablename__ = 'user_logs'
1268 __table_args__ = (
1268 __table_args__ = (
1269 base_table_args,
1269 base_table_args,
1270 )
1270 )
1271
1271
1272 VERSION_1 = 'v1'
1272 VERSION_1 = 'v1'
1273 VERSION_2 = 'v2'
1273 VERSION_2 = 'v2'
1274 VERSIONS = [VERSION_1, VERSION_2]
1274 VERSIONS = [VERSION_1, VERSION_2]
1275
1275
1276 user_log_id = Column("user_log_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1276 user_log_id = Column("user_log_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1277 user_id = Column("user_id", Integer(), ForeignKey('users.user_id',ondelete='SET NULL'), nullable=True, unique=None, default=None)
1277 user_id = Column("user_id", Integer(), ForeignKey('users.user_id',ondelete='SET NULL'), nullable=True, unique=None, default=None)
1278 username = Column("username", String(255), nullable=True, unique=None, default=None)
1278 username = Column("username", String(255), nullable=True, unique=None, default=None)
1279 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id', ondelete='SET NULL'), nullable=True, unique=None, default=None)
1279 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id', ondelete='SET NULL'), nullable=True, unique=None, default=None)
1280 repository_name = Column("repository_name", String(255), nullable=True, unique=None, default=None)
1280 repository_name = Column("repository_name", String(255), nullable=True, unique=None, default=None)
1281 user_ip = Column("user_ip", String(255), nullable=True, unique=None, default=None)
1281 user_ip = Column("user_ip", String(255), nullable=True, unique=None, default=None)
1282 action = Column("action", Text().with_variant(Text(1200000), 'mysql'), nullable=True, unique=None, default=None)
1282 action = Column("action", Text().with_variant(Text(1200000), 'mysql'), nullable=True, unique=None, default=None)
1283 action_date = Column("action_date", DateTime(timezone=False), nullable=True, unique=None, default=None)
1283 action_date = Column("action_date", DateTime(timezone=False), nullable=True, unique=None, default=None)
1284
1284
1285 version = Column("version", String(255), nullable=True, default=VERSION_1)
1285 version = Column("version", String(255), nullable=True, default=VERSION_1)
1286 user_data = Column('user_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=LONGTEXT()))))
1286 user_data = Column('user_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=LONGTEXT()))))
1287 action_data = Column('action_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=LONGTEXT()))))
1287 action_data = Column('action_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=LONGTEXT()))))
1288
1288
1289 def __unicode__(self):
1289 def __unicode__(self):
1290 return u"<%s('id:%s:%s')>" % (
1290 return u"<%s('id:%s:%s')>" % (
1291 self.__class__.__name__, self.repository_name, self.action)
1291 self.__class__.__name__, self.repository_name, self.action)
1292
1292
1293 def __json__(self):
1293 def __json__(self):
1294 return {
1294 return {
1295 'user_id': self.user_id,
1295 'user_id': self.user_id,
1296 'username': self.username,
1296 'username': self.username,
1297 'repository_id': self.repository_id,
1297 'repository_id': self.repository_id,
1298 'repository_name': self.repository_name,
1298 'repository_name': self.repository_name,
1299 'user_ip': self.user_ip,
1299 'user_ip': self.user_ip,
1300 'action_date': self.action_date,
1300 'action_date': self.action_date,
1301 'action': self.action,
1301 'action': self.action,
1302 }
1302 }
1303
1303
1304 @hybrid_property
1304 @hybrid_property
1305 def entry_id(self):
1305 def entry_id(self):
1306 return self.user_log_id
1306 return self.user_log_id
1307
1307
1308 @property
1308 @property
1309 def action_as_day(self):
1309 def action_as_day(self):
1310 return datetime.date(*self.action_date.timetuple()[:3])
1310 return datetime.date(*self.action_date.timetuple()[:3])
1311
1311
1312 user = relationship('User')
1312 user = relationship('User')
1313 repository = relationship('Repository', cascade='')
1313 repository = relationship('Repository', cascade='')
1314
1314
1315
1315
1316 class UserGroup(Base, BaseModel):
1316 class UserGroup(Base, BaseModel):
1317 __tablename__ = 'users_groups'
1317 __tablename__ = 'users_groups'
1318 __table_args__ = (
1318 __table_args__ = (
1319 base_table_args,
1319 base_table_args,
1320 )
1320 )
1321
1321
1322 users_group_id = Column("users_group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1322 users_group_id = Column("users_group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1323 users_group_name = Column("users_group_name", String(255), nullable=False, unique=True, default=None)
1323 users_group_name = Column("users_group_name", String(255), nullable=False, unique=True, default=None)
1324 user_group_description = Column("user_group_description", String(10000), nullable=True, unique=None, default=None)
1324 user_group_description = Column("user_group_description", String(10000), nullable=True, unique=None, default=None)
1325 users_group_active = Column("users_group_active", Boolean(), nullable=True, unique=None, default=None)
1325 users_group_active = Column("users_group_active", Boolean(), nullable=True, unique=None, default=None)
1326 inherit_default_permissions = Column("users_group_inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
1326 inherit_default_permissions = Column("users_group_inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
1327 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
1327 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
1328 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1328 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1329 _group_data = Column("group_data", LargeBinary(), nullable=True) # JSON data
1329 _group_data = Column("group_data", LargeBinary(), nullable=True) # JSON data
1330
1330
1331 members = relationship('UserGroupMember', cascade="all, delete-orphan", lazy="joined")
1331 members = relationship('UserGroupMember', cascade="all, delete-orphan", lazy="joined")
1332 users_group_to_perm = relationship('UserGroupToPerm', cascade='all')
1332 users_group_to_perm = relationship('UserGroupToPerm', cascade='all')
1333 users_group_repo_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1333 users_group_repo_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1334 users_group_repo_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
1334 users_group_repo_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
1335 user_user_group_to_perm = relationship('UserUserGroupToPerm', cascade='all')
1335 user_user_group_to_perm = relationship('UserUserGroupToPerm', cascade='all')
1336 user_group_user_group_to_perm = relationship('UserGroupUserGroupToPerm ', primaryjoin="UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id", cascade='all')
1336 user_group_user_group_to_perm = relationship('UserGroupUserGroupToPerm ', primaryjoin="UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id", cascade='all')
1337
1337
1338 user_group_review_rules = relationship('RepoReviewRuleUserGroup', cascade='all')
1338 user_group_review_rules = relationship('RepoReviewRuleUserGroup', cascade='all')
1339 user = relationship('User', primaryjoin="User.user_id==UserGroup.user_id")
1339 user = relationship('User', primaryjoin="User.user_id==UserGroup.user_id")
1340
1340
1341 @classmethod
1341 @classmethod
1342 def _load_group_data(cls, column):
1342 def _load_group_data(cls, column):
1343 if not column:
1343 if not column:
1344 return {}
1344 return {}
1345
1345
1346 try:
1346 try:
1347 return json.loads(column) or {}
1347 return json.loads(column) or {}
1348 except TypeError:
1348 except TypeError:
1349 return {}
1349 return {}
1350
1350
1351 @hybrid_property
1351 @hybrid_property
1352 def description_safe(self):
1352 def description_safe(self):
1353 from rhodecode.lib import helpers as h
1353 from rhodecode.lib import helpers as h
1354 return h.escape(self.user_group_description)
1354 return h.escape(self.user_group_description)
1355
1355
1356 @hybrid_property
1356 @hybrid_property
1357 def group_data(self):
1357 def group_data(self):
1358 return self._load_group_data(self._group_data)
1358 return self._load_group_data(self._group_data)
1359
1359
1360 @group_data.expression
1360 @group_data.expression
1361 def group_data(self, **kwargs):
1361 def group_data(self, **kwargs):
1362 return self._group_data
1362 return self._group_data
1363
1363
1364 @group_data.setter
1364 @group_data.setter
1365 def group_data(self, val):
1365 def group_data(self, val):
1366 try:
1366 try:
1367 self._group_data = json.dumps(val)
1367 self._group_data = json.dumps(val)
1368 except Exception:
1368 except Exception:
1369 log.error(traceback.format_exc())
1369 log.error(traceback.format_exc())
1370
1370
1371 @classmethod
1371 @classmethod
1372 def _load_sync(cls, group_data):
1372 def _load_sync(cls, group_data):
1373 if group_data:
1373 if group_data:
1374 return group_data.get('extern_type')
1374 return group_data.get('extern_type')
1375
1375
1376 @property
1376 @property
1377 def sync(self):
1377 def sync(self):
1378 return self._load_sync(self.group_data)
1378 return self._load_sync(self.group_data)
1379
1379
1380 def __unicode__(self):
1380 def __unicode__(self):
1381 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
1381 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
1382 self.users_group_id,
1382 self.users_group_id,
1383 self.users_group_name)
1383 self.users_group_name)
1384
1384
1385 @classmethod
1385 @classmethod
1386 def get_by_group_name(cls, group_name, cache=False,
1386 def get_by_group_name(cls, group_name, cache=False,
1387 case_insensitive=False):
1387 case_insensitive=False):
1388 if case_insensitive:
1388 if case_insensitive:
1389 q = cls.query().filter(func.lower(cls.users_group_name) ==
1389 q = cls.query().filter(func.lower(cls.users_group_name) ==
1390 func.lower(group_name))
1390 func.lower(group_name))
1391
1391
1392 else:
1392 else:
1393 q = cls.query().filter(cls.users_group_name == group_name)
1393 q = cls.query().filter(cls.users_group_name == group_name)
1394 if cache:
1394 if cache:
1395 q = q.options(
1395 q = q.options(
1396 FromCache("sql_cache_short", "get_group_%s" % _hash_key(group_name)))
1396 FromCache("sql_cache_short", "get_group_%s" % _hash_key(group_name)))
1397 return q.scalar()
1397 return q.scalar()
1398
1398
1399 @classmethod
1399 @classmethod
1400 def get(cls, user_group_id, cache=False):
1400 def get(cls, user_group_id, cache=False):
1401 if not user_group_id:
1401 if not user_group_id:
1402 return
1402 return
1403
1403
1404 user_group = cls.query()
1404 user_group = cls.query()
1405 if cache:
1405 if cache:
1406 user_group = user_group.options(
1406 user_group = user_group.options(
1407 FromCache("sql_cache_short", "get_users_group_%s" % user_group_id))
1407 FromCache("sql_cache_short", "get_users_group_%s" % user_group_id))
1408 return user_group.get(user_group_id)
1408 return user_group.get(user_group_id)
1409
1409
1410 def permissions(self, with_admins=True, with_owner=True,
1410 def permissions(self, with_admins=True, with_owner=True,
1411 expand_from_user_groups=False):
1411 expand_from_user_groups=False):
1412 """
1412 """
1413 Permissions for user groups
1413 Permissions for user groups
1414 """
1414 """
1415 _admin_perm = 'usergroup.admin'
1415 _admin_perm = 'usergroup.admin'
1416
1416
1417 owner_row = []
1417 owner_row = []
1418 if with_owner:
1418 if with_owner:
1419 usr = AttributeDict(self.user.get_dict())
1419 usr = AttributeDict(self.user.get_dict())
1420 usr.owner_row = True
1420 usr.owner_row = True
1421 usr.permission = _admin_perm
1421 usr.permission = _admin_perm
1422 owner_row.append(usr)
1422 owner_row.append(usr)
1423
1423
1424 super_admin_ids = []
1424 super_admin_ids = []
1425 super_admin_rows = []
1425 super_admin_rows = []
1426 if with_admins:
1426 if with_admins:
1427 for usr in User.get_all_super_admins():
1427 for usr in User.get_all_super_admins():
1428 super_admin_ids.append(usr.user_id)
1428 super_admin_ids.append(usr.user_id)
1429 # if this admin is also owner, don't double the record
1429 # if this admin is also owner, don't double the record
1430 if usr.user_id == owner_row[0].user_id:
1430 if usr.user_id == owner_row[0].user_id:
1431 owner_row[0].admin_row = True
1431 owner_row[0].admin_row = True
1432 else:
1432 else:
1433 usr = AttributeDict(usr.get_dict())
1433 usr = AttributeDict(usr.get_dict())
1434 usr.admin_row = True
1434 usr.admin_row = True
1435 usr.permission = _admin_perm
1435 usr.permission = _admin_perm
1436 super_admin_rows.append(usr)
1436 super_admin_rows.append(usr)
1437
1437
1438 q = UserUserGroupToPerm.query().filter(UserUserGroupToPerm.user_group == self)
1438 q = UserUserGroupToPerm.query().filter(UserUserGroupToPerm.user_group == self)
1439 q = q.options(joinedload(UserUserGroupToPerm.user_group),
1439 q = q.options(joinedload(UserUserGroupToPerm.user_group),
1440 joinedload(UserUserGroupToPerm.user),
1440 joinedload(UserUserGroupToPerm.user),
1441 joinedload(UserUserGroupToPerm.permission),)
1441 joinedload(UserUserGroupToPerm.permission),)
1442
1442
1443 # get owners and admins and permissions. We do a trick of re-writing
1443 # get owners and admins and permissions. We do a trick of re-writing
1444 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1444 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1445 # has a global reference and changing one object propagates to all
1445 # has a global reference and changing one object propagates to all
1446 # others. This means if admin is also an owner admin_row that change
1446 # others. This means if admin is also an owner admin_row that change
1447 # would propagate to both objects
1447 # would propagate to both objects
1448 perm_rows = []
1448 perm_rows = []
1449 for _usr in q.all():
1449 for _usr in q.all():
1450 usr = AttributeDict(_usr.user.get_dict())
1450 usr = AttributeDict(_usr.user.get_dict())
1451 # if this user is also owner/admin, mark as duplicate record
1451 # if this user is also owner/admin, mark as duplicate record
1452 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
1452 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
1453 usr.duplicate_perm = True
1453 usr.duplicate_perm = True
1454 usr.permission = _usr.permission.permission_name
1454 usr.permission = _usr.permission.permission_name
1455 perm_rows.append(usr)
1455 perm_rows.append(usr)
1456
1456
1457 # filter the perm rows by 'default' first and then sort them by
1457 # filter the perm rows by 'default' first and then sort them by
1458 # admin,write,read,none permissions sorted again alphabetically in
1458 # admin,write,read,none permissions sorted again alphabetically in
1459 # each group
1459 # each group
1460 perm_rows = sorted(perm_rows, key=display_user_sort)
1460 perm_rows = sorted(perm_rows, key=display_user_sort)
1461
1461
1462 user_groups_rows = []
1462 user_groups_rows = []
1463 if expand_from_user_groups:
1463 if expand_from_user_groups:
1464 for ug in self.permission_user_groups(with_members=True):
1464 for ug in self.permission_user_groups(with_members=True):
1465 for user_data in ug.members:
1465 for user_data in ug.members:
1466 user_groups_rows.append(user_data)
1466 user_groups_rows.append(user_data)
1467
1467
1468 return super_admin_rows + owner_row + perm_rows + user_groups_rows
1468 return super_admin_rows + owner_row + perm_rows + user_groups_rows
1469
1469
1470 def permission_user_groups(self, with_members=False):
1470 def permission_user_groups(self, with_members=False):
1471 q = UserGroupUserGroupToPerm.query()\
1471 q = UserGroupUserGroupToPerm.query()\
1472 .filter(UserGroupUserGroupToPerm.target_user_group == self)
1472 .filter(UserGroupUserGroupToPerm.target_user_group == self)
1473 q = q.options(joinedload(UserGroupUserGroupToPerm.user_group),
1473 q = q.options(joinedload(UserGroupUserGroupToPerm.user_group),
1474 joinedload(UserGroupUserGroupToPerm.target_user_group),
1474 joinedload(UserGroupUserGroupToPerm.target_user_group),
1475 joinedload(UserGroupUserGroupToPerm.permission),)
1475 joinedload(UserGroupUserGroupToPerm.permission),)
1476
1476
1477 perm_rows = []
1477 perm_rows = []
1478 for _user_group in q.all():
1478 for _user_group in q.all():
1479 entry = AttributeDict(_user_group.user_group.get_dict())
1479 entry = AttributeDict(_user_group.user_group.get_dict())
1480 entry.permission = _user_group.permission.permission_name
1480 entry.permission = _user_group.permission.permission_name
1481 if with_members:
1481 if with_members:
1482 entry.members = [x.user.get_dict()
1482 entry.members = [x.user.get_dict()
1483 for x in _user_group.user_group.members]
1483 for x in _user_group.user_group.members]
1484 perm_rows.append(entry)
1484 perm_rows.append(entry)
1485
1485
1486 perm_rows = sorted(perm_rows, key=display_user_group_sort)
1486 perm_rows = sorted(perm_rows, key=display_user_group_sort)
1487 return perm_rows
1487 return perm_rows
1488
1488
1489 def _get_default_perms(self, user_group, suffix=''):
1489 def _get_default_perms(self, user_group, suffix=''):
1490 from rhodecode.model.permission import PermissionModel
1490 from rhodecode.model.permission import PermissionModel
1491 return PermissionModel().get_default_perms(user_group.users_group_to_perm, suffix)
1491 return PermissionModel().get_default_perms(user_group.users_group_to_perm, suffix)
1492
1492
1493 def get_default_perms(self, suffix=''):
1493 def get_default_perms(self, suffix=''):
1494 return self._get_default_perms(self, suffix)
1494 return self._get_default_perms(self, suffix)
1495
1495
1496 def get_api_data(self, with_group_members=True, include_secrets=False):
1496 def get_api_data(self, with_group_members=True, include_secrets=False):
1497 """
1497 """
1498 :param include_secrets: See :meth:`User.get_api_data`, this parameter is
1498 :param include_secrets: See :meth:`User.get_api_data`, this parameter is
1499 basically forwarded.
1499 basically forwarded.
1500
1500
1501 """
1501 """
1502 user_group = self
1502 user_group = self
1503 data = {
1503 data = {
1504 'users_group_id': user_group.users_group_id,
1504 'users_group_id': user_group.users_group_id,
1505 'group_name': user_group.users_group_name,
1505 'group_name': user_group.users_group_name,
1506 'group_description': user_group.user_group_description,
1506 'group_description': user_group.user_group_description,
1507 'active': user_group.users_group_active,
1507 'active': user_group.users_group_active,
1508 'owner': user_group.user.username,
1508 'owner': user_group.user.username,
1509 'sync': user_group.sync,
1509 'sync': user_group.sync,
1510 'owner_email': user_group.user.email,
1510 'owner_email': user_group.user.email,
1511 }
1511 }
1512
1512
1513 if with_group_members:
1513 if with_group_members:
1514 users = []
1514 users = []
1515 for user in user_group.members:
1515 for user in user_group.members:
1516 user = user.user
1516 user = user.user
1517 users.append(user.get_api_data(include_secrets=include_secrets))
1517 users.append(user.get_api_data(include_secrets=include_secrets))
1518 data['users'] = users
1518 data['users'] = users
1519
1519
1520 return data
1520 return data
1521
1521
1522
1522
1523 class UserGroupMember(Base, BaseModel):
1523 class UserGroupMember(Base, BaseModel):
1524 __tablename__ = 'users_groups_members'
1524 __tablename__ = 'users_groups_members'
1525 __table_args__ = (
1525 __table_args__ = (
1526 base_table_args,
1526 base_table_args,
1527 )
1527 )
1528
1528
1529 users_group_member_id = Column("users_group_member_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1529 users_group_member_id = Column("users_group_member_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1530 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
1530 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
1531 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
1531 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
1532
1532
1533 user = relationship('User', lazy='joined')
1533 user = relationship('User', lazy='joined')
1534 users_group = relationship('UserGroup')
1534 users_group = relationship('UserGroup')
1535
1535
1536 def __init__(self, gr_id='', u_id=''):
1536 def __init__(self, gr_id='', u_id=''):
1537 self.users_group_id = gr_id
1537 self.users_group_id = gr_id
1538 self.user_id = u_id
1538 self.user_id = u_id
1539
1539
1540
1540
1541 class RepositoryField(Base, BaseModel):
1541 class RepositoryField(Base, BaseModel):
1542 __tablename__ = 'repositories_fields'
1542 __tablename__ = 'repositories_fields'
1543 __table_args__ = (
1543 __table_args__ = (
1544 UniqueConstraint('repository_id', 'field_key'), # no-multi field
1544 UniqueConstraint('repository_id', 'field_key'), # no-multi field
1545 base_table_args,
1545 base_table_args,
1546 )
1546 )
1547
1547
1548 PREFIX = 'ex_' # prefix used in form to not conflict with already existing fields
1548 PREFIX = 'ex_' # prefix used in form to not conflict with already existing fields
1549
1549
1550 repo_field_id = Column("repo_field_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1550 repo_field_id = Column("repo_field_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1551 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
1551 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
1552 field_key = Column("field_key", String(250))
1552 field_key = Column("field_key", String(250))
1553 field_label = Column("field_label", String(1024), nullable=False)
1553 field_label = Column("field_label", String(1024), nullable=False)
1554 field_value = Column("field_value", String(10000), nullable=False)
1554 field_value = Column("field_value", String(10000), nullable=False)
1555 field_desc = Column("field_desc", String(1024), nullable=False)
1555 field_desc = Column("field_desc", String(1024), nullable=False)
1556 field_type = Column("field_type", String(255), nullable=False, unique=None)
1556 field_type = Column("field_type", String(255), nullable=False, unique=None)
1557 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1557 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1558
1558
1559 repository = relationship('Repository')
1559 repository = relationship('Repository')
1560
1560
1561 @property
1561 @property
1562 def field_key_prefixed(self):
1562 def field_key_prefixed(self):
1563 return 'ex_%s' % self.field_key
1563 return 'ex_%s' % self.field_key
1564
1564
1565 @classmethod
1565 @classmethod
1566 def un_prefix_key(cls, key):
1566 def un_prefix_key(cls, key):
1567 if key.startswith(cls.PREFIX):
1567 if key.startswith(cls.PREFIX):
1568 return key[len(cls.PREFIX):]
1568 return key[len(cls.PREFIX):]
1569 return key
1569 return key
1570
1570
1571 @classmethod
1571 @classmethod
1572 def get_by_key_name(cls, key, repo):
1572 def get_by_key_name(cls, key, repo):
1573 row = cls.query()\
1573 row = cls.query()\
1574 .filter(cls.repository == repo)\
1574 .filter(cls.repository == repo)\
1575 .filter(cls.field_key == key).scalar()
1575 .filter(cls.field_key == key).scalar()
1576 return row
1576 return row
1577
1577
1578
1578
1579 class Repository(Base, BaseModel):
1579 class Repository(Base, BaseModel):
1580 __tablename__ = 'repositories'
1580 __tablename__ = 'repositories'
1581 __table_args__ = (
1581 __table_args__ = (
1582 Index('r_repo_name_idx', 'repo_name', mysql_length=255),
1582 Index('r_repo_name_idx', 'repo_name', mysql_length=255),
1583 base_table_args,
1583 base_table_args,
1584 )
1584 )
1585 DEFAULT_CLONE_URI = '{scheme}://{user}@{netloc}/{repo}'
1585 DEFAULT_CLONE_URI = '{scheme}://{user}@{netloc}/{repo}'
1586 DEFAULT_CLONE_URI_ID = '{scheme}://{user}@{netloc}/_{repoid}'
1586 DEFAULT_CLONE_URI_ID = '{scheme}://{user}@{netloc}/_{repoid}'
1587 DEFAULT_CLONE_URI_SSH = 'ssh://{sys_user}@{hostname}/{repo}'
1587 DEFAULT_CLONE_URI_SSH = 'ssh://{sys_user}@{hostname}/{repo}'
1588
1588
1589 STATE_CREATED = 'repo_state_created'
1589 STATE_CREATED = 'repo_state_created'
1590 STATE_PENDING = 'repo_state_pending'
1590 STATE_PENDING = 'repo_state_pending'
1591 STATE_ERROR = 'repo_state_error'
1591 STATE_ERROR = 'repo_state_error'
1592
1592
1593 LOCK_AUTOMATIC = 'lock_auto'
1593 LOCK_AUTOMATIC = 'lock_auto'
1594 LOCK_API = 'lock_api'
1594 LOCK_API = 'lock_api'
1595 LOCK_WEB = 'lock_web'
1595 LOCK_WEB = 'lock_web'
1596 LOCK_PULL = 'lock_pull'
1596 LOCK_PULL = 'lock_pull'
1597
1597
1598 NAME_SEP = URL_SEP
1598 NAME_SEP = URL_SEP
1599
1599
1600 repo_id = Column(
1600 repo_id = Column(
1601 "repo_id", Integer(), nullable=False, unique=True, default=None,
1601 "repo_id", Integer(), nullable=False, unique=True, default=None,
1602 primary_key=True)
1602 primary_key=True)
1603 _repo_name = Column(
1603 _repo_name = Column(
1604 "repo_name", Text(), nullable=False, default=None)
1604 "repo_name", Text(), nullable=False, default=None)
1605 _repo_name_hash = Column(
1605 _repo_name_hash = Column(
1606 "repo_name_hash", String(255), nullable=False, unique=True)
1606 "repo_name_hash", String(255), nullable=False, unique=True)
1607 repo_state = Column("repo_state", String(255), nullable=True)
1607 repo_state = Column("repo_state", String(255), nullable=True)
1608
1608
1609 clone_uri = Column(
1609 clone_uri = Column(
1610 "clone_uri", EncryptedTextValue(), nullable=True, unique=False,
1610 "clone_uri", EncryptedTextValue(), nullable=True, unique=False,
1611 default=None)
1611 default=None)
1612 push_uri = Column(
1612 push_uri = Column(
1613 "push_uri", EncryptedTextValue(), nullable=True, unique=False,
1613 "push_uri", EncryptedTextValue(), nullable=True, unique=False,
1614 default=None)
1614 default=None)
1615 repo_type = Column(
1615 repo_type = Column(
1616 "repo_type", String(255), nullable=False, unique=False, default=None)
1616 "repo_type", String(255), nullable=False, unique=False, default=None)
1617 user_id = Column(
1617 user_id = Column(
1618 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
1618 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
1619 unique=False, default=None)
1619 unique=False, default=None)
1620 private = Column(
1620 private = Column(
1621 "private", Boolean(), nullable=True, unique=None, default=None)
1621 "private", Boolean(), nullable=True, unique=None, default=None)
1622 archived = Column(
1622 archived = Column(
1623 "archived", Boolean(), nullable=True, unique=None, default=None)
1623 "archived", Boolean(), nullable=True, unique=None, default=None)
1624 enable_statistics = Column(
1624 enable_statistics = Column(
1625 "statistics", Boolean(), nullable=True, unique=None, default=True)
1625 "statistics", Boolean(), nullable=True, unique=None, default=True)
1626 enable_downloads = Column(
1626 enable_downloads = Column(
1627 "downloads", Boolean(), nullable=True, unique=None, default=True)
1627 "downloads", Boolean(), nullable=True, unique=None, default=True)
1628 description = Column(
1628 description = Column(
1629 "description", String(10000), nullable=True, unique=None, default=None)
1629 "description", String(10000), nullable=True, unique=None, default=None)
1630 created_on = Column(
1630 created_on = Column(
1631 'created_on', DateTime(timezone=False), nullable=True, unique=None,
1631 'created_on', DateTime(timezone=False), nullable=True, unique=None,
1632 default=datetime.datetime.now)
1632 default=datetime.datetime.now)
1633 updated_on = Column(
1633 updated_on = Column(
1634 'updated_on', DateTime(timezone=False), nullable=True, unique=None,
1634 'updated_on', DateTime(timezone=False), nullable=True, unique=None,
1635 default=datetime.datetime.now)
1635 default=datetime.datetime.now)
1636 _landing_revision = Column(
1636 _landing_revision = Column(
1637 "landing_revision", String(255), nullable=False, unique=False,
1637 "landing_revision", String(255), nullable=False, unique=False,
1638 default=None)
1638 default=None)
1639 enable_locking = Column(
1639 enable_locking = Column(
1640 "enable_locking", Boolean(), nullable=False, unique=None,
1640 "enable_locking", Boolean(), nullable=False, unique=None,
1641 default=False)
1641 default=False)
1642 _locked = Column(
1642 _locked = Column(
1643 "locked", String(255), nullable=True, unique=False, default=None)
1643 "locked", String(255), nullable=True, unique=False, default=None)
1644 _changeset_cache = Column(
1644 _changeset_cache = Column(
1645 "changeset_cache", LargeBinary(), nullable=True) # JSON data
1645 "changeset_cache", LargeBinary(), nullable=True) # JSON data
1646
1646
1647 fork_id = Column(
1647 fork_id = Column(
1648 "fork_id", Integer(), ForeignKey('repositories.repo_id'),
1648 "fork_id", Integer(), ForeignKey('repositories.repo_id'),
1649 nullable=True, unique=False, default=None)
1649 nullable=True, unique=False, default=None)
1650 group_id = Column(
1650 group_id = Column(
1651 "group_id", Integer(), ForeignKey('groups.group_id'), nullable=True,
1651 "group_id", Integer(), ForeignKey('groups.group_id'), nullable=True,
1652 unique=False, default=None)
1652 unique=False, default=None)
1653
1653
1654 user = relationship('User', lazy='joined')
1654 user = relationship('User', lazy='joined')
1655 fork = relationship('Repository', remote_side=repo_id, lazy='joined')
1655 fork = relationship('Repository', remote_side=repo_id, lazy='joined')
1656 group = relationship('RepoGroup', lazy='joined')
1656 group = relationship('RepoGroup', lazy='joined')
1657 repo_to_perm = relationship(
1657 repo_to_perm = relationship(
1658 'UserRepoToPerm', cascade='all',
1658 'UserRepoToPerm', cascade='all',
1659 order_by='UserRepoToPerm.repo_to_perm_id')
1659 order_by='UserRepoToPerm.repo_to_perm_id')
1660 users_group_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1660 users_group_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1661 stats = relationship('Statistics', cascade='all', uselist=False)
1661 stats = relationship('Statistics', cascade='all', uselist=False)
1662
1662
1663 followers = relationship(
1663 followers = relationship(
1664 'UserFollowing',
1664 'UserFollowing',
1665 primaryjoin='UserFollowing.follows_repo_id==Repository.repo_id',
1665 primaryjoin='UserFollowing.follows_repo_id==Repository.repo_id',
1666 cascade='all')
1666 cascade='all')
1667 extra_fields = relationship(
1667 extra_fields = relationship(
1668 'RepositoryField', cascade="all, delete-orphan")
1668 'RepositoryField', cascade="all, delete-orphan")
1669 logs = relationship('UserLog')
1669 logs = relationship('UserLog')
1670 comments = relationship(
1670 comments = relationship(
1671 'ChangesetComment', cascade="all, delete-orphan")
1671 'ChangesetComment', cascade="all, delete-orphan")
1672 pull_requests_source = relationship(
1672 pull_requests_source = relationship(
1673 'PullRequest',
1673 'PullRequest',
1674 primaryjoin='PullRequest.source_repo_id==Repository.repo_id',
1674 primaryjoin='PullRequest.source_repo_id==Repository.repo_id',
1675 cascade="all, delete-orphan")
1675 cascade="all, delete-orphan")
1676 pull_requests_target = relationship(
1676 pull_requests_target = relationship(
1677 'PullRequest',
1677 'PullRequest',
1678 primaryjoin='PullRequest.target_repo_id==Repository.repo_id',
1678 primaryjoin='PullRequest.target_repo_id==Repository.repo_id',
1679 cascade="all, delete-orphan")
1679 cascade="all, delete-orphan")
1680 ui = relationship('RepoRhodeCodeUi', cascade="all")
1680 ui = relationship('RepoRhodeCodeUi', cascade="all")
1681 settings = relationship('RepoRhodeCodeSetting', cascade="all")
1681 settings = relationship('RepoRhodeCodeSetting', cascade="all")
1682 integrations = relationship('Integration', cascade="all, delete-orphan")
1682 integrations = relationship('Integration', cascade="all, delete-orphan")
1683
1683
1684 scoped_tokens = relationship('UserApiKeys', cascade="all")
1684 scoped_tokens = relationship('UserApiKeys', cascade="all")
1685
1685
1686 artifacts = relationship('FileStore', cascade="all")
1686 artifacts = relationship('FileStore', cascade="all")
1687
1687
1688 def __unicode__(self):
1688 def __unicode__(self):
1689 return u"<%s('%s:%s')>" % (self.__class__.__name__, self.repo_id,
1689 return u"<%s('%s:%s')>" % (self.__class__.__name__, self.repo_id,
1690 safe_unicode(self.repo_name))
1690 safe_unicode(self.repo_name))
1691
1691
1692 @hybrid_property
1692 @hybrid_property
1693 def description_safe(self):
1693 def description_safe(self):
1694 from rhodecode.lib import helpers as h
1694 from rhodecode.lib import helpers as h
1695 return h.escape(self.description)
1695 return h.escape(self.description)
1696
1696
1697 @hybrid_property
1697 @hybrid_property
1698 def landing_rev(self):
1698 def landing_rev(self):
1699 # always should return [rev_type, rev]
1699 # always should return [rev_type, rev]
1700 if self._landing_revision:
1700 if self._landing_revision:
1701 _rev_info = self._landing_revision.split(':')
1701 _rev_info = self._landing_revision.split(':')
1702 if len(_rev_info) < 2:
1702 if len(_rev_info) < 2:
1703 _rev_info.insert(0, 'rev')
1703 _rev_info.insert(0, 'rev')
1704 return [_rev_info[0], _rev_info[1]]
1704 return [_rev_info[0], _rev_info[1]]
1705 return [None, None]
1705 return [None, None]
1706
1706
1707 @landing_rev.setter
1707 @landing_rev.setter
1708 def landing_rev(self, val):
1708 def landing_rev(self, val):
1709 if ':' not in val:
1709 if ':' not in val:
1710 raise ValueError('value must be delimited with `:` and consist '
1710 raise ValueError('value must be delimited with `:` and consist '
1711 'of <rev_type>:<rev>, got %s instead' % val)
1711 'of <rev_type>:<rev>, got %s instead' % val)
1712 self._landing_revision = val
1712 self._landing_revision = val
1713
1713
1714 @hybrid_property
1714 @hybrid_property
1715 def locked(self):
1715 def locked(self):
1716 if self._locked:
1716 if self._locked:
1717 user_id, timelocked, reason = self._locked.split(':')
1717 user_id, timelocked, reason = self._locked.split(':')
1718 lock_values = int(user_id), timelocked, reason
1718 lock_values = int(user_id), timelocked, reason
1719 else:
1719 else:
1720 lock_values = [None, None, None]
1720 lock_values = [None, None, None]
1721 return lock_values
1721 return lock_values
1722
1722
1723 @locked.setter
1723 @locked.setter
1724 def locked(self, val):
1724 def locked(self, val):
1725 if val and isinstance(val, (list, tuple)):
1725 if val and isinstance(val, (list, tuple)):
1726 self._locked = ':'.join(map(str, val))
1726 self._locked = ':'.join(map(str, val))
1727 else:
1727 else:
1728 self._locked = None
1728 self._locked = None
1729
1729
1730 @hybrid_property
1730 @hybrid_property
1731 def changeset_cache(self):
1731 def changeset_cache(self):
1732 from rhodecode.lib.vcs.backends.base import EmptyCommit
1732 from rhodecode.lib.vcs.backends.base import EmptyCommit
1733 dummy = EmptyCommit().__json__()
1733 dummy = EmptyCommit().__json__()
1734 if not self._changeset_cache:
1734 if not self._changeset_cache:
1735 dummy['source_repo_id'] = self.repo_id
1735 dummy['source_repo_id'] = self.repo_id
1736 return json.loads(json.dumps(dummy))
1736 return json.loads(json.dumps(dummy))
1737
1737
1738 try:
1738 try:
1739 return json.loads(self._changeset_cache)
1739 return json.loads(self._changeset_cache)
1740 except TypeError:
1740 except TypeError:
1741 return dummy
1741 return dummy
1742 except Exception:
1742 except Exception:
1743 log.error(traceback.format_exc())
1743 log.error(traceback.format_exc())
1744 return dummy
1744 return dummy
1745
1745
1746 @changeset_cache.setter
1746 @changeset_cache.setter
1747 def changeset_cache(self, val):
1747 def changeset_cache(self, val):
1748 try:
1748 try:
1749 self._changeset_cache = json.dumps(val)
1749 self._changeset_cache = json.dumps(val)
1750 except Exception:
1750 except Exception:
1751 log.error(traceback.format_exc())
1751 log.error(traceback.format_exc())
1752
1752
1753 @hybrid_property
1753 @hybrid_property
1754 def repo_name(self):
1754 def repo_name(self):
1755 return self._repo_name
1755 return self._repo_name
1756
1756
1757 @repo_name.setter
1757 @repo_name.setter
1758 def repo_name(self, value):
1758 def repo_name(self, value):
1759 self._repo_name = value
1759 self._repo_name = value
1760 self._repo_name_hash = hashlib.sha1(safe_str(value)).hexdigest()
1760 self._repo_name_hash = hashlib.sha1(safe_str(value)).hexdigest()
1761
1761
1762 @classmethod
1762 @classmethod
1763 def normalize_repo_name(cls, repo_name):
1763 def normalize_repo_name(cls, repo_name):
1764 """
1764 """
1765 Normalizes os specific repo_name to the format internally stored inside
1765 Normalizes os specific repo_name to the format internally stored inside
1766 database using URL_SEP
1766 database using URL_SEP
1767
1767
1768 :param cls:
1768 :param cls:
1769 :param repo_name:
1769 :param repo_name:
1770 """
1770 """
1771 return cls.NAME_SEP.join(repo_name.split(os.sep))
1771 return cls.NAME_SEP.join(repo_name.split(os.sep))
1772
1772
1773 @classmethod
1773 @classmethod
1774 def get_by_repo_name(cls, repo_name, cache=False, identity_cache=False):
1774 def get_by_repo_name(cls, repo_name, cache=False, identity_cache=False):
1775 session = Session()
1775 session = Session()
1776 q = session.query(cls).filter(cls.repo_name == repo_name)
1776 q = session.query(cls).filter(cls.repo_name == repo_name)
1777
1777
1778 if cache:
1778 if cache:
1779 if identity_cache:
1779 if identity_cache:
1780 val = cls.identity_cache(session, 'repo_name', repo_name)
1780 val = cls.identity_cache(session, 'repo_name', repo_name)
1781 if val:
1781 if val:
1782 return val
1782 return val
1783 else:
1783 else:
1784 cache_key = "get_repo_by_name_%s" % _hash_key(repo_name)
1784 cache_key = "get_repo_by_name_%s" % _hash_key(repo_name)
1785 q = q.options(
1785 q = q.options(
1786 FromCache("sql_cache_short", cache_key))
1786 FromCache("sql_cache_short", cache_key))
1787
1787
1788 return q.scalar()
1788 return q.scalar()
1789
1789
1790 @classmethod
1790 @classmethod
1791 def get_by_id_or_repo_name(cls, repoid):
1791 def get_by_id_or_repo_name(cls, repoid):
1792 if isinstance(repoid, (int, long)):
1792 if isinstance(repoid, (int, long)):
1793 try:
1793 try:
1794 repo = cls.get(repoid)
1794 repo = cls.get(repoid)
1795 except ValueError:
1795 except ValueError:
1796 repo = None
1796 repo = None
1797 else:
1797 else:
1798 repo = cls.get_by_repo_name(repoid)
1798 repo = cls.get_by_repo_name(repoid)
1799 return repo
1799 return repo
1800
1800
1801 @classmethod
1801 @classmethod
1802 def get_by_full_path(cls, repo_full_path):
1802 def get_by_full_path(cls, repo_full_path):
1803 repo_name = repo_full_path.split(cls.base_path(), 1)[-1]
1803 repo_name = repo_full_path.split(cls.base_path(), 1)[-1]
1804 repo_name = cls.normalize_repo_name(repo_name)
1804 repo_name = cls.normalize_repo_name(repo_name)
1805 return cls.get_by_repo_name(repo_name.strip(URL_SEP))
1805 return cls.get_by_repo_name(repo_name.strip(URL_SEP))
1806
1806
1807 @classmethod
1807 @classmethod
1808 def get_repo_forks(cls, repo_id):
1808 def get_repo_forks(cls, repo_id):
1809 return cls.query().filter(Repository.fork_id == repo_id)
1809 return cls.query().filter(Repository.fork_id == repo_id)
1810
1810
1811 @classmethod
1811 @classmethod
1812 def base_path(cls):
1812 def base_path(cls):
1813 """
1813 """
1814 Returns base path when all repos are stored
1814 Returns base path when all repos are stored
1815
1815
1816 :param cls:
1816 :param cls:
1817 """
1817 """
1818 q = Session().query(RhodeCodeUi)\
1818 q = Session().query(RhodeCodeUi)\
1819 .filter(RhodeCodeUi.ui_key == cls.NAME_SEP)
1819 .filter(RhodeCodeUi.ui_key == cls.NAME_SEP)
1820 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1820 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1821 return q.one().ui_value
1821 return q.one().ui_value
1822
1822
1823 @classmethod
1823 @classmethod
1824 def get_all_repos(cls, user_id=Optional(None), group_id=Optional(None),
1824 def get_all_repos(cls, user_id=Optional(None), group_id=Optional(None),
1825 case_insensitive=True, archived=False):
1825 case_insensitive=True, archived=False):
1826 q = Repository.query()
1826 q = Repository.query()
1827
1827
1828 if not archived:
1828 if not archived:
1829 q = q.filter(Repository.archived.isnot(true()))
1829 q = q.filter(Repository.archived.isnot(true()))
1830
1830
1831 if not isinstance(user_id, Optional):
1831 if not isinstance(user_id, Optional):
1832 q = q.filter(Repository.user_id == user_id)
1832 q = q.filter(Repository.user_id == user_id)
1833
1833
1834 if not isinstance(group_id, Optional):
1834 if not isinstance(group_id, Optional):
1835 q = q.filter(Repository.group_id == group_id)
1835 q = q.filter(Repository.group_id == group_id)
1836
1836
1837 if case_insensitive:
1837 if case_insensitive:
1838 q = q.order_by(func.lower(Repository.repo_name))
1838 q = q.order_by(func.lower(Repository.repo_name))
1839 else:
1839 else:
1840 q = q.order_by(Repository.repo_name)
1840 q = q.order_by(Repository.repo_name)
1841
1841
1842 return q.all()
1842 return q.all()
1843
1843
1844 @property
1844 @property
1845 def repo_uid(self):
1845 def repo_uid(self):
1846 return '_{}'.format(self.repo_id)
1846 return '_{}'.format(self.repo_id)
1847
1847
1848 @property
1848 @property
1849 def forks(self):
1849 def forks(self):
1850 """
1850 """
1851 Return forks of this repo
1851 Return forks of this repo
1852 """
1852 """
1853 return Repository.get_repo_forks(self.repo_id)
1853 return Repository.get_repo_forks(self.repo_id)
1854
1854
1855 @property
1855 @property
1856 def parent(self):
1856 def parent(self):
1857 """
1857 """
1858 Returns fork parent
1858 Returns fork parent
1859 """
1859 """
1860 return self.fork
1860 return self.fork
1861
1861
1862 @property
1862 @property
1863 def just_name(self):
1863 def just_name(self):
1864 return self.repo_name.split(self.NAME_SEP)[-1]
1864 return self.repo_name.split(self.NAME_SEP)[-1]
1865
1865
1866 @property
1866 @property
1867 def groups_with_parents(self):
1867 def groups_with_parents(self):
1868 groups = []
1868 groups = []
1869 if self.group is None:
1869 if self.group is None:
1870 return groups
1870 return groups
1871
1871
1872 cur_gr = self.group
1872 cur_gr = self.group
1873 groups.insert(0, cur_gr)
1873 groups.insert(0, cur_gr)
1874 while 1:
1874 while 1:
1875 gr = getattr(cur_gr, 'parent_group', None)
1875 gr = getattr(cur_gr, 'parent_group', None)
1876 cur_gr = cur_gr.parent_group
1876 cur_gr = cur_gr.parent_group
1877 if gr is None:
1877 if gr is None:
1878 break
1878 break
1879 groups.insert(0, gr)
1879 groups.insert(0, gr)
1880
1880
1881 return groups
1881 return groups
1882
1882
1883 @property
1883 @property
1884 def groups_and_repo(self):
1884 def groups_and_repo(self):
1885 return self.groups_with_parents, self
1885 return self.groups_with_parents, self
1886
1886
1887 @LazyProperty
1887 @LazyProperty
1888 def repo_path(self):
1888 def repo_path(self):
1889 """
1889 """
1890 Returns base full path for that repository means where it actually
1890 Returns base full path for that repository means where it actually
1891 exists on a filesystem
1891 exists on a filesystem
1892 """
1892 """
1893 q = Session().query(RhodeCodeUi).filter(
1893 q = Session().query(RhodeCodeUi).filter(
1894 RhodeCodeUi.ui_key == self.NAME_SEP)
1894 RhodeCodeUi.ui_key == self.NAME_SEP)
1895 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1895 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1896 return q.one().ui_value
1896 return q.one().ui_value
1897
1897
1898 @property
1898 @property
1899 def repo_full_path(self):
1899 def repo_full_path(self):
1900 p = [self.repo_path]
1900 p = [self.repo_path]
1901 # we need to split the name by / since this is how we store the
1901 # we need to split the name by / since this is how we store the
1902 # names in the database, but that eventually needs to be converted
1902 # names in the database, but that eventually needs to be converted
1903 # into a valid system path
1903 # into a valid system path
1904 p += self.repo_name.split(self.NAME_SEP)
1904 p += self.repo_name.split(self.NAME_SEP)
1905 return os.path.join(*map(safe_unicode, p))
1905 return os.path.join(*map(safe_unicode, p))
1906
1906
1907 @property
1907 @property
1908 def cache_keys(self):
1908 def cache_keys(self):
1909 """
1909 """
1910 Returns associated cache keys for that repo
1910 Returns associated cache keys for that repo
1911 """
1911 """
1912 invalidation_namespace = CacheKey.REPO_INVALIDATION_NAMESPACE.format(
1912 invalidation_namespace = CacheKey.REPO_INVALIDATION_NAMESPACE.format(
1913 repo_id=self.repo_id)
1913 repo_id=self.repo_id)
1914 return CacheKey.query()\
1914 return CacheKey.query()\
1915 .filter(CacheKey.cache_args == invalidation_namespace)\
1915 .filter(CacheKey.cache_args == invalidation_namespace)\
1916 .order_by(CacheKey.cache_key)\
1916 .order_by(CacheKey.cache_key)\
1917 .all()
1917 .all()
1918
1918
1919 @property
1919 @property
1920 def cached_diffs_relative_dir(self):
1920 def cached_diffs_relative_dir(self):
1921 """
1921 """
1922 Return a relative to the repository store path of cached diffs
1922 Return a relative to the repository store path of cached diffs
1923 used for safe display for users, who shouldn't know the absolute store
1923 used for safe display for users, who shouldn't know the absolute store
1924 path
1924 path
1925 """
1925 """
1926 return os.path.join(
1926 return os.path.join(
1927 os.path.dirname(self.repo_name),
1927 os.path.dirname(self.repo_name),
1928 self.cached_diffs_dir.split(os.path.sep)[-1])
1928 self.cached_diffs_dir.split(os.path.sep)[-1])
1929
1929
1930 @property
1930 @property
1931 def cached_diffs_dir(self):
1931 def cached_diffs_dir(self):
1932 path = self.repo_full_path
1932 path = self.repo_full_path
1933 return os.path.join(
1933 return os.path.join(
1934 os.path.dirname(path),
1934 os.path.dirname(path),
1935 '.__shadow_diff_cache_repo_{}'.format(self.repo_id))
1935 '.__shadow_diff_cache_repo_{}'.format(self.repo_id))
1936
1936
1937 def cached_diffs(self):
1937 def cached_diffs(self):
1938 diff_cache_dir = self.cached_diffs_dir
1938 diff_cache_dir = self.cached_diffs_dir
1939 if os.path.isdir(diff_cache_dir):
1939 if os.path.isdir(diff_cache_dir):
1940 return os.listdir(diff_cache_dir)
1940 return os.listdir(diff_cache_dir)
1941 return []
1941 return []
1942
1942
1943 def shadow_repos(self):
1943 def shadow_repos(self):
1944 shadow_repos_pattern = '.__shadow_repo_{}'.format(self.repo_id)
1944 shadow_repos_pattern = '.__shadow_repo_{}'.format(self.repo_id)
1945 return [
1945 return [
1946 x for x in os.listdir(os.path.dirname(self.repo_full_path))
1946 x for x in os.listdir(os.path.dirname(self.repo_full_path))
1947 if x.startswith(shadow_repos_pattern)]
1947 if x.startswith(shadow_repos_pattern)]
1948
1948
1949 def get_new_name(self, repo_name):
1949 def get_new_name(self, repo_name):
1950 """
1950 """
1951 returns new full repository name based on assigned group and new new
1951 returns new full repository name based on assigned group and new new
1952
1952
1953 :param group_name:
1953 :param group_name:
1954 """
1954 """
1955 path_prefix = self.group.full_path_splitted if self.group else []
1955 path_prefix = self.group.full_path_splitted if self.group else []
1956 return self.NAME_SEP.join(path_prefix + [repo_name])
1956 return self.NAME_SEP.join(path_prefix + [repo_name])
1957
1957
1958 @property
1958 @property
1959 def _config(self):
1959 def _config(self):
1960 """
1960 """
1961 Returns db based config object.
1961 Returns db based config object.
1962 """
1962 """
1963 from rhodecode.lib.utils import make_db_config
1963 from rhodecode.lib.utils import make_db_config
1964 return make_db_config(clear_session=False, repo=self)
1964 return make_db_config(clear_session=False, repo=self)
1965
1965
1966 def permissions(self, with_admins=True, with_owner=True,
1966 def permissions(self, with_admins=True, with_owner=True,
1967 expand_from_user_groups=False):
1967 expand_from_user_groups=False):
1968 """
1968 """
1969 Permissions for repositories
1969 Permissions for repositories
1970 """
1970 """
1971 _admin_perm = 'repository.admin'
1971 _admin_perm = 'repository.admin'
1972
1972
1973 owner_row = []
1973 owner_row = []
1974 if with_owner:
1974 if with_owner:
1975 usr = AttributeDict(self.user.get_dict())
1975 usr = AttributeDict(self.user.get_dict())
1976 usr.owner_row = True
1976 usr.owner_row = True
1977 usr.permission = _admin_perm
1977 usr.permission = _admin_perm
1978 usr.permission_id = None
1978 usr.permission_id = None
1979 owner_row.append(usr)
1979 owner_row.append(usr)
1980
1980
1981 super_admin_ids = []
1981 super_admin_ids = []
1982 super_admin_rows = []
1982 super_admin_rows = []
1983 if with_admins:
1983 if with_admins:
1984 for usr in User.get_all_super_admins():
1984 for usr in User.get_all_super_admins():
1985 super_admin_ids.append(usr.user_id)
1985 super_admin_ids.append(usr.user_id)
1986 # if this admin is also owner, don't double the record
1986 # if this admin is also owner, don't double the record
1987 if usr.user_id == owner_row[0].user_id:
1987 if usr.user_id == owner_row[0].user_id:
1988 owner_row[0].admin_row = True
1988 owner_row[0].admin_row = True
1989 else:
1989 else:
1990 usr = AttributeDict(usr.get_dict())
1990 usr = AttributeDict(usr.get_dict())
1991 usr.admin_row = True
1991 usr.admin_row = True
1992 usr.permission = _admin_perm
1992 usr.permission = _admin_perm
1993 usr.permission_id = None
1993 usr.permission_id = None
1994 super_admin_rows.append(usr)
1994 super_admin_rows.append(usr)
1995
1995
1996 q = UserRepoToPerm.query().filter(UserRepoToPerm.repository == self)
1996 q = UserRepoToPerm.query().filter(UserRepoToPerm.repository == self)
1997 q = q.options(joinedload(UserRepoToPerm.repository),
1997 q = q.options(joinedload(UserRepoToPerm.repository),
1998 joinedload(UserRepoToPerm.user),
1998 joinedload(UserRepoToPerm.user),
1999 joinedload(UserRepoToPerm.permission),)
1999 joinedload(UserRepoToPerm.permission),)
2000
2000
2001 # get owners and admins and permissions. We do a trick of re-writing
2001 # get owners and admins and permissions. We do a trick of re-writing
2002 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2002 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2003 # has a global reference and changing one object propagates to all
2003 # has a global reference and changing one object propagates to all
2004 # others. This means if admin is also an owner admin_row that change
2004 # others. This means if admin is also an owner admin_row that change
2005 # would propagate to both objects
2005 # would propagate to both objects
2006 perm_rows = []
2006 perm_rows = []
2007 for _usr in q.all():
2007 for _usr in q.all():
2008 usr = AttributeDict(_usr.user.get_dict())
2008 usr = AttributeDict(_usr.user.get_dict())
2009 # if this user is also owner/admin, mark as duplicate record
2009 # if this user is also owner/admin, mark as duplicate record
2010 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
2010 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
2011 usr.duplicate_perm = True
2011 usr.duplicate_perm = True
2012 # also check if this permission is maybe used by branch_permissions
2012 # also check if this permission is maybe used by branch_permissions
2013 if _usr.branch_perm_entry:
2013 if _usr.branch_perm_entry:
2014 usr.branch_rules = [x.branch_rule_id for x in _usr.branch_perm_entry]
2014 usr.branch_rules = [x.branch_rule_id for x in _usr.branch_perm_entry]
2015
2015
2016 usr.permission = _usr.permission.permission_name
2016 usr.permission = _usr.permission.permission_name
2017 usr.permission_id = _usr.repo_to_perm_id
2017 usr.permission_id = _usr.repo_to_perm_id
2018 perm_rows.append(usr)
2018 perm_rows.append(usr)
2019
2019
2020 # filter the perm rows by 'default' first and then sort them by
2020 # filter the perm rows by 'default' first and then sort them by
2021 # admin,write,read,none permissions sorted again alphabetically in
2021 # admin,write,read,none permissions sorted again alphabetically in
2022 # each group
2022 # each group
2023 perm_rows = sorted(perm_rows, key=display_user_sort)
2023 perm_rows = sorted(perm_rows, key=display_user_sort)
2024
2024
2025 user_groups_rows = []
2025 user_groups_rows = []
2026 if expand_from_user_groups:
2026 if expand_from_user_groups:
2027 for ug in self.permission_user_groups(with_members=True):
2027 for ug in self.permission_user_groups(with_members=True):
2028 for user_data in ug.members:
2028 for user_data in ug.members:
2029 user_groups_rows.append(user_data)
2029 user_groups_rows.append(user_data)
2030
2030
2031 return super_admin_rows + owner_row + perm_rows + user_groups_rows
2031 return super_admin_rows + owner_row + perm_rows + user_groups_rows
2032
2032
2033 def permission_user_groups(self, with_members=True):
2033 def permission_user_groups(self, with_members=True):
2034 q = UserGroupRepoToPerm.query()\
2034 q = UserGroupRepoToPerm.query()\
2035 .filter(UserGroupRepoToPerm.repository == self)
2035 .filter(UserGroupRepoToPerm.repository == self)
2036 q = q.options(joinedload(UserGroupRepoToPerm.repository),
2036 q = q.options(joinedload(UserGroupRepoToPerm.repository),
2037 joinedload(UserGroupRepoToPerm.users_group),
2037 joinedload(UserGroupRepoToPerm.users_group),
2038 joinedload(UserGroupRepoToPerm.permission),)
2038 joinedload(UserGroupRepoToPerm.permission),)
2039
2039
2040 perm_rows = []
2040 perm_rows = []
2041 for _user_group in q.all():
2041 for _user_group in q.all():
2042 entry = AttributeDict(_user_group.users_group.get_dict())
2042 entry = AttributeDict(_user_group.users_group.get_dict())
2043 entry.permission = _user_group.permission.permission_name
2043 entry.permission = _user_group.permission.permission_name
2044 if with_members:
2044 if with_members:
2045 entry.members = [x.user.get_dict()
2045 entry.members = [x.user.get_dict()
2046 for x in _user_group.users_group.members]
2046 for x in _user_group.users_group.members]
2047 perm_rows.append(entry)
2047 perm_rows.append(entry)
2048
2048
2049 perm_rows = sorted(perm_rows, key=display_user_group_sort)
2049 perm_rows = sorted(perm_rows, key=display_user_group_sort)
2050 return perm_rows
2050 return perm_rows
2051
2051
2052 def get_api_data(self, include_secrets=False):
2052 def get_api_data(self, include_secrets=False):
2053 """
2053 """
2054 Common function for generating repo api data
2054 Common function for generating repo api data
2055
2055
2056 :param include_secrets: See :meth:`User.get_api_data`.
2056 :param include_secrets: See :meth:`User.get_api_data`.
2057
2057
2058 """
2058 """
2059 # TODO: mikhail: Here there is an anti-pattern, we probably need to
2059 # TODO: mikhail: Here there is an anti-pattern, we probably need to
2060 # move this methods on models level.
2060 # move this methods on models level.
2061 from rhodecode.model.settings import SettingsModel
2061 from rhodecode.model.settings import SettingsModel
2062 from rhodecode.model.repo import RepoModel
2062 from rhodecode.model.repo import RepoModel
2063
2063
2064 repo = self
2064 repo = self
2065 _user_id, _time, _reason = self.locked
2065 _user_id, _time, _reason = self.locked
2066
2066
2067 data = {
2067 data = {
2068 'repo_id': repo.repo_id,
2068 'repo_id': repo.repo_id,
2069 'repo_name': repo.repo_name,
2069 'repo_name': repo.repo_name,
2070 'repo_type': repo.repo_type,
2070 'repo_type': repo.repo_type,
2071 'clone_uri': repo.clone_uri or '',
2071 'clone_uri': repo.clone_uri or '',
2072 'push_uri': repo.push_uri or '',
2072 'push_uri': repo.push_uri or '',
2073 'url': RepoModel().get_url(self),
2073 'url': RepoModel().get_url(self),
2074 'private': repo.private,
2074 'private': repo.private,
2075 'created_on': repo.created_on,
2075 'created_on': repo.created_on,
2076 'description': repo.description_safe,
2076 'description': repo.description_safe,
2077 'landing_rev': repo.landing_rev,
2077 'landing_rev': repo.landing_rev,
2078 'owner': repo.user.username,
2078 'owner': repo.user.username,
2079 'fork_of': repo.fork.repo_name if repo.fork else None,
2079 'fork_of': repo.fork.repo_name if repo.fork else None,
2080 'fork_of_id': repo.fork.repo_id if repo.fork else None,
2080 'fork_of_id': repo.fork.repo_id if repo.fork else None,
2081 'enable_statistics': repo.enable_statistics,
2081 'enable_statistics': repo.enable_statistics,
2082 'enable_locking': repo.enable_locking,
2082 'enable_locking': repo.enable_locking,
2083 'enable_downloads': repo.enable_downloads,
2083 'enable_downloads': repo.enable_downloads,
2084 'last_changeset': repo.changeset_cache,
2084 'last_changeset': repo.changeset_cache,
2085 'locked_by': User.get(_user_id).get_api_data(
2085 'locked_by': User.get(_user_id).get_api_data(
2086 include_secrets=include_secrets) if _user_id else None,
2086 include_secrets=include_secrets) if _user_id else None,
2087 'locked_date': time_to_datetime(_time) if _time else None,
2087 'locked_date': time_to_datetime(_time) if _time else None,
2088 'lock_reason': _reason if _reason else None,
2088 'lock_reason': _reason if _reason else None,
2089 }
2089 }
2090
2090
2091 # TODO: mikhail: should be per-repo settings here
2091 # TODO: mikhail: should be per-repo settings here
2092 rc_config = SettingsModel().get_all_settings()
2092 rc_config = SettingsModel().get_all_settings()
2093 repository_fields = str2bool(
2093 repository_fields = str2bool(
2094 rc_config.get('rhodecode_repository_fields'))
2094 rc_config.get('rhodecode_repository_fields'))
2095 if repository_fields:
2095 if repository_fields:
2096 for f in self.extra_fields:
2096 for f in self.extra_fields:
2097 data[f.field_key_prefixed] = f.field_value
2097 data[f.field_key_prefixed] = f.field_value
2098
2098
2099 return data
2099 return data
2100
2100
2101 @classmethod
2101 @classmethod
2102 def lock(cls, repo, user_id, lock_time=None, lock_reason=None):
2102 def lock(cls, repo, user_id, lock_time=None, lock_reason=None):
2103 if not lock_time:
2103 if not lock_time:
2104 lock_time = time.time()
2104 lock_time = time.time()
2105 if not lock_reason:
2105 if not lock_reason:
2106 lock_reason = cls.LOCK_AUTOMATIC
2106 lock_reason = cls.LOCK_AUTOMATIC
2107 repo.locked = [user_id, lock_time, lock_reason]
2107 repo.locked = [user_id, lock_time, lock_reason]
2108 Session().add(repo)
2108 Session().add(repo)
2109 Session().commit()
2109 Session().commit()
2110
2110
2111 @classmethod
2111 @classmethod
2112 def unlock(cls, repo):
2112 def unlock(cls, repo):
2113 repo.locked = None
2113 repo.locked = None
2114 Session().add(repo)
2114 Session().add(repo)
2115 Session().commit()
2115 Session().commit()
2116
2116
2117 @classmethod
2117 @classmethod
2118 def getlock(cls, repo):
2118 def getlock(cls, repo):
2119 return repo.locked
2119 return repo.locked
2120
2120
2121 def is_user_lock(self, user_id):
2121 def is_user_lock(self, user_id):
2122 if self.lock[0]:
2122 if self.lock[0]:
2123 lock_user_id = safe_int(self.lock[0])
2123 lock_user_id = safe_int(self.lock[0])
2124 user_id = safe_int(user_id)
2124 user_id = safe_int(user_id)
2125 # both are ints, and they are equal
2125 # both are ints, and they are equal
2126 return all([lock_user_id, user_id]) and lock_user_id == user_id
2126 return all([lock_user_id, user_id]) and lock_user_id == user_id
2127
2127
2128 return False
2128 return False
2129
2129
2130 def get_locking_state(self, action, user_id, only_when_enabled=True):
2130 def get_locking_state(self, action, user_id, only_when_enabled=True):
2131 """
2131 """
2132 Checks locking on this repository, if locking is enabled and lock is
2132 Checks locking on this repository, if locking is enabled and lock is
2133 present returns a tuple of make_lock, locked, locked_by.
2133 present returns a tuple of make_lock, locked, locked_by.
2134 make_lock can have 3 states None (do nothing) True, make lock
2134 make_lock can have 3 states None (do nothing) True, make lock
2135 False release lock, This value is later propagated to hooks, which
2135 False release lock, This value is later propagated to hooks, which
2136 do the locking. Think about this as signals passed to hooks what to do.
2136 do the locking. Think about this as signals passed to hooks what to do.
2137
2137
2138 """
2138 """
2139 # TODO: johbo: This is part of the business logic and should be moved
2139 # TODO: johbo: This is part of the business logic and should be moved
2140 # into the RepositoryModel.
2140 # into the RepositoryModel.
2141
2141
2142 if action not in ('push', 'pull'):
2142 if action not in ('push', 'pull'):
2143 raise ValueError("Invalid action value: %s" % repr(action))
2143 raise ValueError("Invalid action value: %s" % repr(action))
2144
2144
2145 # defines if locked error should be thrown to user
2145 # defines if locked error should be thrown to user
2146 currently_locked = False
2146 currently_locked = False
2147 # defines if new lock should be made, tri-state
2147 # defines if new lock should be made, tri-state
2148 make_lock = None
2148 make_lock = None
2149 repo = self
2149 repo = self
2150 user = User.get(user_id)
2150 user = User.get(user_id)
2151
2151
2152 lock_info = repo.locked
2152 lock_info = repo.locked
2153
2153
2154 if repo and (repo.enable_locking or not only_when_enabled):
2154 if repo and (repo.enable_locking or not only_when_enabled):
2155 if action == 'push':
2155 if action == 'push':
2156 # check if it's already locked !, if it is compare users
2156 # check if it's already locked !, if it is compare users
2157 locked_by_user_id = lock_info[0]
2157 locked_by_user_id = lock_info[0]
2158 if user.user_id == locked_by_user_id:
2158 if user.user_id == locked_by_user_id:
2159 log.debug(
2159 log.debug(
2160 'Got `push` action from user %s, now unlocking', user)
2160 'Got `push` action from user %s, now unlocking', user)
2161 # unlock if we have push from user who locked
2161 # unlock if we have push from user who locked
2162 make_lock = False
2162 make_lock = False
2163 else:
2163 else:
2164 # we're not the same user who locked, ban with
2164 # we're not the same user who locked, ban with
2165 # code defined in settings (default is 423 HTTP Locked) !
2165 # code defined in settings (default is 423 HTTP Locked) !
2166 log.debug('Repo %s is currently locked by %s', repo, user)
2166 log.debug('Repo %s is currently locked by %s', repo, user)
2167 currently_locked = True
2167 currently_locked = True
2168 elif action == 'pull':
2168 elif action == 'pull':
2169 # [0] user [1] date
2169 # [0] user [1] date
2170 if lock_info[0] and lock_info[1]:
2170 if lock_info[0] and lock_info[1]:
2171 log.debug('Repo %s is currently locked by %s', repo, user)
2171 log.debug('Repo %s is currently locked by %s', repo, user)
2172 currently_locked = True
2172 currently_locked = True
2173 else:
2173 else:
2174 log.debug('Setting lock on repo %s by %s', repo, user)
2174 log.debug('Setting lock on repo %s by %s', repo, user)
2175 make_lock = True
2175 make_lock = True
2176
2176
2177 else:
2177 else:
2178 log.debug('Repository %s do not have locking enabled', repo)
2178 log.debug('Repository %s do not have locking enabled', repo)
2179
2179
2180 log.debug('FINAL locking values make_lock:%s,locked:%s,locked_by:%s',
2180 log.debug('FINAL locking values make_lock:%s,locked:%s,locked_by:%s',
2181 make_lock, currently_locked, lock_info)
2181 make_lock, currently_locked, lock_info)
2182
2182
2183 from rhodecode.lib.auth import HasRepoPermissionAny
2183 from rhodecode.lib.auth import HasRepoPermissionAny
2184 perm_check = HasRepoPermissionAny('repository.write', 'repository.admin')
2184 perm_check = HasRepoPermissionAny('repository.write', 'repository.admin')
2185 if make_lock and not perm_check(repo_name=repo.repo_name, user=user):
2185 if make_lock and not perm_check(repo_name=repo.repo_name, user=user):
2186 # if we don't have at least write permission we cannot make a lock
2186 # if we don't have at least write permission we cannot make a lock
2187 log.debug('lock state reset back to FALSE due to lack '
2187 log.debug('lock state reset back to FALSE due to lack '
2188 'of at least read permission')
2188 'of at least read permission')
2189 make_lock = False
2189 make_lock = False
2190
2190
2191 return make_lock, currently_locked, lock_info
2191 return make_lock, currently_locked, lock_info
2192
2192
2193 @property
2193 @property
2194 def last_commit_cache_update_diff(self):
2194 def last_commit_cache_update_diff(self):
2195 return time.time() - (safe_int(self.changeset_cache.get('updated_on')) or 0)
2195 return time.time() - (safe_int(self.changeset_cache.get('updated_on')) or 0)
2196
2196
2197 @property
2197 @property
2198 def last_commit_change(self):
2198 def last_commit_change(self):
2199 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2199 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2200 empty_date = datetime.datetime.fromtimestamp(0)
2200 empty_date = datetime.datetime.fromtimestamp(0)
2201 date_latest = self.changeset_cache.get('date', empty_date)
2201 date_latest = self.changeset_cache.get('date', empty_date)
2202 try:
2202 try:
2203 return parse_datetime(date_latest)
2203 return parse_datetime(date_latest)
2204 except Exception:
2204 except Exception:
2205 return empty_date
2205 return empty_date
2206
2206
2207 @property
2207 @property
2208 def last_db_change(self):
2208 def last_db_change(self):
2209 return self.updated_on
2209 return self.updated_on
2210
2210
2211 @property
2211 @property
2212 def clone_uri_hidden(self):
2212 def clone_uri_hidden(self):
2213 clone_uri = self.clone_uri
2213 clone_uri = self.clone_uri
2214 if clone_uri:
2214 if clone_uri:
2215 import urlobject
2215 import urlobject
2216 url_obj = urlobject.URLObject(cleaned_uri(clone_uri))
2216 url_obj = urlobject.URLObject(cleaned_uri(clone_uri))
2217 if url_obj.password:
2217 if url_obj.password:
2218 clone_uri = url_obj.with_password('*****')
2218 clone_uri = url_obj.with_password('*****')
2219 return clone_uri
2219 return clone_uri
2220
2220
2221 @property
2221 @property
2222 def push_uri_hidden(self):
2222 def push_uri_hidden(self):
2223 push_uri = self.push_uri
2223 push_uri = self.push_uri
2224 if push_uri:
2224 if push_uri:
2225 import urlobject
2225 import urlobject
2226 url_obj = urlobject.URLObject(cleaned_uri(push_uri))
2226 url_obj = urlobject.URLObject(cleaned_uri(push_uri))
2227 if url_obj.password:
2227 if url_obj.password:
2228 push_uri = url_obj.with_password('*****')
2228 push_uri = url_obj.with_password('*****')
2229 return push_uri
2229 return push_uri
2230
2230
2231 def clone_url(self, **override):
2231 def clone_url(self, **override):
2232 from rhodecode.model.settings import SettingsModel
2232 from rhodecode.model.settings import SettingsModel
2233
2233
2234 uri_tmpl = None
2234 uri_tmpl = None
2235 if 'with_id' in override:
2235 if 'with_id' in override:
2236 uri_tmpl = self.DEFAULT_CLONE_URI_ID
2236 uri_tmpl = self.DEFAULT_CLONE_URI_ID
2237 del override['with_id']
2237 del override['with_id']
2238
2238
2239 if 'uri_tmpl' in override:
2239 if 'uri_tmpl' in override:
2240 uri_tmpl = override['uri_tmpl']
2240 uri_tmpl = override['uri_tmpl']
2241 del override['uri_tmpl']
2241 del override['uri_tmpl']
2242
2242
2243 ssh = False
2243 ssh = False
2244 if 'ssh' in override:
2244 if 'ssh' in override:
2245 ssh = True
2245 ssh = True
2246 del override['ssh']
2246 del override['ssh']
2247
2247
2248 # we didn't override our tmpl from **overrides
2248 # we didn't override our tmpl from **overrides
2249 request = get_current_request()
2249 request = get_current_request()
2250 if not uri_tmpl:
2250 if not uri_tmpl:
2251 if hasattr(request, 'call_context') and hasattr(request.call_context, 'rc_config'):
2251 if hasattr(request, 'call_context') and hasattr(request.call_context, 'rc_config'):
2252 rc_config = request.call_context.rc_config
2252 rc_config = request.call_context.rc_config
2253 else:
2253 else:
2254 rc_config = SettingsModel().get_all_settings(cache=True)
2254 rc_config = SettingsModel().get_all_settings(cache=True)
2255 if ssh:
2255 if ssh:
2256 uri_tmpl = rc_config.get(
2256 uri_tmpl = rc_config.get(
2257 'rhodecode_clone_uri_ssh_tmpl') or self.DEFAULT_CLONE_URI_SSH
2257 'rhodecode_clone_uri_ssh_tmpl') or self.DEFAULT_CLONE_URI_SSH
2258 else:
2258 else:
2259 uri_tmpl = rc_config.get(
2259 uri_tmpl = rc_config.get(
2260 'rhodecode_clone_uri_tmpl') or self.DEFAULT_CLONE_URI
2260 'rhodecode_clone_uri_tmpl') or self.DEFAULT_CLONE_URI
2261
2261
2262 return get_clone_url(request=request,
2262 return get_clone_url(request=request,
2263 uri_tmpl=uri_tmpl,
2263 uri_tmpl=uri_tmpl,
2264 repo_name=self.repo_name,
2264 repo_name=self.repo_name,
2265 repo_id=self.repo_id, **override)
2265 repo_id=self.repo_id, **override)
2266
2266
2267 def set_state(self, state):
2267 def set_state(self, state):
2268 self.repo_state = state
2268 self.repo_state = state
2269 Session().add(self)
2269 Session().add(self)
2270 #==========================================================================
2270 #==========================================================================
2271 # SCM PROPERTIES
2271 # SCM PROPERTIES
2272 #==========================================================================
2272 #==========================================================================
2273
2273
2274 def get_commit(self, commit_id=None, commit_idx=None, pre_load=None):
2274 def get_commit(self, commit_id=None, commit_idx=None, pre_load=None):
2275 return get_commit_safe(
2275 return get_commit_safe(
2276 self.scm_instance(), commit_id, commit_idx, pre_load=pre_load)
2276 self.scm_instance(), commit_id, commit_idx, pre_load=pre_load)
2277
2277
2278 def get_changeset(self, rev=None, pre_load=None):
2278 def get_changeset(self, rev=None, pre_load=None):
2279 warnings.warn("Use get_commit", DeprecationWarning)
2279 warnings.warn("Use get_commit", DeprecationWarning)
2280 commit_id = None
2280 commit_id = None
2281 commit_idx = None
2281 commit_idx = None
2282 if isinstance(rev, compat.string_types):
2282 if isinstance(rev, compat.string_types):
2283 commit_id = rev
2283 commit_id = rev
2284 else:
2284 else:
2285 commit_idx = rev
2285 commit_idx = rev
2286 return self.get_commit(commit_id=commit_id, commit_idx=commit_idx,
2286 return self.get_commit(commit_id=commit_id, commit_idx=commit_idx,
2287 pre_load=pre_load)
2287 pre_load=pre_load)
2288
2288
2289 def get_landing_commit(self):
2289 def get_landing_commit(self):
2290 """
2290 """
2291 Returns landing commit, or if that doesn't exist returns the tip
2291 Returns landing commit, or if that doesn't exist returns the tip
2292 """
2292 """
2293 _rev_type, _rev = self.landing_rev
2293 _rev_type, _rev = self.landing_rev
2294 commit = self.get_commit(_rev)
2294 commit = self.get_commit(_rev)
2295 if isinstance(commit, EmptyCommit):
2295 if isinstance(commit, EmptyCommit):
2296 return self.get_commit()
2296 return self.get_commit()
2297 return commit
2297 return commit
2298
2298
2299 def update_commit_cache(self, cs_cache=None, config=None):
2299 def update_commit_cache(self, cs_cache=None, config=None):
2300 """
2300 """
2301 Update cache of last commit for repository, keys should be::
2301 Update cache of last commit for repository, keys should be::
2302
2302
2303 source_repo_id
2303 source_repo_id
2304 short_id
2304 short_id
2305 raw_id
2305 raw_id
2306 revision
2306 revision
2307 parents
2307 parents
2308 message
2308 message
2309 date
2309 date
2310 author
2310 author
2311 updated_on
2311 updated_on
2312
2312
2313 """
2313 """
2314 from rhodecode.lib.vcs.backends.base import BaseChangeset
2314 from rhodecode.lib.vcs.backends.base import BaseChangeset
2315 if cs_cache is None:
2315 if cs_cache is None:
2316 # use no-cache version here
2316 # use no-cache version here
2317 scm_repo = self.scm_instance(cache=False, config=config)
2317 scm_repo = self.scm_instance(cache=False, config=config)
2318
2318
2319 empty = scm_repo is None or scm_repo.is_empty()
2319 empty = scm_repo is None or scm_repo.is_empty()
2320 if not empty:
2320 if not empty:
2321 cs_cache = scm_repo.get_commit(
2321 cs_cache = scm_repo.get_commit(
2322 pre_load=["author", "date", "message", "parents", "branch"])
2322 pre_load=["author", "date", "message", "parents", "branch"])
2323 else:
2323 else:
2324 cs_cache = EmptyCommit()
2324 cs_cache = EmptyCommit()
2325
2325
2326 if isinstance(cs_cache, BaseChangeset):
2326 if isinstance(cs_cache, BaseChangeset):
2327 cs_cache = cs_cache.__json__()
2327 cs_cache = cs_cache.__json__()
2328
2328
2329 def is_outdated(new_cs_cache):
2329 def is_outdated(new_cs_cache):
2330 if (new_cs_cache['raw_id'] != self.changeset_cache['raw_id'] or
2330 if (new_cs_cache['raw_id'] != self.changeset_cache['raw_id'] or
2331 new_cs_cache['revision'] != self.changeset_cache['revision']):
2331 new_cs_cache['revision'] != self.changeset_cache['revision']):
2332 return True
2332 return True
2333 return False
2333 return False
2334
2334
2335 # check if we have maybe already latest cached revision
2335 # check if we have maybe already latest cached revision
2336 if is_outdated(cs_cache) or not self.changeset_cache:
2336 if is_outdated(cs_cache) or not self.changeset_cache:
2337 _default = datetime.datetime.utcnow()
2337 _default = datetime.datetime.utcnow()
2338 last_change = cs_cache.get('date') or _default
2338 last_change = cs_cache.get('date') or _default
2339 # we check if last update is newer than the new value
2339 # we check if last update is newer than the new value
2340 # if yes, we use the current timestamp instead. Imagine you get
2340 # if yes, we use the current timestamp instead. Imagine you get
2341 # old commit pushed 1y ago, we'd set last update 1y to ago.
2341 # old commit pushed 1y ago, we'd set last update 1y to ago.
2342 last_change_timestamp = datetime_to_time(last_change)
2342 last_change_timestamp = datetime_to_time(last_change)
2343 current_timestamp = datetime_to_time(last_change)
2343 current_timestamp = datetime_to_time(last_change)
2344 if last_change_timestamp > current_timestamp:
2344 if last_change_timestamp > current_timestamp:
2345 cs_cache['date'] = _default
2345 cs_cache['date'] = _default
2346
2346
2347 cs_cache['updated_on'] = time.time()
2347 cs_cache['updated_on'] = time.time()
2348 self.changeset_cache = cs_cache
2348 self.changeset_cache = cs_cache
2349 Session().add(self)
2349 Session().add(self)
2350 Session().commit()
2350 Session().commit()
2351
2351
2352 log.debug('updated repo %s with new commit cache %s',
2352 log.debug('updated repo %s with new commit cache %s',
2353 self.repo_name, cs_cache)
2353 self.repo_name, cs_cache)
2354 else:
2354 else:
2355 cs_cache = self.changeset_cache
2355 cs_cache = self.changeset_cache
2356 cs_cache['updated_on'] = time.time()
2356 cs_cache['updated_on'] = time.time()
2357 self.changeset_cache = cs_cache
2357 self.changeset_cache = cs_cache
2358 Session().add(self)
2358 Session().add(self)
2359 Session().commit()
2359 Session().commit()
2360
2360
2361 log.debug('Skipping update_commit_cache for repo:`%s` '
2361 log.debug('Skipping update_commit_cache for repo:`%s` '
2362 'commit already with latest changes', self.repo_name)
2362 'commit already with latest changes', self.repo_name)
2363
2363
2364 @property
2364 @property
2365 def tip(self):
2365 def tip(self):
2366 return self.get_commit('tip')
2366 return self.get_commit('tip')
2367
2367
2368 @property
2368 @property
2369 def author(self):
2369 def author(self):
2370 return self.tip.author
2370 return self.tip.author
2371
2371
2372 @property
2372 @property
2373 def last_change(self):
2373 def last_change(self):
2374 return self.scm_instance().last_change
2374 return self.scm_instance().last_change
2375
2375
2376 def get_comments(self, revisions=None):
2376 def get_comments(self, revisions=None):
2377 """
2377 """
2378 Returns comments for this repository grouped by revisions
2378 Returns comments for this repository grouped by revisions
2379
2379
2380 :param revisions: filter query by revisions only
2380 :param revisions: filter query by revisions only
2381 """
2381 """
2382 cmts = ChangesetComment.query()\
2382 cmts = ChangesetComment.query()\
2383 .filter(ChangesetComment.repo == self)
2383 .filter(ChangesetComment.repo == self)
2384 if revisions:
2384 if revisions:
2385 cmts = cmts.filter(ChangesetComment.revision.in_(revisions))
2385 cmts = cmts.filter(ChangesetComment.revision.in_(revisions))
2386 grouped = collections.defaultdict(list)
2386 grouped = collections.defaultdict(list)
2387 for cmt in cmts.all():
2387 for cmt in cmts.all():
2388 grouped[cmt.revision].append(cmt)
2388 grouped[cmt.revision].append(cmt)
2389 return grouped
2389 return grouped
2390
2390
2391 def statuses(self, revisions=None):
2391 def statuses(self, revisions=None):
2392 """
2392 """
2393 Returns statuses for this repository
2393 Returns statuses for this repository
2394
2394
2395 :param revisions: list of revisions to get statuses for
2395 :param revisions: list of revisions to get statuses for
2396 """
2396 """
2397 statuses = ChangesetStatus.query()\
2397 statuses = ChangesetStatus.query()\
2398 .filter(ChangesetStatus.repo == self)\
2398 .filter(ChangesetStatus.repo == self)\
2399 .filter(ChangesetStatus.version == 0)
2399 .filter(ChangesetStatus.version == 0)
2400
2400
2401 if revisions:
2401 if revisions:
2402 # Try doing the filtering in chunks to avoid hitting limits
2402 # Try doing the filtering in chunks to avoid hitting limits
2403 size = 500
2403 size = 500
2404 status_results = []
2404 status_results = []
2405 for chunk in xrange(0, len(revisions), size):
2405 for chunk in xrange(0, len(revisions), size):
2406 status_results += statuses.filter(
2406 status_results += statuses.filter(
2407 ChangesetStatus.revision.in_(
2407 ChangesetStatus.revision.in_(
2408 revisions[chunk: chunk+size])
2408 revisions[chunk: chunk+size])
2409 ).all()
2409 ).all()
2410 else:
2410 else:
2411 status_results = statuses.all()
2411 status_results = statuses.all()
2412
2412
2413 grouped = {}
2413 grouped = {}
2414
2414
2415 # maybe we have open new pullrequest without a status?
2415 # maybe we have open new pullrequest without a status?
2416 stat = ChangesetStatus.STATUS_UNDER_REVIEW
2416 stat = ChangesetStatus.STATUS_UNDER_REVIEW
2417 status_lbl = ChangesetStatus.get_status_lbl(stat)
2417 status_lbl = ChangesetStatus.get_status_lbl(stat)
2418 for pr in PullRequest.query().filter(PullRequest.source_repo == self).all():
2418 for pr in PullRequest.query().filter(PullRequest.source_repo == self).all():
2419 for rev in pr.revisions:
2419 for rev in pr.revisions:
2420 pr_id = pr.pull_request_id
2420 pr_id = pr.pull_request_id
2421 pr_repo = pr.target_repo.repo_name
2421 pr_repo = pr.target_repo.repo_name
2422 grouped[rev] = [stat, status_lbl, pr_id, pr_repo]
2422 grouped[rev] = [stat, status_lbl, pr_id, pr_repo]
2423
2423
2424 for stat in status_results:
2424 for stat in status_results:
2425 pr_id = pr_repo = None
2425 pr_id = pr_repo = None
2426 if stat.pull_request:
2426 if stat.pull_request:
2427 pr_id = stat.pull_request.pull_request_id
2427 pr_id = stat.pull_request.pull_request_id
2428 pr_repo = stat.pull_request.target_repo.repo_name
2428 pr_repo = stat.pull_request.target_repo.repo_name
2429 grouped[stat.revision] = [str(stat.status), stat.status_lbl,
2429 grouped[stat.revision] = [str(stat.status), stat.status_lbl,
2430 pr_id, pr_repo]
2430 pr_id, pr_repo]
2431 return grouped
2431 return grouped
2432
2432
2433 # ==========================================================================
2433 # ==========================================================================
2434 # SCM CACHE INSTANCE
2434 # SCM CACHE INSTANCE
2435 # ==========================================================================
2435 # ==========================================================================
2436
2436
2437 def scm_instance(self, **kwargs):
2437 def scm_instance(self, **kwargs):
2438 import rhodecode
2438 import rhodecode
2439
2439
2440 # Passing a config will not hit the cache currently only used
2440 # Passing a config will not hit the cache currently only used
2441 # for repo2dbmapper
2441 # for repo2dbmapper
2442 config = kwargs.pop('config', None)
2442 config = kwargs.pop('config', None)
2443 cache = kwargs.pop('cache', None)
2443 cache = kwargs.pop('cache', None)
2444 vcs_full_cache = kwargs.pop('vcs_full_cache', None)
2444 vcs_full_cache = kwargs.pop('vcs_full_cache', None)
2445 if vcs_full_cache is not None:
2445 if vcs_full_cache is not None:
2446 # allows override global config
2446 # allows override global config
2447 full_cache = vcs_full_cache
2447 full_cache = vcs_full_cache
2448 else:
2448 else:
2449 full_cache = str2bool(rhodecode.CONFIG.get('vcs_full_cache'))
2449 full_cache = str2bool(rhodecode.CONFIG.get('vcs_full_cache'))
2450 # if cache is NOT defined use default global, else we have a full
2450 # if cache is NOT defined use default global, else we have a full
2451 # control over cache behaviour
2451 # control over cache behaviour
2452 if cache is None and full_cache and not config:
2452 if cache is None and full_cache and not config:
2453 log.debug('Initializing pure cached instance for %s', self.repo_path)
2453 log.debug('Initializing pure cached instance for %s', self.repo_path)
2454 return self._get_instance_cached()
2454 return self._get_instance_cached()
2455
2455
2456 # cache here is sent to the "vcs server"
2456 # cache here is sent to the "vcs server"
2457 return self._get_instance(cache=bool(cache), config=config)
2457 return self._get_instance(cache=bool(cache), config=config)
2458
2458
2459 def _get_instance_cached(self):
2459 def _get_instance_cached(self):
2460 from rhodecode.lib import rc_cache
2460 from rhodecode.lib import rc_cache
2461
2461
2462 cache_namespace_uid = 'cache_repo_instance.{}'.format(self.repo_id)
2462 cache_namespace_uid = 'cache_repo_instance.{}'.format(self.repo_id)
2463 invalidation_namespace = CacheKey.REPO_INVALIDATION_NAMESPACE.format(
2463 invalidation_namespace = CacheKey.REPO_INVALIDATION_NAMESPACE.format(
2464 repo_id=self.repo_id)
2464 repo_id=self.repo_id)
2465 region = rc_cache.get_or_create_region('cache_repo_longterm', cache_namespace_uid)
2465 region = rc_cache.get_or_create_region('cache_repo_longterm', cache_namespace_uid)
2466
2466
2467 @region.conditional_cache_on_arguments(namespace=cache_namespace_uid)
2467 @region.conditional_cache_on_arguments(namespace=cache_namespace_uid)
2468 def get_instance_cached(repo_id, context_id, _cache_state_uid):
2468 def get_instance_cached(repo_id, context_id, _cache_state_uid):
2469 return self._get_instance(repo_state_uid=_cache_state_uid)
2469 return self._get_instance(repo_state_uid=_cache_state_uid)
2470
2470
2471 # we must use thread scoped cache here,
2471 # we must use thread scoped cache here,
2472 # because each thread of gevent needs it's own not shared connection and cache
2472 # because each thread of gevent needs it's own not shared connection and cache
2473 # we also alter `args` so the cache key is individual for every green thread.
2473 # we also alter `args` so the cache key is individual for every green thread.
2474 inv_context_manager = rc_cache.InvalidationContext(
2474 inv_context_manager = rc_cache.InvalidationContext(
2475 uid=cache_namespace_uid, invalidation_namespace=invalidation_namespace,
2475 uid=cache_namespace_uid, invalidation_namespace=invalidation_namespace,
2476 thread_scoped=True)
2476 thread_scoped=True)
2477 with inv_context_manager as invalidation_context:
2477 with inv_context_manager as invalidation_context:
2478 cache_state_uid = invalidation_context.cache_data['cache_state_uid']
2478 cache_state_uid = invalidation_context.cache_data['cache_state_uid']
2479 args = (self.repo_id, inv_context_manager.cache_key, cache_state_uid)
2479 args = (self.repo_id, inv_context_manager.cache_key, cache_state_uid)
2480
2480
2481 # re-compute and store cache if we get invalidate signal
2481 # re-compute and store cache if we get invalidate signal
2482 if invalidation_context.should_invalidate():
2482 if invalidation_context.should_invalidate():
2483 instance = get_instance_cached.refresh(*args)
2483 instance = get_instance_cached.refresh(*args)
2484 else:
2484 else:
2485 instance = get_instance_cached(*args)
2485 instance = get_instance_cached(*args)
2486
2486
2487 log.debug('Repo instance fetched in %.4fs', inv_context_manager.compute_time)
2487 log.debug('Repo instance fetched in %.4fs', inv_context_manager.compute_time)
2488 return instance
2488 return instance
2489
2489
2490 def _get_instance(self, cache=True, config=None, repo_state_uid=None):
2490 def _get_instance(self, cache=True, config=None, repo_state_uid=None):
2491 log.debug('Initializing %s instance `%s` with cache flag set to: %s',
2491 log.debug('Initializing %s instance `%s` with cache flag set to: %s',
2492 self.repo_type, self.repo_path, cache)
2492 self.repo_type, self.repo_path, cache)
2493 config = config or self._config
2493 config = config or self._config
2494 custom_wire = {
2494 custom_wire = {
2495 'cache': cache, # controls the vcs.remote cache
2495 'cache': cache, # controls the vcs.remote cache
2496 'repo_state_uid': repo_state_uid
2496 'repo_state_uid': repo_state_uid
2497 }
2497 }
2498 repo = get_vcs_instance(
2498 repo = get_vcs_instance(
2499 repo_path=safe_str(self.repo_full_path),
2499 repo_path=safe_str(self.repo_full_path),
2500 config=config,
2500 config=config,
2501 with_wire=custom_wire,
2501 with_wire=custom_wire,
2502 create=False,
2502 create=False,
2503 _vcs_alias=self.repo_type)
2503 _vcs_alias=self.repo_type)
2504 if repo is not None:
2504 if repo is not None:
2505 repo.count() # cache rebuild
2505 repo.count() # cache rebuild
2506 return repo
2506 return repo
2507
2507
2508 def get_shadow_repository_path(self, workspace_id):
2508 def get_shadow_repository_path(self, workspace_id):
2509 from rhodecode.lib.vcs.backends.base import BaseRepository
2509 from rhodecode.lib.vcs.backends.base import BaseRepository
2510 shadow_repo_path = BaseRepository._get_shadow_repository_path(
2510 shadow_repo_path = BaseRepository._get_shadow_repository_path(
2511 self.repo_full_path, self.repo_id, workspace_id)
2511 self.repo_full_path, self.repo_id, workspace_id)
2512 return shadow_repo_path
2512 return shadow_repo_path
2513
2513
2514 def __json__(self):
2514 def __json__(self):
2515 return {'landing_rev': self.landing_rev}
2515 return {'landing_rev': self.landing_rev}
2516
2516
2517 def get_dict(self):
2517 def get_dict(self):
2518
2518
2519 # Since we transformed `repo_name` to a hybrid property, we need to
2519 # Since we transformed `repo_name` to a hybrid property, we need to
2520 # keep compatibility with the code which uses `repo_name` field.
2520 # keep compatibility with the code which uses `repo_name` field.
2521
2521
2522 result = super(Repository, self).get_dict()
2522 result = super(Repository, self).get_dict()
2523 result['repo_name'] = result.pop('_repo_name', None)
2523 result['repo_name'] = result.pop('_repo_name', None)
2524 return result
2524 return result
2525
2525
2526
2526
2527 class RepoGroup(Base, BaseModel):
2527 class RepoGroup(Base, BaseModel):
2528 __tablename__ = 'groups'
2528 __tablename__ = 'groups'
2529 __table_args__ = (
2529 __table_args__ = (
2530 UniqueConstraint('group_name', 'group_parent_id'),
2530 UniqueConstraint('group_name', 'group_parent_id'),
2531 base_table_args,
2531 base_table_args,
2532 )
2532 )
2533 __mapper_args__ = {'order_by': 'group_name'}
2533 __mapper_args__ = {'order_by': 'group_name'}
2534
2534
2535 CHOICES_SEPARATOR = '/' # used to generate select2 choices for nested groups
2535 CHOICES_SEPARATOR = '/' # used to generate select2 choices for nested groups
2536
2536
2537 group_id = Column("group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2537 group_id = Column("group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2538 _group_name = Column("group_name", String(255), nullable=False, unique=True, default=None)
2538 _group_name = Column("group_name", String(255), nullable=False, unique=True, default=None)
2539 group_name_hash = Column("repo_group_name_hash", String(1024), nullable=False, unique=False)
2539 group_name_hash = Column("repo_group_name_hash", String(1024), nullable=False, unique=False)
2540 group_parent_id = Column("group_parent_id", Integer(), ForeignKey('groups.group_id'), nullable=True, unique=None, default=None)
2540 group_parent_id = Column("group_parent_id", Integer(), ForeignKey('groups.group_id'), nullable=True, unique=None, default=None)
2541 group_description = Column("group_description", String(10000), nullable=True, unique=None, default=None)
2541 group_description = Column("group_description", String(10000), nullable=True, unique=None, default=None)
2542 enable_locking = Column("enable_locking", Boolean(), nullable=False, unique=None, default=False)
2542 enable_locking = Column("enable_locking", Boolean(), nullable=False, unique=None, default=False)
2543 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
2543 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
2544 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
2544 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
2545 updated_on = Column('updated_on', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
2545 updated_on = Column('updated_on', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
2546 personal = Column('personal', Boolean(), nullable=True, unique=None, default=None)
2546 personal = Column('personal', Boolean(), nullable=True, unique=None, default=None)
2547 _changeset_cache = Column(
2547 _changeset_cache = Column(
2548 "changeset_cache", LargeBinary(), nullable=True) # JSON data
2548 "changeset_cache", LargeBinary(), nullable=True) # JSON data
2549
2549
2550 repo_group_to_perm = relationship('UserRepoGroupToPerm', cascade='all', order_by='UserRepoGroupToPerm.group_to_perm_id')
2550 repo_group_to_perm = relationship('UserRepoGroupToPerm', cascade='all', order_by='UserRepoGroupToPerm.group_to_perm_id')
2551 users_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
2551 users_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
2552 parent_group = relationship('RepoGroup', remote_side=group_id)
2552 parent_group = relationship('RepoGroup', remote_side=group_id)
2553 user = relationship('User')
2553 user = relationship('User')
2554 integrations = relationship('Integration', cascade="all, delete-orphan")
2554 integrations = relationship('Integration', cascade="all, delete-orphan")
2555
2555
2556 def __init__(self, group_name='', parent_group=None):
2556 def __init__(self, group_name='', parent_group=None):
2557 self.group_name = group_name
2557 self.group_name = group_name
2558 self.parent_group = parent_group
2558 self.parent_group = parent_group
2559
2559
2560 def __unicode__(self):
2560 def __unicode__(self):
2561 return u"<%s('id:%s:%s')>" % (
2561 return u"<%s('id:%s:%s')>" % (
2562 self.__class__.__name__, self.group_id, self.group_name)
2562 self.__class__.__name__, self.group_id, self.group_name)
2563
2563
2564 @hybrid_property
2564 @hybrid_property
2565 def group_name(self):
2565 def group_name(self):
2566 return self._group_name
2566 return self._group_name
2567
2567
2568 @group_name.setter
2568 @group_name.setter
2569 def group_name(self, value):
2569 def group_name(self, value):
2570 self._group_name = value
2570 self._group_name = value
2571 self.group_name_hash = self.hash_repo_group_name(value)
2571 self.group_name_hash = self.hash_repo_group_name(value)
2572
2572
2573 @hybrid_property
2573 @hybrid_property
2574 def changeset_cache(self):
2574 def changeset_cache(self):
2575 from rhodecode.lib.vcs.backends.base import EmptyCommit
2575 from rhodecode.lib.vcs.backends.base import EmptyCommit
2576 dummy = EmptyCommit().__json__()
2576 dummy = EmptyCommit().__json__()
2577 if not self._changeset_cache:
2577 if not self._changeset_cache:
2578 dummy['source_repo_id'] = ''
2578 dummy['source_repo_id'] = ''
2579 return json.loads(json.dumps(dummy))
2579 return json.loads(json.dumps(dummy))
2580
2580
2581 try:
2581 try:
2582 return json.loads(self._changeset_cache)
2582 return json.loads(self._changeset_cache)
2583 except TypeError:
2583 except TypeError:
2584 return dummy
2584 return dummy
2585 except Exception:
2585 except Exception:
2586 log.error(traceback.format_exc())
2586 log.error(traceback.format_exc())
2587 return dummy
2587 return dummy
2588
2588
2589 @changeset_cache.setter
2589 @changeset_cache.setter
2590 def changeset_cache(self, val):
2590 def changeset_cache(self, val):
2591 try:
2591 try:
2592 self._changeset_cache = json.dumps(val)
2592 self._changeset_cache = json.dumps(val)
2593 except Exception:
2593 except Exception:
2594 log.error(traceback.format_exc())
2594 log.error(traceback.format_exc())
2595
2595
2596 @validates('group_parent_id')
2596 @validates('group_parent_id')
2597 def validate_group_parent_id(self, key, val):
2597 def validate_group_parent_id(self, key, val):
2598 """
2598 """
2599 Check cycle references for a parent group to self
2599 Check cycle references for a parent group to self
2600 """
2600 """
2601 if self.group_id and val:
2601 if self.group_id and val:
2602 assert val != self.group_id
2602 assert val != self.group_id
2603
2603
2604 return val
2604 return val
2605
2605
2606 @hybrid_property
2606 @hybrid_property
2607 def description_safe(self):
2607 def description_safe(self):
2608 from rhodecode.lib import helpers as h
2608 from rhodecode.lib import helpers as h
2609 return h.escape(self.group_description)
2609 return h.escape(self.group_description)
2610
2610
2611 @classmethod
2611 @classmethod
2612 def hash_repo_group_name(cls, repo_group_name):
2612 def hash_repo_group_name(cls, repo_group_name):
2613 val = remove_formatting(repo_group_name)
2613 val = remove_formatting(repo_group_name)
2614 val = safe_str(val).lower()
2614 val = safe_str(val).lower()
2615 chars = []
2615 chars = []
2616 for c in val:
2616 for c in val:
2617 if c not in string.ascii_letters:
2617 if c not in string.ascii_letters:
2618 c = str(ord(c))
2618 c = str(ord(c))
2619 chars.append(c)
2619 chars.append(c)
2620
2620
2621 return ''.join(chars)
2621 return ''.join(chars)
2622
2622
2623 @classmethod
2623 @classmethod
2624 def _generate_choice(cls, repo_group):
2624 def _generate_choice(cls, repo_group):
2625 from webhelpers.html import literal as _literal
2625 from webhelpers.html import literal as _literal
2626 _name = lambda k: _literal(cls.CHOICES_SEPARATOR.join(k))
2626 _name = lambda k: _literal(cls.CHOICES_SEPARATOR.join(k))
2627 return repo_group.group_id, _name(repo_group.full_path_splitted)
2627 return repo_group.group_id, _name(repo_group.full_path_splitted)
2628
2628
2629 @classmethod
2629 @classmethod
2630 def groups_choices(cls, groups=None, show_empty_group=True):
2630 def groups_choices(cls, groups=None, show_empty_group=True):
2631 if not groups:
2631 if not groups:
2632 groups = cls.query().all()
2632 groups = cls.query().all()
2633
2633
2634 repo_groups = []
2634 repo_groups = []
2635 if show_empty_group:
2635 if show_empty_group:
2636 repo_groups = [(-1, u'-- %s --' % _('No parent'))]
2636 repo_groups = [(-1, u'-- %s --' % _('No parent'))]
2637
2637
2638 repo_groups.extend([cls._generate_choice(x) for x in groups])
2638 repo_groups.extend([cls._generate_choice(x) for x in groups])
2639
2639
2640 repo_groups = sorted(
2640 repo_groups = sorted(
2641 repo_groups, key=lambda t: t[1].split(cls.CHOICES_SEPARATOR)[0])
2641 repo_groups, key=lambda t: t[1].split(cls.CHOICES_SEPARATOR)[0])
2642 return repo_groups
2642 return repo_groups
2643
2643
2644 @classmethod
2644 @classmethod
2645 def url_sep(cls):
2645 def url_sep(cls):
2646 return URL_SEP
2646 return URL_SEP
2647
2647
2648 @classmethod
2648 @classmethod
2649 def get_by_group_name(cls, group_name, cache=False, case_insensitive=False):
2649 def get_by_group_name(cls, group_name, cache=False, case_insensitive=False):
2650 if case_insensitive:
2650 if case_insensitive:
2651 gr = cls.query().filter(func.lower(cls.group_name)
2651 gr = cls.query().filter(func.lower(cls.group_name)
2652 == func.lower(group_name))
2652 == func.lower(group_name))
2653 else:
2653 else:
2654 gr = cls.query().filter(cls.group_name == group_name)
2654 gr = cls.query().filter(cls.group_name == group_name)
2655 if cache:
2655 if cache:
2656 name_key = _hash_key(group_name)
2656 name_key = _hash_key(group_name)
2657 gr = gr.options(
2657 gr = gr.options(
2658 FromCache("sql_cache_short", "get_group_%s" % name_key))
2658 FromCache("sql_cache_short", "get_group_%s" % name_key))
2659 return gr.scalar()
2659 return gr.scalar()
2660
2660
2661 @classmethod
2661 @classmethod
2662 def get_user_personal_repo_group(cls, user_id):
2662 def get_user_personal_repo_group(cls, user_id):
2663 user = User.get(user_id)
2663 user = User.get(user_id)
2664 if user.username == User.DEFAULT_USER:
2664 if user.username == User.DEFAULT_USER:
2665 return None
2665 return None
2666
2666
2667 return cls.query()\
2667 return cls.query()\
2668 .filter(cls.personal == true()) \
2668 .filter(cls.personal == true()) \
2669 .filter(cls.user == user) \
2669 .filter(cls.user == user) \
2670 .order_by(cls.group_id.asc()) \
2670 .order_by(cls.group_id.asc()) \
2671 .first()
2671 .first()
2672
2672
2673 @classmethod
2673 @classmethod
2674 def get_all_repo_groups(cls, user_id=Optional(None), group_id=Optional(None),
2674 def get_all_repo_groups(cls, user_id=Optional(None), group_id=Optional(None),
2675 case_insensitive=True):
2675 case_insensitive=True):
2676 q = RepoGroup.query()
2676 q = RepoGroup.query()
2677
2677
2678 if not isinstance(user_id, Optional):
2678 if not isinstance(user_id, Optional):
2679 q = q.filter(RepoGroup.user_id == user_id)
2679 q = q.filter(RepoGroup.user_id == user_id)
2680
2680
2681 if not isinstance(group_id, Optional):
2681 if not isinstance(group_id, Optional):
2682 q = q.filter(RepoGroup.group_parent_id == group_id)
2682 q = q.filter(RepoGroup.group_parent_id == group_id)
2683
2683
2684 if case_insensitive:
2684 if case_insensitive:
2685 q = q.order_by(func.lower(RepoGroup.group_name))
2685 q = q.order_by(func.lower(RepoGroup.group_name))
2686 else:
2686 else:
2687 q = q.order_by(RepoGroup.group_name)
2687 q = q.order_by(RepoGroup.group_name)
2688 return q.all()
2688 return q.all()
2689
2689
2690 @property
2690 @property
2691 def parents(self, parents_recursion_limit = 10):
2691 def parents(self, parents_recursion_limit = 10):
2692 groups = []
2692 groups = []
2693 if self.parent_group is None:
2693 if self.parent_group is None:
2694 return groups
2694 return groups
2695 cur_gr = self.parent_group
2695 cur_gr = self.parent_group
2696 groups.insert(0, cur_gr)
2696 groups.insert(0, cur_gr)
2697 cnt = 0
2697 cnt = 0
2698 while 1:
2698 while 1:
2699 cnt += 1
2699 cnt += 1
2700 gr = getattr(cur_gr, 'parent_group', None)
2700 gr = getattr(cur_gr, 'parent_group', None)
2701 cur_gr = cur_gr.parent_group
2701 cur_gr = cur_gr.parent_group
2702 if gr is None:
2702 if gr is None:
2703 break
2703 break
2704 if cnt == parents_recursion_limit:
2704 if cnt == parents_recursion_limit:
2705 # this will prevent accidental infinit loops
2705 # this will prevent accidental infinit loops
2706 log.error('more than %s parents found for group %s, stopping '
2706 log.error('more than %s parents found for group %s, stopping '
2707 'recursive parent fetching', parents_recursion_limit, self)
2707 'recursive parent fetching', parents_recursion_limit, self)
2708 break
2708 break
2709
2709
2710 groups.insert(0, gr)
2710 groups.insert(0, gr)
2711 return groups
2711 return groups
2712
2712
2713 @property
2713 @property
2714 def last_commit_cache_update_diff(self):
2714 def last_commit_cache_update_diff(self):
2715 return time.time() - (safe_int(self.changeset_cache.get('updated_on')) or 0)
2715 return time.time() - (safe_int(self.changeset_cache.get('updated_on')) or 0)
2716
2716
2717 @property
2717 @property
2718 def last_commit_change(self):
2718 def last_commit_change(self):
2719 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2719 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2720 empty_date = datetime.datetime.fromtimestamp(0)
2720 empty_date = datetime.datetime.fromtimestamp(0)
2721 date_latest = self.changeset_cache.get('date', empty_date)
2721 date_latest = self.changeset_cache.get('date', empty_date)
2722 try:
2722 try:
2723 return parse_datetime(date_latest)
2723 return parse_datetime(date_latest)
2724 except Exception:
2724 except Exception:
2725 return empty_date
2725 return empty_date
2726
2726
2727 @property
2727 @property
2728 def last_db_change(self):
2728 def last_db_change(self):
2729 return self.updated_on
2729 return self.updated_on
2730
2730
2731 @property
2731 @property
2732 def children(self):
2732 def children(self):
2733 return RepoGroup.query().filter(RepoGroup.parent_group == self)
2733 return RepoGroup.query().filter(RepoGroup.parent_group == self)
2734
2734
2735 @property
2735 @property
2736 def name(self):
2736 def name(self):
2737 return self.group_name.split(RepoGroup.url_sep())[-1]
2737 return self.group_name.split(RepoGroup.url_sep())[-1]
2738
2738
2739 @property
2739 @property
2740 def full_path(self):
2740 def full_path(self):
2741 return self.group_name
2741 return self.group_name
2742
2742
2743 @property
2743 @property
2744 def full_path_splitted(self):
2744 def full_path_splitted(self):
2745 return self.group_name.split(RepoGroup.url_sep())
2745 return self.group_name.split(RepoGroup.url_sep())
2746
2746
2747 @property
2747 @property
2748 def repositories(self):
2748 def repositories(self):
2749 return Repository.query()\
2749 return Repository.query()\
2750 .filter(Repository.group == self)\
2750 .filter(Repository.group == self)\
2751 .order_by(Repository.repo_name)
2751 .order_by(Repository.repo_name)
2752
2752
2753 @property
2753 @property
2754 def repositories_recursive_count(self):
2754 def repositories_recursive_count(self):
2755 cnt = self.repositories.count()
2755 cnt = self.repositories.count()
2756
2756
2757 def children_count(group):
2757 def children_count(group):
2758 cnt = 0
2758 cnt = 0
2759 for child in group.children:
2759 for child in group.children:
2760 cnt += child.repositories.count()
2760 cnt += child.repositories.count()
2761 cnt += children_count(child)
2761 cnt += children_count(child)
2762 return cnt
2762 return cnt
2763
2763
2764 return cnt + children_count(self)
2764 return cnt + children_count(self)
2765
2765
2766 def _recursive_objects(self, include_repos=True, include_groups=True):
2766 def _recursive_objects(self, include_repos=True, include_groups=True):
2767 all_ = []
2767 all_ = []
2768
2768
2769 def _get_members(root_gr):
2769 def _get_members(root_gr):
2770 if include_repos:
2770 if include_repos:
2771 for r in root_gr.repositories:
2771 for r in root_gr.repositories:
2772 all_.append(r)
2772 all_.append(r)
2773 childs = root_gr.children.all()
2773 childs = root_gr.children.all()
2774 if childs:
2774 if childs:
2775 for gr in childs:
2775 for gr in childs:
2776 if include_groups:
2776 if include_groups:
2777 all_.append(gr)
2777 all_.append(gr)
2778 _get_members(gr)
2778 _get_members(gr)
2779
2779
2780 root_group = []
2780 root_group = []
2781 if include_groups:
2781 if include_groups:
2782 root_group = [self]
2782 root_group = [self]
2783
2783
2784 _get_members(self)
2784 _get_members(self)
2785 return root_group + all_
2785 return root_group + all_
2786
2786
2787 def recursive_groups_and_repos(self):
2787 def recursive_groups_and_repos(self):
2788 """
2788 """
2789 Recursive return all groups, with repositories in those groups
2789 Recursive return all groups, with repositories in those groups
2790 """
2790 """
2791 return self._recursive_objects()
2791 return self._recursive_objects()
2792
2792
2793 def recursive_groups(self):
2793 def recursive_groups(self):
2794 """
2794 """
2795 Returns all children groups for this group including children of children
2795 Returns all children groups for this group including children of children
2796 """
2796 """
2797 return self._recursive_objects(include_repos=False)
2797 return self._recursive_objects(include_repos=False)
2798
2798
2799 def recursive_repos(self):
2799 def recursive_repos(self):
2800 """
2800 """
2801 Returns all children repositories for this group
2801 Returns all children repositories for this group
2802 """
2802 """
2803 return self._recursive_objects(include_groups=False)
2803 return self._recursive_objects(include_groups=False)
2804
2804
2805 def get_new_name(self, group_name):
2805 def get_new_name(self, group_name):
2806 """
2806 """
2807 returns new full group name based on parent and new name
2807 returns new full group name based on parent and new name
2808
2808
2809 :param group_name:
2809 :param group_name:
2810 """
2810 """
2811 path_prefix = (self.parent_group.full_path_splitted if
2811 path_prefix = (self.parent_group.full_path_splitted if
2812 self.parent_group else [])
2812 self.parent_group else [])
2813 return RepoGroup.url_sep().join(path_prefix + [group_name])
2813 return RepoGroup.url_sep().join(path_prefix + [group_name])
2814
2814
2815 def update_commit_cache(self, config=None):
2815 def update_commit_cache(self, config=None):
2816 """
2816 """
2817 Update cache of last changeset for newest repository inside this group, keys should be::
2817 Update cache of last changeset for newest repository inside this group, keys should be::
2818
2818
2819 source_repo_id
2819 source_repo_id
2820 short_id
2820 short_id
2821 raw_id
2821 raw_id
2822 revision
2822 revision
2823 parents
2823 parents
2824 message
2824 message
2825 date
2825 date
2826 author
2826 author
2827
2827
2828 """
2828 """
2829 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2829 from rhodecode.lib.vcs.utils.helpers import parse_datetime
2830
2830
2831 def repo_groups_and_repos():
2831 def repo_groups_and_repos():
2832 all_entries = OrderedDefaultDict(list)
2832 all_entries = OrderedDefaultDict(list)
2833
2833
2834 def _get_members(root_gr, pos=0):
2834 def _get_members(root_gr, pos=0):
2835
2835
2836 for repo in root_gr.repositories:
2836 for repo in root_gr.repositories:
2837 all_entries[root_gr].append(repo)
2837 all_entries[root_gr].append(repo)
2838
2838
2839 # fill in all parent positions
2839 # fill in all parent positions
2840 for parent_group in root_gr.parents:
2840 for parent_group in root_gr.parents:
2841 all_entries[parent_group].extend(all_entries[root_gr])
2841 all_entries[parent_group].extend(all_entries[root_gr])
2842
2842
2843 children_groups = root_gr.children.all()
2843 children_groups = root_gr.children.all()
2844 if children_groups:
2844 if children_groups:
2845 for cnt, gr in enumerate(children_groups, 1):
2845 for cnt, gr in enumerate(children_groups, 1):
2846 _get_members(gr, pos=pos+cnt)
2846 _get_members(gr, pos=pos+cnt)
2847
2847
2848 _get_members(root_gr=self)
2848 _get_members(root_gr=self)
2849 return all_entries
2849 return all_entries
2850
2850
2851 empty_date = datetime.datetime.fromtimestamp(0)
2851 empty_date = datetime.datetime.fromtimestamp(0)
2852 for repo_group, repos in repo_groups_and_repos().items():
2852 for repo_group, repos in repo_groups_and_repos().items():
2853
2853
2854 latest_repo_cs_cache = {}
2854 latest_repo_cs_cache = {}
2855 for repo in repos:
2855 for repo in repos:
2856 repo_cs_cache = repo.changeset_cache
2856 repo_cs_cache = repo.changeset_cache
2857 date_latest = latest_repo_cs_cache.get('date', empty_date)
2857 date_latest = latest_repo_cs_cache.get('date', empty_date)
2858 date_current = repo_cs_cache.get('date', empty_date)
2858 date_current = repo_cs_cache.get('date', empty_date)
2859 current_timestamp = datetime_to_time(parse_datetime(date_latest))
2859 current_timestamp = datetime_to_time(parse_datetime(date_latest))
2860 if current_timestamp < datetime_to_time(parse_datetime(date_current)):
2860 if current_timestamp < datetime_to_time(parse_datetime(date_current)):
2861 latest_repo_cs_cache = repo_cs_cache
2861 latest_repo_cs_cache = repo_cs_cache
2862 latest_repo_cs_cache['source_repo_id'] = repo.repo_id
2862 latest_repo_cs_cache['source_repo_id'] = repo.repo_id
2863
2863
2864 latest_repo_cs_cache['updated_on'] = time.time()
2864 latest_repo_cs_cache['updated_on'] = time.time()
2865 repo_group.changeset_cache = latest_repo_cs_cache
2865 repo_group.changeset_cache = latest_repo_cs_cache
2866 Session().add(repo_group)
2866 Session().add(repo_group)
2867 Session().commit()
2867 Session().commit()
2868
2868
2869 log.debug('updated repo group %s with new commit cache %s',
2869 log.debug('updated repo group %s with new commit cache %s',
2870 repo_group.group_name, latest_repo_cs_cache)
2870 repo_group.group_name, latest_repo_cs_cache)
2871
2871
2872 def permissions(self, with_admins=True, with_owner=True,
2872 def permissions(self, with_admins=True, with_owner=True,
2873 expand_from_user_groups=False):
2873 expand_from_user_groups=False):
2874 """
2874 """
2875 Permissions for repository groups
2875 Permissions for repository groups
2876 """
2876 """
2877 _admin_perm = 'group.admin'
2877 _admin_perm = 'group.admin'
2878
2878
2879 owner_row = []
2879 owner_row = []
2880 if with_owner:
2880 if with_owner:
2881 usr = AttributeDict(self.user.get_dict())
2881 usr = AttributeDict(self.user.get_dict())
2882 usr.owner_row = True
2882 usr.owner_row = True
2883 usr.permission = _admin_perm
2883 usr.permission = _admin_perm
2884 owner_row.append(usr)
2884 owner_row.append(usr)
2885
2885
2886 super_admin_ids = []
2886 super_admin_ids = []
2887 super_admin_rows = []
2887 super_admin_rows = []
2888 if with_admins:
2888 if with_admins:
2889 for usr in User.get_all_super_admins():
2889 for usr in User.get_all_super_admins():
2890 super_admin_ids.append(usr.user_id)
2890 super_admin_ids.append(usr.user_id)
2891 # if this admin is also owner, don't double the record
2891 # if this admin is also owner, don't double the record
2892 if usr.user_id == owner_row[0].user_id:
2892 if usr.user_id == owner_row[0].user_id:
2893 owner_row[0].admin_row = True
2893 owner_row[0].admin_row = True
2894 else:
2894 else:
2895 usr = AttributeDict(usr.get_dict())
2895 usr = AttributeDict(usr.get_dict())
2896 usr.admin_row = True
2896 usr.admin_row = True
2897 usr.permission = _admin_perm
2897 usr.permission = _admin_perm
2898 super_admin_rows.append(usr)
2898 super_admin_rows.append(usr)
2899
2899
2900 q = UserRepoGroupToPerm.query().filter(UserRepoGroupToPerm.group == self)
2900 q = UserRepoGroupToPerm.query().filter(UserRepoGroupToPerm.group == self)
2901 q = q.options(joinedload(UserRepoGroupToPerm.group),
2901 q = q.options(joinedload(UserRepoGroupToPerm.group),
2902 joinedload(UserRepoGroupToPerm.user),
2902 joinedload(UserRepoGroupToPerm.user),
2903 joinedload(UserRepoGroupToPerm.permission),)
2903 joinedload(UserRepoGroupToPerm.permission),)
2904
2904
2905 # get owners and admins and permissions. We do a trick of re-writing
2905 # get owners and admins and permissions. We do a trick of re-writing
2906 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2906 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2907 # has a global reference and changing one object propagates to all
2907 # has a global reference and changing one object propagates to all
2908 # others. This means if admin is also an owner admin_row that change
2908 # others. This means if admin is also an owner admin_row that change
2909 # would propagate to both objects
2909 # would propagate to both objects
2910 perm_rows = []
2910 perm_rows = []
2911 for _usr in q.all():
2911 for _usr in q.all():
2912 usr = AttributeDict(_usr.user.get_dict())
2912 usr = AttributeDict(_usr.user.get_dict())
2913 # if this user is also owner/admin, mark as duplicate record
2913 # if this user is also owner/admin, mark as duplicate record
2914 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
2914 if usr.user_id == owner_row[0].user_id or usr.user_id in super_admin_ids:
2915 usr.duplicate_perm = True
2915 usr.duplicate_perm = True
2916 usr.permission = _usr.permission.permission_name
2916 usr.permission = _usr.permission.permission_name
2917 perm_rows.append(usr)
2917 perm_rows.append(usr)
2918
2918
2919 # filter the perm rows by 'default' first and then sort them by
2919 # filter the perm rows by 'default' first and then sort them by
2920 # admin,write,read,none permissions sorted again alphabetically in
2920 # admin,write,read,none permissions sorted again alphabetically in
2921 # each group
2921 # each group
2922 perm_rows = sorted(perm_rows, key=display_user_sort)
2922 perm_rows = sorted(perm_rows, key=display_user_sort)
2923
2923
2924 user_groups_rows = []
2924 user_groups_rows = []
2925 if expand_from_user_groups:
2925 if expand_from_user_groups:
2926 for ug in self.permission_user_groups(with_members=True):
2926 for ug in self.permission_user_groups(with_members=True):
2927 for user_data in ug.members:
2927 for user_data in ug.members:
2928 user_groups_rows.append(user_data)
2928 user_groups_rows.append(user_data)
2929
2929
2930 return super_admin_rows + owner_row + perm_rows + user_groups_rows
2930 return super_admin_rows + owner_row + perm_rows + user_groups_rows
2931
2931
2932 def permission_user_groups(self, with_members=False):
2932 def permission_user_groups(self, with_members=False):
2933 q = UserGroupRepoGroupToPerm.query()\
2933 q = UserGroupRepoGroupToPerm.query()\
2934 .filter(UserGroupRepoGroupToPerm.group == self)
2934 .filter(UserGroupRepoGroupToPerm.group == self)
2935 q = q.options(joinedload(UserGroupRepoGroupToPerm.group),
2935 q = q.options(joinedload(UserGroupRepoGroupToPerm.group),
2936 joinedload(UserGroupRepoGroupToPerm.users_group),
2936 joinedload(UserGroupRepoGroupToPerm.users_group),
2937 joinedload(UserGroupRepoGroupToPerm.permission),)
2937 joinedload(UserGroupRepoGroupToPerm.permission),)
2938
2938
2939 perm_rows = []
2939 perm_rows = []
2940 for _user_group in q.all():
2940 for _user_group in q.all():
2941 entry = AttributeDict(_user_group.users_group.get_dict())
2941 entry = AttributeDict(_user_group.users_group.get_dict())
2942 entry.permission = _user_group.permission.permission_name
2942 entry.permission = _user_group.permission.permission_name
2943 if with_members:
2943 if with_members:
2944 entry.members = [x.user.get_dict()
2944 entry.members = [x.user.get_dict()
2945 for x in _user_group.users_group.members]
2945 for x in _user_group.users_group.members]
2946 perm_rows.append(entry)
2946 perm_rows.append(entry)
2947
2947
2948 perm_rows = sorted(perm_rows, key=display_user_group_sort)
2948 perm_rows = sorted(perm_rows, key=display_user_group_sort)
2949 return perm_rows
2949 return perm_rows
2950
2950
2951 def get_api_data(self):
2951 def get_api_data(self):
2952 """
2952 """
2953 Common function for generating api data
2953 Common function for generating api data
2954
2954
2955 """
2955 """
2956 group = self
2956 group = self
2957 data = {
2957 data = {
2958 'group_id': group.group_id,
2958 'group_id': group.group_id,
2959 'group_name': group.group_name,
2959 'group_name': group.group_name,
2960 'group_description': group.description_safe,
2960 'group_description': group.description_safe,
2961 'parent_group': group.parent_group.group_name if group.parent_group else None,
2961 'parent_group': group.parent_group.group_name if group.parent_group else None,
2962 'repositories': [x.repo_name for x in group.repositories],
2962 'repositories': [x.repo_name for x in group.repositories],
2963 'owner': group.user.username,
2963 'owner': group.user.username,
2964 }
2964 }
2965 return data
2965 return data
2966
2966
2967 def get_dict(self):
2967 def get_dict(self):
2968 # Since we transformed `group_name` to a hybrid property, we need to
2968 # Since we transformed `group_name` to a hybrid property, we need to
2969 # keep compatibility with the code which uses `group_name` field.
2969 # keep compatibility with the code which uses `group_name` field.
2970 result = super(RepoGroup, self).get_dict()
2970 result = super(RepoGroup, self).get_dict()
2971 result['group_name'] = result.pop('_group_name', None)
2971 result['group_name'] = result.pop('_group_name', None)
2972 return result
2972 return result
2973
2973
2974
2974
2975 class Permission(Base, BaseModel):
2975 class Permission(Base, BaseModel):
2976 __tablename__ = 'permissions'
2976 __tablename__ = 'permissions'
2977 __table_args__ = (
2977 __table_args__ = (
2978 Index('p_perm_name_idx', 'permission_name'),
2978 Index('p_perm_name_idx', 'permission_name'),
2979 base_table_args,
2979 base_table_args,
2980 )
2980 )
2981
2981
2982 PERMS = [
2982 PERMS = [
2983 ('hg.admin', _('RhodeCode Super Administrator')),
2983 ('hg.admin', _('RhodeCode Super Administrator')),
2984
2984
2985 ('repository.none', _('Repository no access')),
2985 ('repository.none', _('Repository no access')),
2986 ('repository.read', _('Repository read access')),
2986 ('repository.read', _('Repository read access')),
2987 ('repository.write', _('Repository write access')),
2987 ('repository.write', _('Repository write access')),
2988 ('repository.admin', _('Repository admin access')),
2988 ('repository.admin', _('Repository admin access')),
2989
2989
2990 ('group.none', _('Repository group no access')),
2990 ('group.none', _('Repository group no access')),
2991 ('group.read', _('Repository group read access')),
2991 ('group.read', _('Repository group read access')),
2992 ('group.write', _('Repository group write access')),
2992 ('group.write', _('Repository group write access')),
2993 ('group.admin', _('Repository group admin access')),
2993 ('group.admin', _('Repository group admin access')),
2994
2994
2995 ('usergroup.none', _('User group no access')),
2995 ('usergroup.none', _('User group no access')),
2996 ('usergroup.read', _('User group read access')),
2996 ('usergroup.read', _('User group read access')),
2997 ('usergroup.write', _('User group write access')),
2997 ('usergroup.write', _('User group write access')),
2998 ('usergroup.admin', _('User group admin access')),
2998 ('usergroup.admin', _('User group admin access')),
2999
2999
3000 ('branch.none', _('Branch no permissions')),
3000 ('branch.none', _('Branch no permissions')),
3001 ('branch.merge', _('Branch access by web merge')),
3001 ('branch.merge', _('Branch access by web merge')),
3002 ('branch.push', _('Branch access by push')),
3002 ('branch.push', _('Branch access by push')),
3003 ('branch.push_force', _('Branch access by push with force')),
3003 ('branch.push_force', _('Branch access by push with force')),
3004
3004
3005 ('hg.repogroup.create.false', _('Repository Group creation disabled')),
3005 ('hg.repogroup.create.false', _('Repository Group creation disabled')),
3006 ('hg.repogroup.create.true', _('Repository Group creation enabled')),
3006 ('hg.repogroup.create.true', _('Repository Group creation enabled')),
3007
3007
3008 ('hg.usergroup.create.false', _('User Group creation disabled')),
3008 ('hg.usergroup.create.false', _('User Group creation disabled')),
3009 ('hg.usergroup.create.true', _('User Group creation enabled')),
3009 ('hg.usergroup.create.true', _('User Group creation enabled')),
3010
3010
3011 ('hg.create.none', _('Repository creation disabled')),
3011 ('hg.create.none', _('Repository creation disabled')),
3012 ('hg.create.repository', _('Repository creation enabled')),
3012 ('hg.create.repository', _('Repository creation enabled')),
3013 ('hg.create.write_on_repogroup.true', _('Repository creation enabled with write permission to a repository group')),
3013 ('hg.create.write_on_repogroup.true', _('Repository creation enabled with write permission to a repository group')),
3014 ('hg.create.write_on_repogroup.false', _('Repository creation disabled with write permission to a repository group')),
3014 ('hg.create.write_on_repogroup.false', _('Repository creation disabled with write permission to a repository group')),
3015
3015
3016 ('hg.fork.none', _('Repository forking disabled')),
3016 ('hg.fork.none', _('Repository forking disabled')),
3017 ('hg.fork.repository', _('Repository forking enabled')),
3017 ('hg.fork.repository', _('Repository forking enabled')),
3018
3018
3019 ('hg.register.none', _('Registration disabled')),
3019 ('hg.register.none', _('Registration disabled')),
3020 ('hg.register.manual_activate', _('User Registration with manual account activation')),
3020 ('hg.register.manual_activate', _('User Registration with manual account activation')),
3021 ('hg.register.auto_activate', _('User Registration with automatic account activation')),
3021 ('hg.register.auto_activate', _('User Registration with automatic account activation')),
3022
3022
3023 ('hg.password_reset.enabled', _('Password reset enabled')),
3023 ('hg.password_reset.enabled', _('Password reset enabled')),
3024 ('hg.password_reset.hidden', _('Password reset hidden')),
3024 ('hg.password_reset.hidden', _('Password reset hidden')),
3025 ('hg.password_reset.disabled', _('Password reset disabled')),
3025 ('hg.password_reset.disabled', _('Password reset disabled')),
3026
3026
3027 ('hg.extern_activate.manual', _('Manual activation of external account')),
3027 ('hg.extern_activate.manual', _('Manual activation of external account')),
3028 ('hg.extern_activate.auto', _('Automatic activation of external account')),
3028 ('hg.extern_activate.auto', _('Automatic activation of external account')),
3029
3029
3030 ('hg.inherit_default_perms.false', _('Inherit object permissions from default user disabled')),
3030 ('hg.inherit_default_perms.false', _('Inherit object permissions from default user disabled')),
3031 ('hg.inherit_default_perms.true', _('Inherit object permissions from default user enabled')),
3031 ('hg.inherit_default_perms.true', _('Inherit object permissions from default user enabled')),
3032 ]
3032 ]
3033
3033
3034 # definition of system default permissions for DEFAULT user, created on
3034 # definition of system default permissions for DEFAULT user, created on
3035 # system setup
3035 # system setup
3036 DEFAULT_USER_PERMISSIONS = [
3036 DEFAULT_USER_PERMISSIONS = [
3037 # object perms
3037 # object perms
3038 'repository.read',
3038 'repository.read',
3039 'group.read',
3039 'group.read',
3040 'usergroup.read',
3040 'usergroup.read',
3041 # branch, for backward compat we need same value as before so forced pushed
3041 # branch, for backward compat we need same value as before so forced pushed
3042 'branch.push_force',
3042 'branch.push_force',
3043 # global
3043 # global
3044 'hg.create.repository',
3044 'hg.create.repository',
3045 'hg.repogroup.create.false',
3045 'hg.repogroup.create.false',
3046 'hg.usergroup.create.false',
3046 'hg.usergroup.create.false',
3047 'hg.create.write_on_repogroup.true',
3047 'hg.create.write_on_repogroup.true',
3048 'hg.fork.repository',
3048 'hg.fork.repository',
3049 'hg.register.manual_activate',
3049 'hg.register.manual_activate',
3050 'hg.password_reset.enabled',
3050 'hg.password_reset.enabled',
3051 'hg.extern_activate.auto',
3051 'hg.extern_activate.auto',
3052 'hg.inherit_default_perms.true',
3052 'hg.inherit_default_perms.true',
3053 ]
3053 ]
3054
3054
3055 # defines which permissions are more important higher the more important
3055 # defines which permissions are more important higher the more important
3056 # Weight defines which permissions are more important.
3056 # Weight defines which permissions are more important.
3057 # The higher number the more important.
3057 # The higher number the more important.
3058 PERM_WEIGHTS = {
3058 PERM_WEIGHTS = {
3059 'repository.none': 0,
3059 'repository.none': 0,
3060 'repository.read': 1,
3060 'repository.read': 1,
3061 'repository.write': 3,
3061 'repository.write': 3,
3062 'repository.admin': 4,
3062 'repository.admin': 4,
3063
3063
3064 'group.none': 0,
3064 'group.none': 0,
3065 'group.read': 1,
3065 'group.read': 1,
3066 'group.write': 3,
3066 'group.write': 3,
3067 'group.admin': 4,
3067 'group.admin': 4,
3068
3068
3069 'usergroup.none': 0,
3069 'usergroup.none': 0,
3070 'usergroup.read': 1,
3070 'usergroup.read': 1,
3071 'usergroup.write': 3,
3071 'usergroup.write': 3,
3072 'usergroup.admin': 4,
3072 'usergroup.admin': 4,
3073
3073
3074 'branch.none': 0,
3074 'branch.none': 0,
3075 'branch.merge': 1,
3075 'branch.merge': 1,
3076 'branch.push': 3,
3076 'branch.push': 3,
3077 'branch.push_force': 4,
3077 'branch.push_force': 4,
3078
3078
3079 'hg.repogroup.create.false': 0,
3079 'hg.repogroup.create.false': 0,
3080 'hg.repogroup.create.true': 1,
3080 'hg.repogroup.create.true': 1,
3081
3081
3082 'hg.usergroup.create.false': 0,
3082 'hg.usergroup.create.false': 0,
3083 'hg.usergroup.create.true': 1,
3083 'hg.usergroup.create.true': 1,
3084
3084
3085 'hg.fork.none': 0,
3085 'hg.fork.none': 0,
3086 'hg.fork.repository': 1,
3086 'hg.fork.repository': 1,
3087 'hg.create.none': 0,
3087 'hg.create.none': 0,
3088 'hg.create.repository': 1
3088 'hg.create.repository': 1
3089 }
3089 }
3090
3090
3091 permission_id = Column("permission_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3091 permission_id = Column("permission_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3092 permission_name = Column("permission_name", String(255), nullable=True, unique=None, default=None)
3092 permission_name = Column("permission_name", String(255), nullable=True, unique=None, default=None)
3093 permission_longname = Column("permission_longname", String(255), nullable=True, unique=None, default=None)
3093 permission_longname = Column("permission_longname", String(255), nullable=True, unique=None, default=None)
3094
3094
3095 def __unicode__(self):
3095 def __unicode__(self):
3096 return u"<%s('%s:%s')>" % (
3096 return u"<%s('%s:%s')>" % (
3097 self.__class__.__name__, self.permission_id, self.permission_name
3097 self.__class__.__name__, self.permission_id, self.permission_name
3098 )
3098 )
3099
3099
3100 @classmethod
3100 @classmethod
3101 def get_by_key(cls, key):
3101 def get_by_key(cls, key):
3102 return cls.query().filter(cls.permission_name == key).scalar()
3102 return cls.query().filter(cls.permission_name == key).scalar()
3103
3103
3104 @classmethod
3104 @classmethod
3105 def get_default_repo_perms(cls, user_id, repo_id=None):
3105 def get_default_repo_perms(cls, user_id, repo_id=None):
3106 q = Session().query(UserRepoToPerm, Repository, Permission)\
3106 q = Session().query(UserRepoToPerm, Repository, Permission)\
3107 .join((Permission, UserRepoToPerm.permission_id == Permission.permission_id))\
3107 .join((Permission, UserRepoToPerm.permission_id == Permission.permission_id))\
3108 .join((Repository, UserRepoToPerm.repository_id == Repository.repo_id))\
3108 .join((Repository, UserRepoToPerm.repository_id == Repository.repo_id))\
3109 .filter(UserRepoToPerm.user_id == user_id)
3109 .filter(UserRepoToPerm.user_id == user_id)
3110 if repo_id:
3110 if repo_id:
3111 q = q.filter(UserRepoToPerm.repository_id == repo_id)
3111 q = q.filter(UserRepoToPerm.repository_id == repo_id)
3112 return q.all()
3112 return q.all()
3113
3113
3114 @classmethod
3114 @classmethod
3115 def get_default_repo_branch_perms(cls, user_id, repo_id=None):
3115 def get_default_repo_branch_perms(cls, user_id, repo_id=None):
3116 q = Session().query(UserToRepoBranchPermission, UserRepoToPerm, Permission) \
3116 q = Session().query(UserToRepoBranchPermission, UserRepoToPerm, Permission) \
3117 .join(
3117 .join(
3118 Permission,
3118 Permission,
3119 UserToRepoBranchPermission.permission_id == Permission.permission_id) \
3119 UserToRepoBranchPermission.permission_id == Permission.permission_id) \
3120 .join(
3120 .join(
3121 UserRepoToPerm,
3121 UserRepoToPerm,
3122 UserToRepoBranchPermission.rule_to_perm_id == UserRepoToPerm.repo_to_perm_id) \
3122 UserToRepoBranchPermission.rule_to_perm_id == UserRepoToPerm.repo_to_perm_id) \
3123 .filter(UserRepoToPerm.user_id == user_id)
3123 .filter(UserRepoToPerm.user_id == user_id)
3124
3124
3125 if repo_id:
3125 if repo_id:
3126 q = q.filter(UserToRepoBranchPermission.repository_id == repo_id)
3126 q = q.filter(UserToRepoBranchPermission.repository_id == repo_id)
3127 return q.order_by(UserToRepoBranchPermission.rule_order).all()
3127 return q.order_by(UserToRepoBranchPermission.rule_order).all()
3128
3128
3129 @classmethod
3129 @classmethod
3130 def get_default_repo_perms_from_user_group(cls, user_id, repo_id=None):
3130 def get_default_repo_perms_from_user_group(cls, user_id, repo_id=None):
3131 q = Session().query(UserGroupRepoToPerm, Repository, Permission)\
3131 q = Session().query(UserGroupRepoToPerm, Repository, Permission)\
3132 .join(
3132 .join(
3133 Permission,
3133 Permission,
3134 UserGroupRepoToPerm.permission_id == Permission.permission_id)\
3134 UserGroupRepoToPerm.permission_id == Permission.permission_id)\
3135 .join(
3135 .join(
3136 Repository,
3136 Repository,
3137 UserGroupRepoToPerm.repository_id == Repository.repo_id)\
3137 UserGroupRepoToPerm.repository_id == Repository.repo_id)\
3138 .join(
3138 .join(
3139 UserGroup,
3139 UserGroup,
3140 UserGroupRepoToPerm.users_group_id ==
3140 UserGroupRepoToPerm.users_group_id ==
3141 UserGroup.users_group_id)\
3141 UserGroup.users_group_id)\
3142 .join(
3142 .join(
3143 UserGroupMember,
3143 UserGroupMember,
3144 UserGroupRepoToPerm.users_group_id ==
3144 UserGroupRepoToPerm.users_group_id ==
3145 UserGroupMember.users_group_id)\
3145 UserGroupMember.users_group_id)\
3146 .filter(
3146 .filter(
3147 UserGroupMember.user_id == user_id,
3147 UserGroupMember.user_id == user_id,
3148 UserGroup.users_group_active == true())
3148 UserGroup.users_group_active == true())
3149 if repo_id:
3149 if repo_id:
3150 q = q.filter(UserGroupRepoToPerm.repository_id == repo_id)
3150 q = q.filter(UserGroupRepoToPerm.repository_id == repo_id)
3151 return q.all()
3151 return q.all()
3152
3152
3153 @classmethod
3153 @classmethod
3154 def get_default_repo_branch_perms_from_user_group(cls, user_id, repo_id=None):
3154 def get_default_repo_branch_perms_from_user_group(cls, user_id, repo_id=None):
3155 q = Session().query(UserGroupToRepoBranchPermission, UserGroupRepoToPerm, Permission) \
3155 q = Session().query(UserGroupToRepoBranchPermission, UserGroupRepoToPerm, Permission) \
3156 .join(
3156 .join(
3157 Permission,
3157 Permission,
3158 UserGroupToRepoBranchPermission.permission_id == Permission.permission_id) \
3158 UserGroupToRepoBranchPermission.permission_id == Permission.permission_id) \
3159 .join(
3159 .join(
3160 UserGroupRepoToPerm,
3160 UserGroupRepoToPerm,
3161 UserGroupToRepoBranchPermission.rule_to_perm_id == UserGroupRepoToPerm.users_group_to_perm_id) \
3161 UserGroupToRepoBranchPermission.rule_to_perm_id == UserGroupRepoToPerm.users_group_to_perm_id) \
3162 .join(
3162 .join(
3163 UserGroup,
3163 UserGroup,
3164 UserGroupRepoToPerm.users_group_id == UserGroup.users_group_id) \
3164 UserGroupRepoToPerm.users_group_id == UserGroup.users_group_id) \
3165 .join(
3165 .join(
3166 UserGroupMember,
3166 UserGroupMember,
3167 UserGroupRepoToPerm.users_group_id == UserGroupMember.users_group_id) \
3167 UserGroupRepoToPerm.users_group_id == UserGroupMember.users_group_id) \
3168 .filter(
3168 .filter(
3169 UserGroupMember.user_id == user_id,
3169 UserGroupMember.user_id == user_id,
3170 UserGroup.users_group_active == true())
3170 UserGroup.users_group_active == true())
3171
3171
3172 if repo_id:
3172 if repo_id:
3173 q = q.filter(UserGroupToRepoBranchPermission.repository_id == repo_id)
3173 q = q.filter(UserGroupToRepoBranchPermission.repository_id == repo_id)
3174 return q.order_by(UserGroupToRepoBranchPermission.rule_order).all()
3174 return q.order_by(UserGroupToRepoBranchPermission.rule_order).all()
3175
3175
3176 @classmethod
3176 @classmethod
3177 def get_default_group_perms(cls, user_id, repo_group_id=None):
3177 def get_default_group_perms(cls, user_id, repo_group_id=None):
3178 q = Session().query(UserRepoGroupToPerm, RepoGroup, Permission)\
3178 q = Session().query(UserRepoGroupToPerm, RepoGroup, Permission)\
3179 .join(
3179 .join(
3180 Permission,
3180 Permission,
3181 UserRepoGroupToPerm.permission_id == Permission.permission_id)\
3181 UserRepoGroupToPerm.permission_id == Permission.permission_id)\
3182 .join(
3182 .join(
3183 RepoGroup,
3183 RepoGroup,
3184 UserRepoGroupToPerm.group_id == RepoGroup.group_id)\
3184 UserRepoGroupToPerm.group_id == RepoGroup.group_id)\
3185 .filter(UserRepoGroupToPerm.user_id == user_id)
3185 .filter(UserRepoGroupToPerm.user_id == user_id)
3186 if repo_group_id:
3186 if repo_group_id:
3187 q = q.filter(UserRepoGroupToPerm.group_id == repo_group_id)
3187 q = q.filter(UserRepoGroupToPerm.group_id == repo_group_id)
3188 return q.all()
3188 return q.all()
3189
3189
3190 @classmethod
3190 @classmethod
3191 def get_default_group_perms_from_user_group(
3191 def get_default_group_perms_from_user_group(
3192 cls, user_id, repo_group_id=None):
3192 cls, user_id, repo_group_id=None):
3193 q = Session().query(UserGroupRepoGroupToPerm, RepoGroup, Permission)\
3193 q = Session().query(UserGroupRepoGroupToPerm, RepoGroup, Permission)\
3194 .join(
3194 .join(
3195 Permission,
3195 Permission,
3196 UserGroupRepoGroupToPerm.permission_id ==
3196 UserGroupRepoGroupToPerm.permission_id ==
3197 Permission.permission_id)\
3197 Permission.permission_id)\
3198 .join(
3198 .join(
3199 RepoGroup,
3199 RepoGroup,
3200 UserGroupRepoGroupToPerm.group_id == RepoGroup.group_id)\
3200 UserGroupRepoGroupToPerm.group_id == RepoGroup.group_id)\
3201 .join(
3201 .join(
3202 UserGroup,
3202 UserGroup,
3203 UserGroupRepoGroupToPerm.users_group_id ==
3203 UserGroupRepoGroupToPerm.users_group_id ==
3204 UserGroup.users_group_id)\
3204 UserGroup.users_group_id)\
3205 .join(
3205 .join(
3206 UserGroupMember,
3206 UserGroupMember,
3207 UserGroupRepoGroupToPerm.users_group_id ==
3207 UserGroupRepoGroupToPerm.users_group_id ==
3208 UserGroupMember.users_group_id)\
3208 UserGroupMember.users_group_id)\
3209 .filter(
3209 .filter(
3210 UserGroupMember.user_id == user_id,
3210 UserGroupMember.user_id == user_id,
3211 UserGroup.users_group_active == true())
3211 UserGroup.users_group_active == true())
3212 if repo_group_id:
3212 if repo_group_id:
3213 q = q.filter(UserGroupRepoGroupToPerm.group_id == repo_group_id)
3213 q = q.filter(UserGroupRepoGroupToPerm.group_id == repo_group_id)
3214 return q.all()
3214 return q.all()
3215
3215
3216 @classmethod
3216 @classmethod
3217 def get_default_user_group_perms(cls, user_id, user_group_id=None):
3217 def get_default_user_group_perms(cls, user_id, user_group_id=None):
3218 q = Session().query(UserUserGroupToPerm, UserGroup, Permission)\
3218 q = Session().query(UserUserGroupToPerm, UserGroup, Permission)\
3219 .join((Permission, UserUserGroupToPerm.permission_id == Permission.permission_id))\
3219 .join((Permission, UserUserGroupToPerm.permission_id == Permission.permission_id))\
3220 .join((UserGroup, UserUserGroupToPerm.user_group_id == UserGroup.users_group_id))\
3220 .join((UserGroup, UserUserGroupToPerm.user_group_id == UserGroup.users_group_id))\
3221 .filter(UserUserGroupToPerm.user_id == user_id)
3221 .filter(UserUserGroupToPerm.user_id == user_id)
3222 if user_group_id:
3222 if user_group_id:
3223 q = q.filter(UserUserGroupToPerm.user_group_id == user_group_id)
3223 q = q.filter(UserUserGroupToPerm.user_group_id == user_group_id)
3224 return q.all()
3224 return q.all()
3225
3225
3226 @classmethod
3226 @classmethod
3227 def get_default_user_group_perms_from_user_group(
3227 def get_default_user_group_perms_from_user_group(
3228 cls, user_id, user_group_id=None):
3228 cls, user_id, user_group_id=None):
3229 TargetUserGroup = aliased(UserGroup, name='target_user_group')
3229 TargetUserGroup = aliased(UserGroup, name='target_user_group')
3230 q = Session().query(UserGroupUserGroupToPerm, UserGroup, Permission)\
3230 q = Session().query(UserGroupUserGroupToPerm, UserGroup, Permission)\
3231 .join(
3231 .join(
3232 Permission,
3232 Permission,
3233 UserGroupUserGroupToPerm.permission_id ==
3233 UserGroupUserGroupToPerm.permission_id ==
3234 Permission.permission_id)\
3234 Permission.permission_id)\
3235 .join(
3235 .join(
3236 TargetUserGroup,
3236 TargetUserGroup,
3237 UserGroupUserGroupToPerm.target_user_group_id ==
3237 UserGroupUserGroupToPerm.target_user_group_id ==
3238 TargetUserGroup.users_group_id)\
3238 TargetUserGroup.users_group_id)\
3239 .join(
3239 .join(
3240 UserGroup,
3240 UserGroup,
3241 UserGroupUserGroupToPerm.user_group_id ==
3241 UserGroupUserGroupToPerm.user_group_id ==
3242 UserGroup.users_group_id)\
3242 UserGroup.users_group_id)\
3243 .join(
3243 .join(
3244 UserGroupMember,
3244 UserGroupMember,
3245 UserGroupUserGroupToPerm.user_group_id ==
3245 UserGroupUserGroupToPerm.user_group_id ==
3246 UserGroupMember.users_group_id)\
3246 UserGroupMember.users_group_id)\
3247 .filter(
3247 .filter(
3248 UserGroupMember.user_id == user_id,
3248 UserGroupMember.user_id == user_id,
3249 UserGroup.users_group_active == true())
3249 UserGroup.users_group_active == true())
3250 if user_group_id:
3250 if user_group_id:
3251 q = q.filter(
3251 q = q.filter(
3252 UserGroupUserGroupToPerm.user_group_id == user_group_id)
3252 UserGroupUserGroupToPerm.user_group_id == user_group_id)
3253
3253
3254 return q.all()
3254 return q.all()
3255
3255
3256
3256
3257 class UserRepoToPerm(Base, BaseModel):
3257 class UserRepoToPerm(Base, BaseModel):
3258 __tablename__ = 'repo_to_perm'
3258 __tablename__ = 'repo_to_perm'
3259 __table_args__ = (
3259 __table_args__ = (
3260 UniqueConstraint('user_id', 'repository_id', 'permission_id'),
3260 UniqueConstraint('user_id', 'repository_id', 'permission_id'),
3261 base_table_args
3261 base_table_args
3262 )
3262 )
3263
3263
3264 repo_to_perm_id = Column("repo_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3264 repo_to_perm_id = Column("repo_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3265 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3265 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3266 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3266 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3267 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
3267 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
3268
3268
3269 user = relationship('User')
3269 user = relationship('User')
3270 repository = relationship('Repository')
3270 repository = relationship('Repository')
3271 permission = relationship('Permission')
3271 permission = relationship('Permission')
3272
3272
3273 branch_perm_entry = relationship('UserToRepoBranchPermission', cascade="all, delete-orphan", lazy='joined')
3273 branch_perm_entry = relationship('UserToRepoBranchPermission', cascade="all, delete-orphan", lazy='joined')
3274
3274
3275 @classmethod
3275 @classmethod
3276 def create(cls, user, repository, permission):
3276 def create(cls, user, repository, permission):
3277 n = cls()
3277 n = cls()
3278 n.user = user
3278 n.user = user
3279 n.repository = repository
3279 n.repository = repository
3280 n.permission = permission
3280 n.permission = permission
3281 Session().add(n)
3281 Session().add(n)
3282 return n
3282 return n
3283
3283
3284 def __unicode__(self):
3284 def __unicode__(self):
3285 return u'<%s => %s >' % (self.user, self.repository)
3285 return u'<%s => %s >' % (self.user, self.repository)
3286
3286
3287
3287
3288 class UserUserGroupToPerm(Base, BaseModel):
3288 class UserUserGroupToPerm(Base, BaseModel):
3289 __tablename__ = 'user_user_group_to_perm'
3289 __tablename__ = 'user_user_group_to_perm'
3290 __table_args__ = (
3290 __table_args__ = (
3291 UniqueConstraint('user_id', 'user_group_id', 'permission_id'),
3291 UniqueConstraint('user_id', 'user_group_id', 'permission_id'),
3292 base_table_args
3292 base_table_args
3293 )
3293 )
3294
3294
3295 user_user_group_to_perm_id = Column("user_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3295 user_user_group_to_perm_id = Column("user_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3296 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3296 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3297 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3297 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3298 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3298 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3299
3299
3300 user = relationship('User')
3300 user = relationship('User')
3301 user_group = relationship('UserGroup')
3301 user_group = relationship('UserGroup')
3302 permission = relationship('Permission')
3302 permission = relationship('Permission')
3303
3303
3304 @classmethod
3304 @classmethod
3305 def create(cls, user, user_group, permission):
3305 def create(cls, user, user_group, permission):
3306 n = cls()
3306 n = cls()
3307 n.user = user
3307 n.user = user
3308 n.user_group = user_group
3308 n.user_group = user_group
3309 n.permission = permission
3309 n.permission = permission
3310 Session().add(n)
3310 Session().add(n)
3311 return n
3311 return n
3312
3312
3313 def __unicode__(self):
3313 def __unicode__(self):
3314 return u'<%s => %s >' % (self.user, self.user_group)
3314 return u'<%s => %s >' % (self.user, self.user_group)
3315
3315
3316
3316
3317 class UserToPerm(Base, BaseModel):
3317 class UserToPerm(Base, BaseModel):
3318 __tablename__ = 'user_to_perm'
3318 __tablename__ = 'user_to_perm'
3319 __table_args__ = (
3319 __table_args__ = (
3320 UniqueConstraint('user_id', 'permission_id'),
3320 UniqueConstraint('user_id', 'permission_id'),
3321 base_table_args
3321 base_table_args
3322 )
3322 )
3323
3323
3324 user_to_perm_id = Column("user_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3324 user_to_perm_id = Column("user_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3325 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3325 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3326 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3326 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3327
3327
3328 user = relationship('User')
3328 user = relationship('User')
3329 permission = relationship('Permission', lazy='joined')
3329 permission = relationship('Permission', lazy='joined')
3330
3330
3331 def __unicode__(self):
3331 def __unicode__(self):
3332 return u'<%s => %s >' % (self.user, self.permission)
3332 return u'<%s => %s >' % (self.user, self.permission)
3333
3333
3334
3334
3335 class UserGroupRepoToPerm(Base, BaseModel):
3335 class UserGroupRepoToPerm(Base, BaseModel):
3336 __tablename__ = 'users_group_repo_to_perm'
3336 __tablename__ = 'users_group_repo_to_perm'
3337 __table_args__ = (
3337 __table_args__ = (
3338 UniqueConstraint('repository_id', 'users_group_id', 'permission_id'),
3338 UniqueConstraint('repository_id', 'users_group_id', 'permission_id'),
3339 base_table_args
3339 base_table_args
3340 )
3340 )
3341
3341
3342 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3342 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3343 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3343 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3344 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3344 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3345 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
3345 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
3346
3346
3347 users_group = relationship('UserGroup')
3347 users_group = relationship('UserGroup')
3348 permission = relationship('Permission')
3348 permission = relationship('Permission')
3349 repository = relationship('Repository')
3349 repository = relationship('Repository')
3350 user_group_branch_perms = relationship('UserGroupToRepoBranchPermission', cascade='all')
3350 user_group_branch_perms = relationship('UserGroupToRepoBranchPermission', cascade='all')
3351
3351
3352 @classmethod
3352 @classmethod
3353 def create(cls, users_group, repository, permission):
3353 def create(cls, users_group, repository, permission):
3354 n = cls()
3354 n = cls()
3355 n.users_group = users_group
3355 n.users_group = users_group
3356 n.repository = repository
3356 n.repository = repository
3357 n.permission = permission
3357 n.permission = permission
3358 Session().add(n)
3358 Session().add(n)
3359 return n
3359 return n
3360
3360
3361 def __unicode__(self):
3361 def __unicode__(self):
3362 return u'<UserGroupRepoToPerm:%s => %s >' % (self.users_group, self.repository)
3362 return u'<UserGroupRepoToPerm:%s => %s >' % (self.users_group, self.repository)
3363
3363
3364
3364
3365 class UserGroupUserGroupToPerm(Base, BaseModel):
3365 class UserGroupUserGroupToPerm(Base, BaseModel):
3366 __tablename__ = 'user_group_user_group_to_perm'
3366 __tablename__ = 'user_group_user_group_to_perm'
3367 __table_args__ = (
3367 __table_args__ = (
3368 UniqueConstraint('target_user_group_id', 'user_group_id', 'permission_id'),
3368 UniqueConstraint('target_user_group_id', 'user_group_id', 'permission_id'),
3369 CheckConstraint('target_user_group_id != user_group_id'),
3369 CheckConstraint('target_user_group_id != user_group_id'),
3370 base_table_args
3370 base_table_args
3371 )
3371 )
3372
3372
3373 user_group_user_group_to_perm_id = Column("user_group_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3373 user_group_user_group_to_perm_id = Column("user_group_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3374 target_user_group_id = Column("target_user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3374 target_user_group_id = Column("target_user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3375 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3375 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3376 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3376 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3377
3377
3378 target_user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id')
3378 target_user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id')
3379 user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.user_group_id==UserGroup.users_group_id')
3379 user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.user_group_id==UserGroup.users_group_id')
3380 permission = relationship('Permission')
3380 permission = relationship('Permission')
3381
3381
3382 @classmethod
3382 @classmethod
3383 def create(cls, target_user_group, user_group, permission):
3383 def create(cls, target_user_group, user_group, permission):
3384 n = cls()
3384 n = cls()
3385 n.target_user_group = target_user_group
3385 n.target_user_group = target_user_group
3386 n.user_group = user_group
3386 n.user_group = user_group
3387 n.permission = permission
3387 n.permission = permission
3388 Session().add(n)
3388 Session().add(n)
3389 return n
3389 return n
3390
3390
3391 def __unicode__(self):
3391 def __unicode__(self):
3392 return u'<UserGroupUserGroup:%s => %s >' % (self.target_user_group, self.user_group)
3392 return u'<UserGroupUserGroup:%s => %s >' % (self.target_user_group, self.user_group)
3393
3393
3394
3394
3395 class UserGroupToPerm(Base, BaseModel):
3395 class UserGroupToPerm(Base, BaseModel):
3396 __tablename__ = 'users_group_to_perm'
3396 __tablename__ = 'users_group_to_perm'
3397 __table_args__ = (
3397 __table_args__ = (
3398 UniqueConstraint('users_group_id', 'permission_id',),
3398 UniqueConstraint('users_group_id', 'permission_id',),
3399 base_table_args
3399 base_table_args
3400 )
3400 )
3401
3401
3402 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3402 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3403 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3403 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3404 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3404 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3405
3405
3406 users_group = relationship('UserGroup')
3406 users_group = relationship('UserGroup')
3407 permission = relationship('Permission')
3407 permission = relationship('Permission')
3408
3408
3409
3409
3410 class UserRepoGroupToPerm(Base, BaseModel):
3410 class UserRepoGroupToPerm(Base, BaseModel):
3411 __tablename__ = 'user_repo_group_to_perm'
3411 __tablename__ = 'user_repo_group_to_perm'
3412 __table_args__ = (
3412 __table_args__ = (
3413 UniqueConstraint('user_id', 'group_id', 'permission_id'),
3413 UniqueConstraint('user_id', 'group_id', 'permission_id'),
3414 base_table_args
3414 base_table_args
3415 )
3415 )
3416
3416
3417 group_to_perm_id = Column("group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3417 group_to_perm_id = Column("group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3418 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3418 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3419 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
3419 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
3420 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3420 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3421
3421
3422 user = relationship('User')
3422 user = relationship('User')
3423 group = relationship('RepoGroup')
3423 group = relationship('RepoGroup')
3424 permission = relationship('Permission')
3424 permission = relationship('Permission')
3425
3425
3426 @classmethod
3426 @classmethod
3427 def create(cls, user, repository_group, permission):
3427 def create(cls, user, repository_group, permission):
3428 n = cls()
3428 n = cls()
3429 n.user = user
3429 n.user = user
3430 n.group = repository_group
3430 n.group = repository_group
3431 n.permission = permission
3431 n.permission = permission
3432 Session().add(n)
3432 Session().add(n)
3433 return n
3433 return n
3434
3434
3435
3435
3436 class UserGroupRepoGroupToPerm(Base, BaseModel):
3436 class UserGroupRepoGroupToPerm(Base, BaseModel):
3437 __tablename__ = 'users_group_repo_group_to_perm'
3437 __tablename__ = 'users_group_repo_group_to_perm'
3438 __table_args__ = (
3438 __table_args__ = (
3439 UniqueConstraint('users_group_id', 'group_id'),
3439 UniqueConstraint('users_group_id', 'group_id'),
3440 base_table_args
3440 base_table_args
3441 )
3441 )
3442
3442
3443 users_group_repo_group_to_perm_id = Column("users_group_repo_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3443 users_group_repo_group_to_perm_id = Column("users_group_repo_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3444 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3444 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
3445 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
3445 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
3446 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3446 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
3447
3447
3448 users_group = relationship('UserGroup')
3448 users_group = relationship('UserGroup')
3449 permission = relationship('Permission')
3449 permission = relationship('Permission')
3450 group = relationship('RepoGroup')
3450 group = relationship('RepoGroup')
3451
3451
3452 @classmethod
3452 @classmethod
3453 def create(cls, user_group, repository_group, permission):
3453 def create(cls, user_group, repository_group, permission):
3454 n = cls()
3454 n = cls()
3455 n.users_group = user_group
3455 n.users_group = user_group
3456 n.group = repository_group
3456 n.group = repository_group
3457 n.permission = permission
3457 n.permission = permission
3458 Session().add(n)
3458 Session().add(n)
3459 return n
3459 return n
3460
3460
3461 def __unicode__(self):
3461 def __unicode__(self):
3462 return u'<UserGroupRepoGroupToPerm:%s => %s >' % (self.users_group, self.group)
3462 return u'<UserGroupRepoGroupToPerm:%s => %s >' % (self.users_group, self.group)
3463
3463
3464
3464
3465 class Statistics(Base, BaseModel):
3465 class Statistics(Base, BaseModel):
3466 __tablename__ = 'statistics'
3466 __tablename__ = 'statistics'
3467 __table_args__ = (
3467 __table_args__ = (
3468 base_table_args
3468 base_table_args
3469 )
3469 )
3470
3470
3471 stat_id = Column("stat_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3471 stat_id = Column("stat_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3472 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=True, default=None)
3472 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=True, default=None)
3473 stat_on_revision = Column("stat_on_revision", Integer(), nullable=False)
3473 stat_on_revision = Column("stat_on_revision", Integer(), nullable=False)
3474 commit_activity = Column("commit_activity", LargeBinary(1000000), nullable=False)#JSON data
3474 commit_activity = Column("commit_activity", LargeBinary(1000000), nullable=False)#JSON data
3475 commit_activity_combined = Column("commit_activity_combined", LargeBinary(), nullable=False)#JSON data
3475 commit_activity_combined = Column("commit_activity_combined", LargeBinary(), nullable=False)#JSON data
3476 languages = Column("languages", LargeBinary(1000000), nullable=False)#JSON data
3476 languages = Column("languages", LargeBinary(1000000), nullable=False)#JSON data
3477
3477
3478 repository = relationship('Repository', single_parent=True)
3478 repository = relationship('Repository', single_parent=True)
3479
3479
3480
3480
3481 class UserFollowing(Base, BaseModel):
3481 class UserFollowing(Base, BaseModel):
3482 __tablename__ = 'user_followings'
3482 __tablename__ = 'user_followings'
3483 __table_args__ = (
3483 __table_args__ = (
3484 UniqueConstraint('user_id', 'follows_repository_id'),
3484 UniqueConstraint('user_id', 'follows_repository_id'),
3485 UniqueConstraint('user_id', 'follows_user_id'),
3485 UniqueConstraint('user_id', 'follows_user_id'),
3486 base_table_args
3486 base_table_args
3487 )
3487 )
3488
3488
3489 user_following_id = Column("user_following_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3489 user_following_id = Column("user_following_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3490 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3490 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
3491 follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True, unique=None, default=None)
3491 follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True, unique=None, default=None)
3492 follows_user_id = Column("follows_user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
3492 follows_user_id = Column("follows_user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
3493 follows_from = Column('follows_from', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
3493 follows_from = Column('follows_from', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
3494
3494
3495 user = relationship('User', primaryjoin='User.user_id==UserFollowing.user_id')
3495 user = relationship('User', primaryjoin='User.user_id==UserFollowing.user_id')
3496
3496
3497 follows_user = relationship('User', primaryjoin='User.user_id==UserFollowing.follows_user_id')
3497 follows_user = relationship('User', primaryjoin='User.user_id==UserFollowing.follows_user_id')
3498 follows_repository = relationship('Repository', order_by='Repository.repo_name')
3498 follows_repository = relationship('Repository', order_by='Repository.repo_name')
3499
3499
3500 @classmethod
3500 @classmethod
3501 def get_repo_followers(cls, repo_id):
3501 def get_repo_followers(cls, repo_id):
3502 return cls.query().filter(cls.follows_repo_id == repo_id)
3502 return cls.query().filter(cls.follows_repo_id == repo_id)
3503
3503
3504
3504
3505 class CacheKey(Base, BaseModel):
3505 class CacheKey(Base, BaseModel):
3506 __tablename__ = 'cache_invalidation'
3506 __tablename__ = 'cache_invalidation'
3507 __table_args__ = (
3507 __table_args__ = (
3508 UniqueConstraint('cache_key'),
3508 UniqueConstraint('cache_key'),
3509 Index('key_idx', 'cache_key'),
3509 Index('key_idx', 'cache_key'),
3510 base_table_args,
3510 base_table_args,
3511 )
3511 )
3512
3512
3513 CACHE_TYPE_FEED = 'FEED'
3513 CACHE_TYPE_FEED = 'FEED'
3514
3514
3515 # namespaces used to register process/thread aware caches
3515 # namespaces used to register process/thread aware caches
3516 REPO_INVALIDATION_NAMESPACE = 'repo_cache:{repo_id}'
3516 REPO_INVALIDATION_NAMESPACE = 'repo_cache:{repo_id}'
3517 SETTINGS_INVALIDATION_NAMESPACE = 'system_settings'
3517 SETTINGS_INVALIDATION_NAMESPACE = 'system_settings'
3518
3518
3519 cache_id = Column("cache_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3519 cache_id = Column("cache_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
3520 cache_key = Column("cache_key", String(255), nullable=True, unique=None, default=None)
3520 cache_key = Column("cache_key", String(255), nullable=True, unique=None, default=None)
3521 cache_args = Column("cache_args", String(255), nullable=True, unique=None, default=None)
3521 cache_args = Column("cache_args", String(255), nullable=True, unique=None, default=None)
3522 cache_state_uid = Column("cache_state_uid", String(255), nullable=True, unique=None, default=None)
3522 cache_state_uid = Column("cache_state_uid", String(255), nullable=True, unique=None, default=None)
3523 cache_active = Column("cache_active", Boolean(), nullable=True, unique=None, default=False)
3523 cache_active = Column("cache_active", Boolean(), nullable=True, unique=None, default=False)
3524
3524
3525 def __init__(self, cache_key, cache_args='', cache_state_uid=None):
3525 def __init__(self, cache_key, cache_args='', cache_state_uid=None):
3526 self.cache_key = cache_key
3526 self.cache_key = cache_key
3527 self.cache_args = cache_args
3527 self.cache_args = cache_args
3528 self.cache_active = False
3528 self.cache_active = False
3529 # first key should be same for all entries, since all workers should share it
3529 # first key should be same for all entries, since all workers should share it
3530 self.cache_state_uid = cache_state_uid or self.generate_new_state_uid()
3530 self.cache_state_uid = cache_state_uid or self.generate_new_state_uid()
3531
3531
3532 def __unicode__(self):
3532 def __unicode__(self):
3533 return u"<%s('%s:%s[%s]')>" % (
3533 return u"<%s('%s:%s[%s]')>" % (
3534 self.__class__.__name__,
3534 self.__class__.__name__,
3535 self.cache_id, self.cache_key, self.cache_active)
3535 self.cache_id, self.cache_key, self.cache_active)
3536
3536
3537 def _cache_key_partition(self):
3537 def _cache_key_partition(self):
3538 prefix, repo_name, suffix = self.cache_key.partition(self.cache_args)
3538 prefix, repo_name, suffix = self.cache_key.partition(self.cache_args)
3539 return prefix, repo_name, suffix
3539 return prefix, repo_name, suffix
3540
3540
3541 def get_prefix(self):
3541 def get_prefix(self):
3542 """
3542 """
3543 Try to extract prefix from existing cache key. The key could consist
3543 Try to extract prefix from existing cache key. The key could consist
3544 of prefix, repo_name, suffix
3544 of prefix, repo_name, suffix
3545 """
3545 """
3546 # this returns prefix, repo_name, suffix
3546 # this returns prefix, repo_name, suffix
3547 return self._cache_key_partition()[0]
3547 return self._cache_key_partition()[0]
3548
3548
3549 def get_suffix(self):
3549 def get_suffix(self):
3550 """
3550 """
3551 get suffix that might have been used in _get_cache_key to
3551 get suffix that might have been used in _get_cache_key to
3552 generate self.cache_key. Only used for informational purposes
3552 generate self.cache_key. Only used for informational purposes
3553 in repo_edit.mako.
3553 in repo_edit.mako.
3554 """
3554 """
3555 # prefix, repo_name, suffix
3555 # prefix, repo_name, suffix
3556 return self._cache_key_partition()[2]
3556 return self._cache_key_partition()[2]
3557
3557
3558 @classmethod
3558 @classmethod
3559 def generate_new_state_uid(cls, based_on=None):
3559 def generate_new_state_uid(cls, based_on=None):
3560 if based_on:
3560 if based_on:
3561 return str(uuid.uuid5(uuid.NAMESPACE_URL, safe_str(based_on)))
3561 return str(uuid.uuid5(uuid.NAMESPACE_URL, safe_str(based_on)))
3562 else:
3562 else:
3563 return str(uuid.uuid4())
3563 return str(uuid.uuid4())
3564
3564
3565 @classmethod
3565 @classmethod
3566 def delete_all_cache(cls):
3566 def delete_all_cache(cls):
3567 """
3567 """
3568 Delete all cache keys from database.
3568 Delete all cache keys from database.
3569 Should only be run when all instances are down and all entries
3569 Should only be run when all instances are down and all entries
3570 thus stale.
3570 thus stale.
3571 """
3571 """
3572 cls.query().delete()
3572 cls.query().delete()
3573 Session().commit()
3573 Session().commit()
3574
3574
3575 @classmethod
3575 @classmethod
3576 def set_invalidate(cls, cache_uid, delete=False):
3576 def set_invalidate(cls, cache_uid, delete=False):
3577 """
3577 """
3578 Mark all caches of a repo as invalid in the database.
3578 Mark all caches of a repo as invalid in the database.
3579 """
3579 """
3580
3580
3581 try:
3581 try:
3582 qry = Session().query(cls).filter(cls.cache_args == cache_uid)
3582 qry = Session().query(cls).filter(cls.cache_args == cache_uid)
3583 if delete:
3583 if delete:
3584 qry.delete()
3584 qry.delete()
3585 log.debug('cache objects deleted for cache args %s',
3585 log.debug('cache objects deleted for cache args %s',
3586 safe_str(cache_uid))
3586 safe_str(cache_uid))
3587 else:
3587 else:
3588 qry.update({"cache_active": False,
3588 qry.update({"cache_active": False,
3589 "cache_state_uid": cls.generate_new_state_uid()})
3589 "cache_state_uid": cls.generate_new_state_uid()})
3590 log.debug('cache objects marked as invalid for cache args %s',
3590 log.debug('cache objects marked as invalid for cache args %s',
3591 safe_str(cache_uid))
3591 safe_str(cache_uid))
3592
3592
3593 Session().commit()
3593 Session().commit()
3594 except Exception:
3594 except Exception:
3595 log.exception(
3595 log.exception(
3596 'Cache key invalidation failed for cache args %s',
3596 'Cache key invalidation failed for cache args %s',
3597 safe_str(cache_uid))
3597 safe_str(cache_uid))
3598 Session().rollback()
3598 Session().rollback()
3599
3599
3600 @classmethod
3600 @classmethod
3601 def get_active_cache(cls, cache_key):
3601 def get_active_cache(cls, cache_key):
3602 inv_obj = cls.query().filter(cls.cache_key == cache_key).scalar()
3602 inv_obj = cls.query().filter(cls.cache_key == cache_key).scalar()
3603 if inv_obj:
3603 if inv_obj:
3604 return inv_obj
3604 return inv_obj
3605 return None
3605 return None
3606
3606
3607 @classmethod
3607 @classmethod
3608 def get_namespace_map(cls, namespace):
3608 def get_namespace_map(cls, namespace):
3609 return {
3609 return {
3610 x.cache_key: x
3610 x.cache_key: x
3611 for x in cls.query().filter(cls.cache_args == namespace)}
3611 for x in cls.query().filter(cls.cache_args == namespace)}
3612
3612
3613
3613
3614 class ChangesetComment(Base, BaseModel):
3614 class ChangesetComment(Base, BaseModel):
3615 __tablename__ = 'changeset_comments'
3615 __tablename__ = 'changeset_comments'
3616 __table_args__ = (
3616 __table_args__ = (
3617 Index('cc_revision_idx', 'revision'),
3617 Index('cc_revision_idx', 'revision'),
3618 base_table_args,
3618 base_table_args,
3619 )
3619 )
3620
3620
3621 COMMENT_OUTDATED = u'comment_outdated'
3621 COMMENT_OUTDATED = u'comment_outdated'
3622 COMMENT_TYPE_NOTE = u'note'
3622 COMMENT_TYPE_NOTE = u'note'
3623 COMMENT_TYPE_TODO = u'todo'
3623 COMMENT_TYPE_TODO = u'todo'
3624 COMMENT_TYPES = [COMMENT_TYPE_NOTE, COMMENT_TYPE_TODO]
3624 COMMENT_TYPES = [COMMENT_TYPE_NOTE, COMMENT_TYPE_TODO]
3625
3625
3626 comment_id = Column('comment_id', Integer(), nullable=False, primary_key=True)
3626 comment_id = Column('comment_id', Integer(), nullable=False, primary_key=True)
3627 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3627 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3628 revision = Column('revision', String(40), nullable=True)
3628 revision = Column('revision', String(40), nullable=True)
3629 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3629 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3630 pull_request_version_id = Column("pull_request_version_id", Integer(), ForeignKey('pull_request_versions.pull_request_version_id'), nullable=True)
3630 pull_request_version_id = Column("pull_request_version_id", Integer(), ForeignKey('pull_request_versions.pull_request_version_id'), nullable=True)
3631 line_no = Column('line_no', Unicode(10), nullable=True)
3631 line_no = Column('line_no', Unicode(10), nullable=True)
3632 hl_lines = Column('hl_lines', Unicode(512), nullable=True)
3632 hl_lines = Column('hl_lines', Unicode(512), nullable=True)
3633 f_path = Column('f_path', Unicode(1000), nullable=True)
3633 f_path = Column('f_path', Unicode(1000), nullable=True)
3634 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
3634 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
3635 text = Column('text', UnicodeText().with_variant(UnicodeText(25000), 'mysql'), nullable=False)
3635 text = Column('text', UnicodeText().with_variant(UnicodeText(25000), 'mysql'), nullable=False)
3636 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3636 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3637 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3637 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3638 renderer = Column('renderer', Unicode(64), nullable=True)
3638 renderer = Column('renderer', Unicode(64), nullable=True)
3639 display_state = Column('display_state', Unicode(128), nullable=True)
3639 display_state = Column('display_state', Unicode(128), nullable=True)
3640
3640
3641 comment_type = Column('comment_type', Unicode(128), nullable=True, default=COMMENT_TYPE_NOTE)
3641 comment_type = Column('comment_type', Unicode(128), nullable=True, default=COMMENT_TYPE_NOTE)
3642 resolved_comment_id = Column('resolved_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'), nullable=True)
3642 resolved_comment_id = Column('resolved_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'), nullable=True)
3643
3643
3644 resolved_comment = relationship('ChangesetComment', remote_side=comment_id, back_populates='resolved_by')
3644 resolved_comment = relationship('ChangesetComment', remote_side=comment_id, back_populates='resolved_by')
3645 resolved_by = relationship('ChangesetComment', back_populates='resolved_comment')
3645 resolved_by = relationship('ChangesetComment', back_populates='resolved_comment')
3646
3646
3647 author = relationship('User', lazy='joined')
3647 author = relationship('User', lazy='joined')
3648 repo = relationship('Repository')
3648 repo = relationship('Repository')
3649 status_change = relationship('ChangesetStatus', cascade="all, delete-orphan", lazy='joined')
3649 status_change = relationship('ChangesetStatus', cascade="all, delete-orphan", lazy='joined')
3650 pull_request = relationship('PullRequest', lazy='joined')
3650 pull_request = relationship('PullRequest', lazy='joined')
3651 pull_request_version = relationship('PullRequestVersion')
3651 pull_request_version = relationship('PullRequestVersion')
3652
3652
3653 @classmethod
3653 @classmethod
3654 def get_users(cls, revision=None, pull_request_id=None):
3654 def get_users(cls, revision=None, pull_request_id=None):
3655 """
3655 """
3656 Returns user associated with this ChangesetComment. ie those
3656 Returns user associated with this ChangesetComment. ie those
3657 who actually commented
3657 who actually commented
3658
3658
3659 :param cls:
3659 :param cls:
3660 :param revision:
3660 :param revision:
3661 """
3661 """
3662 q = Session().query(User)\
3662 q = Session().query(User)\
3663 .join(ChangesetComment.author)
3663 .join(ChangesetComment.author)
3664 if revision:
3664 if revision:
3665 q = q.filter(cls.revision == revision)
3665 q = q.filter(cls.revision == revision)
3666 elif pull_request_id:
3666 elif pull_request_id:
3667 q = q.filter(cls.pull_request_id == pull_request_id)
3667 q = q.filter(cls.pull_request_id == pull_request_id)
3668 return q.all()
3668 return q.all()
3669
3669
3670 @classmethod
3670 @classmethod
3671 def get_index_from_version(cls, pr_version, versions):
3671 def get_index_from_version(cls, pr_version, versions):
3672 num_versions = [x.pull_request_version_id for x in versions]
3672 num_versions = [x.pull_request_version_id for x in versions]
3673 try:
3673 try:
3674 return num_versions.index(pr_version) +1
3674 return num_versions.index(pr_version) +1
3675 except (IndexError, ValueError):
3675 except (IndexError, ValueError):
3676 return
3676 return
3677
3677
3678 @property
3678 @property
3679 def outdated(self):
3679 def outdated(self):
3680 return self.display_state == self.COMMENT_OUTDATED
3680 return self.display_state == self.COMMENT_OUTDATED
3681
3681
3682 def outdated_at_version(self, version):
3682 def outdated_at_version(self, version):
3683 """
3683 """
3684 Checks if comment is outdated for given pull request version
3684 Checks if comment is outdated for given pull request version
3685 """
3685 """
3686 return self.outdated and self.pull_request_version_id != version
3686 return self.outdated and self.pull_request_version_id != version
3687
3687
3688 def older_than_version(self, version):
3688 def older_than_version(self, version):
3689 """
3689 """
3690 Checks if comment is made from previous version than given
3690 Checks if comment is made from previous version than given
3691 """
3691 """
3692 if version is None:
3692 if version is None:
3693 return self.pull_request_version_id is not None
3693 return self.pull_request_version_id is not None
3694
3694
3695 return self.pull_request_version_id < version
3695 return self.pull_request_version_id < version
3696
3696
3697 @property
3697 @property
3698 def resolved(self):
3698 def resolved(self):
3699 return self.resolved_by[0] if self.resolved_by else None
3699 return self.resolved_by[0] if self.resolved_by else None
3700
3700
3701 @property
3701 @property
3702 def is_todo(self):
3702 def is_todo(self):
3703 return self.comment_type == self.COMMENT_TYPE_TODO
3703 return self.comment_type == self.COMMENT_TYPE_TODO
3704
3704
3705 @property
3705 @property
3706 def is_inline(self):
3706 def is_inline(self):
3707 return self.line_no and self.f_path
3707 return self.line_no and self.f_path
3708
3708
3709 def get_index_version(self, versions):
3709 def get_index_version(self, versions):
3710 return self.get_index_from_version(
3710 return self.get_index_from_version(
3711 self.pull_request_version_id, versions)
3711 self.pull_request_version_id, versions)
3712
3712
3713 def __repr__(self):
3713 def __repr__(self):
3714 if self.comment_id:
3714 if self.comment_id:
3715 return '<DB:Comment #%s>' % self.comment_id
3715 return '<DB:Comment #%s>' % self.comment_id
3716 else:
3716 else:
3717 return '<DB:Comment at %#x>' % id(self)
3717 return '<DB:Comment at %#x>' % id(self)
3718
3718
3719 def get_api_data(self):
3719 def get_api_data(self):
3720 comment = self
3720 comment = self
3721 data = {
3721 data = {
3722 'comment_id': comment.comment_id,
3722 'comment_id': comment.comment_id,
3723 'comment_type': comment.comment_type,
3723 'comment_type': comment.comment_type,
3724 'comment_text': comment.text,
3724 'comment_text': comment.text,
3725 'comment_status': comment.status_change,
3725 'comment_status': comment.status_change,
3726 'comment_f_path': comment.f_path,
3726 'comment_f_path': comment.f_path,
3727 'comment_lineno': comment.line_no,
3727 'comment_lineno': comment.line_no,
3728 'comment_author': comment.author,
3728 'comment_author': comment.author,
3729 'comment_created_on': comment.created_on,
3729 'comment_created_on': comment.created_on,
3730 'comment_resolved_by': self.resolved
3730 'comment_resolved_by': self.resolved
3731 }
3731 }
3732 return data
3732 return data
3733
3733
3734 def __json__(self):
3734 def __json__(self):
3735 data = dict()
3735 data = dict()
3736 data.update(self.get_api_data())
3736 data.update(self.get_api_data())
3737 return data
3737 return data
3738
3738
3739
3739
3740 class ChangesetStatus(Base, BaseModel):
3740 class ChangesetStatus(Base, BaseModel):
3741 __tablename__ = 'changeset_statuses'
3741 __tablename__ = 'changeset_statuses'
3742 __table_args__ = (
3742 __table_args__ = (
3743 Index('cs_revision_idx', 'revision'),
3743 Index('cs_revision_idx', 'revision'),
3744 Index('cs_version_idx', 'version'),
3744 Index('cs_version_idx', 'version'),
3745 UniqueConstraint('repo_id', 'revision', 'version'),
3745 UniqueConstraint('repo_id', 'revision', 'version'),
3746 base_table_args
3746 base_table_args
3747 )
3747 )
3748
3748
3749 STATUS_NOT_REVIEWED = DEFAULT = 'not_reviewed'
3749 STATUS_NOT_REVIEWED = DEFAULT = 'not_reviewed'
3750 STATUS_APPROVED = 'approved'
3750 STATUS_APPROVED = 'approved'
3751 STATUS_REJECTED = 'rejected'
3751 STATUS_REJECTED = 'rejected'
3752 STATUS_UNDER_REVIEW = 'under_review'
3752 STATUS_UNDER_REVIEW = 'under_review'
3753
3753
3754 STATUSES = [
3754 STATUSES = [
3755 (STATUS_NOT_REVIEWED, _("Not Reviewed")), # (no icon) and default
3755 (STATUS_NOT_REVIEWED, _("Not Reviewed")), # (no icon) and default
3756 (STATUS_APPROVED, _("Approved")),
3756 (STATUS_APPROVED, _("Approved")),
3757 (STATUS_REJECTED, _("Rejected")),
3757 (STATUS_REJECTED, _("Rejected")),
3758 (STATUS_UNDER_REVIEW, _("Under Review")),
3758 (STATUS_UNDER_REVIEW, _("Under Review")),
3759 ]
3759 ]
3760
3760
3761 changeset_status_id = Column('changeset_status_id', Integer(), nullable=False, primary_key=True)
3761 changeset_status_id = Column('changeset_status_id', Integer(), nullable=False, primary_key=True)
3762 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3762 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3763 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None)
3763 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None)
3764 revision = Column('revision', String(40), nullable=False)
3764 revision = Column('revision', String(40), nullable=False)
3765 status = Column('status', String(128), nullable=False, default=DEFAULT)
3765 status = Column('status', String(128), nullable=False, default=DEFAULT)
3766 changeset_comment_id = Column('changeset_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'))
3766 changeset_comment_id = Column('changeset_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'))
3767 modified_at = Column('modified_at', DateTime(), nullable=False, default=datetime.datetime.now)
3767 modified_at = Column('modified_at', DateTime(), nullable=False, default=datetime.datetime.now)
3768 version = Column('version', Integer(), nullable=False, default=0)
3768 version = Column('version', Integer(), nullable=False, default=0)
3769 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3769 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3770
3770
3771 author = relationship('User', lazy='joined')
3771 author = relationship('User', lazy='joined')
3772 repo = relationship('Repository')
3772 repo = relationship('Repository')
3773 comment = relationship('ChangesetComment', lazy='joined')
3773 comment = relationship('ChangesetComment', lazy='joined')
3774 pull_request = relationship('PullRequest', lazy='joined')
3774 pull_request = relationship('PullRequest', lazy='joined')
3775
3775
3776 def __unicode__(self):
3776 def __unicode__(self):
3777 return u"<%s('%s[v%s]:%s')>" % (
3777 return u"<%s('%s[v%s]:%s')>" % (
3778 self.__class__.__name__,
3778 self.__class__.__name__,
3779 self.status, self.version, self.author
3779 self.status, self.version, self.author
3780 )
3780 )
3781
3781
3782 @classmethod
3782 @classmethod
3783 def get_status_lbl(cls, value):
3783 def get_status_lbl(cls, value):
3784 return dict(cls.STATUSES).get(value)
3784 return dict(cls.STATUSES).get(value)
3785
3785
3786 @property
3786 @property
3787 def status_lbl(self):
3787 def status_lbl(self):
3788 return ChangesetStatus.get_status_lbl(self.status)
3788 return ChangesetStatus.get_status_lbl(self.status)
3789
3789
3790 def get_api_data(self):
3790 def get_api_data(self):
3791 status = self
3791 status = self
3792 data = {
3792 data = {
3793 'status_id': status.changeset_status_id,
3793 'status_id': status.changeset_status_id,
3794 'status': status.status,
3794 'status': status.status,
3795 }
3795 }
3796 return data
3796 return data
3797
3797
3798 def __json__(self):
3798 def __json__(self):
3799 data = dict()
3799 data = dict()
3800 data.update(self.get_api_data())
3800 data.update(self.get_api_data())
3801 return data
3801 return data
3802
3802
3803
3803
3804 class _SetState(object):
3804 class _SetState(object):
3805 """
3805 """
3806 Context processor allowing changing state for sensitive operation such as
3806 Context processor allowing changing state for sensitive operation such as
3807 pull request update or merge
3807 pull request update or merge
3808 """
3808 """
3809
3809
3810 def __init__(self, pull_request, pr_state, back_state=None):
3810 def __init__(self, pull_request, pr_state, back_state=None):
3811 self._pr = pull_request
3811 self._pr = pull_request
3812 self._org_state = back_state or pull_request.pull_request_state
3812 self._org_state = back_state or pull_request.pull_request_state
3813 self._pr_state = pr_state
3813 self._pr_state = pr_state
3814 self._current_state = None
3814 self._current_state = None
3815
3815
3816 def __enter__(self):
3816 def __enter__(self):
3817 log.debug('StateLock: entering set state context, setting state to: `%s`',
3817 log.debug('StateLock: entering set state context, setting state to: `%s`',
3818 self._pr_state)
3818 self._pr_state)
3819 self.set_pr_state(self._pr_state)
3819 self.set_pr_state(self._pr_state)
3820 return self
3820 return self
3821
3821
3822 def __exit__(self, exc_type, exc_val, exc_tb):
3822 def __exit__(self, exc_type, exc_val, exc_tb):
3823 if exc_val is not None:
3823 if exc_val is not None:
3824 log.error(traceback.format_exc(exc_tb))
3824 log.error(traceback.format_exc(exc_tb))
3825 return None
3825 return None
3826
3826
3827 self.set_pr_state(self._org_state)
3827 self.set_pr_state(self._org_state)
3828 log.debug('StateLock: exiting set state context, setting state to: `%s`',
3828 log.debug('StateLock: exiting set state context, setting state to: `%s`',
3829 self._org_state)
3829 self._org_state)
3830 @property
3830 @property
3831 def state(self):
3831 def state(self):
3832 return self._current_state
3832 return self._current_state
3833
3833
3834 def set_pr_state(self, pr_state):
3834 def set_pr_state(self, pr_state):
3835 try:
3835 try:
3836 self._pr.pull_request_state = pr_state
3836 self._pr.pull_request_state = pr_state
3837 Session().add(self._pr)
3837 Session().add(self._pr)
3838 Session().commit()
3838 Session().commit()
3839 self._current_state = pr_state
3839 self._current_state = pr_state
3840 except Exception:
3840 except Exception:
3841 log.exception('Failed to set PullRequest %s state to %s', self._pr, pr_state)
3841 log.exception('Failed to set PullRequest %s state to %s', self._pr, pr_state)
3842 raise
3842 raise
3843
3843
3844 class _PullRequestBase(BaseModel):
3844 class _PullRequestBase(BaseModel):
3845 """
3845 """
3846 Common attributes of pull request and version entries.
3846 Common attributes of pull request and version entries.
3847 """
3847 """
3848
3848
3849 # .status values
3849 # .status values
3850 STATUS_NEW = u'new'
3850 STATUS_NEW = u'new'
3851 STATUS_OPEN = u'open'
3851 STATUS_OPEN = u'open'
3852 STATUS_CLOSED = u'closed'
3852 STATUS_CLOSED = u'closed'
3853
3853
3854 # available states
3854 # available states
3855 STATE_CREATING = u'creating'
3855 STATE_CREATING = u'creating'
3856 STATE_UPDATING = u'updating'
3856 STATE_UPDATING = u'updating'
3857 STATE_MERGING = u'merging'
3857 STATE_MERGING = u'merging'
3858 STATE_CREATED = u'created'
3858 STATE_CREATED = u'created'
3859
3859
3860 title = Column('title', Unicode(255), nullable=True)
3860 title = Column('title', Unicode(255), nullable=True)
3861 description = Column(
3861 description = Column(
3862 'description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'),
3862 'description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'),
3863 nullable=True)
3863 nullable=True)
3864 description_renderer = Column('description_renderer', Unicode(64), nullable=True)
3864 description_renderer = Column('description_renderer', Unicode(64), nullable=True)
3865
3865
3866 # new/open/closed status of pull request (not approve/reject/etc)
3866 # new/open/closed status of pull request (not approve/reject/etc)
3867 status = Column('status', Unicode(255), nullable=False, default=STATUS_NEW)
3867 status = Column('status', Unicode(255), nullable=False, default=STATUS_NEW)
3868 created_on = Column(
3868 created_on = Column(
3869 'created_on', DateTime(timezone=False), nullable=False,
3869 'created_on', DateTime(timezone=False), nullable=False,
3870 default=datetime.datetime.now)
3870 default=datetime.datetime.now)
3871 updated_on = Column(
3871 updated_on = Column(
3872 'updated_on', DateTime(timezone=False), nullable=False,
3872 'updated_on', DateTime(timezone=False), nullable=False,
3873 default=datetime.datetime.now)
3873 default=datetime.datetime.now)
3874
3874
3875 pull_request_state = Column("pull_request_state", String(255), nullable=True)
3875 pull_request_state = Column("pull_request_state", String(255), nullable=True)
3876
3876
3877 @declared_attr
3877 @declared_attr
3878 def user_id(cls):
3878 def user_id(cls):
3879 return Column(
3879 return Column(
3880 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
3880 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
3881 unique=None)
3881 unique=None)
3882
3882
3883 # 500 revisions max
3883 # 500 revisions max
3884 _revisions = Column(
3884 _revisions = Column(
3885 'revisions', UnicodeText().with_variant(UnicodeText(20500), 'mysql'))
3885 'revisions', UnicodeText().with_variant(UnicodeText(20500), 'mysql'))
3886
3886
3887 @declared_attr
3887 @declared_attr
3888 def source_repo_id(cls):
3888 def source_repo_id(cls):
3889 # TODO: dan: rename column to source_repo_id
3889 # TODO: dan: rename column to source_repo_id
3890 return Column(
3890 return Column(
3891 'org_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3891 'org_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3892 nullable=False)
3892 nullable=False)
3893
3893
3894 _source_ref = Column('org_ref', Unicode(255), nullable=False)
3894 _source_ref = Column('org_ref', Unicode(255), nullable=False)
3895
3895
3896 @hybrid_property
3896 @hybrid_property
3897 def source_ref(self):
3897 def source_ref(self):
3898 return self._source_ref
3898 return self._source_ref
3899
3899
3900 @source_ref.setter
3900 @source_ref.setter
3901 def source_ref(self, val):
3901 def source_ref(self, val):
3902 parts = (val or '').split(':')
3902 parts = (val or '').split(':')
3903 if len(parts) != 3:
3903 if len(parts) != 3:
3904 raise ValueError(
3904 raise ValueError(
3905 'Invalid reference format given: {}, expected X:Y:Z'.format(val))
3905 'Invalid reference format given: {}, expected X:Y:Z'.format(val))
3906 self._source_ref = safe_unicode(val)
3906 self._source_ref = safe_unicode(val)
3907
3907
3908 _target_ref = Column('other_ref', Unicode(255), nullable=False)
3908 _target_ref = Column('other_ref', Unicode(255), nullable=False)
3909
3909
3910 @hybrid_property
3910 @hybrid_property
3911 def target_ref(self):
3911 def target_ref(self):
3912 return self._target_ref
3912 return self._target_ref
3913
3913
3914 @target_ref.setter
3914 @target_ref.setter
3915 def target_ref(self, val):
3915 def target_ref(self, val):
3916 parts = (val or '').split(':')
3916 parts = (val or '').split(':')
3917 if len(parts) != 3:
3917 if len(parts) != 3:
3918 raise ValueError(
3918 raise ValueError(
3919 'Invalid reference format given: {}, expected X:Y:Z'.format(val))
3919 'Invalid reference format given: {}, expected X:Y:Z'.format(val))
3920 self._target_ref = safe_unicode(val)
3920 self._target_ref = safe_unicode(val)
3921
3921
3922 @declared_attr
3922 @declared_attr
3923 def target_repo_id(cls):
3923 def target_repo_id(cls):
3924 # TODO: dan: rename column to target_repo_id
3924 # TODO: dan: rename column to target_repo_id
3925 return Column(
3925 return Column(
3926 'other_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3926 'other_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3927 nullable=False)
3927 nullable=False)
3928
3928
3929 _shadow_merge_ref = Column('shadow_merge_ref', Unicode(255), nullable=True)
3929 _shadow_merge_ref = Column('shadow_merge_ref', Unicode(255), nullable=True)
3930
3930
3931 # TODO: dan: rename column to last_merge_source_rev
3931 # TODO: dan: rename column to last_merge_source_rev
3932 _last_merge_source_rev = Column(
3932 _last_merge_source_rev = Column(
3933 'last_merge_org_rev', String(40), nullable=True)
3933 'last_merge_org_rev', String(40), nullable=True)
3934 # TODO: dan: rename column to last_merge_target_rev
3934 # TODO: dan: rename column to last_merge_target_rev
3935 _last_merge_target_rev = Column(
3935 _last_merge_target_rev = Column(
3936 'last_merge_other_rev', String(40), nullable=True)
3936 'last_merge_other_rev', String(40), nullable=True)
3937 _last_merge_status = Column('merge_status', Integer(), nullable=True)
3937 _last_merge_status = Column('merge_status', Integer(), nullable=True)
3938 merge_rev = Column('merge_rev', String(40), nullable=True)
3938 merge_rev = Column('merge_rev', String(40), nullable=True)
3939
3939
3940 reviewer_data = Column(
3940 reviewer_data = Column(
3941 'reviewer_data_json', MutationObj.as_mutable(
3941 'reviewer_data_json', MutationObj.as_mutable(
3942 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3942 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3943
3943
3944 @property
3944 @property
3945 def reviewer_data_json(self):
3945 def reviewer_data_json(self):
3946 return json.dumps(self.reviewer_data)
3946 return json.dumps(self.reviewer_data)
3947
3947
3948 @hybrid_property
3948 @hybrid_property
3949 def description_safe(self):
3949 def description_safe(self):
3950 from rhodecode.lib import helpers as h
3950 from rhodecode.lib import helpers as h
3951 return h.escape(self.description)
3951 return h.escape(self.description)
3952
3952
3953 @hybrid_property
3953 @hybrid_property
3954 def revisions(self):
3954 def revisions(self):
3955 return self._revisions.split(':') if self._revisions else []
3955 return self._revisions.split(':') if self._revisions else []
3956
3956
3957 @revisions.setter
3957 @revisions.setter
3958 def revisions(self, val):
3958 def revisions(self, val):
3959 self._revisions = u':'.join(val)
3959 self._revisions = u':'.join(val)
3960
3960
3961 @hybrid_property
3961 @hybrid_property
3962 def last_merge_status(self):
3962 def last_merge_status(self):
3963 return safe_int(self._last_merge_status)
3963 return safe_int(self._last_merge_status)
3964
3964
3965 @last_merge_status.setter
3965 @last_merge_status.setter
3966 def last_merge_status(self, val):
3966 def last_merge_status(self, val):
3967 self._last_merge_status = val
3967 self._last_merge_status = val
3968
3968
3969 @declared_attr
3969 @declared_attr
3970 def author(cls):
3970 def author(cls):
3971 return relationship('User', lazy='joined')
3971 return relationship('User', lazy='joined')
3972
3972
3973 @declared_attr
3973 @declared_attr
3974 def source_repo(cls):
3974 def source_repo(cls):
3975 return relationship(
3975 return relationship(
3976 'Repository',
3976 'Repository',
3977 primaryjoin='%s.source_repo_id==Repository.repo_id' % cls.__name__)
3977 primaryjoin='%s.source_repo_id==Repository.repo_id' % cls.__name__)
3978
3978
3979 @property
3979 @property
3980 def source_ref_parts(self):
3980 def source_ref_parts(self):
3981 return self.unicode_to_reference(self.source_ref)
3981 return self.unicode_to_reference(self.source_ref)
3982
3982
3983 @declared_attr
3983 @declared_attr
3984 def target_repo(cls):
3984 def target_repo(cls):
3985 return relationship(
3985 return relationship(
3986 'Repository',
3986 'Repository',
3987 primaryjoin='%s.target_repo_id==Repository.repo_id' % cls.__name__)
3987 primaryjoin='%s.target_repo_id==Repository.repo_id' % cls.__name__)
3988
3988
3989 @property
3989 @property
3990 def target_ref_parts(self):
3990 def target_ref_parts(self):
3991 return self.unicode_to_reference(self.target_ref)
3991 return self.unicode_to_reference(self.target_ref)
3992
3992
3993 @property
3993 @property
3994 def shadow_merge_ref(self):
3994 def shadow_merge_ref(self):
3995 return self.unicode_to_reference(self._shadow_merge_ref)
3995 return self.unicode_to_reference(self._shadow_merge_ref)
3996
3996
3997 @shadow_merge_ref.setter
3997 @shadow_merge_ref.setter
3998 def shadow_merge_ref(self, ref):
3998 def shadow_merge_ref(self, ref):
3999 self._shadow_merge_ref = self.reference_to_unicode(ref)
3999 self._shadow_merge_ref = self.reference_to_unicode(ref)
4000
4000
4001 @staticmethod
4001 @staticmethod
4002 def unicode_to_reference(raw):
4002 def unicode_to_reference(raw):
4003 """
4003 """
4004 Convert a unicode (or string) to a reference object.
4004 Convert a unicode (or string) to a reference object.
4005 If unicode evaluates to False it returns None.
4005 If unicode evaluates to False it returns None.
4006 """
4006 """
4007 if raw:
4007 if raw:
4008 refs = raw.split(':')
4008 refs = raw.split(':')
4009 return Reference(*refs)
4009 return Reference(*refs)
4010 else:
4010 else:
4011 return None
4011 return None
4012
4012
4013 @staticmethod
4013 @staticmethod
4014 def reference_to_unicode(ref):
4014 def reference_to_unicode(ref):
4015 """
4015 """
4016 Convert a reference object to unicode.
4016 Convert a reference object to unicode.
4017 If reference is None it returns None.
4017 If reference is None it returns None.
4018 """
4018 """
4019 if ref:
4019 if ref:
4020 return u':'.join(ref)
4020 return u':'.join(ref)
4021 else:
4021 else:
4022 return None
4022 return None
4023
4023
4024 def get_api_data(self, with_merge_state=True):
4024 def get_api_data(self, with_merge_state=True):
4025 from rhodecode.model.pull_request import PullRequestModel
4025 from rhodecode.model.pull_request import PullRequestModel
4026
4026
4027 pull_request = self
4027 pull_request = self
4028 if with_merge_state:
4028 if with_merge_state:
4029 merge_status = PullRequestModel().merge_status(pull_request)
4029 merge_status = PullRequestModel().merge_status(pull_request)
4030 merge_state = {
4030 merge_state = {
4031 'status': merge_status[0],
4031 'status': merge_status[0],
4032 'message': safe_unicode(merge_status[1]),
4032 'message': safe_unicode(merge_status[1]),
4033 }
4033 }
4034 else:
4034 else:
4035 merge_state = {'status': 'not_available',
4035 merge_state = {'status': 'not_available',
4036 'message': 'not_available'}
4036 'message': 'not_available'}
4037
4037
4038 merge_data = {
4038 merge_data = {
4039 'clone_url': PullRequestModel().get_shadow_clone_url(pull_request),
4039 'clone_url': PullRequestModel().get_shadow_clone_url(pull_request),
4040 'reference': (
4040 'reference': (
4041 pull_request.shadow_merge_ref._asdict()
4041 pull_request.shadow_merge_ref._asdict()
4042 if pull_request.shadow_merge_ref else None),
4042 if pull_request.shadow_merge_ref else None),
4043 }
4043 }
4044
4044
4045 data = {
4045 data = {
4046 'pull_request_id': pull_request.pull_request_id,
4046 'pull_request_id': pull_request.pull_request_id,
4047 'url': PullRequestModel().get_url(pull_request),
4047 'url': PullRequestModel().get_url(pull_request),
4048 'title': pull_request.title,
4048 'title': pull_request.title,
4049 'description': pull_request.description,
4049 'description': pull_request.description,
4050 'status': pull_request.status,
4050 'status': pull_request.status,
4051 'state': pull_request.pull_request_state,
4051 'state': pull_request.pull_request_state,
4052 'created_on': pull_request.created_on,
4052 'created_on': pull_request.created_on,
4053 'updated_on': pull_request.updated_on,
4053 'updated_on': pull_request.updated_on,
4054 'commit_ids': pull_request.revisions,
4054 'commit_ids': pull_request.revisions,
4055 'review_status': pull_request.calculated_review_status(),
4055 'review_status': pull_request.calculated_review_status(),
4056 'mergeable': merge_state,
4056 'mergeable': merge_state,
4057 'source': {
4057 'source': {
4058 'clone_url': pull_request.source_repo.clone_url(),
4058 'clone_url': pull_request.source_repo.clone_url(),
4059 'repository': pull_request.source_repo.repo_name,
4059 'repository': pull_request.source_repo.repo_name,
4060 'reference': {
4060 'reference': {
4061 'name': pull_request.source_ref_parts.name,
4061 'name': pull_request.source_ref_parts.name,
4062 'type': pull_request.source_ref_parts.type,
4062 'type': pull_request.source_ref_parts.type,
4063 'commit_id': pull_request.source_ref_parts.commit_id,
4063 'commit_id': pull_request.source_ref_parts.commit_id,
4064 },
4064 },
4065 },
4065 },
4066 'target': {
4066 'target': {
4067 'clone_url': pull_request.target_repo.clone_url(),
4067 'clone_url': pull_request.target_repo.clone_url(),
4068 'repository': pull_request.target_repo.repo_name,
4068 'repository': pull_request.target_repo.repo_name,
4069 'reference': {
4069 'reference': {
4070 'name': pull_request.target_ref_parts.name,
4070 'name': pull_request.target_ref_parts.name,
4071 'type': pull_request.target_ref_parts.type,
4071 'type': pull_request.target_ref_parts.type,
4072 'commit_id': pull_request.target_ref_parts.commit_id,
4072 'commit_id': pull_request.target_ref_parts.commit_id,
4073 },
4073 },
4074 },
4074 },
4075 'merge': merge_data,
4075 'merge': merge_data,
4076 'author': pull_request.author.get_api_data(include_secrets=False,
4076 'author': pull_request.author.get_api_data(include_secrets=False,
4077 details='basic'),
4077 details='basic'),
4078 'reviewers': [
4078 'reviewers': [
4079 {
4079 {
4080 'user': reviewer.get_api_data(include_secrets=False,
4080 'user': reviewer.get_api_data(include_secrets=False,
4081 details='basic'),
4081 details='basic'),
4082 'reasons': reasons,
4082 'reasons': reasons,
4083 'review_status': st[0][1].status if st else 'not_reviewed',
4083 'review_status': st[0][1].status if st else 'not_reviewed',
4084 }
4084 }
4085 for obj, reviewer, reasons, mandatory, st in
4085 for obj, reviewer, reasons, mandatory, st in
4086 pull_request.reviewers_statuses()
4086 pull_request.reviewers_statuses()
4087 ]
4087 ]
4088 }
4088 }
4089
4089
4090 return data
4090 return data
4091
4091
4092 def set_state(self, pull_request_state, final_state=None):
4092 def set_state(self, pull_request_state, final_state=None):
4093 """
4093 """
4094 # goes from initial state to updating to initial state.
4094 # goes from initial state to updating to initial state.
4095 # initial state can be changed by specifying back_state=
4095 # initial state can be changed by specifying back_state=
4096 with pull_request_obj.set_state(PullRequest.STATE_UPDATING):
4096 with pull_request_obj.set_state(PullRequest.STATE_UPDATING):
4097 pull_request.merge()
4097 pull_request.merge()
4098
4098
4099 :param pull_request_state:
4099 :param pull_request_state:
4100 :param final_state:
4100 :param final_state:
4101
4101
4102 """
4102 """
4103
4103
4104 return _SetState(self, pull_request_state, back_state=final_state)
4104 return _SetState(self, pull_request_state, back_state=final_state)
4105
4105
4106
4106
4107 class PullRequest(Base, _PullRequestBase):
4107 class PullRequest(Base, _PullRequestBase):
4108 __tablename__ = 'pull_requests'
4108 __tablename__ = 'pull_requests'
4109 __table_args__ = (
4109 __table_args__ = (
4110 base_table_args,
4110 base_table_args,
4111 )
4111 )
4112
4112
4113 pull_request_id = Column(
4113 pull_request_id = Column(
4114 'pull_request_id', Integer(), nullable=False, primary_key=True)
4114 'pull_request_id', Integer(), nullable=False, primary_key=True)
4115
4115
4116 def __repr__(self):
4116 def __repr__(self):
4117 if self.pull_request_id:
4117 if self.pull_request_id:
4118 return '<DB:PullRequest #%s>' % self.pull_request_id
4118 return '<DB:PullRequest #%s>' % self.pull_request_id
4119 else:
4119 else:
4120 return '<DB:PullRequest at %#x>' % id(self)
4120 return '<DB:PullRequest at %#x>' % id(self)
4121
4121
4122 reviewers = relationship('PullRequestReviewers',
4122 reviewers = relationship('PullRequestReviewers',
4123 cascade="all, delete-orphan")
4123 cascade="all, delete-orphan")
4124 statuses = relationship('ChangesetStatus',
4124 statuses = relationship('ChangesetStatus',
4125 cascade="all, delete-orphan")
4125 cascade="all, delete-orphan")
4126 comments = relationship('ChangesetComment',
4126 comments = relationship('ChangesetComment',
4127 cascade="all, delete-orphan")
4127 cascade="all, delete-orphan")
4128 versions = relationship('PullRequestVersion',
4128 versions = relationship('PullRequestVersion',
4129 cascade="all, delete-orphan",
4129 cascade="all, delete-orphan",
4130 lazy='dynamic')
4130 lazy='dynamic')
4131
4131
4132 @classmethod
4132 @classmethod
4133 def get_pr_display_object(cls, pull_request_obj, org_pull_request_obj,
4133 def get_pr_display_object(cls, pull_request_obj, org_pull_request_obj,
4134 internal_methods=None):
4134 internal_methods=None):
4135
4135
4136 class PullRequestDisplay(object):
4136 class PullRequestDisplay(object):
4137 """
4137 """
4138 Special object wrapper for showing PullRequest data via Versions
4138 Special object wrapper for showing PullRequest data via Versions
4139 It mimics PR object as close as possible. This is read only object
4139 It mimics PR object as close as possible. This is read only object
4140 just for display
4140 just for display
4141 """
4141 """
4142
4142
4143 def __init__(self, attrs, internal=None):
4143 def __init__(self, attrs, internal=None):
4144 self.attrs = attrs
4144 self.attrs = attrs
4145 # internal have priority over the given ones via attrs
4145 # internal have priority over the given ones via attrs
4146 self.internal = internal or ['versions']
4146 self.internal = internal or ['versions']
4147
4147
4148 def __getattr__(self, item):
4148 def __getattr__(self, item):
4149 if item in self.internal:
4149 if item in self.internal:
4150 return getattr(self, item)
4150 return getattr(self, item)
4151 try:
4151 try:
4152 return self.attrs[item]
4152 return self.attrs[item]
4153 except KeyError:
4153 except KeyError:
4154 raise AttributeError(
4154 raise AttributeError(
4155 '%s object has no attribute %s' % (self, item))
4155 '%s object has no attribute %s' % (self, item))
4156
4156
4157 def __repr__(self):
4157 def __repr__(self):
4158 return '<DB:PullRequestDisplay #%s>' % self.attrs.get('pull_request_id')
4158 return '<DB:PullRequestDisplay #%s>' % self.attrs.get('pull_request_id')
4159
4159
4160 def versions(self):
4160 def versions(self):
4161 return pull_request_obj.versions.order_by(
4161 return pull_request_obj.versions.order_by(
4162 PullRequestVersion.pull_request_version_id).all()
4162 PullRequestVersion.pull_request_version_id).all()
4163
4163
4164 def is_closed(self):
4164 def is_closed(self):
4165 return pull_request_obj.is_closed()
4165 return pull_request_obj.is_closed()
4166
4166
4167 @property
4167 @property
4168 def pull_request_version_id(self):
4168 def pull_request_version_id(self):
4169 return getattr(pull_request_obj, 'pull_request_version_id', None)
4169 return getattr(pull_request_obj, 'pull_request_version_id', None)
4170
4170
4171 attrs = StrictAttributeDict(pull_request_obj.get_api_data(with_merge_state=False))
4171 attrs = StrictAttributeDict(pull_request_obj.get_api_data(with_merge_state=False))
4172
4172
4173 attrs.author = StrictAttributeDict(
4173 attrs.author = StrictAttributeDict(
4174 pull_request_obj.author.get_api_data())
4174 pull_request_obj.author.get_api_data())
4175 if pull_request_obj.target_repo:
4175 if pull_request_obj.target_repo:
4176 attrs.target_repo = StrictAttributeDict(
4176 attrs.target_repo = StrictAttributeDict(
4177 pull_request_obj.target_repo.get_api_data())
4177 pull_request_obj.target_repo.get_api_data())
4178 attrs.target_repo.clone_url = pull_request_obj.target_repo.clone_url
4178 attrs.target_repo.clone_url = pull_request_obj.target_repo.clone_url
4179
4179
4180 if pull_request_obj.source_repo:
4180 if pull_request_obj.source_repo:
4181 attrs.source_repo = StrictAttributeDict(
4181 attrs.source_repo = StrictAttributeDict(
4182 pull_request_obj.source_repo.get_api_data())
4182 pull_request_obj.source_repo.get_api_data())
4183 attrs.source_repo.clone_url = pull_request_obj.source_repo.clone_url
4183 attrs.source_repo.clone_url = pull_request_obj.source_repo.clone_url
4184
4184
4185 attrs.source_ref_parts = pull_request_obj.source_ref_parts
4185 attrs.source_ref_parts = pull_request_obj.source_ref_parts
4186 attrs.target_ref_parts = pull_request_obj.target_ref_parts
4186 attrs.target_ref_parts = pull_request_obj.target_ref_parts
4187 attrs.revisions = pull_request_obj.revisions
4187 attrs.revisions = pull_request_obj.revisions
4188
4188
4189 attrs.shadow_merge_ref = org_pull_request_obj.shadow_merge_ref
4189 attrs.shadow_merge_ref = org_pull_request_obj.shadow_merge_ref
4190 attrs.reviewer_data = org_pull_request_obj.reviewer_data
4190 attrs.reviewer_data = org_pull_request_obj.reviewer_data
4191 attrs.reviewer_data_json = org_pull_request_obj.reviewer_data_json
4191 attrs.reviewer_data_json = org_pull_request_obj.reviewer_data_json
4192
4192
4193 return PullRequestDisplay(attrs, internal=internal_methods)
4193 return PullRequestDisplay(attrs, internal=internal_methods)
4194
4194
4195 def is_closed(self):
4195 def is_closed(self):
4196 return self.status == self.STATUS_CLOSED
4196 return self.status == self.STATUS_CLOSED
4197
4197
4198 def __json__(self):
4198 def __json__(self):
4199 return {
4199 return {
4200 'revisions': self.revisions,
4200 'revisions': self.revisions,
4201 }
4201 }
4202
4202
4203 def calculated_review_status(self):
4203 def calculated_review_status(self):
4204 from rhodecode.model.changeset_status import ChangesetStatusModel
4204 from rhodecode.model.changeset_status import ChangesetStatusModel
4205 return ChangesetStatusModel().calculated_review_status(self)
4205 return ChangesetStatusModel().calculated_review_status(self)
4206
4206
4207 def reviewers_statuses(self):
4207 def reviewers_statuses(self):
4208 from rhodecode.model.changeset_status import ChangesetStatusModel
4208 from rhodecode.model.changeset_status import ChangesetStatusModel
4209 return ChangesetStatusModel().reviewers_statuses(self)
4209 return ChangesetStatusModel().reviewers_statuses(self)
4210
4210
4211 @property
4211 @property
4212 def workspace_id(self):
4212 def workspace_id(self):
4213 from rhodecode.model.pull_request import PullRequestModel
4213 from rhodecode.model.pull_request import PullRequestModel
4214 return PullRequestModel()._workspace_id(self)
4214 return PullRequestModel()._workspace_id(self)
4215
4215
4216 def get_shadow_repo(self):
4216 def get_shadow_repo(self):
4217 workspace_id = self.workspace_id
4217 workspace_id = self.workspace_id
4218 shadow_repository_path = self.target_repo.get_shadow_repository_path(workspace_id)
4218 shadow_repository_path = self.target_repo.get_shadow_repository_path(workspace_id)
4219 if os.path.isdir(shadow_repository_path):
4219 if os.path.isdir(shadow_repository_path):
4220 vcs_obj = self.target_repo.scm_instance()
4220 vcs_obj = self.target_repo.scm_instance()
4221 return vcs_obj.get_shadow_instance(shadow_repository_path)
4221 return vcs_obj.get_shadow_instance(shadow_repository_path)
4222
4222
4223
4223
4224 class PullRequestVersion(Base, _PullRequestBase):
4224 class PullRequestVersion(Base, _PullRequestBase):
4225 __tablename__ = 'pull_request_versions'
4225 __tablename__ = 'pull_request_versions'
4226 __table_args__ = (
4226 __table_args__ = (
4227 base_table_args,
4227 base_table_args,
4228 )
4228 )
4229
4229
4230 pull_request_version_id = Column(
4230 pull_request_version_id = Column(
4231 'pull_request_version_id', Integer(), nullable=False, primary_key=True)
4231 'pull_request_version_id', Integer(), nullable=False, primary_key=True)
4232 pull_request_id = Column(
4232 pull_request_id = Column(
4233 'pull_request_id', Integer(),
4233 'pull_request_id', Integer(),
4234 ForeignKey('pull_requests.pull_request_id'), nullable=False)
4234 ForeignKey('pull_requests.pull_request_id'), nullable=False)
4235 pull_request = relationship('PullRequest')
4235 pull_request = relationship('PullRequest')
4236
4236
4237 def __repr__(self):
4237 def __repr__(self):
4238 if self.pull_request_version_id:
4238 if self.pull_request_version_id:
4239 return '<DB:PullRequestVersion #%s>' % self.pull_request_version_id
4239 return '<DB:PullRequestVersion #%s>' % self.pull_request_version_id
4240 else:
4240 else:
4241 return '<DB:PullRequestVersion at %#x>' % id(self)
4241 return '<DB:PullRequestVersion at %#x>' % id(self)
4242
4242
4243 @property
4243 @property
4244 def reviewers(self):
4244 def reviewers(self):
4245 return self.pull_request.reviewers
4245 return self.pull_request.reviewers
4246
4246
4247 @property
4247 @property
4248 def versions(self):
4248 def versions(self):
4249 return self.pull_request.versions
4249 return self.pull_request.versions
4250
4250
4251 def is_closed(self):
4251 def is_closed(self):
4252 # calculate from original
4252 # calculate from original
4253 return self.pull_request.status == self.STATUS_CLOSED
4253 return self.pull_request.status == self.STATUS_CLOSED
4254
4254
4255 def calculated_review_status(self):
4255 def calculated_review_status(self):
4256 return self.pull_request.calculated_review_status()
4256 return self.pull_request.calculated_review_status()
4257
4257
4258 def reviewers_statuses(self):
4258 def reviewers_statuses(self):
4259 return self.pull_request.reviewers_statuses()
4259 return self.pull_request.reviewers_statuses()
4260
4260
4261
4261
4262 class PullRequestReviewers(Base, BaseModel):
4262 class PullRequestReviewers(Base, BaseModel):
4263 __tablename__ = 'pull_request_reviewers'
4263 __tablename__ = 'pull_request_reviewers'
4264 __table_args__ = (
4264 __table_args__ = (
4265 base_table_args,
4265 base_table_args,
4266 )
4266 )
4267
4267
4268 @hybrid_property
4268 @hybrid_property
4269 def reasons(self):
4269 def reasons(self):
4270 if not self._reasons:
4270 if not self._reasons:
4271 return []
4271 return []
4272 return self._reasons
4272 return self._reasons
4273
4273
4274 @reasons.setter
4274 @reasons.setter
4275 def reasons(self, val):
4275 def reasons(self, val):
4276 val = val or []
4276 val = val or []
4277 if any(not isinstance(x, compat.string_types) for x in val):
4277 if any(not isinstance(x, compat.string_types) for x in val):
4278 raise Exception('invalid reasons type, must be list of strings')
4278 raise Exception('invalid reasons type, must be list of strings')
4279 self._reasons = val
4279 self._reasons = val
4280
4280
4281 pull_requests_reviewers_id = Column(
4281 pull_requests_reviewers_id = Column(
4282 'pull_requests_reviewers_id', Integer(), nullable=False,
4282 'pull_requests_reviewers_id', Integer(), nullable=False,
4283 primary_key=True)
4283 primary_key=True)
4284 pull_request_id = Column(
4284 pull_request_id = Column(
4285 "pull_request_id", Integer(),
4285 "pull_request_id", Integer(),
4286 ForeignKey('pull_requests.pull_request_id'), nullable=False)
4286 ForeignKey('pull_requests.pull_request_id'), nullable=False)
4287 user_id = Column(
4287 user_id = Column(
4288 "user_id", Integer(), ForeignKey('users.user_id'), nullable=True)
4288 "user_id", Integer(), ForeignKey('users.user_id'), nullable=True)
4289 _reasons = Column(
4289 _reasons = Column(
4290 'reason', MutationList.as_mutable(
4290 'reason', MutationList.as_mutable(
4291 JsonType('list', dialect_map=dict(mysql=UnicodeText(16384)))))
4291 JsonType('list', dialect_map=dict(mysql=UnicodeText(16384)))))
4292
4292
4293 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4293 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4294 user = relationship('User')
4294 user = relationship('User')
4295 pull_request = relationship('PullRequest')
4295 pull_request = relationship('PullRequest')
4296
4296
4297 rule_data = Column(
4297 rule_data = Column(
4298 'rule_data_json',
4298 'rule_data_json',
4299 JsonType(dialect_map=dict(mysql=UnicodeText(16384))))
4299 JsonType(dialect_map=dict(mysql=UnicodeText(16384))))
4300
4300
4301 def rule_user_group_data(self):
4301 def rule_user_group_data(self):
4302 """
4302 """
4303 Returns the voting user group rule data for this reviewer
4303 Returns the voting user group rule data for this reviewer
4304 """
4304 """
4305
4305
4306 if self.rule_data and 'vote_rule' in self.rule_data:
4306 if self.rule_data and 'vote_rule' in self.rule_data:
4307 user_group_data = {}
4307 user_group_data = {}
4308 if 'rule_user_group_entry_id' in self.rule_data:
4308 if 'rule_user_group_entry_id' in self.rule_data:
4309 # means a group with voting rules !
4309 # means a group with voting rules !
4310 user_group_data['id'] = self.rule_data['rule_user_group_entry_id']
4310 user_group_data['id'] = self.rule_data['rule_user_group_entry_id']
4311 user_group_data['name'] = self.rule_data['rule_name']
4311 user_group_data['name'] = self.rule_data['rule_name']
4312 user_group_data['vote_rule'] = self.rule_data['vote_rule']
4312 user_group_data['vote_rule'] = self.rule_data['vote_rule']
4313
4313
4314 return user_group_data
4314 return user_group_data
4315
4315
4316 def __unicode__(self):
4316 def __unicode__(self):
4317 return u"<%s('id:%s')>" % (self.__class__.__name__,
4317 return u"<%s('id:%s')>" % (self.__class__.__name__,
4318 self.pull_requests_reviewers_id)
4318 self.pull_requests_reviewers_id)
4319
4319
4320
4320
4321 class Notification(Base, BaseModel):
4321 class Notification(Base, BaseModel):
4322 __tablename__ = 'notifications'
4322 __tablename__ = 'notifications'
4323 __table_args__ = (
4323 __table_args__ = (
4324 Index('notification_type_idx', 'type'),
4324 Index('notification_type_idx', 'type'),
4325 base_table_args,
4325 base_table_args,
4326 )
4326 )
4327
4327
4328 TYPE_CHANGESET_COMMENT = u'cs_comment'
4328 TYPE_CHANGESET_COMMENT = u'cs_comment'
4329 TYPE_MESSAGE = u'message'
4329 TYPE_MESSAGE = u'message'
4330 TYPE_MENTION = u'mention'
4330 TYPE_MENTION = u'mention'
4331 TYPE_REGISTRATION = u'registration'
4331 TYPE_REGISTRATION = u'registration'
4332 TYPE_PULL_REQUEST = u'pull_request'
4332 TYPE_PULL_REQUEST = u'pull_request'
4333 TYPE_PULL_REQUEST_COMMENT = u'pull_request_comment'
4333 TYPE_PULL_REQUEST_COMMENT = u'pull_request_comment'
4334
4334
4335 notification_id = Column('notification_id', Integer(), nullable=False, primary_key=True)
4335 notification_id = Column('notification_id', Integer(), nullable=False, primary_key=True)
4336 subject = Column('subject', Unicode(512), nullable=True)
4336 subject = Column('subject', Unicode(512), nullable=True)
4337 body = Column('body', UnicodeText().with_variant(UnicodeText(50000), 'mysql'), nullable=True)
4337 body = Column('body', UnicodeText().with_variant(UnicodeText(50000), 'mysql'), nullable=True)
4338 created_by = Column("created_by", Integer(), ForeignKey('users.user_id'), nullable=True)
4338 created_by = Column("created_by", Integer(), ForeignKey('users.user_id'), nullable=True)
4339 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4339 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4340 type_ = Column('type', Unicode(255))
4340 type_ = Column('type', Unicode(255))
4341
4341
4342 created_by_user = relationship('User')
4342 created_by_user = relationship('User')
4343 notifications_to_users = relationship('UserNotification', lazy='joined',
4343 notifications_to_users = relationship('UserNotification', lazy='joined',
4344 cascade="all, delete-orphan")
4344 cascade="all, delete-orphan")
4345
4345
4346 @property
4346 @property
4347 def recipients(self):
4347 def recipients(self):
4348 return [x.user for x in UserNotification.query()\
4348 return [x.user for x in UserNotification.query()\
4349 .filter(UserNotification.notification == self)\
4349 .filter(UserNotification.notification == self)\
4350 .order_by(UserNotification.user_id.asc()).all()]
4350 .order_by(UserNotification.user_id.asc()).all()]
4351
4351
4352 @classmethod
4352 @classmethod
4353 def create(cls, created_by, subject, body, recipients, type_=None):
4353 def create(cls, created_by, subject, body, recipients, type_=None):
4354 if type_ is None:
4354 if type_ is None:
4355 type_ = Notification.TYPE_MESSAGE
4355 type_ = Notification.TYPE_MESSAGE
4356
4356
4357 notification = cls()
4357 notification = cls()
4358 notification.created_by_user = created_by
4358 notification.created_by_user = created_by
4359 notification.subject = subject
4359 notification.subject = subject
4360 notification.body = body
4360 notification.body = body
4361 notification.type_ = type_
4361 notification.type_ = type_
4362 notification.created_on = datetime.datetime.now()
4362 notification.created_on = datetime.datetime.now()
4363
4363
4364 # For each recipient link the created notification to his account
4364 # For each recipient link the created notification to his account
4365 for u in recipients:
4365 for u in recipients:
4366 assoc = UserNotification()
4366 assoc = UserNotification()
4367 assoc.user_id = u.user_id
4367 assoc.user_id = u.user_id
4368 assoc.notification = notification
4368 assoc.notification = notification
4369
4369
4370 # if created_by is inside recipients mark his notification
4370 # if created_by is inside recipients mark his notification
4371 # as read
4371 # as read
4372 if u.user_id == created_by.user_id:
4372 if u.user_id == created_by.user_id:
4373 assoc.read = True
4373 assoc.read = True
4374 Session().add(assoc)
4374 Session().add(assoc)
4375
4375
4376 Session().add(notification)
4376 Session().add(notification)
4377
4377
4378 return notification
4378 return notification
4379
4379
4380
4380
4381 class UserNotification(Base, BaseModel):
4381 class UserNotification(Base, BaseModel):
4382 __tablename__ = 'user_to_notification'
4382 __tablename__ = 'user_to_notification'
4383 __table_args__ = (
4383 __table_args__ = (
4384 UniqueConstraint('user_id', 'notification_id'),
4384 UniqueConstraint('user_id', 'notification_id'),
4385 base_table_args
4385 base_table_args
4386 )
4386 )
4387
4387
4388 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
4388 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
4389 notification_id = Column("notification_id", Integer(), ForeignKey('notifications.notification_id'), primary_key=True)
4389 notification_id = Column("notification_id", Integer(), ForeignKey('notifications.notification_id'), primary_key=True)
4390 read = Column('read', Boolean, default=False)
4390 read = Column('read', Boolean, default=False)
4391 sent_on = Column('sent_on', DateTime(timezone=False), nullable=True, unique=None)
4391 sent_on = Column('sent_on', DateTime(timezone=False), nullable=True, unique=None)
4392
4392
4393 user = relationship('User', lazy="joined")
4393 user = relationship('User', lazy="joined")
4394 notification = relationship('Notification', lazy="joined",
4394 notification = relationship('Notification', lazy="joined",
4395 order_by=lambda: Notification.created_on.desc(),)
4395 order_by=lambda: Notification.created_on.desc(),)
4396
4396
4397 def mark_as_read(self):
4397 def mark_as_read(self):
4398 self.read = True
4398 self.read = True
4399 Session().add(self)
4399 Session().add(self)
4400
4400
4401
4401
4402 class Gist(Base, BaseModel):
4402 class Gist(Base, BaseModel):
4403 __tablename__ = 'gists'
4403 __tablename__ = 'gists'
4404 __table_args__ = (
4404 __table_args__ = (
4405 Index('g_gist_access_id_idx', 'gist_access_id'),
4405 Index('g_gist_access_id_idx', 'gist_access_id'),
4406 Index('g_created_on_idx', 'created_on'),
4406 Index('g_created_on_idx', 'created_on'),
4407 base_table_args
4407 base_table_args
4408 )
4408 )
4409
4409
4410 GIST_PUBLIC = u'public'
4410 GIST_PUBLIC = u'public'
4411 GIST_PRIVATE = u'private'
4411 GIST_PRIVATE = u'private'
4412 DEFAULT_FILENAME = u'gistfile1.txt'
4412 DEFAULT_FILENAME = u'gistfile1.txt'
4413
4413
4414 ACL_LEVEL_PUBLIC = u'acl_public'
4414 ACL_LEVEL_PUBLIC = u'acl_public'
4415 ACL_LEVEL_PRIVATE = u'acl_private'
4415 ACL_LEVEL_PRIVATE = u'acl_private'
4416
4416
4417 gist_id = Column('gist_id', Integer(), primary_key=True)
4417 gist_id = Column('gist_id', Integer(), primary_key=True)
4418 gist_access_id = Column('gist_access_id', Unicode(250))
4418 gist_access_id = Column('gist_access_id', Unicode(250))
4419 gist_description = Column('gist_description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
4419 gist_description = Column('gist_description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
4420 gist_owner = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True)
4420 gist_owner = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True)
4421 gist_expires = Column('gist_expires', Float(53), nullable=False)
4421 gist_expires = Column('gist_expires', Float(53), nullable=False)
4422 gist_type = Column('gist_type', Unicode(128), nullable=False)
4422 gist_type = Column('gist_type', Unicode(128), nullable=False)
4423 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4423 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4424 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4424 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4425 acl_level = Column('acl_level', Unicode(128), nullable=True)
4425 acl_level = Column('acl_level', Unicode(128), nullable=True)
4426
4426
4427 owner = relationship('User')
4427 owner = relationship('User')
4428
4428
4429 def __repr__(self):
4429 def __repr__(self):
4430 return '<Gist:[%s]%s>' % (self.gist_type, self.gist_access_id)
4430 return '<Gist:[%s]%s>' % (self.gist_type, self.gist_access_id)
4431
4431
4432 @hybrid_property
4432 @hybrid_property
4433 def description_safe(self):
4433 def description_safe(self):
4434 from rhodecode.lib import helpers as h
4434 from rhodecode.lib import helpers as h
4435 return h.escape(self.gist_description)
4435 return h.escape(self.gist_description)
4436
4436
4437 @classmethod
4437 @classmethod
4438 def get_or_404(cls, id_):
4438 def get_or_404(cls, id_):
4439 from pyramid.httpexceptions import HTTPNotFound
4439 from pyramid.httpexceptions import HTTPNotFound
4440
4440
4441 res = cls.query().filter(cls.gist_access_id == id_).scalar()
4441 res = cls.query().filter(cls.gist_access_id == id_).scalar()
4442 if not res:
4442 if not res:
4443 raise HTTPNotFound()
4443 raise HTTPNotFound()
4444 return res
4444 return res
4445
4445
4446 @classmethod
4446 @classmethod
4447 def get_by_access_id(cls, gist_access_id):
4447 def get_by_access_id(cls, gist_access_id):
4448 return cls.query().filter(cls.gist_access_id == gist_access_id).scalar()
4448 return cls.query().filter(cls.gist_access_id == gist_access_id).scalar()
4449
4449
4450 def gist_url(self):
4450 def gist_url(self):
4451 from rhodecode.model.gist import GistModel
4451 from rhodecode.model.gist import GistModel
4452 return GistModel().get_url(self)
4452 return GistModel().get_url(self)
4453
4453
4454 @classmethod
4454 @classmethod
4455 def base_path(cls):
4455 def base_path(cls):
4456 """
4456 """
4457 Returns base path when all gists are stored
4457 Returns base path when all gists are stored
4458
4458
4459 :param cls:
4459 :param cls:
4460 """
4460 """
4461 from rhodecode.model.gist import GIST_STORE_LOC
4461 from rhodecode.model.gist import GIST_STORE_LOC
4462 q = Session().query(RhodeCodeUi)\
4462 q = Session().query(RhodeCodeUi)\
4463 .filter(RhodeCodeUi.ui_key == URL_SEP)
4463 .filter(RhodeCodeUi.ui_key == URL_SEP)
4464 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
4464 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
4465 return os.path.join(q.one().ui_value, GIST_STORE_LOC)
4465 return os.path.join(q.one().ui_value, GIST_STORE_LOC)
4466
4466
4467 def get_api_data(self):
4467 def get_api_data(self):
4468 """
4468 """
4469 Common function for generating gist related data for API
4469 Common function for generating gist related data for API
4470 """
4470 """
4471 gist = self
4471 gist = self
4472 data = {
4472 data = {
4473 'gist_id': gist.gist_id,
4473 'gist_id': gist.gist_id,
4474 'type': gist.gist_type,
4474 'type': gist.gist_type,
4475 'access_id': gist.gist_access_id,
4475 'access_id': gist.gist_access_id,
4476 'description': gist.gist_description,
4476 'description': gist.gist_description,
4477 'url': gist.gist_url(),
4477 'url': gist.gist_url(),
4478 'expires': gist.gist_expires,
4478 'expires': gist.gist_expires,
4479 'created_on': gist.created_on,
4479 'created_on': gist.created_on,
4480 'modified_at': gist.modified_at,
4480 'modified_at': gist.modified_at,
4481 'content': None,
4481 'content': None,
4482 'acl_level': gist.acl_level,
4482 'acl_level': gist.acl_level,
4483 }
4483 }
4484 return data
4484 return data
4485
4485
4486 def __json__(self):
4486 def __json__(self):
4487 data = dict(
4487 data = dict(
4488 )
4488 )
4489 data.update(self.get_api_data())
4489 data.update(self.get_api_data())
4490 return data
4490 return data
4491 # SCM functions
4491 # SCM functions
4492
4492
4493 def scm_instance(self, **kwargs):
4493 def scm_instance(self, **kwargs):
4494 """
4494 """
4495 Get an instance of VCS Repository
4495 Get an instance of VCS Repository
4496
4496
4497 :param kwargs:
4497 :param kwargs:
4498 """
4498 """
4499 from rhodecode.model.gist import GistModel
4499 from rhodecode.model.gist import GistModel
4500 full_repo_path = os.path.join(self.base_path(), self.gist_access_id)
4500 full_repo_path = os.path.join(self.base_path(), self.gist_access_id)
4501 return get_vcs_instance(
4501 return get_vcs_instance(
4502 repo_path=safe_str(full_repo_path), create=False,
4502 repo_path=safe_str(full_repo_path), create=False,
4503 _vcs_alias=GistModel.vcs_backend)
4503 _vcs_alias=GistModel.vcs_backend)
4504
4504
4505
4505
4506 class ExternalIdentity(Base, BaseModel):
4506 class ExternalIdentity(Base, BaseModel):
4507 __tablename__ = 'external_identities'
4507 __tablename__ = 'external_identities'
4508 __table_args__ = (
4508 __table_args__ = (
4509 Index('local_user_id_idx', 'local_user_id'),
4509 Index('local_user_id_idx', 'local_user_id'),
4510 Index('external_id_idx', 'external_id'),
4510 Index('external_id_idx', 'external_id'),
4511 base_table_args
4511 base_table_args
4512 )
4512 )
4513
4513
4514 external_id = Column('external_id', Unicode(255), default=u'', primary_key=True)
4514 external_id = Column('external_id', Unicode(255), default=u'', primary_key=True)
4515 external_username = Column('external_username', Unicode(1024), default=u'')
4515 external_username = Column('external_username', Unicode(1024), default=u'')
4516 local_user_id = Column('local_user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
4516 local_user_id = Column('local_user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
4517 provider_name = Column('provider_name', Unicode(255), default=u'', primary_key=True)
4517 provider_name = Column('provider_name', Unicode(255), default=u'', primary_key=True)
4518 access_token = Column('access_token', String(1024), default=u'')
4518 access_token = Column('access_token', String(1024), default=u'')
4519 alt_token = Column('alt_token', String(1024), default=u'')
4519 alt_token = Column('alt_token', String(1024), default=u'')
4520 token_secret = Column('token_secret', String(1024), default=u'')
4520 token_secret = Column('token_secret', String(1024), default=u'')
4521
4521
4522 @classmethod
4522 @classmethod
4523 def by_external_id_and_provider(cls, external_id, provider_name, local_user_id=None):
4523 def by_external_id_and_provider(cls, external_id, provider_name, local_user_id=None):
4524 """
4524 """
4525 Returns ExternalIdentity instance based on search params
4525 Returns ExternalIdentity instance based on search params
4526
4526
4527 :param external_id:
4527 :param external_id:
4528 :param provider_name:
4528 :param provider_name:
4529 :return: ExternalIdentity
4529 :return: ExternalIdentity
4530 """
4530 """
4531 query = cls.query()
4531 query = cls.query()
4532 query = query.filter(cls.external_id == external_id)
4532 query = query.filter(cls.external_id == external_id)
4533 query = query.filter(cls.provider_name == provider_name)
4533 query = query.filter(cls.provider_name == provider_name)
4534 if local_user_id:
4534 if local_user_id:
4535 query = query.filter(cls.local_user_id == local_user_id)
4535 query = query.filter(cls.local_user_id == local_user_id)
4536 return query.first()
4536 return query.first()
4537
4537
4538 @classmethod
4538 @classmethod
4539 def user_by_external_id_and_provider(cls, external_id, provider_name):
4539 def user_by_external_id_and_provider(cls, external_id, provider_name):
4540 """
4540 """
4541 Returns User instance based on search params
4541 Returns User instance based on search params
4542
4542
4543 :param external_id:
4543 :param external_id:
4544 :param provider_name:
4544 :param provider_name:
4545 :return: User
4545 :return: User
4546 """
4546 """
4547 query = User.query()
4547 query = User.query()
4548 query = query.filter(cls.external_id == external_id)
4548 query = query.filter(cls.external_id == external_id)
4549 query = query.filter(cls.provider_name == provider_name)
4549 query = query.filter(cls.provider_name == provider_name)
4550 query = query.filter(User.user_id == cls.local_user_id)
4550 query = query.filter(User.user_id == cls.local_user_id)
4551 return query.first()
4551 return query.first()
4552
4552
4553 @classmethod
4553 @classmethod
4554 def by_local_user_id(cls, local_user_id):
4554 def by_local_user_id(cls, local_user_id):
4555 """
4555 """
4556 Returns all tokens for user
4556 Returns all tokens for user
4557
4557
4558 :param local_user_id:
4558 :param local_user_id:
4559 :return: ExternalIdentity
4559 :return: ExternalIdentity
4560 """
4560 """
4561 query = cls.query()
4561 query = cls.query()
4562 query = query.filter(cls.local_user_id == local_user_id)
4562 query = query.filter(cls.local_user_id == local_user_id)
4563 return query
4563 return query
4564
4564
4565 @classmethod
4565 @classmethod
4566 def load_provider_plugin(cls, plugin_id):
4566 def load_provider_plugin(cls, plugin_id):
4567 from rhodecode.authentication.base import loadplugin
4567 from rhodecode.authentication.base import loadplugin
4568 _plugin_id = 'egg:rhodecode-enterprise-ee#{}'.format(plugin_id)
4568 _plugin_id = 'egg:rhodecode-enterprise-ee#{}'.format(plugin_id)
4569 auth_plugin = loadplugin(_plugin_id)
4569 auth_plugin = loadplugin(_plugin_id)
4570 return auth_plugin
4570 return auth_plugin
4571
4571
4572
4572
4573 class Integration(Base, BaseModel):
4573 class Integration(Base, BaseModel):
4574 __tablename__ = 'integrations'
4574 __tablename__ = 'integrations'
4575 __table_args__ = (
4575 __table_args__ = (
4576 base_table_args
4576 base_table_args
4577 )
4577 )
4578
4578
4579 integration_id = Column('integration_id', Integer(), primary_key=True)
4579 integration_id = Column('integration_id', Integer(), primary_key=True)
4580 integration_type = Column('integration_type', String(255))
4580 integration_type = Column('integration_type', String(255))
4581 enabled = Column('enabled', Boolean(), nullable=False)
4581 enabled = Column('enabled', Boolean(), nullable=False)
4582 name = Column('name', String(255), nullable=False)
4582 name = Column('name', String(255), nullable=False)
4583 child_repos_only = Column('child_repos_only', Boolean(), nullable=False,
4583 child_repos_only = Column('child_repos_only', Boolean(), nullable=False,
4584 default=False)
4584 default=False)
4585
4585
4586 settings = Column(
4586 settings = Column(
4587 'settings_json', MutationObj.as_mutable(
4587 'settings_json', MutationObj.as_mutable(
4588 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
4588 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
4589 repo_id = Column(
4589 repo_id = Column(
4590 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
4590 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
4591 nullable=True, unique=None, default=None)
4591 nullable=True, unique=None, default=None)
4592 repo = relationship('Repository', lazy='joined')
4592 repo = relationship('Repository', lazy='joined')
4593
4593
4594 repo_group_id = Column(
4594 repo_group_id = Column(
4595 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
4595 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
4596 nullable=True, unique=None, default=None)
4596 nullable=True, unique=None, default=None)
4597 repo_group = relationship('RepoGroup', lazy='joined')
4597 repo_group = relationship('RepoGroup', lazy='joined')
4598
4598
4599 @property
4599 @property
4600 def scope(self):
4600 def scope(self):
4601 if self.repo:
4601 if self.repo:
4602 return repr(self.repo)
4602 return repr(self.repo)
4603 if self.repo_group:
4603 if self.repo_group:
4604 if self.child_repos_only:
4604 if self.child_repos_only:
4605 return repr(self.repo_group) + ' (child repos only)'
4605 return repr(self.repo_group) + ' (child repos only)'
4606 else:
4606 else:
4607 return repr(self.repo_group) + ' (recursive)'
4607 return repr(self.repo_group) + ' (recursive)'
4608 if self.child_repos_only:
4608 if self.child_repos_only:
4609 return 'root_repos'
4609 return 'root_repos'
4610 return 'global'
4610 return 'global'
4611
4611
4612 def __repr__(self):
4612 def __repr__(self):
4613 return '<Integration(%r, %r)>' % (self.integration_type, self.scope)
4613 return '<Integration(%r, %r)>' % (self.integration_type, self.scope)
4614
4614
4615
4615
4616 class RepoReviewRuleUser(Base, BaseModel):
4616 class RepoReviewRuleUser(Base, BaseModel):
4617 __tablename__ = 'repo_review_rules_users'
4617 __tablename__ = 'repo_review_rules_users'
4618 __table_args__ = (
4618 __table_args__ = (
4619 base_table_args
4619 base_table_args
4620 )
4620 )
4621
4621
4622 repo_review_rule_user_id = Column('repo_review_rule_user_id', Integer(), primary_key=True)
4622 repo_review_rule_user_id = Column('repo_review_rule_user_id', Integer(), primary_key=True)
4623 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
4623 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
4624 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False)
4624 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False)
4625 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4625 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4626 user = relationship('User')
4626 user = relationship('User')
4627
4627
4628 def rule_data(self):
4628 def rule_data(self):
4629 return {
4629 return {
4630 'mandatory': self.mandatory
4630 'mandatory': self.mandatory
4631 }
4631 }
4632
4632
4633
4633
4634 class RepoReviewRuleUserGroup(Base, BaseModel):
4634 class RepoReviewRuleUserGroup(Base, BaseModel):
4635 __tablename__ = 'repo_review_rules_users_groups'
4635 __tablename__ = 'repo_review_rules_users_groups'
4636 __table_args__ = (
4636 __table_args__ = (
4637 base_table_args
4637 base_table_args
4638 )
4638 )
4639
4639
4640 VOTE_RULE_ALL = -1
4640 VOTE_RULE_ALL = -1
4641
4641
4642 repo_review_rule_users_group_id = Column('repo_review_rule_users_group_id', Integer(), primary_key=True)
4642 repo_review_rule_users_group_id = Column('repo_review_rule_users_group_id', Integer(), primary_key=True)
4643 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
4643 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
4644 users_group_id = Column("users_group_id", Integer(),ForeignKey('users_groups.users_group_id'), nullable=False)
4644 users_group_id = Column("users_group_id", Integer(),ForeignKey('users_groups.users_group_id'), nullable=False)
4645 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4645 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
4646 vote_rule = Column("vote_rule", Integer(), nullable=True, default=VOTE_RULE_ALL)
4646 vote_rule = Column("vote_rule", Integer(), nullable=True, default=VOTE_RULE_ALL)
4647 users_group = relationship('UserGroup')
4647 users_group = relationship('UserGroup')
4648
4648
4649 def rule_data(self):
4649 def rule_data(self):
4650 return {
4650 return {
4651 'mandatory': self.mandatory,
4651 'mandatory': self.mandatory,
4652 'vote_rule': self.vote_rule
4652 'vote_rule': self.vote_rule
4653 }
4653 }
4654
4654
4655 @property
4655 @property
4656 def vote_rule_label(self):
4656 def vote_rule_label(self):
4657 if not self.vote_rule or self.vote_rule == self.VOTE_RULE_ALL:
4657 if not self.vote_rule or self.vote_rule == self.VOTE_RULE_ALL:
4658 return 'all must vote'
4658 return 'all must vote'
4659 else:
4659 else:
4660 return 'min. vote {}'.format(self.vote_rule)
4660 return 'min. vote {}'.format(self.vote_rule)
4661
4661
4662
4662
4663 class RepoReviewRule(Base, BaseModel):
4663 class RepoReviewRule(Base, BaseModel):
4664 __tablename__ = 'repo_review_rules'
4664 __tablename__ = 'repo_review_rules'
4665 __table_args__ = (
4665 __table_args__ = (
4666 base_table_args
4666 base_table_args
4667 )
4667 )
4668
4668
4669 repo_review_rule_id = Column(
4669 repo_review_rule_id = Column(
4670 'repo_review_rule_id', Integer(), primary_key=True)
4670 'repo_review_rule_id', Integer(), primary_key=True)
4671 repo_id = Column(
4671 repo_id = Column(
4672 "repo_id", Integer(), ForeignKey('repositories.repo_id'))
4672 "repo_id", Integer(), ForeignKey('repositories.repo_id'))
4673 repo = relationship('Repository', backref='review_rules')
4673 repo = relationship('Repository', backref='review_rules')
4674
4674
4675 review_rule_name = Column('review_rule_name', String(255))
4675 review_rule_name = Column('review_rule_name', String(255))
4676 _branch_pattern = Column("branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4676 _branch_pattern = Column("branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4677 _target_branch_pattern = Column("target_branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4677 _target_branch_pattern = Column("target_branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4678 _file_pattern = Column("file_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4678 _file_pattern = Column("file_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
4679
4679
4680 use_authors_for_review = Column("use_authors_for_review", Boolean(), nullable=False, default=False)
4680 use_authors_for_review = Column("use_authors_for_review", Boolean(), nullable=False, default=False)
4681 forbid_author_to_review = Column("forbid_author_to_review", Boolean(), nullable=False, default=False)
4681 forbid_author_to_review = Column("forbid_author_to_review", Boolean(), nullable=False, default=False)
4682 forbid_commit_author_to_review = Column("forbid_commit_author_to_review", Boolean(), nullable=False, default=False)
4682 forbid_commit_author_to_review = Column("forbid_commit_author_to_review", Boolean(), nullable=False, default=False)
4683 forbid_adding_reviewers = Column("forbid_adding_reviewers", Boolean(), nullable=False, default=False)
4683 forbid_adding_reviewers = Column("forbid_adding_reviewers", Boolean(), nullable=False, default=False)
4684
4684
4685 rule_users = relationship('RepoReviewRuleUser')
4685 rule_users = relationship('RepoReviewRuleUser')
4686 rule_user_groups = relationship('RepoReviewRuleUserGroup')
4686 rule_user_groups = relationship('RepoReviewRuleUserGroup')
4687
4687
4688 def _validate_pattern(self, value):
4688 def _validate_pattern(self, value):
4689 re.compile('^' + glob2re(value) + '$')
4689 re.compile('^' + glob2re(value) + '$')
4690
4690
4691 @hybrid_property
4691 @hybrid_property
4692 def source_branch_pattern(self):
4692 def source_branch_pattern(self):
4693 return self._branch_pattern or '*'
4693 return self._branch_pattern or '*'
4694
4694
4695 @source_branch_pattern.setter
4695 @source_branch_pattern.setter
4696 def source_branch_pattern(self, value):
4696 def source_branch_pattern(self, value):
4697 self._validate_pattern(value)
4697 self._validate_pattern(value)
4698 self._branch_pattern = value or '*'
4698 self._branch_pattern = value or '*'
4699
4699
4700 @hybrid_property
4700 @hybrid_property
4701 def target_branch_pattern(self):
4701 def target_branch_pattern(self):
4702 return self._target_branch_pattern or '*'
4702 return self._target_branch_pattern or '*'
4703
4703
4704 @target_branch_pattern.setter
4704 @target_branch_pattern.setter
4705 def target_branch_pattern(self, value):
4705 def target_branch_pattern(self, value):
4706 self._validate_pattern(value)
4706 self._validate_pattern(value)
4707 self._target_branch_pattern = value or '*'
4707 self._target_branch_pattern = value or '*'
4708
4708
4709 @hybrid_property
4709 @hybrid_property
4710 def file_pattern(self):
4710 def file_pattern(self):
4711 return self._file_pattern or '*'
4711 return self._file_pattern or '*'
4712
4712
4713 @file_pattern.setter
4713 @file_pattern.setter
4714 def file_pattern(self, value):
4714 def file_pattern(self, value):
4715 self._validate_pattern(value)
4715 self._validate_pattern(value)
4716 self._file_pattern = value or '*'
4716 self._file_pattern = value or '*'
4717
4717
4718 def matches(self, source_branch, target_branch, files_changed):
4718 def matches(self, source_branch, target_branch, files_changed):
4719 """
4719 """
4720 Check if this review rule matches a branch/files in a pull request
4720 Check if this review rule matches a branch/files in a pull request
4721
4721
4722 :param source_branch: source branch name for the commit
4722 :param source_branch: source branch name for the commit
4723 :param target_branch: target branch name for the commit
4723 :param target_branch: target branch name for the commit
4724 :param files_changed: list of file paths changed in the pull request
4724 :param files_changed: list of file paths changed in the pull request
4725 """
4725 """
4726
4726
4727 source_branch = source_branch or ''
4727 source_branch = source_branch or ''
4728 target_branch = target_branch or ''
4728 target_branch = target_branch or ''
4729 files_changed = files_changed or []
4729 files_changed = files_changed or []
4730
4730
4731 branch_matches = True
4731 branch_matches = True
4732 if source_branch or target_branch:
4732 if source_branch or target_branch:
4733 if self.source_branch_pattern == '*':
4733 if self.source_branch_pattern == '*':
4734 source_branch_match = True
4734 source_branch_match = True
4735 else:
4735 else:
4736 if self.source_branch_pattern.startswith('re:'):
4736 if self.source_branch_pattern.startswith('re:'):
4737 source_pattern = self.source_branch_pattern[3:]
4737 source_pattern = self.source_branch_pattern[3:]
4738 else:
4738 else:
4739 source_pattern = '^' + glob2re(self.source_branch_pattern) + '$'
4739 source_pattern = '^' + glob2re(self.source_branch_pattern) + '$'
4740 source_branch_regex = re.compile(source_pattern)
4740 source_branch_regex = re.compile(source_pattern)
4741 source_branch_match = bool(source_branch_regex.search(source_branch))
4741 source_branch_match = bool(source_branch_regex.search(source_branch))
4742 if self.target_branch_pattern == '*':
4742 if self.target_branch_pattern == '*':
4743 target_branch_match = True
4743 target_branch_match = True
4744 else:
4744 else:
4745 if self.target_branch_pattern.startswith('re:'):
4745 if self.target_branch_pattern.startswith('re:'):
4746 target_pattern = self.target_branch_pattern[3:]
4746 target_pattern = self.target_branch_pattern[3:]
4747 else:
4747 else:
4748 target_pattern = '^' + glob2re(self.target_branch_pattern) + '$'
4748 target_pattern = '^' + glob2re(self.target_branch_pattern) + '$'
4749 target_branch_regex = re.compile(target_pattern)
4749 target_branch_regex = re.compile(target_pattern)
4750 target_branch_match = bool(target_branch_regex.search(target_branch))
4750 target_branch_match = bool(target_branch_regex.search(target_branch))
4751
4751
4752 branch_matches = source_branch_match and target_branch_match
4752 branch_matches = source_branch_match and target_branch_match
4753
4753
4754 files_matches = True
4754 files_matches = True
4755 if self.file_pattern != '*':
4755 if self.file_pattern != '*':
4756 files_matches = False
4756 files_matches = False
4757 if self.file_pattern.startswith('re:'):
4757 if self.file_pattern.startswith('re:'):
4758 file_pattern = self.file_pattern[3:]
4758 file_pattern = self.file_pattern[3:]
4759 else:
4759 else:
4760 file_pattern = glob2re(self.file_pattern)
4760 file_pattern = glob2re(self.file_pattern)
4761 file_regex = re.compile(file_pattern)
4761 file_regex = re.compile(file_pattern)
4762 for filename in files_changed:
4762 for filename in files_changed:
4763 if file_regex.search(filename):
4763 if file_regex.search(filename):
4764 files_matches = True
4764 files_matches = True
4765 break
4765 break
4766
4766
4767 return branch_matches and files_matches
4767 return branch_matches and files_matches
4768
4768
4769 @property
4769 @property
4770 def review_users(self):
4770 def review_users(self):
4771 """ Returns the users which this rule applies to """
4771 """ Returns the users which this rule applies to """
4772
4772
4773 users = collections.OrderedDict()
4773 users = collections.OrderedDict()
4774
4774
4775 for rule_user in self.rule_users:
4775 for rule_user in self.rule_users:
4776 if rule_user.user.active:
4776 if rule_user.user.active:
4777 if rule_user.user not in users:
4777 if rule_user.user not in users:
4778 users[rule_user.user.username] = {
4778 users[rule_user.user.username] = {
4779 'user': rule_user.user,
4779 'user': rule_user.user,
4780 'source': 'user',
4780 'source': 'user',
4781 'source_data': {},
4781 'source_data': {},
4782 'data': rule_user.rule_data()
4782 'data': rule_user.rule_data()
4783 }
4783 }
4784
4784
4785 for rule_user_group in self.rule_user_groups:
4785 for rule_user_group in self.rule_user_groups:
4786 source_data = {
4786 source_data = {
4787 'user_group_id': rule_user_group.users_group.users_group_id,
4787 'user_group_id': rule_user_group.users_group.users_group_id,
4788 'name': rule_user_group.users_group.users_group_name,
4788 'name': rule_user_group.users_group.users_group_name,
4789 'members': len(rule_user_group.users_group.members)
4789 'members': len(rule_user_group.users_group.members)
4790 }
4790 }
4791 for member in rule_user_group.users_group.members:
4791 for member in rule_user_group.users_group.members:
4792 if member.user.active:
4792 if member.user.active:
4793 key = member.user.username
4793 key = member.user.username
4794 if key in users:
4794 if key in users:
4795 # skip this member as we have him already
4795 # skip this member as we have him already
4796 # this prevents from override the "first" matched
4796 # this prevents from override the "first" matched
4797 # users with duplicates in multiple groups
4797 # users with duplicates in multiple groups
4798 continue
4798 continue
4799
4799
4800 users[key] = {
4800 users[key] = {
4801 'user': member.user,
4801 'user': member.user,
4802 'source': 'user_group',
4802 'source': 'user_group',
4803 'source_data': source_data,
4803 'source_data': source_data,
4804 'data': rule_user_group.rule_data()
4804 'data': rule_user_group.rule_data()
4805 }
4805 }
4806
4806
4807 return users
4807 return users
4808
4808
4809 def user_group_vote_rule(self, user_id):
4809 def user_group_vote_rule(self, user_id):
4810
4810
4811 rules = []
4811 rules = []
4812 if not self.rule_user_groups:
4812 if not self.rule_user_groups:
4813 return rules
4813 return rules
4814
4814
4815 for user_group in self.rule_user_groups:
4815 for user_group in self.rule_user_groups:
4816 user_group_members = [x.user_id for x in user_group.users_group.members]
4816 user_group_members = [x.user_id for x in user_group.users_group.members]
4817 if user_id in user_group_members:
4817 if user_id in user_group_members:
4818 rules.append(user_group)
4818 rules.append(user_group)
4819 return rules
4819 return rules
4820
4820
4821 def __repr__(self):
4821 def __repr__(self):
4822 return '<RepoReviewerRule(id=%r, repo=%r)>' % (
4822 return '<RepoReviewerRule(id=%r, repo=%r)>' % (
4823 self.repo_review_rule_id, self.repo)
4823 self.repo_review_rule_id, self.repo)
4824
4824
4825
4825
4826 class ScheduleEntry(Base, BaseModel):
4826 class ScheduleEntry(Base, BaseModel):
4827 __tablename__ = 'schedule_entries'
4827 __tablename__ = 'schedule_entries'
4828 __table_args__ = (
4828 __table_args__ = (
4829 UniqueConstraint('schedule_name', name='s_schedule_name_idx'),
4829 UniqueConstraint('schedule_name', name='s_schedule_name_idx'),
4830 UniqueConstraint('task_uid', name='s_task_uid_idx'),
4830 UniqueConstraint('task_uid', name='s_task_uid_idx'),
4831 base_table_args,
4831 base_table_args,
4832 )
4832 )
4833
4833
4834 schedule_types = ['crontab', 'timedelta', 'integer']
4834 schedule_types = ['crontab', 'timedelta', 'integer']
4835 schedule_entry_id = Column('schedule_entry_id', Integer(), primary_key=True)
4835 schedule_entry_id = Column('schedule_entry_id', Integer(), primary_key=True)
4836
4836
4837 schedule_name = Column("schedule_name", String(255), nullable=False, unique=None, default=None)
4837 schedule_name = Column("schedule_name", String(255), nullable=False, unique=None, default=None)
4838 schedule_description = Column("schedule_description", String(10000), nullable=True, unique=None, default=None)
4838 schedule_description = Column("schedule_description", String(10000), nullable=True, unique=None, default=None)
4839 schedule_enabled = Column("schedule_enabled", Boolean(), nullable=False, unique=None, default=True)
4839 schedule_enabled = Column("schedule_enabled", Boolean(), nullable=False, unique=None, default=True)
4840
4840
4841 _schedule_type = Column("schedule_type", String(255), nullable=False, unique=None, default=None)
4841 _schedule_type = Column("schedule_type", String(255), nullable=False, unique=None, default=None)
4842 schedule_definition = Column('schedule_definition_json', MutationObj.as_mutable(JsonType(default=lambda: "", dialect_map=dict(mysql=LONGTEXT()))))
4842 schedule_definition = Column('schedule_definition_json', MutationObj.as_mutable(JsonType(default=lambda: "", dialect_map=dict(mysql=LONGTEXT()))))
4843
4843
4844 schedule_last_run = Column('schedule_last_run', DateTime(timezone=False), nullable=True, unique=None, default=None)
4844 schedule_last_run = Column('schedule_last_run', DateTime(timezone=False), nullable=True, unique=None, default=None)
4845 schedule_total_run_count = Column('schedule_total_run_count', Integer(), nullable=True, unique=None, default=0)
4845 schedule_total_run_count = Column('schedule_total_run_count', Integer(), nullable=True, unique=None, default=0)
4846
4846
4847 # task
4847 # task
4848 task_uid = Column("task_uid", String(255), nullable=False, unique=None, default=None)
4848 task_uid = Column("task_uid", String(255), nullable=False, unique=None, default=None)
4849 task_dot_notation = Column("task_dot_notation", String(4096), nullable=False, unique=None, default=None)
4849 task_dot_notation = Column("task_dot_notation", String(4096), nullable=False, unique=None, default=None)
4850 task_args = Column('task_args_json', MutationObj.as_mutable(JsonType(default=list, dialect_map=dict(mysql=LONGTEXT()))))
4850 task_args = Column('task_args_json', MutationObj.as_mutable(JsonType(default=list, dialect_map=dict(mysql=LONGTEXT()))))
4851 task_kwargs = Column('task_kwargs_json', MutationObj.as_mutable(JsonType(default=dict, dialect_map=dict(mysql=LONGTEXT()))))
4851 task_kwargs = Column('task_kwargs_json', MutationObj.as_mutable(JsonType(default=dict, dialect_map=dict(mysql=LONGTEXT()))))
4852
4852
4853 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4853 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
4854 updated_on = Column('updated_on', DateTime(timezone=False), nullable=True, unique=None, default=None)
4854 updated_on = Column('updated_on', DateTime(timezone=False), nullable=True, unique=None, default=None)
4855
4855
4856 @hybrid_property
4856 @hybrid_property
4857 def schedule_type(self):
4857 def schedule_type(self):
4858 return self._schedule_type
4858 return self._schedule_type
4859
4859
4860 @schedule_type.setter
4860 @schedule_type.setter
4861 def schedule_type(self, val):
4861 def schedule_type(self, val):
4862 if val not in self.schedule_types:
4862 if val not in self.schedule_types:
4863 raise ValueError('Value must be on of `{}` and got `{}`'.format(
4863 raise ValueError('Value must be on of `{}` and got `{}`'.format(
4864 val, self.schedule_type))
4864 val, self.schedule_type))
4865
4865
4866 self._schedule_type = val
4866 self._schedule_type = val
4867
4867
4868 @classmethod
4868 @classmethod
4869 def get_uid(cls, obj):
4869 def get_uid(cls, obj):
4870 args = obj.task_args
4870 args = obj.task_args
4871 kwargs = obj.task_kwargs
4871 kwargs = obj.task_kwargs
4872 if isinstance(args, JsonRaw):
4872 if isinstance(args, JsonRaw):
4873 try:
4873 try:
4874 args = json.loads(args)
4874 args = json.loads(args)
4875 except ValueError:
4875 except ValueError:
4876 args = tuple()
4876 args = tuple()
4877
4877
4878 if isinstance(kwargs, JsonRaw):
4878 if isinstance(kwargs, JsonRaw):
4879 try:
4879 try:
4880 kwargs = json.loads(kwargs)
4880 kwargs = json.loads(kwargs)
4881 except ValueError:
4881 except ValueError:
4882 kwargs = dict()
4882 kwargs = dict()
4883
4883
4884 dot_notation = obj.task_dot_notation
4884 dot_notation = obj.task_dot_notation
4885 val = '.'.join(map(safe_str, [
4885 val = '.'.join(map(safe_str, [
4886 sorted(dot_notation), args, sorted(kwargs.items())]))
4886 sorted(dot_notation), args, sorted(kwargs.items())]))
4887 return hashlib.sha1(val).hexdigest()
4887 return hashlib.sha1(val).hexdigest()
4888
4888
4889 @classmethod
4889 @classmethod
4890 def get_by_schedule_name(cls, schedule_name):
4890 def get_by_schedule_name(cls, schedule_name):
4891 return cls.query().filter(cls.schedule_name == schedule_name).scalar()
4891 return cls.query().filter(cls.schedule_name == schedule_name).scalar()
4892
4892
4893 @classmethod
4893 @classmethod
4894 def get_by_schedule_id(cls, schedule_id):
4894 def get_by_schedule_id(cls, schedule_id):
4895 return cls.query().filter(cls.schedule_entry_id == schedule_id).scalar()
4895 return cls.query().filter(cls.schedule_entry_id == schedule_id).scalar()
4896
4896
4897 @property
4897 @property
4898 def task(self):
4898 def task(self):
4899 return self.task_dot_notation
4899 return self.task_dot_notation
4900
4900
4901 @property
4901 @property
4902 def schedule(self):
4902 def schedule(self):
4903 from rhodecode.lib.celerylib.utils import raw_2_schedule
4903 from rhodecode.lib.celerylib.utils import raw_2_schedule
4904 schedule = raw_2_schedule(self.schedule_definition, self.schedule_type)
4904 schedule = raw_2_schedule(self.schedule_definition, self.schedule_type)
4905 return schedule
4905 return schedule
4906
4906
4907 @property
4907 @property
4908 def args(self):
4908 def args(self):
4909 try:
4909 try:
4910 return list(self.task_args or [])
4910 return list(self.task_args or [])
4911 except ValueError:
4911 except ValueError:
4912 return list()
4912 return list()
4913
4913
4914 @property
4914 @property
4915 def kwargs(self):
4915 def kwargs(self):
4916 try:
4916 try:
4917 return dict(self.task_kwargs or {})
4917 return dict(self.task_kwargs or {})
4918 except ValueError:
4918 except ValueError:
4919 return dict()
4919 return dict()
4920
4920
4921 def _as_raw(self, val):
4921 def _as_raw(self, val):
4922 if hasattr(val, 'de_coerce'):
4922 if hasattr(val, 'de_coerce'):
4923 val = val.de_coerce()
4923 val = val.de_coerce()
4924 if val:
4924 if val:
4925 val = json.dumps(val)
4925 val = json.dumps(val)
4926
4926
4927 return val
4927 return val
4928
4928
4929 @property
4929 @property
4930 def schedule_definition_raw(self):
4930 def schedule_definition_raw(self):
4931 return self._as_raw(self.schedule_definition)
4931 return self._as_raw(self.schedule_definition)
4932
4932
4933 @property
4933 @property
4934 def args_raw(self):
4934 def args_raw(self):
4935 return self._as_raw(self.task_args)
4935 return self._as_raw(self.task_args)
4936
4936
4937 @property
4937 @property
4938 def kwargs_raw(self):
4938 def kwargs_raw(self):
4939 return self._as_raw(self.task_kwargs)
4939 return self._as_raw(self.task_kwargs)
4940
4940
4941 def __repr__(self):
4941 def __repr__(self):
4942 return '<DB:ScheduleEntry({}:{})>'.format(
4942 return '<DB:ScheduleEntry({}:{})>'.format(
4943 self.schedule_entry_id, self.schedule_name)
4943 self.schedule_entry_id, self.schedule_name)
4944
4944
4945
4945
4946 @event.listens_for(ScheduleEntry, 'before_update')
4946 @event.listens_for(ScheduleEntry, 'before_update')
4947 def update_task_uid(mapper, connection, target):
4947 def update_task_uid(mapper, connection, target):
4948 target.task_uid = ScheduleEntry.get_uid(target)
4948 target.task_uid = ScheduleEntry.get_uid(target)
4949
4949
4950
4950
4951 @event.listens_for(ScheduleEntry, 'before_insert')
4951 @event.listens_for(ScheduleEntry, 'before_insert')
4952 def set_task_uid(mapper, connection, target):
4952 def set_task_uid(mapper, connection, target):
4953 target.task_uid = ScheduleEntry.get_uid(target)
4953 target.task_uid = ScheduleEntry.get_uid(target)
4954
4954
4955
4955
4956 class _BaseBranchPerms(BaseModel):
4956 class _BaseBranchPerms(BaseModel):
4957 @classmethod
4957 @classmethod
4958 def compute_hash(cls, value):
4958 def compute_hash(cls, value):
4959 return sha1_safe(value)
4959 return sha1_safe(value)
4960
4960
4961 @hybrid_property
4961 @hybrid_property
4962 def branch_pattern(self):
4962 def branch_pattern(self):
4963 return self._branch_pattern or '*'
4963 return self._branch_pattern or '*'
4964
4964
4965 @hybrid_property
4965 @hybrid_property
4966 def branch_hash(self):
4966 def branch_hash(self):
4967 return self._branch_hash
4967 return self._branch_hash
4968
4968
4969 def _validate_glob(self, value):
4969 def _validate_glob(self, value):
4970 re.compile('^' + glob2re(value) + '$')
4970 re.compile('^' + glob2re(value) + '$')
4971
4971
4972 @branch_pattern.setter
4972 @branch_pattern.setter
4973 def branch_pattern(self, value):
4973 def branch_pattern(self, value):
4974 self._validate_glob(value)
4974 self._validate_glob(value)
4975 self._branch_pattern = value or '*'
4975 self._branch_pattern = value or '*'
4976 # set the Hash when setting the branch pattern
4976 # set the Hash when setting the branch pattern
4977 self._branch_hash = self.compute_hash(self._branch_pattern)
4977 self._branch_hash = self.compute_hash(self._branch_pattern)
4978
4978
4979 def matches(self, branch):
4979 def matches(self, branch):
4980 """
4980 """
4981 Check if this the branch matches entry
4981 Check if this the branch matches entry
4982
4982
4983 :param branch: branch name for the commit
4983 :param branch: branch name for the commit
4984 """
4984 """
4985
4985
4986 branch = branch or ''
4986 branch = branch or ''
4987
4987
4988 branch_matches = True
4988 branch_matches = True
4989 if branch:
4989 if branch:
4990 branch_regex = re.compile('^' + glob2re(self.branch_pattern) + '$')
4990 branch_regex = re.compile('^' + glob2re(self.branch_pattern) + '$')
4991 branch_matches = bool(branch_regex.search(branch))
4991 branch_matches = bool(branch_regex.search(branch))
4992
4992
4993 return branch_matches
4993 return branch_matches
4994
4994
4995
4995
4996 class UserToRepoBranchPermission(Base, _BaseBranchPerms):
4996 class UserToRepoBranchPermission(Base, _BaseBranchPerms):
4997 __tablename__ = 'user_to_repo_branch_permissions'
4997 __tablename__ = 'user_to_repo_branch_permissions'
4998 __table_args__ = (
4998 __table_args__ = (
4999 base_table_args
4999 base_table_args
5000 )
5000 )
5001
5001
5002 branch_rule_id = Column('branch_rule_id', Integer(), primary_key=True)
5002 branch_rule_id = Column('branch_rule_id', Integer(), primary_key=True)
5003
5003
5004 repository_id = Column('repository_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
5004 repository_id = Column('repository_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
5005 repo = relationship('Repository', backref='user_branch_perms')
5005 repo = relationship('Repository', backref='user_branch_perms')
5006
5006
5007 permission_id = Column('permission_id', Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
5007 permission_id = Column('permission_id', Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
5008 permission = relationship('Permission')
5008 permission = relationship('Permission')
5009
5009
5010 rule_to_perm_id = Column('rule_to_perm_id', Integer(), ForeignKey('repo_to_perm.repo_to_perm_id'), nullable=False, unique=None, default=None)
5010 rule_to_perm_id = Column('rule_to_perm_id', Integer(), ForeignKey('repo_to_perm.repo_to_perm_id'), nullable=False, unique=None, default=None)
5011 user_repo_to_perm = relationship('UserRepoToPerm')
5011 user_repo_to_perm = relationship('UserRepoToPerm')
5012
5012
5013 rule_order = Column('rule_order', Integer(), nullable=False)
5013 rule_order = Column('rule_order', Integer(), nullable=False)
5014 _branch_pattern = Column('branch_pattern', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), default=u'*') # glob
5014 _branch_pattern = Column('branch_pattern', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), default=u'*') # glob
5015 _branch_hash = Column('branch_hash', UnicodeText().with_variant(UnicodeText(2048), 'mysql'))
5015 _branch_hash = Column('branch_hash', UnicodeText().with_variant(UnicodeText(2048), 'mysql'))
5016
5016
5017 def __unicode__(self):
5017 def __unicode__(self):
5018 return u'<UserBranchPermission(%s => %r)>' % (
5018 return u'<UserBranchPermission(%s => %r)>' % (
5019 self.user_repo_to_perm, self.branch_pattern)
5019 self.user_repo_to_perm, self.branch_pattern)
5020
5020
5021
5021
5022 class UserGroupToRepoBranchPermission(Base, _BaseBranchPerms):
5022 class UserGroupToRepoBranchPermission(Base, _BaseBranchPerms):
5023 __tablename__ = 'user_group_to_repo_branch_permissions'
5023 __tablename__ = 'user_group_to_repo_branch_permissions'
5024 __table_args__ = (
5024 __table_args__ = (
5025 base_table_args
5025 base_table_args
5026 )
5026 )
5027
5027
5028 branch_rule_id = Column('branch_rule_id', Integer(), primary_key=True)
5028 branch_rule_id = Column('branch_rule_id', Integer(), primary_key=True)
5029
5029
5030 repository_id = Column('repository_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
5030 repository_id = Column('repository_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
5031 repo = relationship('Repository', backref='user_group_branch_perms')
5031 repo = relationship('Repository', backref='user_group_branch_perms')
5032
5032
5033 permission_id = Column('permission_id', Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
5033 permission_id = Column('permission_id', Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
5034 permission = relationship('Permission')
5034 permission = relationship('Permission')
5035
5035
5036 rule_to_perm_id = Column('rule_to_perm_id', Integer(), ForeignKey('users_group_repo_to_perm.users_group_to_perm_id'), nullable=False, unique=None, default=None)
5036 rule_to_perm_id = Column('rule_to_perm_id', Integer(), ForeignKey('users_group_repo_to_perm.users_group_to_perm_id'), nullable=False, unique=None, default=None)
5037 user_group_repo_to_perm = relationship('UserGroupRepoToPerm')
5037 user_group_repo_to_perm = relationship('UserGroupRepoToPerm')
5038
5038
5039 rule_order = Column('rule_order', Integer(), nullable=False)
5039 rule_order = Column('rule_order', Integer(), nullable=False)
5040 _branch_pattern = Column('branch_pattern', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), default=u'*') # glob
5040 _branch_pattern = Column('branch_pattern', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), default=u'*') # glob
5041 _branch_hash = Column('branch_hash', UnicodeText().with_variant(UnicodeText(2048), 'mysql'))
5041 _branch_hash = Column('branch_hash', UnicodeText().with_variant(UnicodeText(2048), 'mysql'))
5042
5042
5043 def __unicode__(self):
5043 def __unicode__(self):
5044 return u'<UserBranchPermission(%s => %r)>' % (
5044 return u'<UserBranchPermission(%s => %r)>' % (
5045 self.user_group_repo_to_perm, self.branch_pattern)
5045 self.user_group_repo_to_perm, self.branch_pattern)
5046
5046
5047
5047
5048 class UserBookmark(Base, BaseModel):
5048 class UserBookmark(Base, BaseModel):
5049 __tablename__ = 'user_bookmarks'
5049 __tablename__ = 'user_bookmarks'
5050 __table_args__ = (
5050 __table_args__ = (
5051 UniqueConstraint('user_id', 'bookmark_repo_id'),
5051 UniqueConstraint('user_id', 'bookmark_repo_id'),
5052 UniqueConstraint('user_id', 'bookmark_repo_group_id'),
5052 UniqueConstraint('user_id', 'bookmark_repo_group_id'),
5053 UniqueConstraint('user_id', 'bookmark_position'),
5053 UniqueConstraint('user_id', 'bookmark_position'),
5054 base_table_args
5054 base_table_args
5055 )
5055 )
5056
5056
5057 user_bookmark_id = Column("user_bookmark_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
5057 user_bookmark_id = Column("user_bookmark_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
5058 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
5058 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
5059 position = Column("bookmark_position", Integer(), nullable=False)
5059 position = Column("bookmark_position", Integer(), nullable=False)
5060 title = Column("bookmark_title", String(255), nullable=True, unique=None, default=None)
5060 title = Column("bookmark_title", String(255), nullable=True, unique=None, default=None)
5061 redirect_url = Column("bookmark_redirect_url", String(10240), nullable=True, unique=None, default=None)
5061 redirect_url = Column("bookmark_redirect_url", String(10240), nullable=True, unique=None, default=None)
5062 created_on = Column("created_on", DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
5062 created_on = Column("created_on", DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
5063
5063
5064 bookmark_repo_id = Column("bookmark_repo_id", Integer(), ForeignKey("repositories.repo_id"), nullable=True, unique=None, default=None)
5064 bookmark_repo_id = Column("bookmark_repo_id", Integer(), ForeignKey("repositories.repo_id"), nullable=True, unique=None, default=None)
5065 bookmark_repo_group_id = Column("bookmark_repo_group_id", Integer(), ForeignKey("groups.group_id"), nullable=True, unique=None, default=None)
5065 bookmark_repo_group_id = Column("bookmark_repo_group_id", Integer(), ForeignKey("groups.group_id"), nullable=True, unique=None, default=None)
5066
5066
5067 user = relationship("User")
5067 user = relationship("User")
5068
5068
5069 repository = relationship("Repository")
5069 repository = relationship("Repository")
5070 repository_group = relationship("RepoGroup")
5070 repository_group = relationship("RepoGroup")
5071
5071
5072 @classmethod
5072 @classmethod
5073 def get_by_position_for_user(cls, position, user_id):
5073 def get_by_position_for_user(cls, position, user_id):
5074 return cls.query() \
5074 return cls.query() \
5075 .filter(UserBookmark.user_id == user_id) \
5075 .filter(UserBookmark.user_id == user_id) \
5076 .filter(UserBookmark.position == position).scalar()
5076 .filter(UserBookmark.position == position).scalar()
5077
5077
5078 @classmethod
5078 @classmethod
5079 def get_bookmarks_for_user(cls, user_id):
5079 def get_bookmarks_for_user(cls, user_id):
5080 return cls.query() \
5080 return cls.query() \
5081 .filter(UserBookmark.user_id == user_id) \
5081 .filter(UserBookmark.user_id == user_id) \
5082 .options(joinedload(UserBookmark.repository)) \
5082 .options(joinedload(UserBookmark.repository)) \
5083 .options(joinedload(UserBookmark.repository_group)) \
5083 .options(joinedload(UserBookmark.repository_group)) \
5084 .order_by(UserBookmark.position.asc()) \
5084 .order_by(UserBookmark.position.asc()) \
5085 .all()
5085 .all()
5086
5086
5087 def __unicode__(self):
5087 def __unicode__(self):
5088 return u'<UserBookmark(%d @ %r)>' % (self.position, self.redirect_url)
5088 return u'<UserBookmark(%s @ %r)>' % (self.position, self.redirect_url)
5089
5089
5090
5090
5091 class FileStore(Base, BaseModel):
5091 class FileStore(Base, BaseModel):
5092 __tablename__ = 'file_store'
5092 __tablename__ = 'file_store'
5093 __table_args__ = (
5093 __table_args__ = (
5094 base_table_args
5094 base_table_args
5095 )
5095 )
5096
5096
5097 file_store_id = Column('file_store_id', Integer(), primary_key=True)
5097 file_store_id = Column('file_store_id', Integer(), primary_key=True)
5098 file_uid = Column('file_uid', String(1024), nullable=False)
5098 file_uid = Column('file_uid', String(1024), nullable=False)
5099 file_display_name = Column('file_display_name', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), nullable=True)
5099 file_display_name = Column('file_display_name', UnicodeText().with_variant(UnicodeText(2048), 'mysql'), nullable=True)
5100 file_description = Column('file_description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'), nullable=True)
5100 file_description = Column('file_description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'), nullable=True)
5101 file_org_name = Column('file_org_name', UnicodeText().with_variant(UnicodeText(10240), 'mysql'), nullable=False)
5101 file_org_name = Column('file_org_name', UnicodeText().with_variant(UnicodeText(10240), 'mysql'), nullable=False)
5102
5102
5103 # sha256 hash
5103 # sha256 hash
5104 file_hash = Column('file_hash', String(512), nullable=False)
5104 file_hash = Column('file_hash', String(512), nullable=False)
5105 file_size = Column('file_size', Integer(), nullable=False)
5105 file_size = Column('file_size', Integer(), nullable=False)
5106
5106
5107 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
5107 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
5108 accessed_on = Column('accessed_on', DateTime(timezone=False), nullable=True)
5108 accessed_on = Column('accessed_on', DateTime(timezone=False), nullable=True)
5109 accessed_count = Column('accessed_count', Integer(), default=0)
5109 accessed_count = Column('accessed_count', Integer(), default=0)
5110
5110
5111 enabled = Column('enabled', Boolean(), nullable=False, default=True)
5111 enabled = Column('enabled', Boolean(), nullable=False, default=True)
5112
5112
5113 # if repo/repo_group reference is set, check for permissions
5113 # if repo/repo_group reference is set, check for permissions
5114 check_acl = Column('check_acl', Boolean(), nullable=False, default=True)
5114 check_acl = Column('check_acl', Boolean(), nullable=False, default=True)
5115
5115
5116 # hidden defines an attachment that should be hidden from showing in artifact listing
5116 # hidden defines an attachment that should be hidden from showing in artifact listing
5117 hidden = Column('hidden', Boolean(), nullable=False, default=False)
5117 hidden = Column('hidden', Boolean(), nullable=False, default=False)
5118
5118
5119 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
5119 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
5120 upload_user = relationship('User', lazy='joined', primaryjoin='User.user_id==FileStore.user_id')
5120 upload_user = relationship('User', lazy='joined', primaryjoin='User.user_id==FileStore.user_id')
5121
5121
5122 file_metadata = relationship('FileStoreMetadata', lazy='joined')
5122 file_metadata = relationship('FileStoreMetadata', lazy='joined')
5123
5123
5124 # scope limited to user, which requester have access to
5124 # scope limited to user, which requester have access to
5125 scope_user_id = Column(
5125 scope_user_id = Column(
5126 'scope_user_id', Integer(), ForeignKey('users.user_id'),
5126 'scope_user_id', Integer(), ForeignKey('users.user_id'),
5127 nullable=True, unique=None, default=None)
5127 nullable=True, unique=None, default=None)
5128 user = relationship('User', lazy='joined', primaryjoin='User.user_id==FileStore.scope_user_id')
5128 user = relationship('User', lazy='joined', primaryjoin='User.user_id==FileStore.scope_user_id')
5129
5129
5130 # scope limited to user group, which requester have access to
5130 # scope limited to user group, which requester have access to
5131 scope_user_group_id = Column(
5131 scope_user_group_id = Column(
5132 'scope_user_group_id', Integer(), ForeignKey('users_groups.users_group_id'),
5132 'scope_user_group_id', Integer(), ForeignKey('users_groups.users_group_id'),
5133 nullable=True, unique=None, default=None)
5133 nullable=True, unique=None, default=None)
5134 user_group = relationship('UserGroup', lazy='joined')
5134 user_group = relationship('UserGroup', lazy='joined')
5135
5135
5136 # scope limited to repo, which requester have access to
5136 # scope limited to repo, which requester have access to
5137 scope_repo_id = Column(
5137 scope_repo_id = Column(
5138 'scope_repo_id', Integer(), ForeignKey('repositories.repo_id'),
5138 'scope_repo_id', Integer(), ForeignKey('repositories.repo_id'),
5139 nullable=True, unique=None, default=None)
5139 nullable=True, unique=None, default=None)
5140 repo = relationship('Repository', lazy='joined')
5140 repo = relationship('Repository', lazy='joined')
5141
5141
5142 # scope limited to repo group, which requester have access to
5142 # scope limited to repo group, which requester have access to
5143 scope_repo_group_id = Column(
5143 scope_repo_group_id = Column(
5144 'scope_repo_group_id', Integer(), ForeignKey('groups.group_id'),
5144 'scope_repo_group_id', Integer(), ForeignKey('groups.group_id'),
5145 nullable=True, unique=None, default=None)
5145 nullable=True, unique=None, default=None)
5146 repo_group = relationship('RepoGroup', lazy='joined')
5146 repo_group = relationship('RepoGroup', lazy='joined')
5147
5147
5148 @classmethod
5148 @classmethod
5149 def create(cls, file_uid, filename, file_hash, file_size, file_display_name='',
5149 def create(cls, file_uid, filename, file_hash, file_size, file_display_name='',
5150 file_description='', enabled=True, hidden=False, check_acl=True,
5150 file_description='', enabled=True, hidden=False, check_acl=True,
5151 user_id=None, scope_user_id=None, scope_repo_id=None, scope_repo_group_id=None):
5151 user_id=None, scope_user_id=None, scope_repo_id=None, scope_repo_group_id=None):
5152
5152
5153 store_entry = FileStore()
5153 store_entry = FileStore()
5154 store_entry.file_uid = file_uid
5154 store_entry.file_uid = file_uid
5155 store_entry.file_display_name = file_display_name
5155 store_entry.file_display_name = file_display_name
5156 store_entry.file_org_name = filename
5156 store_entry.file_org_name = filename
5157 store_entry.file_size = file_size
5157 store_entry.file_size = file_size
5158 store_entry.file_hash = file_hash
5158 store_entry.file_hash = file_hash
5159 store_entry.file_description = file_description
5159 store_entry.file_description = file_description
5160
5160
5161 store_entry.check_acl = check_acl
5161 store_entry.check_acl = check_acl
5162 store_entry.enabled = enabled
5162 store_entry.enabled = enabled
5163 store_entry.hidden = hidden
5163 store_entry.hidden = hidden
5164
5164
5165 store_entry.user_id = user_id
5165 store_entry.user_id = user_id
5166 store_entry.scope_user_id = scope_user_id
5166 store_entry.scope_user_id = scope_user_id
5167 store_entry.scope_repo_id = scope_repo_id
5167 store_entry.scope_repo_id = scope_repo_id
5168 store_entry.scope_repo_group_id = scope_repo_group_id
5168 store_entry.scope_repo_group_id = scope_repo_group_id
5169
5169
5170 return store_entry
5170 return store_entry
5171
5171
5172 @classmethod
5172 @classmethod
5173 def store_metadata(cls, file_store_id, args, commit=True):
5173 def store_metadata(cls, file_store_id, args, commit=True):
5174 file_store = FileStore.get(file_store_id)
5174 file_store = FileStore.get(file_store_id)
5175 if file_store is None:
5175 if file_store is None:
5176 return
5176 return
5177
5177
5178 for section, key, value, value_type in args:
5178 for section, key, value, value_type in args:
5179 meta_entry = FileStoreMetadata()
5179 meta_entry = FileStoreMetadata()
5180 meta_entry.file_store = file_store
5180 meta_entry.file_store = file_store
5181 meta_entry.file_store_meta_section = section
5181 meta_entry.file_store_meta_section = section
5182 meta_entry.file_store_meta_key = key
5182 meta_entry.file_store_meta_key = key
5183 meta_entry.file_store_meta_value_type = value_type
5183 meta_entry.file_store_meta_value_type = value_type
5184 meta_entry.file_store_meta_value = value
5184 meta_entry.file_store_meta_value = value
5185
5185
5186 Session().add(meta_entry)
5186 Session().add(meta_entry)
5187
5187
5188 if commit:
5188 if commit:
5189 Session().commit()
5189 Session().commit()
5190
5190
5191 @classmethod
5191 @classmethod
5192 def bump_access_counter(cls, file_uid, commit=True):
5192 def bump_access_counter(cls, file_uid, commit=True):
5193 FileStore().query()\
5193 FileStore().query()\
5194 .filter(FileStore.file_uid == file_uid)\
5194 .filter(FileStore.file_uid == file_uid)\
5195 .update({FileStore.accessed_count: (FileStore.accessed_count + 1),
5195 .update({FileStore.accessed_count: (FileStore.accessed_count + 1),
5196 FileStore.accessed_on: datetime.datetime.now()})
5196 FileStore.accessed_on: datetime.datetime.now()})
5197 if commit:
5197 if commit:
5198 Session().commit()
5198 Session().commit()
5199
5199
5200 def __repr__(self):
5200 def __repr__(self):
5201 return '<FileStore({})>'.format(self.file_store_id)
5201 return '<FileStore({})>'.format(self.file_store_id)
5202
5202
5203
5203
5204 class FileStoreMetadata(Base, BaseModel):
5204 class FileStoreMetadata(Base, BaseModel):
5205 __tablename__ = 'file_store_metadata'
5205 __tablename__ = 'file_store_metadata'
5206 __table_args__ = (
5206 __table_args__ = (
5207 UniqueConstraint('file_store_meta_section', 'file_store_meta_key'),
5207 UniqueConstraint('file_store_meta_section', 'file_store_meta_key'),
5208 Index('file_store_meta_section_idx', 'file_store_meta_section'),
5208 Index('file_store_meta_section_idx', 'file_store_meta_section'),
5209 Index('file_store_meta_key_idx', 'file_store_meta_key'),
5209 Index('file_store_meta_key_idx', 'file_store_meta_key'),
5210 base_table_args
5210 base_table_args
5211 )
5211 )
5212 SETTINGS_TYPES = {
5212 SETTINGS_TYPES = {
5213 'str': safe_str,
5213 'str': safe_str,
5214 'int': safe_int,
5214 'int': safe_int,
5215 'unicode': safe_unicode,
5215 'unicode': safe_unicode,
5216 'bool': str2bool,
5216 'bool': str2bool,
5217 'list': functools.partial(aslist, sep=',')
5217 'list': functools.partial(aslist, sep=',')
5218 }
5218 }
5219
5219
5220 file_store_meta_id = Column(
5220 file_store_meta_id = Column(
5221 "file_store_meta_id", Integer(), nullable=False, unique=True, default=None,
5221 "file_store_meta_id", Integer(), nullable=False, unique=True, default=None,
5222 primary_key=True)
5222 primary_key=True)
5223 file_store_meta_section = Column(
5223 file_store_meta_section = Column(
5224 "file_store_meta_section", UnicodeText().with_variant(UnicodeText(1024), 'mysql'),
5224 "file_store_meta_section", UnicodeText().with_variant(UnicodeText(1024), 'mysql'),
5225 nullable=True, unique=None, default=None)
5225 nullable=True, unique=None, default=None)
5226 file_store_meta_key = Column(
5226 file_store_meta_key = Column(
5227 "file_store_meta_key", UnicodeText().with_variant(UnicodeText(1024), 'mysql'),
5227 "file_store_meta_key", UnicodeText().with_variant(UnicodeText(1024), 'mysql'),
5228 nullable=True, unique=None, default=None)
5228 nullable=True, unique=None, default=None)
5229 _file_store_meta_value = Column(
5229 _file_store_meta_value = Column(
5230 "file_store_meta_value", UnicodeText().with_variant(UnicodeText(20480), 'mysql'),
5230 "file_store_meta_value", UnicodeText().with_variant(UnicodeText(20480), 'mysql'),
5231 nullable=True, unique=None, default=None)
5231 nullable=True, unique=None, default=None)
5232 _file_store_meta_value_type = Column(
5232 _file_store_meta_value_type = Column(
5233 "file_store_meta_value_type", String(255), nullable=True, unique=None,
5233 "file_store_meta_value_type", String(255), nullable=True, unique=None,
5234 default='unicode')
5234 default='unicode')
5235
5235
5236 file_store_id = Column(
5236 file_store_id = Column(
5237 'file_store_id', Integer(), ForeignKey('file_store.file_store_id'),
5237 'file_store_id', Integer(), ForeignKey('file_store.file_store_id'),
5238 nullable=True, unique=None, default=None)
5238 nullable=True, unique=None, default=None)
5239
5239
5240 file_store = relationship('FileStore', lazy='joined')
5240 file_store = relationship('FileStore', lazy='joined')
5241
5241
5242 @hybrid_property
5242 @hybrid_property
5243 def file_store_meta_value(self):
5243 def file_store_meta_value(self):
5244 v = self._file_store_meta_value
5244 v = self._file_store_meta_value
5245 _type = self._file_store_meta_value
5245 _type = self._file_store_meta_value
5246 if _type:
5246 if _type:
5247 # e.g unicode.encrypted == unicode
5247 # e.g unicode.encrypted == unicode
5248 _type = self._file_store_meta_value.split('.')[0]
5248 _type = self._file_store_meta_value.split('.')[0]
5249 # decode the encrypted value
5249 # decode the encrypted value
5250 if '.encrypted' in self._file_store_meta_value_type:
5250 if '.encrypted' in self._file_store_meta_value_type:
5251 cipher = EncryptedTextValue()
5251 cipher = EncryptedTextValue()
5252 v = safe_unicode(cipher.process_result_value(v, None))
5252 v = safe_unicode(cipher.process_result_value(v, None))
5253
5253
5254 converter = self.SETTINGS_TYPES.get(_type) or self.SETTINGS_TYPES['unicode']
5254 converter = self.SETTINGS_TYPES.get(_type) or self.SETTINGS_TYPES['unicode']
5255 return converter(v)
5255 return converter(v)
5256
5256
5257 @file_store_meta_value.setter
5257 @file_store_meta_value.setter
5258 def file_store_meta_value(self, val):
5258 def file_store_meta_value(self, val):
5259 val = safe_unicode(val)
5259 val = safe_unicode(val)
5260 # encode the encrypted value
5260 # encode the encrypted value
5261 if '.encrypted' in self.file_store_meta_value_type:
5261 if '.encrypted' in self.file_store_meta_value_type:
5262 cipher = EncryptedTextValue()
5262 cipher = EncryptedTextValue()
5263 val = safe_unicode(cipher.process_bind_param(val, None))
5263 val = safe_unicode(cipher.process_bind_param(val, None))
5264 self._file_store_meta_value = val
5264 self._file_store_meta_value = val
5265
5265
5266 @hybrid_property
5266 @hybrid_property
5267 def file_store_meta_value_type(self):
5267 def file_store_meta_value_type(self):
5268 return self._file_store_meta_value_type
5268 return self._file_store_meta_value_type
5269
5269
5270 @file_store_meta_value_type.setter
5270 @file_store_meta_value_type.setter
5271 def file_store_meta_value_type(self, val):
5271 def file_store_meta_value_type(self, val):
5272 # e.g unicode.encrypted
5272 # e.g unicode.encrypted
5273 if val.split('.')[0] not in self.SETTINGS_TYPES:
5273 if val.split('.')[0] not in self.SETTINGS_TYPES:
5274 raise Exception('type must be one of %s got %s'
5274 raise Exception('type must be one of %s got %s'
5275 % (self.SETTINGS_TYPES.keys(), val))
5275 % (self.SETTINGS_TYPES.keys(), val))
5276 self._file_store_meta_value_type = val
5276 self._file_store_meta_value_type = val
5277
5277
5278 def __repr__(self):
5278 def __repr__(self):
5279 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.file_store_meta_section,
5279 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.file_store_meta_section,
5280 self.file_store_meta_key, self.file_store_meta_value)
5280 self.file_store_meta_key, self.file_store_meta_value)
5281
5281
5282
5282
5283 class DbMigrateVersion(Base, BaseModel):
5283 class DbMigrateVersion(Base, BaseModel):
5284 __tablename__ = 'db_migrate_version'
5284 __tablename__ = 'db_migrate_version'
5285 __table_args__ = (
5285 __table_args__ = (
5286 base_table_args,
5286 base_table_args,
5287 )
5287 )
5288
5288
5289 repository_id = Column('repository_id', String(250), primary_key=True)
5289 repository_id = Column('repository_id', String(250), primary_key=True)
5290 repository_path = Column('repository_path', Text)
5290 repository_path = Column('repository_path', Text)
5291 version = Column('version', Integer)
5291 version = Column('version', Integer)
5292
5292
5293 @classmethod
5293 @classmethod
5294 def set_version(cls, version):
5294 def set_version(cls, version):
5295 """
5295 """
5296 Helper for forcing a different version, usually for debugging purposes via ishell.
5296 Helper for forcing a different version, usually for debugging purposes via ishell.
5297 """
5297 """
5298 ver = DbMigrateVersion.query().first()
5298 ver = DbMigrateVersion.query().first()
5299 ver.version = version
5299 ver.version = version
5300 Session().commit()
5300 Session().commit()
5301
5301
5302
5302
5303 class DbSession(Base, BaseModel):
5303 class DbSession(Base, BaseModel):
5304 __tablename__ = 'db_session'
5304 __tablename__ = 'db_session'
5305 __table_args__ = (
5305 __table_args__ = (
5306 base_table_args,
5306 base_table_args,
5307 )
5307 )
5308
5308
5309 def __repr__(self):
5309 def __repr__(self):
5310 return '<DB:DbSession({})>'.format(self.id)
5310 return '<DB:DbSession({})>'.format(self.id)
5311
5311
5312 id = Column('id', Integer())
5312 id = Column('id', Integer())
5313 namespace = Column('namespace', String(255), primary_key=True)
5313 namespace = Column('namespace', String(255), primary_key=True)
5314 accessed = Column('accessed', DateTime, nullable=False)
5314 accessed = Column('accessed', DateTime, nullable=False)
5315 created = Column('created', DateTime, nullable=False)
5315 created = Column('created', DateTime, nullable=False)
5316 data = Column('data', PickleType, nullable=False)
5316 data = Column('data', PickleType, nullable=False)
@@ -1,225 +1,226 b''
1 <%namespace name="dt" file="/data_table/_dt_elements.mako"/>
1 <%namespace name="dt" file="/data_table/_dt_elements.mako"/>
2
2
3 <%def name="form_item(count, position=None, title=None, redirect_url=None, repo=None, repo_group=None)">
3 <%def name="form_item(position=None, title=None, redirect_url=None, repo=None, repo_group=None)">
4 <tr>
4 <tr>
5 <td class="td-align-top" >
5 <td class="td-align-top" >
6 <div class="label">
6 <div class="label">
7 <label for="position">${_('Position')}:</label>
7 <label for="position">${_('Position')}:</label>
8 </div>
8 </div>
9 <div class="input">
9 <div class="input">
10 <input type="text" name="position" value="${position or count}" style="width: 40px"/>
10 <input type="text" name="position" value="${position}" style="width: 40px"/>
11 ${h.hidden('cur_position', position)}
11 </div>
12 </div>
12 </td>
13 </td>
13
14
14 <td>
15 <td>
15 <div class="label">
16 <div class="label">
16 <label for="title">${_('Bookmark title (max 30 characters, optional)')}:</label>
17 <label for="title">${_('Bookmark title (max 30 characters, optional)')}:</label>
17 </div>
18 </div>
18 <div class="input">
19 <div class="input">
19 <input type="text" name="title" value="${title}" style="width: 300px" maxlength="30"/>
20 <input type="text" name="title" value="${title}" style="width: 300px" maxlength="30"/>
20
21
21 <div class="field pull-right">
22 <div class="field pull-right">
22 <div>
23 <div>
23 <label class="btn-link btn-danger">${_('Clear')}:</label>
24 <label class="btn-link btn-danger">${_('Clear')}:</label>
24 ${h.checkbox('remove', value=True)}
25 ${h.checkbox('remove', value=True)}
25 </div>
26 </div>
26 </div>
27 </div>
27 </div>
28 </div>
28 <p class="help-block help-block-inline" >
29 <p class="help-block help-block-inline" >
29 ${_('Server URL is available as ${server_url} variable. E.g. Redirect url: ${server_url}/_admin/exception_tracker')}
30 ${_('Server URL is available as ${server_url} variable. E.g. Redirect url: ${server_url}/_admin/exception_tracker')}
30 </p>
31 </p>
31
32
32 <div class="label">
33 <div class="label">
33 <label for="redirect_url">${_('Redirect URL')}:</label>
34 <label for="redirect_url">${_('Redirect URL')}:</label>
34 </div>
35 </div>
35 <div class="input">
36 <div class="input">
36 <input type="text" name="redirect_url" value="${redirect_url}" style="width: 600px"/>
37 <input type="text" name="redirect_url" value="${redirect_url}" style="width: 600px"/>
37 </div>
38 </div>
38
39
39
40
40 <div class="select">
41 <div class="select">
41 % if repo:
42 % if repo:
42 <div class="label">
43 <div class="label">
43 <label for="redirect_url">${_('Repository template')}:</label>
44 <label for="redirect_url">${_('Repository template')}:</label>
44 </div>
45 </div>
45 ${dt.repo_name(name=repo.repo_name, rtype=repo.repo_type,rstate=None,private=None,archived=False,fork_of=False)}
46 ${dt.repo_name(name=repo.repo_name, rtype=repo.repo_type,rstate=None,private=None,archived=False,fork_of=False)}
46 ${h.hidden('bookmark_repo', repo.repo_id)}
47 ${h.hidden('bookmark_repo', repo.repo_id)}
47 % elif repo_group:
48 % elif repo_group:
48 <div class="label">
49 <div class="label">
49 <label for="redirect_url">${_('Repository group template')}:</label>
50 <label for="redirect_url">${_('Repository group template')}:</label>
50 </div>
51 </div>
51 ${dt.repo_group_name(repo_group.group_name)}
52 ${dt.repo_group_name(repo_group.group_name)}
52 ${h.hidden('bookmark_repo_group', repo_group.group_id)}
53 ${h.hidden('bookmark_repo_group', repo_group.group_id)}
53 % else:
54 % else:
54 <div class="label">
55 <div class="label">
55 <label for="redirect_url">${_('Template Repository or Repository group')}:</label>
56 <label for="redirect_url">${_('Template Repository or Repository group')}:</label>
56 </div>
57 </div>
57 ${h.hidden('bookmark_repo', class_='bookmark_repo')}
58 ${h.hidden('bookmark_repo', class_='bookmark_repo')}
58 <span style="padding-right:15px">OR</span>
59 <span style="padding-right:15px">OR</span>
59 ${h.hidden('bookmark_repo_group', class_='bookmark_repo_group')}
60 ${h.hidden('bookmark_repo_group', class_='bookmark_repo_group')}
60 % endif
61 % endif
61 </div>
62 </div>
62
63
63 <p class="help-block help-block-inline" >
64 <p class="help-block help-block-inline" >
64 % if repo:
65 % if repo:
65 ${_('Available as ${repo_url} e.g. Redirect url: ${repo_url}/changelog')}
66 ${_('Available as ${repo_url} e.g. Redirect url: ${repo_url}/changelog')}
66 % elif repo_group:
67 % elif repo_group:
67 ${_('Available as ${repo_group_url} e.g. Redirect url: ${repo_group_url}')}
68 ${_('Available as ${repo_group_url} e.g. Redirect url: ${repo_group_url}')}
68 % else:
69 % else:
69 ${_('Available as full url variables in redirect url. i.e: ${repo_url}, ${repo_group_url}.')}
70 ${_('Available as full url variables in redirect url. i.e: ${repo_url}, ${repo_group_url}.')}
70 % endif
71 % endif
71 </p>
72 </p>
72 </td>
73 </td>
73
74
74 </tr>
75 </tr>
75 </%def>
76 </%def>
76
77
77 <div class="panel panel-default">
78 <div class="panel panel-default">
78 <div class="panel-heading">
79 <div class="panel-heading">
79 <h3 class="panel-title">${_('Your Bookmarks')}</h3>
80 <h3 class="panel-title">${_('Your Bookmarks')}</h3>
80 </div>
81 </div>
81
82
82 <div class="panel-body">
83 <div class="panel-body">
83 <p>
84 <p>
84 ${_('Store upto 10 bookmark links to favorite repositories, external issue tracker or CI server. ')}
85 ${_('Store upto 10 bookmark links to favorite repositories, external issue tracker or CI server. ')}
85 <br/>
86 <br/>
86 ${_('Bookmarks are accessible from your username dropdown or by keyboard shortcut `g 0-9`')}
87 ${_('Bookmarks are accessible from your username dropdown or by keyboard shortcut `g 0-9`')}
87 </p>
88 </p>
88
89
89 ${h.secure_form(h.route_path('my_account_bookmarks_update'), request=request)}
90 ${h.secure_form(h.route_path('my_account_bookmarks_update'), request=request)}
90 <div class="form-vertical">
91 <div class="form-vertical">
91 <table class="rctable">
92 <table class="rctable">
92 ## generate always 10 entries
93 ## generate always 10 entries
93 <input type="hidden" name="__start__" value="bookmarks:sequence"/>
94 <input type="hidden" name="__start__" value="bookmarks:sequence"/>
94 % for cnt, item in enumerate((c.bookmark_items + [None for i in range(10)])[:10]):
95 % for item in (c.bookmark_items + [None for i in range(10)])[:10]:
95 <input type="hidden" name="__start__" value="bookmark:mapping"/>
96 <input type="hidden" name="__start__" value="bookmark:mapping"/>
96 % if item is None:
97 % if item is None:
97 ## empty placehodlder
98 ## empty placehodlder
98 ${form_item(cnt)}
99 ${form_item()}
99 % else:
100 % else:
100 ## actual entry
101 ## actual entry
101 ${form_item(cnt, position=item.position, title=item.title, redirect_url=item.redirect_url, repo=item.repository, repo_group=item.repository_group)}
102 ${form_item(position=item.position, title=item.title, redirect_url=item.redirect_url, repo=item.repository, repo_group=item.repository_group)}
102 % endif
103 % endif
103 <input type="hidden" name="__end__" value="bookmark:mapping"/>
104 <input type="hidden" name="__end__" value="bookmark:mapping"/>
104 % endfor
105 % endfor
105 <input type="hidden" name="__end__" value="bookmarks:sequence"/>
106 <input type="hidden" name="__end__" value="bookmarks:sequence"/>
106 </table>
107 </table>
107 <div class="buttons">
108 <div class="buttons">
108 ${h.submit('save',_('Save'),class_="btn")}
109 ${h.submit('save',_('Save'),class_="btn")}
109 </div>
110 </div>
110 </div>
111 </div>
111 ${h.end_form()}
112 ${h.end_form()}
112 </div>
113 </div>
113 </div>
114 </div>
114
115
115 <script>
116 <script>
116 $(document).ready(function(){
117 $(document).ready(function(){
117
118
118
119
119 var repoFilter = function (data) {
120 var repoFilter = function (data) {
120 var results = [];
121 var results = [];
121
122
122 if (!data.results[0]) {
123 if (!data.results[0]) {
123 return data
124 return data
124 }
125 }
125
126
126 $.each(data.results[0].children, function () {
127 $.each(data.results[0].children, function () {
127 // replace name to ID for submision
128 // replace name to ID for submision
128 this.id = this.repo_id;
129 this.id = this.repo_id;
129 results.push(this);
130 results.push(this);
130 });
131 });
131
132
132 data.results[0].children = results;
133 data.results[0].children = results;
133 return data;
134 return data;
134 };
135 };
135
136
136
137
137 $(".bookmark_repo").select2({
138 $(".bookmark_repo").select2({
138 cachedDataSource: {},
139 cachedDataSource: {},
139 minimumInputLength: 2,
140 minimumInputLength: 2,
140 placeholder: "${_('repository')}",
141 placeholder: "${_('repository')}",
141 dropdownAutoWidth: true,
142 dropdownAutoWidth: true,
142 containerCssClass: "drop-menu",
143 containerCssClass: "drop-menu",
143 dropdownCssClass: "drop-menu-dropdown",
144 dropdownCssClass: "drop-menu-dropdown",
144 formatResult: formatRepoResult,
145 formatResult: formatRepoResult,
145 query: $.debounce(250, function (query) {
146 query: $.debounce(250, function (query) {
146 self = this;
147 self = this;
147 var cacheKey = query.term;
148 var cacheKey = query.term;
148 var cachedData = self.cachedDataSource[cacheKey];
149 var cachedData = self.cachedDataSource[cacheKey];
149
150
150 if (cachedData) {
151 if (cachedData) {
151 query.callback({results: cachedData.results});
152 query.callback({results: cachedData.results});
152 } else {
153 } else {
153 $.ajax({
154 $.ajax({
154 url: pyroutes.url('repo_list_data'),
155 url: pyroutes.url('repo_list_data'),
155 data: {'query': query.term},
156 data: {'query': query.term},
156 dataType: 'json',
157 dataType: 'json',
157 type: 'GET',
158 type: 'GET',
158 success: function (data) {
159 success: function (data) {
159 data = repoFilter(data);
160 data = repoFilter(data);
160 self.cachedDataSource[cacheKey] = data;
161 self.cachedDataSource[cacheKey] = data;
161 query.callback({results: data.results});
162 query.callback({results: data.results});
162 },
163 },
163 error: function (data, textStatus, errorThrown) {
164 error: function (data, textStatus, errorThrown) {
164 alert("Error while fetching entries.\nError code {0} ({1}).".format(data.status, data.statusText));
165 alert("Error while fetching entries.\nError code {0} ({1}).".format(data.status, data.statusText));
165 }
166 }
166 })
167 })
167 }
168 }
168 }),
169 }),
169 });
170 });
170
171
171 var repoGroupFilter = function (data) {
172 var repoGroupFilter = function (data) {
172 var results = [];
173 var results = [];
173
174
174 if (!data.results[0]) {
175 if (!data.results[0]) {
175 return data
176 return data
176 }
177 }
177
178
178 $.each(data.results[0].children, function () {
179 $.each(data.results[0].children, function () {
179 // replace name to ID for submision
180 // replace name to ID for submision
180 this.id = this.repo_group_id;
181 this.id = this.repo_group_id;
181 results.push(this);
182 results.push(this);
182 });
183 });
183
184
184 data.results[0].children = results;
185 data.results[0].children = results;
185 return data;
186 return data;
186 };
187 };
187
188
188 $(".bookmark_repo_group").select2({
189 $(".bookmark_repo_group").select2({
189 cachedDataSource: {},
190 cachedDataSource: {},
190 minimumInputLength: 2,
191 minimumInputLength: 2,
191 placeholder: "${_('repository group')}",
192 placeholder: "${_('repository group')}",
192 dropdownAutoWidth: true,
193 dropdownAutoWidth: true,
193 containerCssClass: "drop-menu",
194 containerCssClass: "drop-menu",
194 dropdownCssClass: "drop-menu-dropdown",
195 dropdownCssClass: "drop-menu-dropdown",
195 formatResult: formatRepoGroupResult,
196 formatResult: formatRepoGroupResult,
196 query: $.debounce(250, function (query) {
197 query: $.debounce(250, function (query) {
197 self = this;
198 self = this;
198 var cacheKey = query.term;
199 var cacheKey = query.term;
199 var cachedData = self.cachedDataSource[cacheKey];
200 var cachedData = self.cachedDataSource[cacheKey];
200
201
201 if (cachedData) {
202 if (cachedData) {
202 query.callback({results: cachedData.results});
203 query.callback({results: cachedData.results});
203 } else {
204 } else {
204 $.ajax({
205 $.ajax({
205 url: pyroutes.url('repo_group_list_data'),
206 url: pyroutes.url('repo_group_list_data'),
206 data: {'query': query.term},
207 data: {'query': query.term},
207 dataType: 'json',
208 dataType: 'json',
208 type: 'GET',
209 type: 'GET',
209 success: function (data) {
210 success: function (data) {
210 data = repoGroupFilter(data);
211 data = repoGroupFilter(data);
211 self.cachedDataSource[cacheKey] = data;
212 self.cachedDataSource[cacheKey] = data;
212 query.callback({results: data.results});
213 query.callback({results: data.results});
213 },
214 },
214 error: function (data, textStatus, errorThrown) {
215 error: function (data, textStatus, errorThrown) {
215 alert("Error while fetching entries.\nError code {0} ({1}).".format(data.status, data.statusText));
216 alert("Error while fetching entries.\nError code {0} ({1}).".format(data.status, data.statusText));
216 }
217 }
217 })
218 })
218 }
219 }
219 })
220 })
220 });
221 });
221
222
222
223
223 });
224 });
224
225
225 </script>
226 </script>
General Comments 0
You need to be logged in to leave comments. Login now