##// END OF EJS Templates
pyramid: don't use pylons translation string in db models.
marcink -
r1917:3fcf87f0 default
parent child Browse files
Show More
@@ -1,4117 +1,4117 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2017 RhodeCode GmbH
3 # Copyright (C) 2010-2017 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 hashlib
28 import hashlib
29 import logging
29 import logging
30 import datetime
30 import datetime
31 import warnings
31 import warnings
32 import ipaddress
32 import ipaddress
33 import functools
33 import functools
34 import traceback
34 import traceback
35 import collections
35 import collections
36
36
37
37
38 from sqlalchemy import *
38 from sqlalchemy import *
39 from sqlalchemy.ext.declarative import declared_attr
39 from sqlalchemy.ext.declarative import declared_attr
40 from sqlalchemy.ext.hybrid import hybrid_property
40 from sqlalchemy.ext.hybrid import hybrid_property
41 from sqlalchemy.orm import (
41 from sqlalchemy.orm import (
42 relationship, joinedload, class_mapper, validates, aliased)
42 relationship, joinedload, class_mapper, validates, aliased)
43 from sqlalchemy.sql.expression import true
43 from sqlalchemy.sql.expression import true
44 from beaker.cache import cache_region
44 from beaker.cache import cache_region
45 from zope.cachedescriptors.property import Lazy as LazyProperty
45 from zope.cachedescriptors.property import Lazy as LazyProperty
46
46
47 from pylons.i18n.translation import lazy_ugettext as _
48 from pyramid.threadlocal import get_current_request
47 from pyramid.threadlocal import get_current_request
49
48
49 from rhodecode.translation import _
50 from rhodecode.lib.vcs import get_vcs_instance
50 from rhodecode.lib.vcs import get_vcs_instance
51 from rhodecode.lib.vcs.backends.base import EmptyCommit, Reference
51 from rhodecode.lib.vcs.backends.base import EmptyCommit, Reference
52 from rhodecode.lib.utils2 import (
52 from rhodecode.lib.utils2 import (
53 str2bool, safe_str, get_commit_safe, safe_unicode, md5_safe,
53 str2bool, safe_str, get_commit_safe, safe_unicode, md5_safe,
54 time_to_datetime, aslist, Optional, safe_int, get_clone_url, AttributeDict,
54 time_to_datetime, aslist, Optional, safe_int, get_clone_url, AttributeDict,
55 glob2re, StrictAttributeDict, cleaned_uri)
55 glob2re, StrictAttributeDict, cleaned_uri)
56 from rhodecode.lib.jsonalchemy import MutationObj, MutationList, JsonType
56 from rhodecode.lib.jsonalchemy import MutationObj, MutationList, JsonType
57 from rhodecode.lib.ext_json import json
57 from rhodecode.lib.ext_json import json
58 from rhodecode.lib.caching_query import FromCache
58 from rhodecode.lib.caching_query import FromCache
59 from rhodecode.lib.encrypt import AESCipher
59 from rhodecode.lib.encrypt import AESCipher
60
60
61 from rhodecode.model.meta import Base, Session
61 from rhodecode.model.meta import Base, Session
62
62
63 URL_SEP = '/'
63 URL_SEP = '/'
64 log = logging.getLogger(__name__)
64 log = logging.getLogger(__name__)
65
65
66 # =============================================================================
66 # =============================================================================
67 # BASE CLASSES
67 # BASE CLASSES
68 # =============================================================================
68 # =============================================================================
69
69
70 # this is propagated from .ini file rhodecode.encrypted_values.secret or
70 # this is propagated from .ini file rhodecode.encrypted_values.secret or
71 # beaker.session.secret if first is not set.
71 # beaker.session.secret if first is not set.
72 # and initialized at environment.py
72 # and initialized at environment.py
73 ENCRYPTION_KEY = None
73 ENCRYPTION_KEY = None
74
74
75 # used to sort permissions by types, '#' used here is not allowed to be in
75 # used to sort permissions by types, '#' used here is not allowed to be in
76 # usernames, and it's very early in sorted string.printable table.
76 # usernames, and it's very early in sorted string.printable table.
77 PERMISSION_TYPE_SORT = {
77 PERMISSION_TYPE_SORT = {
78 'admin': '####',
78 'admin': '####',
79 'write': '###',
79 'write': '###',
80 'read': '##',
80 'read': '##',
81 'none': '#',
81 'none': '#',
82 }
82 }
83
83
84
84
85 def display_sort(obj):
85 def display_sort(obj):
86 """
86 """
87 Sort function used to sort permissions in .permissions() function of
87 Sort function used to sort permissions in .permissions() function of
88 Repository, RepoGroup, UserGroup. Also it put the default user in front
88 Repository, RepoGroup, UserGroup. Also it put the default user in front
89 of all other resources
89 of all other resources
90 """
90 """
91
91
92 if obj.username == User.DEFAULT_USER:
92 if obj.username == User.DEFAULT_USER:
93 return '#####'
93 return '#####'
94 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
94 prefix = PERMISSION_TYPE_SORT.get(obj.permission.split('.')[-1], '')
95 return prefix + obj.username
95 return prefix + obj.username
96
96
97
97
98 def _hash_key(k):
98 def _hash_key(k):
99 return md5_safe(k)
99 return md5_safe(k)
100
100
101
101
102 class EncryptedTextValue(TypeDecorator):
102 class EncryptedTextValue(TypeDecorator):
103 """
103 """
104 Special column for encrypted long text data, use like::
104 Special column for encrypted long text data, use like::
105
105
106 value = Column("encrypted_value", EncryptedValue(), nullable=False)
106 value = Column("encrypted_value", EncryptedValue(), nullable=False)
107
107
108 This column is intelligent so if value is in unencrypted form it return
108 This column is intelligent so if value is in unencrypted form it return
109 unencrypted form, but on save it always encrypts
109 unencrypted form, but on save it always encrypts
110 """
110 """
111 impl = Text
111 impl = Text
112
112
113 def process_bind_param(self, value, dialect):
113 def process_bind_param(self, value, dialect):
114 if not value:
114 if not value:
115 return value
115 return value
116 if value.startswith('enc$aes$') or value.startswith('enc$aes_hmac$'):
116 if value.startswith('enc$aes$') or value.startswith('enc$aes_hmac$'):
117 # protect against double encrypting if someone manually starts
117 # protect against double encrypting if someone manually starts
118 # doing
118 # doing
119 raise ValueError('value needs to be in unencrypted format, ie. '
119 raise ValueError('value needs to be in unencrypted format, ie. '
120 'not starting with enc$aes')
120 'not starting with enc$aes')
121 return 'enc$aes_hmac$%s' % AESCipher(
121 return 'enc$aes_hmac$%s' % AESCipher(
122 ENCRYPTION_KEY, hmac=True).encrypt(value)
122 ENCRYPTION_KEY, hmac=True).encrypt(value)
123
123
124 def process_result_value(self, value, dialect):
124 def process_result_value(self, value, dialect):
125 import rhodecode
125 import rhodecode
126
126
127 if not value:
127 if not value:
128 return value
128 return value
129
129
130 parts = value.split('$', 3)
130 parts = value.split('$', 3)
131 if not len(parts) == 3:
131 if not len(parts) == 3:
132 # probably not encrypted values
132 # probably not encrypted values
133 return value
133 return value
134 else:
134 else:
135 if parts[0] != 'enc':
135 if parts[0] != 'enc':
136 # parts ok but without our header ?
136 # parts ok but without our header ?
137 return value
137 return value
138 enc_strict_mode = str2bool(rhodecode.CONFIG.get(
138 enc_strict_mode = str2bool(rhodecode.CONFIG.get(
139 'rhodecode.encrypted_values.strict') or True)
139 'rhodecode.encrypted_values.strict') or True)
140 # at that stage we know it's our encryption
140 # at that stage we know it's our encryption
141 if parts[1] == 'aes':
141 if parts[1] == 'aes':
142 decrypted_data = AESCipher(ENCRYPTION_KEY).decrypt(parts[2])
142 decrypted_data = AESCipher(ENCRYPTION_KEY).decrypt(parts[2])
143 elif parts[1] == 'aes_hmac':
143 elif parts[1] == 'aes_hmac':
144 decrypted_data = AESCipher(
144 decrypted_data = AESCipher(
145 ENCRYPTION_KEY, hmac=True,
145 ENCRYPTION_KEY, hmac=True,
146 strict_verification=enc_strict_mode).decrypt(parts[2])
146 strict_verification=enc_strict_mode).decrypt(parts[2])
147 else:
147 else:
148 raise ValueError(
148 raise ValueError(
149 'Encryption type part is wrong, must be `aes` '
149 'Encryption type part is wrong, must be `aes` '
150 'or `aes_hmac`, got `%s` instead' % (parts[1]))
150 'or `aes_hmac`, got `%s` instead' % (parts[1]))
151 return decrypted_data
151 return decrypted_data
152
152
153
153
154 class BaseModel(object):
154 class BaseModel(object):
155 """
155 """
156 Base Model for all classes
156 Base Model for all classes
157 """
157 """
158
158
159 @classmethod
159 @classmethod
160 def _get_keys(cls):
160 def _get_keys(cls):
161 """return column names for this model """
161 """return column names for this model """
162 return class_mapper(cls).c.keys()
162 return class_mapper(cls).c.keys()
163
163
164 def get_dict(self):
164 def get_dict(self):
165 """
165 """
166 return dict with keys and values corresponding
166 return dict with keys and values corresponding
167 to this model data """
167 to this model data """
168
168
169 d = {}
169 d = {}
170 for k in self._get_keys():
170 for k in self._get_keys():
171 d[k] = getattr(self, k)
171 d[k] = getattr(self, k)
172
172
173 # also use __json__() if present to get additional fields
173 # also use __json__() if present to get additional fields
174 _json_attr = getattr(self, '__json__', None)
174 _json_attr = getattr(self, '__json__', None)
175 if _json_attr:
175 if _json_attr:
176 # update with attributes from __json__
176 # update with attributes from __json__
177 if callable(_json_attr):
177 if callable(_json_attr):
178 _json_attr = _json_attr()
178 _json_attr = _json_attr()
179 for k, val in _json_attr.iteritems():
179 for k, val in _json_attr.iteritems():
180 d[k] = val
180 d[k] = val
181 return d
181 return d
182
182
183 def get_appstruct(self):
183 def get_appstruct(self):
184 """return list with keys and values tuples corresponding
184 """return list with keys and values tuples corresponding
185 to this model data """
185 to this model data """
186
186
187 l = []
187 l = []
188 for k in self._get_keys():
188 for k in self._get_keys():
189 l.append((k, getattr(self, k),))
189 l.append((k, getattr(self, k),))
190 return l
190 return l
191
191
192 def populate_obj(self, populate_dict):
192 def populate_obj(self, populate_dict):
193 """populate model with data from given populate_dict"""
193 """populate model with data from given populate_dict"""
194
194
195 for k in self._get_keys():
195 for k in self._get_keys():
196 if k in populate_dict:
196 if k in populate_dict:
197 setattr(self, k, populate_dict[k])
197 setattr(self, k, populate_dict[k])
198
198
199 @classmethod
199 @classmethod
200 def query(cls):
200 def query(cls):
201 return Session().query(cls)
201 return Session().query(cls)
202
202
203 @classmethod
203 @classmethod
204 def get(cls, id_):
204 def get(cls, id_):
205 if id_:
205 if id_:
206 return cls.query().get(id_)
206 return cls.query().get(id_)
207
207
208 @classmethod
208 @classmethod
209 def get_or_404(cls, id_, pyramid_exc=False):
209 def get_or_404(cls, id_, pyramid_exc=False):
210 if pyramid_exc:
210 if pyramid_exc:
211 # NOTE(marcink): backward compat, once migration to pyramid
211 # NOTE(marcink): backward compat, once migration to pyramid
212 # this should only use pyramid exceptions
212 # this should only use pyramid exceptions
213 from pyramid.httpexceptions import HTTPNotFound
213 from pyramid.httpexceptions import HTTPNotFound
214 else:
214 else:
215 from webob.exc import HTTPNotFound
215 from webob.exc import HTTPNotFound
216
216
217 try:
217 try:
218 id_ = int(id_)
218 id_ = int(id_)
219 except (TypeError, ValueError):
219 except (TypeError, ValueError):
220 raise HTTPNotFound
220 raise HTTPNotFound
221
221
222 res = cls.query().get(id_)
222 res = cls.query().get(id_)
223 if not res:
223 if not res:
224 raise HTTPNotFound
224 raise HTTPNotFound
225 return res
225 return res
226
226
227 @classmethod
227 @classmethod
228 def getAll(cls):
228 def getAll(cls):
229 # deprecated and left for backward compatibility
229 # deprecated and left for backward compatibility
230 return cls.get_all()
230 return cls.get_all()
231
231
232 @classmethod
232 @classmethod
233 def get_all(cls):
233 def get_all(cls):
234 return cls.query().all()
234 return cls.query().all()
235
235
236 @classmethod
236 @classmethod
237 def delete(cls, id_):
237 def delete(cls, id_):
238 obj = cls.query().get(id_)
238 obj = cls.query().get(id_)
239 Session().delete(obj)
239 Session().delete(obj)
240
240
241 @classmethod
241 @classmethod
242 def identity_cache(cls, session, attr_name, value):
242 def identity_cache(cls, session, attr_name, value):
243 exist_in_session = []
243 exist_in_session = []
244 for (item_cls, pkey), instance in session.identity_map.items():
244 for (item_cls, pkey), instance in session.identity_map.items():
245 if cls == item_cls and getattr(instance, attr_name) == value:
245 if cls == item_cls and getattr(instance, attr_name) == value:
246 exist_in_session.append(instance)
246 exist_in_session.append(instance)
247 if exist_in_session:
247 if exist_in_session:
248 if len(exist_in_session) == 1:
248 if len(exist_in_session) == 1:
249 return exist_in_session[0]
249 return exist_in_session[0]
250 log.exception(
250 log.exception(
251 'multiple objects with attr %s and '
251 'multiple objects with attr %s and '
252 'value %s found with same name: %r',
252 'value %s found with same name: %r',
253 attr_name, value, exist_in_session)
253 attr_name, value, exist_in_session)
254
254
255 def __repr__(self):
255 def __repr__(self):
256 if hasattr(self, '__unicode__'):
256 if hasattr(self, '__unicode__'):
257 # python repr needs to return str
257 # python repr needs to return str
258 try:
258 try:
259 return safe_str(self.__unicode__())
259 return safe_str(self.__unicode__())
260 except UnicodeDecodeError:
260 except UnicodeDecodeError:
261 pass
261 pass
262 return '<DB:%s>' % (self.__class__.__name__)
262 return '<DB:%s>' % (self.__class__.__name__)
263
263
264
264
265 class RhodeCodeSetting(Base, BaseModel):
265 class RhodeCodeSetting(Base, BaseModel):
266 __tablename__ = 'rhodecode_settings'
266 __tablename__ = 'rhodecode_settings'
267 __table_args__ = (
267 __table_args__ = (
268 UniqueConstraint('app_settings_name'),
268 UniqueConstraint('app_settings_name'),
269 {'extend_existing': True, 'mysql_engine': 'InnoDB',
269 {'extend_existing': True, 'mysql_engine': 'InnoDB',
270 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
270 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
271 )
271 )
272
272
273 SETTINGS_TYPES = {
273 SETTINGS_TYPES = {
274 'str': safe_str,
274 'str': safe_str,
275 'int': safe_int,
275 'int': safe_int,
276 'unicode': safe_unicode,
276 'unicode': safe_unicode,
277 'bool': str2bool,
277 'bool': str2bool,
278 'list': functools.partial(aslist, sep=',')
278 'list': functools.partial(aslist, sep=',')
279 }
279 }
280 DEFAULT_UPDATE_URL = 'https://rhodecode.com/api/v1/info/versions'
280 DEFAULT_UPDATE_URL = 'https://rhodecode.com/api/v1/info/versions'
281 GLOBAL_CONF_KEY = 'app_settings'
281 GLOBAL_CONF_KEY = 'app_settings'
282
282
283 app_settings_id = Column("app_settings_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
283 app_settings_id = Column("app_settings_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
284 app_settings_name = Column("app_settings_name", String(255), nullable=True, unique=None, default=None)
284 app_settings_name = Column("app_settings_name", String(255), nullable=True, unique=None, default=None)
285 _app_settings_value = Column("app_settings_value", String(4096), nullable=True, unique=None, default=None)
285 _app_settings_value = Column("app_settings_value", String(4096), nullable=True, unique=None, default=None)
286 _app_settings_type = Column("app_settings_type", String(255), nullable=True, unique=None, default=None)
286 _app_settings_type = Column("app_settings_type", String(255), nullable=True, unique=None, default=None)
287
287
288 def __init__(self, key='', val='', type='unicode'):
288 def __init__(self, key='', val='', type='unicode'):
289 self.app_settings_name = key
289 self.app_settings_name = key
290 self.app_settings_type = type
290 self.app_settings_type = type
291 self.app_settings_value = val
291 self.app_settings_value = val
292
292
293 @validates('_app_settings_value')
293 @validates('_app_settings_value')
294 def validate_settings_value(self, key, val):
294 def validate_settings_value(self, key, val):
295 assert type(val) == unicode
295 assert type(val) == unicode
296 return val
296 return val
297
297
298 @hybrid_property
298 @hybrid_property
299 def app_settings_value(self):
299 def app_settings_value(self):
300 v = self._app_settings_value
300 v = self._app_settings_value
301 _type = self.app_settings_type
301 _type = self.app_settings_type
302 if _type:
302 if _type:
303 _type = self.app_settings_type.split('.')[0]
303 _type = self.app_settings_type.split('.')[0]
304 # decode the encrypted value
304 # decode the encrypted value
305 if 'encrypted' in self.app_settings_type:
305 if 'encrypted' in self.app_settings_type:
306 cipher = EncryptedTextValue()
306 cipher = EncryptedTextValue()
307 v = safe_unicode(cipher.process_result_value(v, None))
307 v = safe_unicode(cipher.process_result_value(v, None))
308
308
309 converter = self.SETTINGS_TYPES.get(_type) or \
309 converter = self.SETTINGS_TYPES.get(_type) or \
310 self.SETTINGS_TYPES['unicode']
310 self.SETTINGS_TYPES['unicode']
311 return converter(v)
311 return converter(v)
312
312
313 @app_settings_value.setter
313 @app_settings_value.setter
314 def app_settings_value(self, val):
314 def app_settings_value(self, val):
315 """
315 """
316 Setter that will always make sure we use unicode in app_settings_value
316 Setter that will always make sure we use unicode in app_settings_value
317
317
318 :param val:
318 :param val:
319 """
319 """
320 val = safe_unicode(val)
320 val = safe_unicode(val)
321 # encode the encrypted value
321 # encode the encrypted value
322 if 'encrypted' in self.app_settings_type:
322 if 'encrypted' in self.app_settings_type:
323 cipher = EncryptedTextValue()
323 cipher = EncryptedTextValue()
324 val = safe_unicode(cipher.process_bind_param(val, None))
324 val = safe_unicode(cipher.process_bind_param(val, None))
325 self._app_settings_value = val
325 self._app_settings_value = val
326
326
327 @hybrid_property
327 @hybrid_property
328 def app_settings_type(self):
328 def app_settings_type(self):
329 return self._app_settings_type
329 return self._app_settings_type
330
330
331 @app_settings_type.setter
331 @app_settings_type.setter
332 def app_settings_type(self, val):
332 def app_settings_type(self, val):
333 if val.split('.')[0] not in self.SETTINGS_TYPES:
333 if val.split('.')[0] not in self.SETTINGS_TYPES:
334 raise Exception('type must be one of %s got %s'
334 raise Exception('type must be one of %s got %s'
335 % (self.SETTINGS_TYPES.keys(), val))
335 % (self.SETTINGS_TYPES.keys(), val))
336 self._app_settings_type = val
336 self._app_settings_type = val
337
337
338 def __unicode__(self):
338 def __unicode__(self):
339 return u"<%s('%s:%s[%s]')>" % (
339 return u"<%s('%s:%s[%s]')>" % (
340 self.__class__.__name__,
340 self.__class__.__name__,
341 self.app_settings_name, self.app_settings_value,
341 self.app_settings_name, self.app_settings_value,
342 self.app_settings_type
342 self.app_settings_type
343 )
343 )
344
344
345
345
346 class RhodeCodeUi(Base, BaseModel):
346 class RhodeCodeUi(Base, BaseModel):
347 __tablename__ = 'rhodecode_ui'
347 __tablename__ = 'rhodecode_ui'
348 __table_args__ = (
348 __table_args__ = (
349 UniqueConstraint('ui_key'),
349 UniqueConstraint('ui_key'),
350 {'extend_existing': True, 'mysql_engine': 'InnoDB',
350 {'extend_existing': True, 'mysql_engine': 'InnoDB',
351 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
351 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
352 )
352 )
353
353
354 HOOK_REPO_SIZE = 'changegroup.repo_size'
354 HOOK_REPO_SIZE = 'changegroup.repo_size'
355 # HG
355 # HG
356 HOOK_PRE_PULL = 'preoutgoing.pre_pull'
356 HOOK_PRE_PULL = 'preoutgoing.pre_pull'
357 HOOK_PULL = 'outgoing.pull_logger'
357 HOOK_PULL = 'outgoing.pull_logger'
358 HOOK_PRE_PUSH = 'prechangegroup.pre_push'
358 HOOK_PRE_PUSH = 'prechangegroup.pre_push'
359 HOOK_PRETX_PUSH = 'pretxnchangegroup.pre_push'
359 HOOK_PRETX_PUSH = 'pretxnchangegroup.pre_push'
360 HOOK_PUSH = 'changegroup.push_logger'
360 HOOK_PUSH = 'changegroup.push_logger'
361 HOOK_PUSH_KEY = 'pushkey.key_push'
361 HOOK_PUSH_KEY = 'pushkey.key_push'
362
362
363 # TODO: johbo: Unify way how hooks are configured for git and hg,
363 # TODO: johbo: Unify way how hooks are configured for git and hg,
364 # git part is currently hardcoded.
364 # git part is currently hardcoded.
365
365
366 # SVN PATTERNS
366 # SVN PATTERNS
367 SVN_BRANCH_ID = 'vcs_svn_branch'
367 SVN_BRANCH_ID = 'vcs_svn_branch'
368 SVN_TAG_ID = 'vcs_svn_tag'
368 SVN_TAG_ID = 'vcs_svn_tag'
369
369
370 ui_id = Column(
370 ui_id = Column(
371 "ui_id", Integer(), nullable=False, unique=True, default=None,
371 "ui_id", Integer(), nullable=False, unique=True, default=None,
372 primary_key=True)
372 primary_key=True)
373 ui_section = Column(
373 ui_section = Column(
374 "ui_section", String(255), nullable=True, unique=None, default=None)
374 "ui_section", String(255), nullable=True, unique=None, default=None)
375 ui_key = Column(
375 ui_key = Column(
376 "ui_key", String(255), nullable=True, unique=None, default=None)
376 "ui_key", String(255), nullable=True, unique=None, default=None)
377 ui_value = Column(
377 ui_value = Column(
378 "ui_value", String(255), nullable=True, unique=None, default=None)
378 "ui_value", String(255), nullable=True, unique=None, default=None)
379 ui_active = Column(
379 ui_active = Column(
380 "ui_active", Boolean(), nullable=True, unique=None, default=True)
380 "ui_active", Boolean(), nullable=True, unique=None, default=True)
381
381
382 def __repr__(self):
382 def __repr__(self):
383 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.ui_section,
383 return '<%s[%s]%s=>%s]>' % (self.__class__.__name__, self.ui_section,
384 self.ui_key, self.ui_value)
384 self.ui_key, self.ui_value)
385
385
386
386
387 class RepoRhodeCodeSetting(Base, BaseModel):
387 class RepoRhodeCodeSetting(Base, BaseModel):
388 __tablename__ = 'repo_rhodecode_settings'
388 __tablename__ = 'repo_rhodecode_settings'
389 __table_args__ = (
389 __table_args__ = (
390 UniqueConstraint(
390 UniqueConstraint(
391 'app_settings_name', 'repository_id',
391 'app_settings_name', 'repository_id',
392 name='uq_repo_rhodecode_setting_name_repo_id'),
392 name='uq_repo_rhodecode_setting_name_repo_id'),
393 {'extend_existing': True, 'mysql_engine': 'InnoDB',
393 {'extend_existing': True, 'mysql_engine': 'InnoDB',
394 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
394 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
395 )
395 )
396
396
397 repository_id = Column(
397 repository_id = Column(
398 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
398 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
399 nullable=False)
399 nullable=False)
400 app_settings_id = Column(
400 app_settings_id = Column(
401 "app_settings_id", Integer(), nullable=False, unique=True,
401 "app_settings_id", Integer(), nullable=False, unique=True,
402 default=None, primary_key=True)
402 default=None, primary_key=True)
403 app_settings_name = Column(
403 app_settings_name = Column(
404 "app_settings_name", String(255), nullable=True, unique=None,
404 "app_settings_name", String(255), nullable=True, unique=None,
405 default=None)
405 default=None)
406 _app_settings_value = Column(
406 _app_settings_value = Column(
407 "app_settings_value", String(4096), nullable=True, unique=None,
407 "app_settings_value", String(4096), nullable=True, unique=None,
408 default=None)
408 default=None)
409 _app_settings_type = Column(
409 _app_settings_type = Column(
410 "app_settings_type", String(255), nullable=True, unique=None,
410 "app_settings_type", String(255), nullable=True, unique=None,
411 default=None)
411 default=None)
412
412
413 repository = relationship('Repository')
413 repository = relationship('Repository')
414
414
415 def __init__(self, repository_id, key='', val='', type='unicode'):
415 def __init__(self, repository_id, key='', val='', type='unicode'):
416 self.repository_id = repository_id
416 self.repository_id = repository_id
417 self.app_settings_name = key
417 self.app_settings_name = key
418 self.app_settings_type = type
418 self.app_settings_type = type
419 self.app_settings_value = val
419 self.app_settings_value = val
420
420
421 @validates('_app_settings_value')
421 @validates('_app_settings_value')
422 def validate_settings_value(self, key, val):
422 def validate_settings_value(self, key, val):
423 assert type(val) == unicode
423 assert type(val) == unicode
424 return val
424 return val
425
425
426 @hybrid_property
426 @hybrid_property
427 def app_settings_value(self):
427 def app_settings_value(self):
428 v = self._app_settings_value
428 v = self._app_settings_value
429 type_ = self.app_settings_type
429 type_ = self.app_settings_type
430 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
430 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
431 converter = SETTINGS_TYPES.get(type_) or SETTINGS_TYPES['unicode']
431 converter = SETTINGS_TYPES.get(type_) or SETTINGS_TYPES['unicode']
432 return converter(v)
432 return converter(v)
433
433
434 @app_settings_value.setter
434 @app_settings_value.setter
435 def app_settings_value(self, val):
435 def app_settings_value(self, val):
436 """
436 """
437 Setter that will always make sure we use unicode in app_settings_value
437 Setter that will always make sure we use unicode in app_settings_value
438
438
439 :param val:
439 :param val:
440 """
440 """
441 self._app_settings_value = safe_unicode(val)
441 self._app_settings_value = safe_unicode(val)
442
442
443 @hybrid_property
443 @hybrid_property
444 def app_settings_type(self):
444 def app_settings_type(self):
445 return self._app_settings_type
445 return self._app_settings_type
446
446
447 @app_settings_type.setter
447 @app_settings_type.setter
448 def app_settings_type(self, val):
448 def app_settings_type(self, val):
449 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
449 SETTINGS_TYPES = RhodeCodeSetting.SETTINGS_TYPES
450 if val not in SETTINGS_TYPES:
450 if val not in SETTINGS_TYPES:
451 raise Exception('type must be one of %s got %s'
451 raise Exception('type must be one of %s got %s'
452 % (SETTINGS_TYPES.keys(), val))
452 % (SETTINGS_TYPES.keys(), val))
453 self._app_settings_type = val
453 self._app_settings_type = val
454
454
455 def __unicode__(self):
455 def __unicode__(self):
456 return u"<%s('%s:%s:%s[%s]')>" % (
456 return u"<%s('%s:%s:%s[%s]')>" % (
457 self.__class__.__name__, self.repository.repo_name,
457 self.__class__.__name__, self.repository.repo_name,
458 self.app_settings_name, self.app_settings_value,
458 self.app_settings_name, self.app_settings_value,
459 self.app_settings_type
459 self.app_settings_type
460 )
460 )
461
461
462
462
463 class RepoRhodeCodeUi(Base, BaseModel):
463 class RepoRhodeCodeUi(Base, BaseModel):
464 __tablename__ = 'repo_rhodecode_ui'
464 __tablename__ = 'repo_rhodecode_ui'
465 __table_args__ = (
465 __table_args__ = (
466 UniqueConstraint(
466 UniqueConstraint(
467 'repository_id', 'ui_section', 'ui_key',
467 'repository_id', 'ui_section', 'ui_key',
468 name='uq_repo_rhodecode_ui_repository_id_section_key'),
468 name='uq_repo_rhodecode_ui_repository_id_section_key'),
469 {'extend_existing': True, 'mysql_engine': 'InnoDB',
469 {'extend_existing': True, 'mysql_engine': 'InnoDB',
470 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
470 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
471 )
471 )
472
472
473 repository_id = Column(
473 repository_id = Column(
474 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
474 "repository_id", Integer(), ForeignKey('repositories.repo_id'),
475 nullable=False)
475 nullable=False)
476 ui_id = Column(
476 ui_id = Column(
477 "ui_id", Integer(), nullable=False, unique=True, default=None,
477 "ui_id", Integer(), nullable=False, unique=True, default=None,
478 primary_key=True)
478 primary_key=True)
479 ui_section = Column(
479 ui_section = Column(
480 "ui_section", String(255), nullable=True, unique=None, default=None)
480 "ui_section", String(255), nullable=True, unique=None, default=None)
481 ui_key = Column(
481 ui_key = Column(
482 "ui_key", String(255), nullable=True, unique=None, default=None)
482 "ui_key", String(255), nullable=True, unique=None, default=None)
483 ui_value = Column(
483 ui_value = Column(
484 "ui_value", String(255), nullable=True, unique=None, default=None)
484 "ui_value", String(255), nullable=True, unique=None, default=None)
485 ui_active = Column(
485 ui_active = Column(
486 "ui_active", Boolean(), nullable=True, unique=None, default=True)
486 "ui_active", Boolean(), nullable=True, unique=None, default=True)
487
487
488 repository = relationship('Repository')
488 repository = relationship('Repository')
489
489
490 def __repr__(self):
490 def __repr__(self):
491 return '<%s[%s:%s]%s=>%s]>' % (
491 return '<%s[%s:%s]%s=>%s]>' % (
492 self.__class__.__name__, self.repository.repo_name,
492 self.__class__.__name__, self.repository.repo_name,
493 self.ui_section, self.ui_key, self.ui_value)
493 self.ui_section, self.ui_key, self.ui_value)
494
494
495
495
496 class User(Base, BaseModel):
496 class User(Base, BaseModel):
497 __tablename__ = 'users'
497 __tablename__ = 'users'
498 __table_args__ = (
498 __table_args__ = (
499 UniqueConstraint('username'), UniqueConstraint('email'),
499 UniqueConstraint('username'), UniqueConstraint('email'),
500 Index('u_username_idx', 'username'),
500 Index('u_username_idx', 'username'),
501 Index('u_email_idx', 'email'),
501 Index('u_email_idx', 'email'),
502 {'extend_existing': True, 'mysql_engine': 'InnoDB',
502 {'extend_existing': True, 'mysql_engine': 'InnoDB',
503 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
503 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
504 )
504 )
505 DEFAULT_USER = 'default'
505 DEFAULT_USER = 'default'
506 DEFAULT_USER_EMAIL = 'anonymous@rhodecode.org'
506 DEFAULT_USER_EMAIL = 'anonymous@rhodecode.org'
507 DEFAULT_GRAVATAR_URL = 'https://secure.gravatar.com/avatar/{md5email}?d=identicon&s={size}'
507 DEFAULT_GRAVATAR_URL = 'https://secure.gravatar.com/avatar/{md5email}?d=identicon&s={size}'
508
508
509 user_id = Column("user_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
509 user_id = Column("user_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
510 username = Column("username", String(255), nullable=True, unique=None, default=None)
510 username = Column("username", String(255), nullable=True, unique=None, default=None)
511 password = Column("password", String(255), nullable=True, unique=None, default=None)
511 password = Column("password", String(255), nullable=True, unique=None, default=None)
512 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
512 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
513 admin = Column("admin", Boolean(), nullable=True, unique=None, default=False)
513 admin = Column("admin", Boolean(), nullable=True, unique=None, default=False)
514 name = Column("firstname", String(255), nullable=True, unique=None, default=None)
514 name = Column("firstname", String(255), nullable=True, unique=None, default=None)
515 lastname = Column("lastname", String(255), nullable=True, unique=None, default=None)
515 lastname = Column("lastname", String(255), nullable=True, unique=None, default=None)
516 _email = Column("email", String(255), nullable=True, unique=None, default=None)
516 _email = Column("email", String(255), nullable=True, unique=None, default=None)
517 last_login = Column("last_login", DateTime(timezone=False), nullable=True, unique=None, default=None)
517 last_login = Column("last_login", DateTime(timezone=False), nullable=True, unique=None, default=None)
518 last_activity = Column('last_activity', DateTime(timezone=False), nullable=True, unique=None, default=None)
518 last_activity = Column('last_activity', DateTime(timezone=False), nullable=True, unique=None, default=None)
519
519
520 extern_type = Column("extern_type", String(255), nullable=True, unique=None, default=None)
520 extern_type = Column("extern_type", String(255), nullable=True, unique=None, default=None)
521 extern_name = Column("extern_name", String(255), nullable=True, unique=None, default=None)
521 extern_name = Column("extern_name", String(255), nullable=True, unique=None, default=None)
522 _api_key = Column("api_key", String(255), nullable=True, unique=None, default=None)
522 _api_key = Column("api_key", String(255), nullable=True, unique=None, default=None)
523 inherit_default_permissions = Column("inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
523 inherit_default_permissions = Column("inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
524 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
524 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
525 _user_data = Column("user_data", LargeBinary(), nullable=True) # JSON data
525 _user_data = Column("user_data", LargeBinary(), nullable=True) # JSON data
526
526
527 user_log = relationship('UserLog')
527 user_log = relationship('UserLog')
528 user_perms = relationship('UserToPerm', primaryjoin="User.user_id==UserToPerm.user_id", cascade='all')
528 user_perms = relationship('UserToPerm', primaryjoin="User.user_id==UserToPerm.user_id", cascade='all')
529
529
530 repositories = relationship('Repository')
530 repositories = relationship('Repository')
531 repository_groups = relationship('RepoGroup')
531 repository_groups = relationship('RepoGroup')
532 user_groups = relationship('UserGroup')
532 user_groups = relationship('UserGroup')
533
533
534 user_followers = relationship('UserFollowing', primaryjoin='UserFollowing.follows_user_id==User.user_id', cascade='all')
534 user_followers = relationship('UserFollowing', primaryjoin='UserFollowing.follows_user_id==User.user_id', cascade='all')
535 followings = relationship('UserFollowing', primaryjoin='UserFollowing.user_id==User.user_id', cascade='all')
535 followings = relationship('UserFollowing', primaryjoin='UserFollowing.user_id==User.user_id', cascade='all')
536
536
537 repo_to_perm = relationship('UserRepoToPerm', primaryjoin='UserRepoToPerm.user_id==User.user_id', cascade='all')
537 repo_to_perm = relationship('UserRepoToPerm', primaryjoin='UserRepoToPerm.user_id==User.user_id', cascade='all')
538 repo_group_to_perm = relationship('UserRepoGroupToPerm', primaryjoin='UserRepoGroupToPerm.user_id==User.user_id', cascade='all')
538 repo_group_to_perm = relationship('UserRepoGroupToPerm', primaryjoin='UserRepoGroupToPerm.user_id==User.user_id', cascade='all')
539 user_group_to_perm = relationship('UserUserGroupToPerm', primaryjoin='UserUserGroupToPerm.user_id==User.user_id', cascade='all')
539 user_group_to_perm = relationship('UserUserGroupToPerm', primaryjoin='UserUserGroupToPerm.user_id==User.user_id', cascade='all')
540
540
541 group_member = relationship('UserGroupMember', cascade='all')
541 group_member = relationship('UserGroupMember', cascade='all')
542
542
543 notifications = relationship('UserNotification', cascade='all')
543 notifications = relationship('UserNotification', cascade='all')
544 # notifications assigned to this user
544 # notifications assigned to this user
545 user_created_notifications = relationship('Notification', cascade='all')
545 user_created_notifications = relationship('Notification', cascade='all')
546 # comments created by this user
546 # comments created by this user
547 user_comments = relationship('ChangesetComment', cascade='all')
547 user_comments = relationship('ChangesetComment', cascade='all')
548 # user profile extra info
548 # user profile extra info
549 user_emails = relationship('UserEmailMap', cascade='all')
549 user_emails = relationship('UserEmailMap', cascade='all')
550 user_ip_map = relationship('UserIpMap', cascade='all')
550 user_ip_map = relationship('UserIpMap', cascade='all')
551 user_auth_tokens = relationship('UserApiKeys', cascade='all')
551 user_auth_tokens = relationship('UserApiKeys', cascade='all')
552 # gists
552 # gists
553 user_gists = relationship('Gist', cascade='all')
553 user_gists = relationship('Gist', cascade='all')
554 # user pull requests
554 # user pull requests
555 user_pull_requests = relationship('PullRequest', cascade='all')
555 user_pull_requests = relationship('PullRequest', cascade='all')
556 # external identities
556 # external identities
557 extenal_identities = relationship(
557 extenal_identities = relationship(
558 'ExternalIdentity',
558 'ExternalIdentity',
559 primaryjoin="User.user_id==ExternalIdentity.local_user_id",
559 primaryjoin="User.user_id==ExternalIdentity.local_user_id",
560 cascade='all')
560 cascade='all')
561
561
562 def __unicode__(self):
562 def __unicode__(self):
563 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
563 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
564 self.user_id, self.username)
564 self.user_id, self.username)
565
565
566 @hybrid_property
566 @hybrid_property
567 def email(self):
567 def email(self):
568 return self._email
568 return self._email
569
569
570 @email.setter
570 @email.setter
571 def email(self, val):
571 def email(self, val):
572 self._email = val.lower() if val else None
572 self._email = val.lower() if val else None
573
573
574 @hybrid_property
574 @hybrid_property
575 def first_name(self):
575 def first_name(self):
576 from rhodecode.lib import helpers as h
576 from rhodecode.lib import helpers as h
577 if self.name:
577 if self.name:
578 return h.escape(self.name)
578 return h.escape(self.name)
579 return self.name
579 return self.name
580
580
581 @hybrid_property
581 @hybrid_property
582 def last_name(self):
582 def last_name(self):
583 from rhodecode.lib import helpers as h
583 from rhodecode.lib import helpers as h
584 if self.lastname:
584 if self.lastname:
585 return h.escape(self.lastname)
585 return h.escape(self.lastname)
586 return self.lastname
586 return self.lastname
587
587
588 @hybrid_property
588 @hybrid_property
589 def api_key(self):
589 def api_key(self):
590 """
590 """
591 Fetch if exist an auth-token with role ALL connected to this user
591 Fetch if exist an auth-token with role ALL connected to this user
592 """
592 """
593 user_auth_token = UserApiKeys.query()\
593 user_auth_token = UserApiKeys.query()\
594 .filter(UserApiKeys.user_id == self.user_id)\
594 .filter(UserApiKeys.user_id == self.user_id)\
595 .filter(or_(UserApiKeys.expires == -1,
595 .filter(or_(UserApiKeys.expires == -1,
596 UserApiKeys.expires >= time.time()))\
596 UserApiKeys.expires >= time.time()))\
597 .filter(UserApiKeys.role == UserApiKeys.ROLE_ALL).first()
597 .filter(UserApiKeys.role == UserApiKeys.ROLE_ALL).first()
598 if user_auth_token:
598 if user_auth_token:
599 user_auth_token = user_auth_token.api_key
599 user_auth_token = user_auth_token.api_key
600
600
601 return user_auth_token
601 return user_auth_token
602
602
603 @api_key.setter
603 @api_key.setter
604 def api_key(self, val):
604 def api_key(self, val):
605 # don't allow to set API key this is deprecated for now
605 # don't allow to set API key this is deprecated for now
606 self._api_key = None
606 self._api_key = None
607
607
608 @property
608 @property
609 def firstname(self):
609 def firstname(self):
610 # alias for future
610 # alias for future
611 return self.name
611 return self.name
612
612
613 @property
613 @property
614 def emails(self):
614 def emails(self):
615 other = UserEmailMap.query().filter(UserEmailMap.user==self).all()
615 other = UserEmailMap.query().filter(UserEmailMap.user==self).all()
616 return [self.email] + [x.email for x in other]
616 return [self.email] + [x.email for x in other]
617
617
618 @property
618 @property
619 def auth_tokens(self):
619 def auth_tokens(self):
620 return [x.api_key for x in self.extra_auth_tokens]
620 return [x.api_key for x in self.extra_auth_tokens]
621
621
622 @property
622 @property
623 def extra_auth_tokens(self):
623 def extra_auth_tokens(self):
624 return UserApiKeys.query().filter(UserApiKeys.user == self).all()
624 return UserApiKeys.query().filter(UserApiKeys.user == self).all()
625
625
626 @property
626 @property
627 def feed_token(self):
627 def feed_token(self):
628 return self.get_feed_token()
628 return self.get_feed_token()
629
629
630 def get_feed_token(self):
630 def get_feed_token(self):
631 feed_tokens = UserApiKeys.query()\
631 feed_tokens = UserApiKeys.query()\
632 .filter(UserApiKeys.user == self)\
632 .filter(UserApiKeys.user == self)\
633 .filter(UserApiKeys.role == UserApiKeys.ROLE_FEED)\
633 .filter(UserApiKeys.role == UserApiKeys.ROLE_FEED)\
634 .all()
634 .all()
635 if feed_tokens:
635 if feed_tokens:
636 return feed_tokens[0].api_key
636 return feed_tokens[0].api_key
637 return 'NO_FEED_TOKEN_AVAILABLE'
637 return 'NO_FEED_TOKEN_AVAILABLE'
638
638
639 @classmethod
639 @classmethod
640 def extra_valid_auth_tokens(cls, user, role=None):
640 def extra_valid_auth_tokens(cls, user, role=None):
641 tokens = UserApiKeys.query().filter(UserApiKeys.user == user)\
641 tokens = UserApiKeys.query().filter(UserApiKeys.user == user)\
642 .filter(or_(UserApiKeys.expires == -1,
642 .filter(or_(UserApiKeys.expires == -1,
643 UserApiKeys.expires >= time.time()))
643 UserApiKeys.expires >= time.time()))
644 if role:
644 if role:
645 tokens = tokens.filter(or_(UserApiKeys.role == role,
645 tokens = tokens.filter(or_(UserApiKeys.role == role,
646 UserApiKeys.role == UserApiKeys.ROLE_ALL))
646 UserApiKeys.role == UserApiKeys.ROLE_ALL))
647 return tokens.all()
647 return tokens.all()
648
648
649 def authenticate_by_token(self, auth_token, roles=None, scope_repo_id=None):
649 def authenticate_by_token(self, auth_token, roles=None, scope_repo_id=None):
650 from rhodecode.lib import auth
650 from rhodecode.lib import auth
651
651
652 log.debug('Trying to authenticate user: %s via auth-token, '
652 log.debug('Trying to authenticate user: %s via auth-token, '
653 'and roles: %s', self, roles)
653 'and roles: %s', self, roles)
654
654
655 if not auth_token:
655 if not auth_token:
656 return False
656 return False
657
657
658 crypto_backend = auth.crypto_backend()
658 crypto_backend = auth.crypto_backend()
659
659
660 roles = (roles or []) + [UserApiKeys.ROLE_ALL]
660 roles = (roles or []) + [UserApiKeys.ROLE_ALL]
661 tokens_q = UserApiKeys.query()\
661 tokens_q = UserApiKeys.query()\
662 .filter(UserApiKeys.user_id == self.user_id)\
662 .filter(UserApiKeys.user_id == self.user_id)\
663 .filter(or_(UserApiKeys.expires == -1,
663 .filter(or_(UserApiKeys.expires == -1,
664 UserApiKeys.expires >= time.time()))
664 UserApiKeys.expires >= time.time()))
665
665
666 tokens_q = tokens_q.filter(UserApiKeys.role.in_(roles))
666 tokens_q = tokens_q.filter(UserApiKeys.role.in_(roles))
667
667
668 plain_tokens = []
668 plain_tokens = []
669 hash_tokens = []
669 hash_tokens = []
670
670
671 for token in tokens_q.all():
671 for token in tokens_q.all():
672 # verify scope first
672 # verify scope first
673 if token.repo_id:
673 if token.repo_id:
674 # token has a scope, we need to verify it
674 # token has a scope, we need to verify it
675 if scope_repo_id != token.repo_id:
675 if scope_repo_id != token.repo_id:
676 log.debug(
676 log.debug(
677 'Scope mismatch: token has a set repo scope: %s, '
677 'Scope mismatch: token has a set repo scope: %s, '
678 'and calling scope is:%s, skipping further checks',
678 'and calling scope is:%s, skipping further checks',
679 token.repo, scope_repo_id)
679 token.repo, scope_repo_id)
680 # token has a scope, and it doesn't match, skip token
680 # token has a scope, and it doesn't match, skip token
681 continue
681 continue
682
682
683 if token.api_key.startswith(crypto_backend.ENC_PREF):
683 if token.api_key.startswith(crypto_backend.ENC_PREF):
684 hash_tokens.append(token.api_key)
684 hash_tokens.append(token.api_key)
685 else:
685 else:
686 plain_tokens.append(token.api_key)
686 plain_tokens.append(token.api_key)
687
687
688 is_plain_match = auth_token in plain_tokens
688 is_plain_match = auth_token in plain_tokens
689 if is_plain_match:
689 if is_plain_match:
690 return True
690 return True
691
691
692 for hashed in hash_tokens:
692 for hashed in hash_tokens:
693 # TODO(marcink): this is expensive to calculate, but most secure
693 # TODO(marcink): this is expensive to calculate, but most secure
694 match = crypto_backend.hash_check(auth_token, hashed)
694 match = crypto_backend.hash_check(auth_token, hashed)
695 if match:
695 if match:
696 return True
696 return True
697
697
698 return False
698 return False
699
699
700 @property
700 @property
701 def ip_addresses(self):
701 def ip_addresses(self):
702 ret = UserIpMap.query().filter(UserIpMap.user == self).all()
702 ret = UserIpMap.query().filter(UserIpMap.user == self).all()
703 return [x.ip_addr for x in ret]
703 return [x.ip_addr for x in ret]
704
704
705 @property
705 @property
706 def username_and_name(self):
706 def username_and_name(self):
707 return '%s (%s %s)' % (self.username, self.first_name, self.last_name)
707 return '%s (%s %s)' % (self.username, self.first_name, self.last_name)
708
708
709 @property
709 @property
710 def username_or_name_or_email(self):
710 def username_or_name_or_email(self):
711 full_name = self.full_name if self.full_name is not ' ' else None
711 full_name = self.full_name if self.full_name is not ' ' else None
712 return self.username or full_name or self.email
712 return self.username or full_name or self.email
713
713
714 @property
714 @property
715 def full_name(self):
715 def full_name(self):
716 return '%s %s' % (self.first_name, self.last_name)
716 return '%s %s' % (self.first_name, self.last_name)
717
717
718 @property
718 @property
719 def full_name_or_username(self):
719 def full_name_or_username(self):
720 return ('%s %s' % (self.first_name, self.last_name)
720 return ('%s %s' % (self.first_name, self.last_name)
721 if (self.first_name and self.last_name) else self.username)
721 if (self.first_name and self.last_name) else self.username)
722
722
723 @property
723 @property
724 def full_contact(self):
724 def full_contact(self):
725 return '%s %s <%s>' % (self.first_name, self.last_name, self.email)
725 return '%s %s <%s>' % (self.first_name, self.last_name, self.email)
726
726
727 @property
727 @property
728 def short_contact(self):
728 def short_contact(self):
729 return '%s %s' % (self.first_name, self.last_name)
729 return '%s %s' % (self.first_name, self.last_name)
730
730
731 @property
731 @property
732 def is_admin(self):
732 def is_admin(self):
733 return self.admin
733 return self.admin
734
734
735 @property
735 @property
736 def AuthUser(self):
736 def AuthUser(self):
737 """
737 """
738 Returns instance of AuthUser for this user
738 Returns instance of AuthUser for this user
739 """
739 """
740 from rhodecode.lib.auth import AuthUser
740 from rhodecode.lib.auth import AuthUser
741 return AuthUser(user_id=self.user_id, username=self.username)
741 return AuthUser(user_id=self.user_id, username=self.username)
742
742
743 @hybrid_property
743 @hybrid_property
744 def user_data(self):
744 def user_data(self):
745 if not self._user_data:
745 if not self._user_data:
746 return {}
746 return {}
747
747
748 try:
748 try:
749 return json.loads(self._user_data)
749 return json.loads(self._user_data)
750 except TypeError:
750 except TypeError:
751 return {}
751 return {}
752
752
753 @user_data.setter
753 @user_data.setter
754 def user_data(self, val):
754 def user_data(self, val):
755 if not isinstance(val, dict):
755 if not isinstance(val, dict):
756 raise Exception('user_data must be dict, got %s' % type(val))
756 raise Exception('user_data must be dict, got %s' % type(val))
757 try:
757 try:
758 self._user_data = json.dumps(val)
758 self._user_data = json.dumps(val)
759 except Exception:
759 except Exception:
760 log.error(traceback.format_exc())
760 log.error(traceback.format_exc())
761
761
762 @classmethod
762 @classmethod
763 def get_by_username(cls, username, case_insensitive=False,
763 def get_by_username(cls, username, case_insensitive=False,
764 cache=False, identity_cache=False):
764 cache=False, identity_cache=False):
765 session = Session()
765 session = Session()
766
766
767 if case_insensitive:
767 if case_insensitive:
768 q = cls.query().filter(
768 q = cls.query().filter(
769 func.lower(cls.username) == func.lower(username))
769 func.lower(cls.username) == func.lower(username))
770 else:
770 else:
771 q = cls.query().filter(cls.username == username)
771 q = cls.query().filter(cls.username == username)
772
772
773 if cache:
773 if cache:
774 if identity_cache:
774 if identity_cache:
775 val = cls.identity_cache(session, 'username', username)
775 val = cls.identity_cache(session, 'username', username)
776 if val:
776 if val:
777 return val
777 return val
778 else:
778 else:
779 cache_key = "get_user_by_name_%s" % _hash_key(username)
779 cache_key = "get_user_by_name_%s" % _hash_key(username)
780 q = q.options(
780 q = q.options(
781 FromCache("sql_cache_short", cache_key))
781 FromCache("sql_cache_short", cache_key))
782
782
783 return q.scalar()
783 return q.scalar()
784
784
785 @classmethod
785 @classmethod
786 def get_by_auth_token(cls, auth_token, cache=False):
786 def get_by_auth_token(cls, auth_token, cache=False):
787 q = UserApiKeys.query()\
787 q = UserApiKeys.query()\
788 .filter(UserApiKeys.api_key == auth_token)\
788 .filter(UserApiKeys.api_key == auth_token)\
789 .filter(or_(UserApiKeys.expires == -1,
789 .filter(or_(UserApiKeys.expires == -1,
790 UserApiKeys.expires >= time.time()))
790 UserApiKeys.expires >= time.time()))
791 if cache:
791 if cache:
792 q = q.options(
792 q = q.options(
793 FromCache("sql_cache_short", "get_auth_token_%s" % auth_token))
793 FromCache("sql_cache_short", "get_auth_token_%s" % auth_token))
794
794
795 match = q.first()
795 match = q.first()
796 if match:
796 if match:
797 return match.user
797 return match.user
798
798
799 @classmethod
799 @classmethod
800 def get_by_email(cls, email, case_insensitive=False, cache=False):
800 def get_by_email(cls, email, case_insensitive=False, cache=False):
801
801
802 if case_insensitive:
802 if case_insensitive:
803 q = cls.query().filter(func.lower(cls.email) == func.lower(email))
803 q = cls.query().filter(func.lower(cls.email) == func.lower(email))
804
804
805 else:
805 else:
806 q = cls.query().filter(cls.email == email)
806 q = cls.query().filter(cls.email == email)
807
807
808 email_key = _hash_key(email)
808 email_key = _hash_key(email)
809 if cache:
809 if cache:
810 q = q.options(
810 q = q.options(
811 FromCache("sql_cache_short", "get_email_key_%s" % email_key))
811 FromCache("sql_cache_short", "get_email_key_%s" % email_key))
812
812
813 ret = q.scalar()
813 ret = q.scalar()
814 if ret is None:
814 if ret is None:
815 q = UserEmailMap.query()
815 q = UserEmailMap.query()
816 # try fetching in alternate email map
816 # try fetching in alternate email map
817 if case_insensitive:
817 if case_insensitive:
818 q = q.filter(func.lower(UserEmailMap.email) == func.lower(email))
818 q = q.filter(func.lower(UserEmailMap.email) == func.lower(email))
819 else:
819 else:
820 q = q.filter(UserEmailMap.email == email)
820 q = q.filter(UserEmailMap.email == email)
821 q = q.options(joinedload(UserEmailMap.user))
821 q = q.options(joinedload(UserEmailMap.user))
822 if cache:
822 if cache:
823 q = q.options(
823 q = q.options(
824 FromCache("sql_cache_short", "get_email_map_key_%s" % email_key))
824 FromCache("sql_cache_short", "get_email_map_key_%s" % email_key))
825 ret = getattr(q.scalar(), 'user', None)
825 ret = getattr(q.scalar(), 'user', None)
826
826
827 return ret
827 return ret
828
828
829 @classmethod
829 @classmethod
830 def get_from_cs_author(cls, author):
830 def get_from_cs_author(cls, author):
831 """
831 """
832 Tries to get User objects out of commit author string
832 Tries to get User objects out of commit author string
833
833
834 :param author:
834 :param author:
835 """
835 """
836 from rhodecode.lib.helpers import email, author_name
836 from rhodecode.lib.helpers import email, author_name
837 # Valid email in the attribute passed, see if they're in the system
837 # Valid email in the attribute passed, see if they're in the system
838 _email = email(author)
838 _email = email(author)
839 if _email:
839 if _email:
840 user = cls.get_by_email(_email, case_insensitive=True)
840 user = cls.get_by_email(_email, case_insensitive=True)
841 if user:
841 if user:
842 return user
842 return user
843 # Maybe we can match by username?
843 # Maybe we can match by username?
844 _author = author_name(author)
844 _author = author_name(author)
845 user = cls.get_by_username(_author, case_insensitive=True)
845 user = cls.get_by_username(_author, case_insensitive=True)
846 if user:
846 if user:
847 return user
847 return user
848
848
849 def update_userdata(self, **kwargs):
849 def update_userdata(self, **kwargs):
850 usr = self
850 usr = self
851 old = usr.user_data
851 old = usr.user_data
852 old.update(**kwargs)
852 old.update(**kwargs)
853 usr.user_data = old
853 usr.user_data = old
854 Session().add(usr)
854 Session().add(usr)
855 log.debug('updated userdata with ', kwargs)
855 log.debug('updated userdata with ', kwargs)
856
856
857 def update_lastlogin(self):
857 def update_lastlogin(self):
858 """Update user lastlogin"""
858 """Update user lastlogin"""
859 self.last_login = datetime.datetime.now()
859 self.last_login = datetime.datetime.now()
860 Session().add(self)
860 Session().add(self)
861 log.debug('updated user %s lastlogin', self.username)
861 log.debug('updated user %s lastlogin', self.username)
862
862
863 def update_lastactivity(self):
863 def update_lastactivity(self):
864 """Update user lastactivity"""
864 """Update user lastactivity"""
865 self.last_activity = datetime.datetime.now()
865 self.last_activity = datetime.datetime.now()
866 Session().add(self)
866 Session().add(self)
867 log.debug('updated user %s lastactivity', self.username)
867 log.debug('updated user %s lastactivity', self.username)
868
868
869 def update_password(self, new_password):
869 def update_password(self, new_password):
870 from rhodecode.lib.auth import get_crypt_password
870 from rhodecode.lib.auth import get_crypt_password
871
871
872 self.password = get_crypt_password(new_password)
872 self.password = get_crypt_password(new_password)
873 Session().add(self)
873 Session().add(self)
874
874
875 @classmethod
875 @classmethod
876 def get_first_super_admin(cls):
876 def get_first_super_admin(cls):
877 user = User.query().filter(User.admin == true()).first()
877 user = User.query().filter(User.admin == true()).first()
878 if user is None:
878 if user is None:
879 raise Exception('FATAL: Missing administrative account!')
879 raise Exception('FATAL: Missing administrative account!')
880 return user
880 return user
881
881
882 @classmethod
882 @classmethod
883 def get_all_super_admins(cls):
883 def get_all_super_admins(cls):
884 """
884 """
885 Returns all admin accounts sorted by username
885 Returns all admin accounts sorted by username
886 """
886 """
887 return User.query().filter(User.admin == true())\
887 return User.query().filter(User.admin == true())\
888 .order_by(User.username.asc()).all()
888 .order_by(User.username.asc()).all()
889
889
890 @classmethod
890 @classmethod
891 def get_default_user(cls, cache=False, refresh=False):
891 def get_default_user(cls, cache=False, refresh=False):
892 user = User.get_by_username(User.DEFAULT_USER, cache=cache)
892 user = User.get_by_username(User.DEFAULT_USER, cache=cache)
893 if user is None:
893 if user is None:
894 raise Exception('FATAL: Missing default account!')
894 raise Exception('FATAL: Missing default account!')
895 if refresh:
895 if refresh:
896 # The default user might be based on outdated state which
896 # The default user might be based on outdated state which
897 # has been loaded from the cache.
897 # has been loaded from the cache.
898 # A call to refresh() ensures that the
898 # A call to refresh() ensures that the
899 # latest state from the database is used.
899 # latest state from the database is used.
900 Session().refresh(user)
900 Session().refresh(user)
901 return user
901 return user
902
902
903 def _get_default_perms(self, user, suffix=''):
903 def _get_default_perms(self, user, suffix=''):
904 from rhodecode.model.permission import PermissionModel
904 from rhodecode.model.permission import PermissionModel
905 return PermissionModel().get_default_perms(user.user_perms, suffix)
905 return PermissionModel().get_default_perms(user.user_perms, suffix)
906
906
907 def get_default_perms(self, suffix=''):
907 def get_default_perms(self, suffix=''):
908 return self._get_default_perms(self, suffix)
908 return self._get_default_perms(self, suffix)
909
909
910 def get_api_data(self, include_secrets=False, details='full'):
910 def get_api_data(self, include_secrets=False, details='full'):
911 """
911 """
912 Common function for generating user related data for API
912 Common function for generating user related data for API
913
913
914 :param include_secrets: By default secrets in the API data will be replaced
914 :param include_secrets: By default secrets in the API data will be replaced
915 by a placeholder value to prevent exposing this data by accident. In case
915 by a placeholder value to prevent exposing this data by accident. In case
916 this data shall be exposed, set this flag to ``True``.
916 this data shall be exposed, set this flag to ``True``.
917
917
918 :param details: details can be 'basic|full' basic gives only a subset of
918 :param details: details can be 'basic|full' basic gives only a subset of
919 the available user information that includes user_id, name and emails.
919 the available user information that includes user_id, name and emails.
920 """
920 """
921 user = self
921 user = self
922 user_data = self.user_data
922 user_data = self.user_data
923 data = {
923 data = {
924 'user_id': user.user_id,
924 'user_id': user.user_id,
925 'username': user.username,
925 'username': user.username,
926 'firstname': user.name,
926 'firstname': user.name,
927 'lastname': user.lastname,
927 'lastname': user.lastname,
928 'email': user.email,
928 'email': user.email,
929 'emails': user.emails,
929 'emails': user.emails,
930 }
930 }
931 if details == 'basic':
931 if details == 'basic':
932 return data
932 return data
933
933
934 api_key_length = 40
934 api_key_length = 40
935 api_key_replacement = '*' * api_key_length
935 api_key_replacement = '*' * api_key_length
936
936
937 extras = {
937 extras = {
938 'api_keys': [api_key_replacement],
938 'api_keys': [api_key_replacement],
939 'auth_tokens': [api_key_replacement],
939 'auth_tokens': [api_key_replacement],
940 'active': user.active,
940 'active': user.active,
941 'admin': user.admin,
941 'admin': user.admin,
942 'extern_type': user.extern_type,
942 'extern_type': user.extern_type,
943 'extern_name': user.extern_name,
943 'extern_name': user.extern_name,
944 'last_login': user.last_login,
944 'last_login': user.last_login,
945 'last_activity': user.last_activity,
945 'last_activity': user.last_activity,
946 'ip_addresses': user.ip_addresses,
946 'ip_addresses': user.ip_addresses,
947 'language': user_data.get('language')
947 'language': user_data.get('language')
948 }
948 }
949 data.update(extras)
949 data.update(extras)
950
950
951 if include_secrets:
951 if include_secrets:
952 data['api_keys'] = user.auth_tokens
952 data['api_keys'] = user.auth_tokens
953 data['auth_tokens'] = user.extra_auth_tokens
953 data['auth_tokens'] = user.extra_auth_tokens
954 return data
954 return data
955
955
956 def __json__(self):
956 def __json__(self):
957 data = {
957 data = {
958 'full_name': self.full_name,
958 'full_name': self.full_name,
959 'full_name_or_username': self.full_name_or_username,
959 'full_name_or_username': self.full_name_or_username,
960 'short_contact': self.short_contact,
960 'short_contact': self.short_contact,
961 'full_contact': self.full_contact,
961 'full_contact': self.full_contact,
962 }
962 }
963 data.update(self.get_api_data())
963 data.update(self.get_api_data())
964 return data
964 return data
965
965
966
966
967 class UserApiKeys(Base, BaseModel):
967 class UserApiKeys(Base, BaseModel):
968 __tablename__ = 'user_api_keys'
968 __tablename__ = 'user_api_keys'
969 __table_args__ = (
969 __table_args__ = (
970 Index('uak_api_key_idx', 'api_key'),
970 Index('uak_api_key_idx', 'api_key'),
971 Index('uak_api_key_expires_idx', 'api_key', 'expires'),
971 Index('uak_api_key_expires_idx', 'api_key', 'expires'),
972 UniqueConstraint('api_key'),
972 UniqueConstraint('api_key'),
973 {'extend_existing': True, 'mysql_engine': 'InnoDB',
973 {'extend_existing': True, 'mysql_engine': 'InnoDB',
974 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
974 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
975 )
975 )
976 __mapper_args__ = {}
976 __mapper_args__ = {}
977
977
978 # ApiKey role
978 # ApiKey role
979 ROLE_ALL = 'token_role_all'
979 ROLE_ALL = 'token_role_all'
980 ROLE_HTTP = 'token_role_http'
980 ROLE_HTTP = 'token_role_http'
981 ROLE_VCS = 'token_role_vcs'
981 ROLE_VCS = 'token_role_vcs'
982 ROLE_API = 'token_role_api'
982 ROLE_API = 'token_role_api'
983 ROLE_FEED = 'token_role_feed'
983 ROLE_FEED = 'token_role_feed'
984 ROLE_PASSWORD_RESET = 'token_password_reset'
984 ROLE_PASSWORD_RESET = 'token_password_reset'
985
985
986 ROLES = [ROLE_ALL, ROLE_HTTP, ROLE_VCS, ROLE_API, ROLE_FEED]
986 ROLES = [ROLE_ALL, ROLE_HTTP, ROLE_VCS, ROLE_API, ROLE_FEED]
987
987
988 user_api_key_id = Column("user_api_key_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
988 user_api_key_id = Column("user_api_key_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
989 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
989 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
990 api_key = Column("api_key", String(255), nullable=False, unique=True)
990 api_key = Column("api_key", String(255), nullable=False, unique=True)
991 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
991 description = Column('description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
992 expires = Column('expires', Float(53), nullable=False)
992 expires = Column('expires', Float(53), nullable=False)
993 role = Column('role', String(255), nullable=True)
993 role = Column('role', String(255), nullable=True)
994 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
994 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
995
995
996 # scope columns
996 # scope columns
997 repo_id = Column(
997 repo_id = Column(
998 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
998 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
999 nullable=True, unique=None, default=None)
999 nullable=True, unique=None, default=None)
1000 repo = relationship('Repository', lazy='joined')
1000 repo = relationship('Repository', lazy='joined')
1001
1001
1002 repo_group_id = Column(
1002 repo_group_id = Column(
1003 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
1003 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
1004 nullable=True, unique=None, default=None)
1004 nullable=True, unique=None, default=None)
1005 repo_group = relationship('RepoGroup', lazy='joined')
1005 repo_group = relationship('RepoGroup', lazy='joined')
1006
1006
1007 user = relationship('User', lazy='joined')
1007 user = relationship('User', lazy='joined')
1008
1008
1009 def __unicode__(self):
1009 def __unicode__(self):
1010 return u"<%s('%s')>" % (self.__class__.__name__, self.role)
1010 return u"<%s('%s')>" % (self.__class__.__name__, self.role)
1011
1011
1012 def __json__(self):
1012 def __json__(self):
1013 data = {
1013 data = {
1014 'auth_token': self.api_key,
1014 'auth_token': self.api_key,
1015 'role': self.role,
1015 'role': self.role,
1016 'scope': self.scope_humanized,
1016 'scope': self.scope_humanized,
1017 'expired': self.expired
1017 'expired': self.expired
1018 }
1018 }
1019 return data
1019 return data
1020
1020
1021 def get_api_data(self, include_secrets=False):
1021 def get_api_data(self, include_secrets=False):
1022 data = self.__json__()
1022 data = self.__json__()
1023 if include_secrets:
1023 if include_secrets:
1024 return data
1024 return data
1025 else:
1025 else:
1026 data['auth_token'] = self.token_obfuscated
1026 data['auth_token'] = self.token_obfuscated
1027 return data
1027 return data
1028
1028
1029 @hybrid_property
1029 @hybrid_property
1030 def description_safe(self):
1030 def description_safe(self):
1031 from rhodecode.lib import helpers as h
1031 from rhodecode.lib import helpers as h
1032 return h.escape(self.description)
1032 return h.escape(self.description)
1033
1033
1034 @property
1034 @property
1035 def expired(self):
1035 def expired(self):
1036 if self.expires == -1:
1036 if self.expires == -1:
1037 return False
1037 return False
1038 return time.time() > self.expires
1038 return time.time() > self.expires
1039
1039
1040 @classmethod
1040 @classmethod
1041 def _get_role_name(cls, role):
1041 def _get_role_name(cls, role):
1042 return {
1042 return {
1043 cls.ROLE_ALL: _('all'),
1043 cls.ROLE_ALL: _('all'),
1044 cls.ROLE_HTTP: _('http/web interface'),
1044 cls.ROLE_HTTP: _('http/web interface'),
1045 cls.ROLE_VCS: _('vcs (git/hg/svn protocol)'),
1045 cls.ROLE_VCS: _('vcs (git/hg/svn protocol)'),
1046 cls.ROLE_API: _('api calls'),
1046 cls.ROLE_API: _('api calls'),
1047 cls.ROLE_FEED: _('feed access'),
1047 cls.ROLE_FEED: _('feed access'),
1048 }.get(role, role)
1048 }.get(role, role)
1049
1049
1050 @property
1050 @property
1051 def role_humanized(self):
1051 def role_humanized(self):
1052 return self._get_role_name(self.role)
1052 return self._get_role_name(self.role)
1053
1053
1054 def _get_scope(self):
1054 def _get_scope(self):
1055 if self.repo:
1055 if self.repo:
1056 return repr(self.repo)
1056 return repr(self.repo)
1057 if self.repo_group:
1057 if self.repo_group:
1058 return repr(self.repo_group) + ' (recursive)'
1058 return repr(self.repo_group) + ' (recursive)'
1059 return 'global'
1059 return 'global'
1060
1060
1061 @property
1061 @property
1062 def scope_humanized(self):
1062 def scope_humanized(self):
1063 return self._get_scope()
1063 return self._get_scope()
1064
1064
1065 @property
1065 @property
1066 def token_obfuscated(self):
1066 def token_obfuscated(self):
1067 if self.api_key:
1067 if self.api_key:
1068 return self.api_key[:4] + "****"
1068 return self.api_key[:4] + "****"
1069
1069
1070
1070
1071 class UserEmailMap(Base, BaseModel):
1071 class UserEmailMap(Base, BaseModel):
1072 __tablename__ = 'user_email_map'
1072 __tablename__ = 'user_email_map'
1073 __table_args__ = (
1073 __table_args__ = (
1074 Index('uem_email_idx', 'email'),
1074 Index('uem_email_idx', 'email'),
1075 UniqueConstraint('email'),
1075 UniqueConstraint('email'),
1076 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1076 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1077 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
1077 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
1078 )
1078 )
1079 __mapper_args__ = {}
1079 __mapper_args__ = {}
1080
1080
1081 email_id = Column("email_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1081 email_id = Column("email_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1082 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1082 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1083 _email = Column("email", String(255), nullable=True, unique=False, default=None)
1083 _email = Column("email", String(255), nullable=True, unique=False, default=None)
1084 user = relationship('User', lazy='joined')
1084 user = relationship('User', lazy='joined')
1085
1085
1086 @validates('_email')
1086 @validates('_email')
1087 def validate_email(self, key, email):
1087 def validate_email(self, key, email):
1088 # check if this email is not main one
1088 # check if this email is not main one
1089 main_email = Session().query(User).filter(User.email == email).scalar()
1089 main_email = Session().query(User).filter(User.email == email).scalar()
1090 if main_email is not None:
1090 if main_email is not None:
1091 raise AttributeError('email %s is present is user table' % email)
1091 raise AttributeError('email %s is present is user table' % email)
1092 return email
1092 return email
1093
1093
1094 @hybrid_property
1094 @hybrid_property
1095 def email(self):
1095 def email(self):
1096 return self._email
1096 return self._email
1097
1097
1098 @email.setter
1098 @email.setter
1099 def email(self, val):
1099 def email(self, val):
1100 self._email = val.lower() if val else None
1100 self._email = val.lower() if val else None
1101
1101
1102
1102
1103 class UserIpMap(Base, BaseModel):
1103 class UserIpMap(Base, BaseModel):
1104 __tablename__ = 'user_ip_map'
1104 __tablename__ = 'user_ip_map'
1105 __table_args__ = (
1105 __table_args__ = (
1106 UniqueConstraint('user_id', 'ip_addr'),
1106 UniqueConstraint('user_id', 'ip_addr'),
1107 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1107 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1108 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
1108 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
1109 )
1109 )
1110 __mapper_args__ = {}
1110 __mapper_args__ = {}
1111
1111
1112 ip_id = Column("ip_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1112 ip_id = Column("ip_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1113 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1113 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1114 ip_addr = Column("ip_addr", String(255), nullable=True, unique=False, default=None)
1114 ip_addr = Column("ip_addr", String(255), nullable=True, unique=False, default=None)
1115 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
1115 active = Column("active", Boolean(), nullable=True, unique=None, default=True)
1116 description = Column("description", String(10000), nullable=True, unique=None, default=None)
1116 description = Column("description", String(10000), nullable=True, unique=None, default=None)
1117 user = relationship('User', lazy='joined')
1117 user = relationship('User', lazy='joined')
1118
1118
1119 @hybrid_property
1119 @hybrid_property
1120 def description_safe(self):
1120 def description_safe(self):
1121 from rhodecode.lib import helpers as h
1121 from rhodecode.lib import helpers as h
1122 return h.escape(self.description)
1122 return h.escape(self.description)
1123
1123
1124 @classmethod
1124 @classmethod
1125 def _get_ip_range(cls, ip_addr):
1125 def _get_ip_range(cls, ip_addr):
1126 net = ipaddress.ip_network(safe_unicode(ip_addr), strict=False)
1126 net = ipaddress.ip_network(safe_unicode(ip_addr), strict=False)
1127 return [str(net.network_address), str(net.broadcast_address)]
1127 return [str(net.network_address), str(net.broadcast_address)]
1128
1128
1129 def __json__(self):
1129 def __json__(self):
1130 return {
1130 return {
1131 'ip_addr': self.ip_addr,
1131 'ip_addr': self.ip_addr,
1132 'ip_range': self._get_ip_range(self.ip_addr),
1132 'ip_range': self._get_ip_range(self.ip_addr),
1133 }
1133 }
1134
1134
1135 def __unicode__(self):
1135 def __unicode__(self):
1136 return u"<%s('user_id:%s=>%s')>" % (self.__class__.__name__,
1136 return u"<%s('user_id:%s=>%s')>" % (self.__class__.__name__,
1137 self.user_id, self.ip_addr)
1137 self.user_id, self.ip_addr)
1138
1138
1139
1139
1140 class UserLog(Base, BaseModel):
1140 class UserLog(Base, BaseModel):
1141 __tablename__ = 'user_logs'
1141 __tablename__ = 'user_logs'
1142 __table_args__ = (
1142 __table_args__ = (
1143 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1143 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1144 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1144 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1145 )
1145 )
1146 VERSION_1 = 'v1'
1146 VERSION_1 = 'v1'
1147 VERSION_2 = 'v2'
1147 VERSION_2 = 'v2'
1148 VERSIONS = [VERSION_1, VERSION_2]
1148 VERSIONS = [VERSION_1, VERSION_2]
1149
1149
1150 user_log_id = Column("user_log_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1150 user_log_id = Column("user_log_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1151 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1151 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
1152 username = Column("username", String(255), nullable=True, unique=None, default=None)
1152 username = Column("username", String(255), nullable=True, unique=None, default=None)
1153 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True)
1153 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True)
1154 repository_name = Column("repository_name", String(255), nullable=True, unique=None, default=None)
1154 repository_name = Column("repository_name", String(255), nullable=True, unique=None, default=None)
1155 user_ip = Column("user_ip", String(255), nullable=True, unique=None, default=None)
1155 user_ip = Column("user_ip", String(255), nullable=True, unique=None, default=None)
1156 action = Column("action", Text().with_variant(Text(1200000), 'mysql'), nullable=True, unique=None, default=None)
1156 action = Column("action", Text().with_variant(Text(1200000), 'mysql'), nullable=True, unique=None, default=None)
1157 action_date = Column("action_date", DateTime(timezone=False), nullable=True, unique=None, default=None)
1157 action_date = Column("action_date", DateTime(timezone=False), nullable=True, unique=None, default=None)
1158
1158
1159 version = Column("version", String(255), nullable=True, default=VERSION_1)
1159 version = Column("version", String(255), nullable=True, default=VERSION_1)
1160 user_data = Column('user_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
1160 user_data = Column('user_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
1161 action_data = Column('action_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
1161 action_data = Column('action_data_json', MutationObj.as_mutable(JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
1162
1162
1163 def __unicode__(self):
1163 def __unicode__(self):
1164 return u"<%s('id:%s:%s')>" % (
1164 return u"<%s('id:%s:%s')>" % (
1165 self.__class__.__name__, self.repository_name, self.action)
1165 self.__class__.__name__, self.repository_name, self.action)
1166
1166
1167 def __json__(self):
1167 def __json__(self):
1168 return {
1168 return {
1169 'user_id': self.user_id,
1169 'user_id': self.user_id,
1170 'username': self.username,
1170 'username': self.username,
1171 'repository_id': self.repository_id,
1171 'repository_id': self.repository_id,
1172 'repository_name': self.repository_name,
1172 'repository_name': self.repository_name,
1173 'user_ip': self.user_ip,
1173 'user_ip': self.user_ip,
1174 'action_date': self.action_date,
1174 'action_date': self.action_date,
1175 'action': self.action,
1175 'action': self.action,
1176 }
1176 }
1177
1177
1178 @property
1178 @property
1179 def action_as_day(self):
1179 def action_as_day(self):
1180 return datetime.date(*self.action_date.timetuple()[:3])
1180 return datetime.date(*self.action_date.timetuple()[:3])
1181
1181
1182 user = relationship('User')
1182 user = relationship('User')
1183 repository = relationship('Repository', cascade='')
1183 repository = relationship('Repository', cascade='')
1184
1184
1185
1185
1186 class UserGroup(Base, BaseModel):
1186 class UserGroup(Base, BaseModel):
1187 __tablename__ = 'users_groups'
1187 __tablename__ = 'users_groups'
1188 __table_args__ = (
1188 __table_args__ = (
1189 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1189 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1190 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1190 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1191 )
1191 )
1192
1192
1193 users_group_id = Column("users_group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1193 users_group_id = Column("users_group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1194 users_group_name = Column("users_group_name", String(255), nullable=False, unique=True, default=None)
1194 users_group_name = Column("users_group_name", String(255), nullable=False, unique=True, default=None)
1195 user_group_description = Column("user_group_description", String(10000), nullable=True, unique=None, default=None)
1195 user_group_description = Column("user_group_description", String(10000), nullable=True, unique=None, default=None)
1196 users_group_active = Column("users_group_active", Boolean(), nullable=True, unique=None, default=None)
1196 users_group_active = Column("users_group_active", Boolean(), nullable=True, unique=None, default=None)
1197 inherit_default_permissions = Column("users_group_inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
1197 inherit_default_permissions = Column("users_group_inherit_default_permissions", Boolean(), nullable=False, unique=None, default=True)
1198 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
1198 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
1199 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1199 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1200 _group_data = Column("group_data", LargeBinary(), nullable=True) # JSON data
1200 _group_data = Column("group_data", LargeBinary(), nullable=True) # JSON data
1201
1201
1202 members = relationship('UserGroupMember', cascade="all, delete, delete-orphan", lazy="joined")
1202 members = relationship('UserGroupMember', cascade="all, delete, delete-orphan", lazy="joined")
1203 users_group_to_perm = relationship('UserGroupToPerm', cascade='all')
1203 users_group_to_perm = relationship('UserGroupToPerm', cascade='all')
1204 users_group_repo_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1204 users_group_repo_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1205 users_group_repo_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
1205 users_group_repo_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
1206 user_user_group_to_perm = relationship('UserUserGroupToPerm', cascade='all')
1206 user_user_group_to_perm = relationship('UserUserGroupToPerm', cascade='all')
1207 user_group_user_group_to_perm = relationship('UserGroupUserGroupToPerm ', primaryjoin="UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id", cascade='all')
1207 user_group_user_group_to_perm = relationship('UserGroupUserGroupToPerm ', primaryjoin="UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id", cascade='all')
1208
1208
1209 user = relationship('User')
1209 user = relationship('User')
1210
1210
1211 @hybrid_property
1211 @hybrid_property
1212 def description_safe(self):
1212 def description_safe(self):
1213 from rhodecode.lib import helpers as h
1213 from rhodecode.lib import helpers as h
1214 return h.escape(self.description)
1214 return h.escape(self.description)
1215
1215
1216 @hybrid_property
1216 @hybrid_property
1217 def group_data(self):
1217 def group_data(self):
1218 if not self._group_data:
1218 if not self._group_data:
1219 return {}
1219 return {}
1220
1220
1221 try:
1221 try:
1222 return json.loads(self._group_data)
1222 return json.loads(self._group_data)
1223 except TypeError:
1223 except TypeError:
1224 return {}
1224 return {}
1225
1225
1226 @group_data.setter
1226 @group_data.setter
1227 def group_data(self, val):
1227 def group_data(self, val):
1228 try:
1228 try:
1229 self._group_data = json.dumps(val)
1229 self._group_data = json.dumps(val)
1230 except Exception:
1230 except Exception:
1231 log.error(traceback.format_exc())
1231 log.error(traceback.format_exc())
1232
1232
1233 def __unicode__(self):
1233 def __unicode__(self):
1234 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
1234 return u"<%s('id:%s:%s')>" % (self.__class__.__name__,
1235 self.users_group_id,
1235 self.users_group_id,
1236 self.users_group_name)
1236 self.users_group_name)
1237
1237
1238 @classmethod
1238 @classmethod
1239 def get_by_group_name(cls, group_name, cache=False,
1239 def get_by_group_name(cls, group_name, cache=False,
1240 case_insensitive=False):
1240 case_insensitive=False):
1241 if case_insensitive:
1241 if case_insensitive:
1242 q = cls.query().filter(func.lower(cls.users_group_name) ==
1242 q = cls.query().filter(func.lower(cls.users_group_name) ==
1243 func.lower(group_name))
1243 func.lower(group_name))
1244
1244
1245 else:
1245 else:
1246 q = cls.query().filter(cls.users_group_name == group_name)
1246 q = cls.query().filter(cls.users_group_name == group_name)
1247 if cache:
1247 if cache:
1248 q = q.options(
1248 q = q.options(
1249 FromCache("sql_cache_short", "get_group_%s" % _hash_key(group_name)))
1249 FromCache("sql_cache_short", "get_group_%s" % _hash_key(group_name)))
1250 return q.scalar()
1250 return q.scalar()
1251
1251
1252 @classmethod
1252 @classmethod
1253 def get(cls, user_group_id, cache=False):
1253 def get(cls, user_group_id, cache=False):
1254 user_group = cls.query()
1254 user_group = cls.query()
1255 if cache:
1255 if cache:
1256 user_group = user_group.options(
1256 user_group = user_group.options(
1257 FromCache("sql_cache_short", "get_users_group_%s" % user_group_id))
1257 FromCache("sql_cache_short", "get_users_group_%s" % user_group_id))
1258 return user_group.get(user_group_id)
1258 return user_group.get(user_group_id)
1259
1259
1260 def permissions(self, with_admins=True, with_owner=True):
1260 def permissions(self, with_admins=True, with_owner=True):
1261 q = UserUserGroupToPerm.query().filter(UserUserGroupToPerm.user_group == self)
1261 q = UserUserGroupToPerm.query().filter(UserUserGroupToPerm.user_group == self)
1262 q = q.options(joinedload(UserUserGroupToPerm.user_group),
1262 q = q.options(joinedload(UserUserGroupToPerm.user_group),
1263 joinedload(UserUserGroupToPerm.user),
1263 joinedload(UserUserGroupToPerm.user),
1264 joinedload(UserUserGroupToPerm.permission),)
1264 joinedload(UserUserGroupToPerm.permission),)
1265
1265
1266 # get owners and admins and permissions. We do a trick of re-writing
1266 # get owners and admins and permissions. We do a trick of re-writing
1267 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1267 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1268 # has a global reference and changing one object propagates to all
1268 # has a global reference and changing one object propagates to all
1269 # others. This means if admin is also an owner admin_row that change
1269 # others. This means if admin is also an owner admin_row that change
1270 # would propagate to both objects
1270 # would propagate to both objects
1271 perm_rows = []
1271 perm_rows = []
1272 for _usr in q.all():
1272 for _usr in q.all():
1273 usr = AttributeDict(_usr.user.get_dict())
1273 usr = AttributeDict(_usr.user.get_dict())
1274 usr.permission = _usr.permission.permission_name
1274 usr.permission = _usr.permission.permission_name
1275 perm_rows.append(usr)
1275 perm_rows.append(usr)
1276
1276
1277 # filter the perm rows by 'default' first and then sort them by
1277 # filter the perm rows by 'default' first and then sort them by
1278 # admin,write,read,none permissions sorted again alphabetically in
1278 # admin,write,read,none permissions sorted again alphabetically in
1279 # each group
1279 # each group
1280 perm_rows = sorted(perm_rows, key=display_sort)
1280 perm_rows = sorted(perm_rows, key=display_sort)
1281
1281
1282 _admin_perm = 'usergroup.admin'
1282 _admin_perm = 'usergroup.admin'
1283 owner_row = []
1283 owner_row = []
1284 if with_owner:
1284 if with_owner:
1285 usr = AttributeDict(self.user.get_dict())
1285 usr = AttributeDict(self.user.get_dict())
1286 usr.owner_row = True
1286 usr.owner_row = True
1287 usr.permission = _admin_perm
1287 usr.permission = _admin_perm
1288 owner_row.append(usr)
1288 owner_row.append(usr)
1289
1289
1290 super_admin_rows = []
1290 super_admin_rows = []
1291 if with_admins:
1291 if with_admins:
1292 for usr in User.get_all_super_admins():
1292 for usr in User.get_all_super_admins():
1293 # if this admin is also owner, don't double the record
1293 # if this admin is also owner, don't double the record
1294 if usr.user_id == owner_row[0].user_id:
1294 if usr.user_id == owner_row[0].user_id:
1295 owner_row[0].admin_row = True
1295 owner_row[0].admin_row = True
1296 else:
1296 else:
1297 usr = AttributeDict(usr.get_dict())
1297 usr = AttributeDict(usr.get_dict())
1298 usr.admin_row = True
1298 usr.admin_row = True
1299 usr.permission = _admin_perm
1299 usr.permission = _admin_perm
1300 super_admin_rows.append(usr)
1300 super_admin_rows.append(usr)
1301
1301
1302 return super_admin_rows + owner_row + perm_rows
1302 return super_admin_rows + owner_row + perm_rows
1303
1303
1304 def permission_user_groups(self):
1304 def permission_user_groups(self):
1305 q = UserGroupUserGroupToPerm.query().filter(UserGroupUserGroupToPerm.target_user_group == self)
1305 q = UserGroupUserGroupToPerm.query().filter(UserGroupUserGroupToPerm.target_user_group == self)
1306 q = q.options(joinedload(UserGroupUserGroupToPerm.user_group),
1306 q = q.options(joinedload(UserGroupUserGroupToPerm.user_group),
1307 joinedload(UserGroupUserGroupToPerm.target_user_group),
1307 joinedload(UserGroupUserGroupToPerm.target_user_group),
1308 joinedload(UserGroupUserGroupToPerm.permission),)
1308 joinedload(UserGroupUserGroupToPerm.permission),)
1309
1309
1310 perm_rows = []
1310 perm_rows = []
1311 for _user_group in q.all():
1311 for _user_group in q.all():
1312 usr = AttributeDict(_user_group.user_group.get_dict())
1312 usr = AttributeDict(_user_group.user_group.get_dict())
1313 usr.permission = _user_group.permission.permission_name
1313 usr.permission = _user_group.permission.permission_name
1314 perm_rows.append(usr)
1314 perm_rows.append(usr)
1315
1315
1316 return perm_rows
1316 return perm_rows
1317
1317
1318 def _get_default_perms(self, user_group, suffix=''):
1318 def _get_default_perms(self, user_group, suffix=''):
1319 from rhodecode.model.permission import PermissionModel
1319 from rhodecode.model.permission import PermissionModel
1320 return PermissionModel().get_default_perms(user_group.users_group_to_perm, suffix)
1320 return PermissionModel().get_default_perms(user_group.users_group_to_perm, suffix)
1321
1321
1322 def get_default_perms(self, suffix=''):
1322 def get_default_perms(self, suffix=''):
1323 return self._get_default_perms(self, suffix)
1323 return self._get_default_perms(self, suffix)
1324
1324
1325 def get_api_data(self, with_group_members=True, include_secrets=False):
1325 def get_api_data(self, with_group_members=True, include_secrets=False):
1326 """
1326 """
1327 :param include_secrets: See :meth:`User.get_api_data`, this parameter is
1327 :param include_secrets: See :meth:`User.get_api_data`, this parameter is
1328 basically forwarded.
1328 basically forwarded.
1329
1329
1330 """
1330 """
1331 user_group = self
1331 user_group = self
1332 data = {
1332 data = {
1333 'users_group_id': user_group.users_group_id,
1333 'users_group_id': user_group.users_group_id,
1334 'group_name': user_group.users_group_name,
1334 'group_name': user_group.users_group_name,
1335 'group_description': user_group.user_group_description,
1335 'group_description': user_group.user_group_description,
1336 'active': user_group.users_group_active,
1336 'active': user_group.users_group_active,
1337 'owner': user_group.user.username,
1337 'owner': user_group.user.username,
1338 'owner_email': user_group.user.email,
1338 'owner_email': user_group.user.email,
1339 }
1339 }
1340
1340
1341 if with_group_members:
1341 if with_group_members:
1342 users = []
1342 users = []
1343 for user in user_group.members:
1343 for user in user_group.members:
1344 user = user.user
1344 user = user.user
1345 users.append(user.get_api_data(include_secrets=include_secrets))
1345 users.append(user.get_api_data(include_secrets=include_secrets))
1346 data['users'] = users
1346 data['users'] = users
1347
1347
1348 return data
1348 return data
1349
1349
1350
1350
1351 class UserGroupMember(Base, BaseModel):
1351 class UserGroupMember(Base, BaseModel):
1352 __tablename__ = 'users_groups_members'
1352 __tablename__ = 'users_groups_members'
1353 __table_args__ = (
1353 __table_args__ = (
1354 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1354 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1355 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1355 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1356 )
1356 )
1357
1357
1358 users_group_member_id = Column("users_group_member_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1358 users_group_member_id = Column("users_group_member_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1359 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
1359 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
1360 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
1360 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
1361
1361
1362 user = relationship('User', lazy='joined')
1362 user = relationship('User', lazy='joined')
1363 users_group = relationship('UserGroup')
1363 users_group = relationship('UserGroup')
1364
1364
1365 def __init__(self, gr_id='', u_id=''):
1365 def __init__(self, gr_id='', u_id=''):
1366 self.users_group_id = gr_id
1366 self.users_group_id = gr_id
1367 self.user_id = u_id
1367 self.user_id = u_id
1368
1368
1369
1369
1370 class RepositoryField(Base, BaseModel):
1370 class RepositoryField(Base, BaseModel):
1371 __tablename__ = 'repositories_fields'
1371 __tablename__ = 'repositories_fields'
1372 __table_args__ = (
1372 __table_args__ = (
1373 UniqueConstraint('repository_id', 'field_key'), # no-multi field
1373 UniqueConstraint('repository_id', 'field_key'), # no-multi field
1374 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1374 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1375 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1375 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1376 )
1376 )
1377 PREFIX = 'ex_' # prefix used in form to not conflict with already existing fields
1377 PREFIX = 'ex_' # prefix used in form to not conflict with already existing fields
1378
1378
1379 repo_field_id = Column("repo_field_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1379 repo_field_id = Column("repo_field_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
1380 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
1380 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
1381 field_key = Column("field_key", String(250))
1381 field_key = Column("field_key", String(250))
1382 field_label = Column("field_label", String(1024), nullable=False)
1382 field_label = Column("field_label", String(1024), nullable=False)
1383 field_value = Column("field_value", String(10000), nullable=False)
1383 field_value = Column("field_value", String(10000), nullable=False)
1384 field_desc = Column("field_desc", String(1024), nullable=False)
1384 field_desc = Column("field_desc", String(1024), nullable=False)
1385 field_type = Column("field_type", String(255), nullable=False, unique=None)
1385 field_type = Column("field_type", String(255), nullable=False, unique=None)
1386 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1386 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
1387
1387
1388 repository = relationship('Repository')
1388 repository = relationship('Repository')
1389
1389
1390 @property
1390 @property
1391 def field_key_prefixed(self):
1391 def field_key_prefixed(self):
1392 return 'ex_%s' % self.field_key
1392 return 'ex_%s' % self.field_key
1393
1393
1394 @classmethod
1394 @classmethod
1395 def un_prefix_key(cls, key):
1395 def un_prefix_key(cls, key):
1396 if key.startswith(cls.PREFIX):
1396 if key.startswith(cls.PREFIX):
1397 return key[len(cls.PREFIX):]
1397 return key[len(cls.PREFIX):]
1398 return key
1398 return key
1399
1399
1400 @classmethod
1400 @classmethod
1401 def get_by_key_name(cls, key, repo):
1401 def get_by_key_name(cls, key, repo):
1402 row = cls.query()\
1402 row = cls.query()\
1403 .filter(cls.repository == repo)\
1403 .filter(cls.repository == repo)\
1404 .filter(cls.field_key == key).scalar()
1404 .filter(cls.field_key == key).scalar()
1405 return row
1405 return row
1406
1406
1407
1407
1408 class Repository(Base, BaseModel):
1408 class Repository(Base, BaseModel):
1409 __tablename__ = 'repositories'
1409 __tablename__ = 'repositories'
1410 __table_args__ = (
1410 __table_args__ = (
1411 Index('r_repo_name_idx', 'repo_name', mysql_length=255),
1411 Index('r_repo_name_idx', 'repo_name', mysql_length=255),
1412 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1412 {'extend_existing': True, 'mysql_engine': 'InnoDB',
1413 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1413 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
1414 )
1414 )
1415 DEFAULT_CLONE_URI = '{scheme}://{user}@{netloc}/{repo}'
1415 DEFAULT_CLONE_URI = '{scheme}://{user}@{netloc}/{repo}'
1416 DEFAULT_CLONE_URI_ID = '{scheme}://{user}@{netloc}/_{repoid}'
1416 DEFAULT_CLONE_URI_ID = '{scheme}://{user}@{netloc}/_{repoid}'
1417
1417
1418 STATE_CREATED = 'repo_state_created'
1418 STATE_CREATED = 'repo_state_created'
1419 STATE_PENDING = 'repo_state_pending'
1419 STATE_PENDING = 'repo_state_pending'
1420 STATE_ERROR = 'repo_state_error'
1420 STATE_ERROR = 'repo_state_error'
1421
1421
1422 LOCK_AUTOMATIC = 'lock_auto'
1422 LOCK_AUTOMATIC = 'lock_auto'
1423 LOCK_API = 'lock_api'
1423 LOCK_API = 'lock_api'
1424 LOCK_WEB = 'lock_web'
1424 LOCK_WEB = 'lock_web'
1425 LOCK_PULL = 'lock_pull'
1425 LOCK_PULL = 'lock_pull'
1426
1426
1427 NAME_SEP = URL_SEP
1427 NAME_SEP = URL_SEP
1428
1428
1429 repo_id = Column(
1429 repo_id = Column(
1430 "repo_id", Integer(), nullable=False, unique=True, default=None,
1430 "repo_id", Integer(), nullable=False, unique=True, default=None,
1431 primary_key=True)
1431 primary_key=True)
1432 _repo_name = Column(
1432 _repo_name = Column(
1433 "repo_name", Text(), nullable=False, default=None)
1433 "repo_name", Text(), nullable=False, default=None)
1434 _repo_name_hash = Column(
1434 _repo_name_hash = Column(
1435 "repo_name_hash", String(255), nullable=False, unique=True)
1435 "repo_name_hash", String(255), nullable=False, unique=True)
1436 repo_state = Column("repo_state", String(255), nullable=True)
1436 repo_state = Column("repo_state", String(255), nullable=True)
1437
1437
1438 clone_uri = Column(
1438 clone_uri = Column(
1439 "clone_uri", EncryptedTextValue(), nullable=True, unique=False,
1439 "clone_uri", EncryptedTextValue(), nullable=True, unique=False,
1440 default=None)
1440 default=None)
1441 repo_type = Column(
1441 repo_type = Column(
1442 "repo_type", String(255), nullable=False, unique=False, default=None)
1442 "repo_type", String(255), nullable=False, unique=False, default=None)
1443 user_id = Column(
1443 user_id = Column(
1444 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
1444 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
1445 unique=False, default=None)
1445 unique=False, default=None)
1446 private = Column(
1446 private = Column(
1447 "private", Boolean(), nullable=True, unique=None, default=None)
1447 "private", Boolean(), nullable=True, unique=None, default=None)
1448 enable_statistics = Column(
1448 enable_statistics = Column(
1449 "statistics", Boolean(), nullable=True, unique=None, default=True)
1449 "statistics", Boolean(), nullable=True, unique=None, default=True)
1450 enable_downloads = Column(
1450 enable_downloads = Column(
1451 "downloads", Boolean(), nullable=True, unique=None, default=True)
1451 "downloads", Boolean(), nullable=True, unique=None, default=True)
1452 description = Column(
1452 description = Column(
1453 "description", String(10000), nullable=True, unique=None, default=None)
1453 "description", String(10000), nullable=True, unique=None, default=None)
1454 created_on = Column(
1454 created_on = Column(
1455 'created_on', DateTime(timezone=False), nullable=True, unique=None,
1455 'created_on', DateTime(timezone=False), nullable=True, unique=None,
1456 default=datetime.datetime.now)
1456 default=datetime.datetime.now)
1457 updated_on = Column(
1457 updated_on = Column(
1458 'updated_on', DateTime(timezone=False), nullable=True, unique=None,
1458 'updated_on', DateTime(timezone=False), nullable=True, unique=None,
1459 default=datetime.datetime.now)
1459 default=datetime.datetime.now)
1460 _landing_revision = Column(
1460 _landing_revision = Column(
1461 "landing_revision", String(255), nullable=False, unique=False,
1461 "landing_revision", String(255), nullable=False, unique=False,
1462 default=None)
1462 default=None)
1463 enable_locking = Column(
1463 enable_locking = Column(
1464 "enable_locking", Boolean(), nullable=False, unique=None,
1464 "enable_locking", Boolean(), nullable=False, unique=None,
1465 default=False)
1465 default=False)
1466 _locked = Column(
1466 _locked = Column(
1467 "locked", String(255), nullable=True, unique=False, default=None)
1467 "locked", String(255), nullable=True, unique=False, default=None)
1468 _changeset_cache = Column(
1468 _changeset_cache = Column(
1469 "changeset_cache", LargeBinary(), nullable=True) # JSON data
1469 "changeset_cache", LargeBinary(), nullable=True) # JSON data
1470
1470
1471 fork_id = Column(
1471 fork_id = Column(
1472 "fork_id", Integer(), ForeignKey('repositories.repo_id'),
1472 "fork_id", Integer(), ForeignKey('repositories.repo_id'),
1473 nullable=True, unique=False, default=None)
1473 nullable=True, unique=False, default=None)
1474 group_id = Column(
1474 group_id = Column(
1475 "group_id", Integer(), ForeignKey('groups.group_id'), nullable=True,
1475 "group_id", Integer(), ForeignKey('groups.group_id'), nullable=True,
1476 unique=False, default=None)
1476 unique=False, default=None)
1477
1477
1478 user = relationship('User', lazy='joined')
1478 user = relationship('User', lazy='joined')
1479 fork = relationship('Repository', remote_side=repo_id, lazy='joined')
1479 fork = relationship('Repository', remote_side=repo_id, lazy='joined')
1480 group = relationship('RepoGroup', lazy='joined')
1480 group = relationship('RepoGroup', lazy='joined')
1481 repo_to_perm = relationship(
1481 repo_to_perm = relationship(
1482 'UserRepoToPerm', cascade='all',
1482 'UserRepoToPerm', cascade='all',
1483 order_by='UserRepoToPerm.repo_to_perm_id')
1483 order_by='UserRepoToPerm.repo_to_perm_id')
1484 users_group_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1484 users_group_to_perm = relationship('UserGroupRepoToPerm', cascade='all')
1485 stats = relationship('Statistics', cascade='all', uselist=False)
1485 stats = relationship('Statistics', cascade='all', uselist=False)
1486
1486
1487 followers = relationship(
1487 followers = relationship(
1488 'UserFollowing',
1488 'UserFollowing',
1489 primaryjoin='UserFollowing.follows_repo_id==Repository.repo_id',
1489 primaryjoin='UserFollowing.follows_repo_id==Repository.repo_id',
1490 cascade='all')
1490 cascade='all')
1491 extra_fields = relationship(
1491 extra_fields = relationship(
1492 'RepositoryField', cascade="all, delete, delete-orphan")
1492 'RepositoryField', cascade="all, delete, delete-orphan")
1493 logs = relationship('UserLog')
1493 logs = relationship('UserLog')
1494 comments = relationship(
1494 comments = relationship(
1495 'ChangesetComment', cascade="all, delete, delete-orphan")
1495 'ChangesetComment', cascade="all, delete, delete-orphan")
1496 pull_requests_source = relationship(
1496 pull_requests_source = relationship(
1497 'PullRequest',
1497 'PullRequest',
1498 primaryjoin='PullRequest.source_repo_id==Repository.repo_id',
1498 primaryjoin='PullRequest.source_repo_id==Repository.repo_id',
1499 cascade="all, delete, delete-orphan")
1499 cascade="all, delete, delete-orphan")
1500 pull_requests_target = relationship(
1500 pull_requests_target = relationship(
1501 'PullRequest',
1501 'PullRequest',
1502 primaryjoin='PullRequest.target_repo_id==Repository.repo_id',
1502 primaryjoin='PullRequest.target_repo_id==Repository.repo_id',
1503 cascade="all, delete, delete-orphan")
1503 cascade="all, delete, delete-orphan")
1504 ui = relationship('RepoRhodeCodeUi', cascade="all")
1504 ui = relationship('RepoRhodeCodeUi', cascade="all")
1505 settings = relationship('RepoRhodeCodeSetting', cascade="all")
1505 settings = relationship('RepoRhodeCodeSetting', cascade="all")
1506 integrations = relationship('Integration',
1506 integrations = relationship('Integration',
1507 cascade="all, delete, delete-orphan")
1507 cascade="all, delete, delete-orphan")
1508
1508
1509 def __unicode__(self):
1509 def __unicode__(self):
1510 return u"<%s('%s:%s')>" % (self.__class__.__name__, self.repo_id,
1510 return u"<%s('%s:%s')>" % (self.__class__.__name__, self.repo_id,
1511 safe_unicode(self.repo_name))
1511 safe_unicode(self.repo_name))
1512
1512
1513 @hybrid_property
1513 @hybrid_property
1514 def description_safe(self):
1514 def description_safe(self):
1515 from rhodecode.lib import helpers as h
1515 from rhodecode.lib import helpers as h
1516 return h.escape(self.description)
1516 return h.escape(self.description)
1517
1517
1518 @hybrid_property
1518 @hybrid_property
1519 def landing_rev(self):
1519 def landing_rev(self):
1520 # always should return [rev_type, rev]
1520 # always should return [rev_type, rev]
1521 if self._landing_revision:
1521 if self._landing_revision:
1522 _rev_info = self._landing_revision.split(':')
1522 _rev_info = self._landing_revision.split(':')
1523 if len(_rev_info) < 2:
1523 if len(_rev_info) < 2:
1524 _rev_info.insert(0, 'rev')
1524 _rev_info.insert(0, 'rev')
1525 return [_rev_info[0], _rev_info[1]]
1525 return [_rev_info[0], _rev_info[1]]
1526 return [None, None]
1526 return [None, None]
1527
1527
1528 @landing_rev.setter
1528 @landing_rev.setter
1529 def landing_rev(self, val):
1529 def landing_rev(self, val):
1530 if ':' not in val:
1530 if ':' not in val:
1531 raise ValueError('value must be delimited with `:` and consist '
1531 raise ValueError('value must be delimited with `:` and consist '
1532 'of <rev_type>:<rev>, got %s instead' % val)
1532 'of <rev_type>:<rev>, got %s instead' % val)
1533 self._landing_revision = val
1533 self._landing_revision = val
1534
1534
1535 @hybrid_property
1535 @hybrid_property
1536 def locked(self):
1536 def locked(self):
1537 if self._locked:
1537 if self._locked:
1538 user_id, timelocked, reason = self._locked.split(':')
1538 user_id, timelocked, reason = self._locked.split(':')
1539 lock_values = int(user_id), timelocked, reason
1539 lock_values = int(user_id), timelocked, reason
1540 else:
1540 else:
1541 lock_values = [None, None, None]
1541 lock_values = [None, None, None]
1542 return lock_values
1542 return lock_values
1543
1543
1544 @locked.setter
1544 @locked.setter
1545 def locked(self, val):
1545 def locked(self, val):
1546 if val and isinstance(val, (list, tuple)):
1546 if val and isinstance(val, (list, tuple)):
1547 self._locked = ':'.join(map(str, val))
1547 self._locked = ':'.join(map(str, val))
1548 else:
1548 else:
1549 self._locked = None
1549 self._locked = None
1550
1550
1551 @hybrid_property
1551 @hybrid_property
1552 def changeset_cache(self):
1552 def changeset_cache(self):
1553 from rhodecode.lib.vcs.backends.base import EmptyCommit
1553 from rhodecode.lib.vcs.backends.base import EmptyCommit
1554 dummy = EmptyCommit().__json__()
1554 dummy = EmptyCommit().__json__()
1555 if not self._changeset_cache:
1555 if not self._changeset_cache:
1556 return dummy
1556 return dummy
1557 try:
1557 try:
1558 return json.loads(self._changeset_cache)
1558 return json.loads(self._changeset_cache)
1559 except TypeError:
1559 except TypeError:
1560 return dummy
1560 return dummy
1561 except Exception:
1561 except Exception:
1562 log.error(traceback.format_exc())
1562 log.error(traceback.format_exc())
1563 return dummy
1563 return dummy
1564
1564
1565 @changeset_cache.setter
1565 @changeset_cache.setter
1566 def changeset_cache(self, val):
1566 def changeset_cache(self, val):
1567 try:
1567 try:
1568 self._changeset_cache = json.dumps(val)
1568 self._changeset_cache = json.dumps(val)
1569 except Exception:
1569 except Exception:
1570 log.error(traceback.format_exc())
1570 log.error(traceback.format_exc())
1571
1571
1572 @hybrid_property
1572 @hybrid_property
1573 def repo_name(self):
1573 def repo_name(self):
1574 return self._repo_name
1574 return self._repo_name
1575
1575
1576 @repo_name.setter
1576 @repo_name.setter
1577 def repo_name(self, value):
1577 def repo_name(self, value):
1578 self._repo_name = value
1578 self._repo_name = value
1579 self._repo_name_hash = hashlib.sha1(safe_str(value)).hexdigest()
1579 self._repo_name_hash = hashlib.sha1(safe_str(value)).hexdigest()
1580
1580
1581 @classmethod
1581 @classmethod
1582 def normalize_repo_name(cls, repo_name):
1582 def normalize_repo_name(cls, repo_name):
1583 """
1583 """
1584 Normalizes os specific repo_name to the format internally stored inside
1584 Normalizes os specific repo_name to the format internally stored inside
1585 database using URL_SEP
1585 database using URL_SEP
1586
1586
1587 :param cls:
1587 :param cls:
1588 :param repo_name:
1588 :param repo_name:
1589 """
1589 """
1590 return cls.NAME_SEP.join(repo_name.split(os.sep))
1590 return cls.NAME_SEP.join(repo_name.split(os.sep))
1591
1591
1592 @classmethod
1592 @classmethod
1593 def get_by_repo_name(cls, repo_name, cache=False, identity_cache=False):
1593 def get_by_repo_name(cls, repo_name, cache=False, identity_cache=False):
1594 session = Session()
1594 session = Session()
1595 q = session.query(cls).filter(cls.repo_name == repo_name)
1595 q = session.query(cls).filter(cls.repo_name == repo_name)
1596
1596
1597 if cache:
1597 if cache:
1598 if identity_cache:
1598 if identity_cache:
1599 val = cls.identity_cache(session, 'repo_name', repo_name)
1599 val = cls.identity_cache(session, 'repo_name', repo_name)
1600 if val:
1600 if val:
1601 return val
1601 return val
1602 else:
1602 else:
1603 cache_key = "get_repo_by_name_%s" % _hash_key(repo_name)
1603 cache_key = "get_repo_by_name_%s" % _hash_key(repo_name)
1604 q = q.options(
1604 q = q.options(
1605 FromCache("sql_cache_short", cache_key))
1605 FromCache("sql_cache_short", cache_key))
1606
1606
1607 return q.scalar()
1607 return q.scalar()
1608
1608
1609 @classmethod
1609 @classmethod
1610 def get_by_full_path(cls, repo_full_path):
1610 def get_by_full_path(cls, repo_full_path):
1611 repo_name = repo_full_path.split(cls.base_path(), 1)[-1]
1611 repo_name = repo_full_path.split(cls.base_path(), 1)[-1]
1612 repo_name = cls.normalize_repo_name(repo_name)
1612 repo_name = cls.normalize_repo_name(repo_name)
1613 return cls.get_by_repo_name(repo_name.strip(URL_SEP))
1613 return cls.get_by_repo_name(repo_name.strip(URL_SEP))
1614
1614
1615 @classmethod
1615 @classmethod
1616 def get_repo_forks(cls, repo_id):
1616 def get_repo_forks(cls, repo_id):
1617 return cls.query().filter(Repository.fork_id == repo_id)
1617 return cls.query().filter(Repository.fork_id == repo_id)
1618
1618
1619 @classmethod
1619 @classmethod
1620 def base_path(cls):
1620 def base_path(cls):
1621 """
1621 """
1622 Returns base path when all repos are stored
1622 Returns base path when all repos are stored
1623
1623
1624 :param cls:
1624 :param cls:
1625 """
1625 """
1626 q = Session().query(RhodeCodeUi)\
1626 q = Session().query(RhodeCodeUi)\
1627 .filter(RhodeCodeUi.ui_key == cls.NAME_SEP)
1627 .filter(RhodeCodeUi.ui_key == cls.NAME_SEP)
1628 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1628 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1629 return q.one().ui_value
1629 return q.one().ui_value
1630
1630
1631 @classmethod
1631 @classmethod
1632 def is_valid(cls, repo_name):
1632 def is_valid(cls, repo_name):
1633 """
1633 """
1634 returns True if given repo name is a valid filesystem repository
1634 returns True if given repo name is a valid filesystem repository
1635
1635
1636 :param cls:
1636 :param cls:
1637 :param repo_name:
1637 :param repo_name:
1638 """
1638 """
1639 from rhodecode.lib.utils import is_valid_repo
1639 from rhodecode.lib.utils import is_valid_repo
1640
1640
1641 return is_valid_repo(repo_name, cls.base_path())
1641 return is_valid_repo(repo_name, cls.base_path())
1642
1642
1643 @classmethod
1643 @classmethod
1644 def get_all_repos(cls, user_id=Optional(None), group_id=Optional(None),
1644 def get_all_repos(cls, user_id=Optional(None), group_id=Optional(None),
1645 case_insensitive=True):
1645 case_insensitive=True):
1646 q = Repository.query()
1646 q = Repository.query()
1647
1647
1648 if not isinstance(user_id, Optional):
1648 if not isinstance(user_id, Optional):
1649 q = q.filter(Repository.user_id == user_id)
1649 q = q.filter(Repository.user_id == user_id)
1650
1650
1651 if not isinstance(group_id, Optional):
1651 if not isinstance(group_id, Optional):
1652 q = q.filter(Repository.group_id == group_id)
1652 q = q.filter(Repository.group_id == group_id)
1653
1653
1654 if case_insensitive:
1654 if case_insensitive:
1655 q = q.order_by(func.lower(Repository.repo_name))
1655 q = q.order_by(func.lower(Repository.repo_name))
1656 else:
1656 else:
1657 q = q.order_by(Repository.repo_name)
1657 q = q.order_by(Repository.repo_name)
1658 return q.all()
1658 return q.all()
1659
1659
1660 @property
1660 @property
1661 def forks(self):
1661 def forks(self):
1662 """
1662 """
1663 Return forks of this repo
1663 Return forks of this repo
1664 """
1664 """
1665 return Repository.get_repo_forks(self.repo_id)
1665 return Repository.get_repo_forks(self.repo_id)
1666
1666
1667 @property
1667 @property
1668 def parent(self):
1668 def parent(self):
1669 """
1669 """
1670 Returns fork parent
1670 Returns fork parent
1671 """
1671 """
1672 return self.fork
1672 return self.fork
1673
1673
1674 @property
1674 @property
1675 def just_name(self):
1675 def just_name(self):
1676 return self.repo_name.split(self.NAME_SEP)[-1]
1676 return self.repo_name.split(self.NAME_SEP)[-1]
1677
1677
1678 @property
1678 @property
1679 def groups_with_parents(self):
1679 def groups_with_parents(self):
1680 groups = []
1680 groups = []
1681 if self.group is None:
1681 if self.group is None:
1682 return groups
1682 return groups
1683
1683
1684 cur_gr = self.group
1684 cur_gr = self.group
1685 groups.insert(0, cur_gr)
1685 groups.insert(0, cur_gr)
1686 while 1:
1686 while 1:
1687 gr = getattr(cur_gr, 'parent_group', None)
1687 gr = getattr(cur_gr, 'parent_group', None)
1688 cur_gr = cur_gr.parent_group
1688 cur_gr = cur_gr.parent_group
1689 if gr is None:
1689 if gr is None:
1690 break
1690 break
1691 groups.insert(0, gr)
1691 groups.insert(0, gr)
1692
1692
1693 return groups
1693 return groups
1694
1694
1695 @property
1695 @property
1696 def groups_and_repo(self):
1696 def groups_and_repo(self):
1697 return self.groups_with_parents, self
1697 return self.groups_with_parents, self
1698
1698
1699 @LazyProperty
1699 @LazyProperty
1700 def repo_path(self):
1700 def repo_path(self):
1701 """
1701 """
1702 Returns base full path for that repository means where it actually
1702 Returns base full path for that repository means where it actually
1703 exists on a filesystem
1703 exists on a filesystem
1704 """
1704 """
1705 q = Session().query(RhodeCodeUi).filter(
1705 q = Session().query(RhodeCodeUi).filter(
1706 RhodeCodeUi.ui_key == self.NAME_SEP)
1706 RhodeCodeUi.ui_key == self.NAME_SEP)
1707 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1707 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
1708 return q.one().ui_value
1708 return q.one().ui_value
1709
1709
1710 @property
1710 @property
1711 def repo_full_path(self):
1711 def repo_full_path(self):
1712 p = [self.repo_path]
1712 p = [self.repo_path]
1713 # we need to split the name by / since this is how we store the
1713 # we need to split the name by / since this is how we store the
1714 # names in the database, but that eventually needs to be converted
1714 # names in the database, but that eventually needs to be converted
1715 # into a valid system path
1715 # into a valid system path
1716 p += self.repo_name.split(self.NAME_SEP)
1716 p += self.repo_name.split(self.NAME_SEP)
1717 return os.path.join(*map(safe_unicode, p))
1717 return os.path.join(*map(safe_unicode, p))
1718
1718
1719 @property
1719 @property
1720 def cache_keys(self):
1720 def cache_keys(self):
1721 """
1721 """
1722 Returns associated cache keys for that repo
1722 Returns associated cache keys for that repo
1723 """
1723 """
1724 return CacheKey.query()\
1724 return CacheKey.query()\
1725 .filter(CacheKey.cache_args == self.repo_name)\
1725 .filter(CacheKey.cache_args == self.repo_name)\
1726 .order_by(CacheKey.cache_key)\
1726 .order_by(CacheKey.cache_key)\
1727 .all()
1727 .all()
1728
1728
1729 def get_new_name(self, repo_name):
1729 def get_new_name(self, repo_name):
1730 """
1730 """
1731 returns new full repository name based on assigned group and new new
1731 returns new full repository name based on assigned group and new new
1732
1732
1733 :param group_name:
1733 :param group_name:
1734 """
1734 """
1735 path_prefix = self.group.full_path_splitted if self.group else []
1735 path_prefix = self.group.full_path_splitted if self.group else []
1736 return self.NAME_SEP.join(path_prefix + [repo_name])
1736 return self.NAME_SEP.join(path_prefix + [repo_name])
1737
1737
1738 @property
1738 @property
1739 def _config(self):
1739 def _config(self):
1740 """
1740 """
1741 Returns db based config object.
1741 Returns db based config object.
1742 """
1742 """
1743 from rhodecode.lib.utils import make_db_config
1743 from rhodecode.lib.utils import make_db_config
1744 return make_db_config(clear_session=False, repo=self)
1744 return make_db_config(clear_session=False, repo=self)
1745
1745
1746 def permissions(self, with_admins=True, with_owner=True):
1746 def permissions(self, with_admins=True, with_owner=True):
1747 q = UserRepoToPerm.query().filter(UserRepoToPerm.repository == self)
1747 q = UserRepoToPerm.query().filter(UserRepoToPerm.repository == self)
1748 q = q.options(joinedload(UserRepoToPerm.repository),
1748 q = q.options(joinedload(UserRepoToPerm.repository),
1749 joinedload(UserRepoToPerm.user),
1749 joinedload(UserRepoToPerm.user),
1750 joinedload(UserRepoToPerm.permission),)
1750 joinedload(UserRepoToPerm.permission),)
1751
1751
1752 # get owners and admins and permissions. We do a trick of re-writing
1752 # get owners and admins and permissions. We do a trick of re-writing
1753 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1753 # objects from sqlalchemy to named-tuples due to sqlalchemy session
1754 # has a global reference and changing one object propagates to all
1754 # has a global reference and changing one object propagates to all
1755 # others. This means if admin is also an owner admin_row that change
1755 # others. This means if admin is also an owner admin_row that change
1756 # would propagate to both objects
1756 # would propagate to both objects
1757 perm_rows = []
1757 perm_rows = []
1758 for _usr in q.all():
1758 for _usr in q.all():
1759 usr = AttributeDict(_usr.user.get_dict())
1759 usr = AttributeDict(_usr.user.get_dict())
1760 usr.permission = _usr.permission.permission_name
1760 usr.permission = _usr.permission.permission_name
1761 perm_rows.append(usr)
1761 perm_rows.append(usr)
1762
1762
1763 # filter the perm rows by 'default' first and then sort them by
1763 # filter the perm rows by 'default' first and then sort them by
1764 # admin,write,read,none permissions sorted again alphabetically in
1764 # admin,write,read,none permissions sorted again alphabetically in
1765 # each group
1765 # each group
1766 perm_rows = sorted(perm_rows, key=display_sort)
1766 perm_rows = sorted(perm_rows, key=display_sort)
1767
1767
1768 _admin_perm = 'repository.admin'
1768 _admin_perm = 'repository.admin'
1769 owner_row = []
1769 owner_row = []
1770 if with_owner:
1770 if with_owner:
1771 usr = AttributeDict(self.user.get_dict())
1771 usr = AttributeDict(self.user.get_dict())
1772 usr.owner_row = True
1772 usr.owner_row = True
1773 usr.permission = _admin_perm
1773 usr.permission = _admin_perm
1774 owner_row.append(usr)
1774 owner_row.append(usr)
1775
1775
1776 super_admin_rows = []
1776 super_admin_rows = []
1777 if with_admins:
1777 if with_admins:
1778 for usr in User.get_all_super_admins():
1778 for usr in User.get_all_super_admins():
1779 # if this admin is also owner, don't double the record
1779 # if this admin is also owner, don't double the record
1780 if usr.user_id == owner_row[0].user_id:
1780 if usr.user_id == owner_row[0].user_id:
1781 owner_row[0].admin_row = True
1781 owner_row[0].admin_row = True
1782 else:
1782 else:
1783 usr = AttributeDict(usr.get_dict())
1783 usr = AttributeDict(usr.get_dict())
1784 usr.admin_row = True
1784 usr.admin_row = True
1785 usr.permission = _admin_perm
1785 usr.permission = _admin_perm
1786 super_admin_rows.append(usr)
1786 super_admin_rows.append(usr)
1787
1787
1788 return super_admin_rows + owner_row + perm_rows
1788 return super_admin_rows + owner_row + perm_rows
1789
1789
1790 def permission_user_groups(self):
1790 def permission_user_groups(self):
1791 q = UserGroupRepoToPerm.query().filter(
1791 q = UserGroupRepoToPerm.query().filter(
1792 UserGroupRepoToPerm.repository == self)
1792 UserGroupRepoToPerm.repository == self)
1793 q = q.options(joinedload(UserGroupRepoToPerm.repository),
1793 q = q.options(joinedload(UserGroupRepoToPerm.repository),
1794 joinedload(UserGroupRepoToPerm.users_group),
1794 joinedload(UserGroupRepoToPerm.users_group),
1795 joinedload(UserGroupRepoToPerm.permission),)
1795 joinedload(UserGroupRepoToPerm.permission),)
1796
1796
1797 perm_rows = []
1797 perm_rows = []
1798 for _user_group in q.all():
1798 for _user_group in q.all():
1799 usr = AttributeDict(_user_group.users_group.get_dict())
1799 usr = AttributeDict(_user_group.users_group.get_dict())
1800 usr.permission = _user_group.permission.permission_name
1800 usr.permission = _user_group.permission.permission_name
1801 perm_rows.append(usr)
1801 perm_rows.append(usr)
1802
1802
1803 return perm_rows
1803 return perm_rows
1804
1804
1805 def get_api_data(self, include_secrets=False):
1805 def get_api_data(self, include_secrets=False):
1806 """
1806 """
1807 Common function for generating repo api data
1807 Common function for generating repo api data
1808
1808
1809 :param include_secrets: See :meth:`User.get_api_data`.
1809 :param include_secrets: See :meth:`User.get_api_data`.
1810
1810
1811 """
1811 """
1812 # TODO: mikhail: Here there is an anti-pattern, we probably need to
1812 # TODO: mikhail: Here there is an anti-pattern, we probably need to
1813 # move this methods on models level.
1813 # move this methods on models level.
1814 from rhodecode.model.settings import SettingsModel
1814 from rhodecode.model.settings import SettingsModel
1815 from rhodecode.model.repo import RepoModel
1815 from rhodecode.model.repo import RepoModel
1816
1816
1817 repo = self
1817 repo = self
1818 _user_id, _time, _reason = self.locked
1818 _user_id, _time, _reason = self.locked
1819
1819
1820 data = {
1820 data = {
1821 'repo_id': repo.repo_id,
1821 'repo_id': repo.repo_id,
1822 'repo_name': repo.repo_name,
1822 'repo_name': repo.repo_name,
1823 'repo_type': repo.repo_type,
1823 'repo_type': repo.repo_type,
1824 'clone_uri': repo.clone_uri or '',
1824 'clone_uri': repo.clone_uri or '',
1825 'url': RepoModel().get_url(self),
1825 'url': RepoModel().get_url(self),
1826 'private': repo.private,
1826 'private': repo.private,
1827 'created_on': repo.created_on,
1827 'created_on': repo.created_on,
1828 'description': repo.description_safe,
1828 'description': repo.description_safe,
1829 'landing_rev': repo.landing_rev,
1829 'landing_rev': repo.landing_rev,
1830 'owner': repo.user.username,
1830 'owner': repo.user.username,
1831 'fork_of': repo.fork.repo_name if repo.fork else None,
1831 'fork_of': repo.fork.repo_name if repo.fork else None,
1832 'fork_of_id': repo.fork.repo_id if repo.fork else None,
1832 'fork_of_id': repo.fork.repo_id if repo.fork else None,
1833 'enable_statistics': repo.enable_statistics,
1833 'enable_statistics': repo.enable_statistics,
1834 'enable_locking': repo.enable_locking,
1834 'enable_locking': repo.enable_locking,
1835 'enable_downloads': repo.enable_downloads,
1835 'enable_downloads': repo.enable_downloads,
1836 'last_changeset': repo.changeset_cache,
1836 'last_changeset': repo.changeset_cache,
1837 'locked_by': User.get(_user_id).get_api_data(
1837 'locked_by': User.get(_user_id).get_api_data(
1838 include_secrets=include_secrets) if _user_id else None,
1838 include_secrets=include_secrets) if _user_id else None,
1839 'locked_date': time_to_datetime(_time) if _time else None,
1839 'locked_date': time_to_datetime(_time) if _time else None,
1840 'lock_reason': _reason if _reason else None,
1840 'lock_reason': _reason if _reason else None,
1841 }
1841 }
1842
1842
1843 # TODO: mikhail: should be per-repo settings here
1843 # TODO: mikhail: should be per-repo settings here
1844 rc_config = SettingsModel().get_all_settings()
1844 rc_config = SettingsModel().get_all_settings()
1845 repository_fields = str2bool(
1845 repository_fields = str2bool(
1846 rc_config.get('rhodecode_repository_fields'))
1846 rc_config.get('rhodecode_repository_fields'))
1847 if repository_fields:
1847 if repository_fields:
1848 for f in self.extra_fields:
1848 for f in self.extra_fields:
1849 data[f.field_key_prefixed] = f.field_value
1849 data[f.field_key_prefixed] = f.field_value
1850
1850
1851 return data
1851 return data
1852
1852
1853 @classmethod
1853 @classmethod
1854 def lock(cls, repo, user_id, lock_time=None, lock_reason=None):
1854 def lock(cls, repo, user_id, lock_time=None, lock_reason=None):
1855 if not lock_time:
1855 if not lock_time:
1856 lock_time = time.time()
1856 lock_time = time.time()
1857 if not lock_reason:
1857 if not lock_reason:
1858 lock_reason = cls.LOCK_AUTOMATIC
1858 lock_reason = cls.LOCK_AUTOMATIC
1859 repo.locked = [user_id, lock_time, lock_reason]
1859 repo.locked = [user_id, lock_time, lock_reason]
1860 Session().add(repo)
1860 Session().add(repo)
1861 Session().commit()
1861 Session().commit()
1862
1862
1863 @classmethod
1863 @classmethod
1864 def unlock(cls, repo):
1864 def unlock(cls, repo):
1865 repo.locked = None
1865 repo.locked = None
1866 Session().add(repo)
1866 Session().add(repo)
1867 Session().commit()
1867 Session().commit()
1868
1868
1869 @classmethod
1869 @classmethod
1870 def getlock(cls, repo):
1870 def getlock(cls, repo):
1871 return repo.locked
1871 return repo.locked
1872
1872
1873 def is_user_lock(self, user_id):
1873 def is_user_lock(self, user_id):
1874 if self.lock[0]:
1874 if self.lock[0]:
1875 lock_user_id = safe_int(self.lock[0])
1875 lock_user_id = safe_int(self.lock[0])
1876 user_id = safe_int(user_id)
1876 user_id = safe_int(user_id)
1877 # both are ints, and they are equal
1877 # both are ints, and they are equal
1878 return all([lock_user_id, user_id]) and lock_user_id == user_id
1878 return all([lock_user_id, user_id]) and lock_user_id == user_id
1879
1879
1880 return False
1880 return False
1881
1881
1882 def get_locking_state(self, action, user_id, only_when_enabled=True):
1882 def get_locking_state(self, action, user_id, only_when_enabled=True):
1883 """
1883 """
1884 Checks locking on this repository, if locking is enabled and lock is
1884 Checks locking on this repository, if locking is enabled and lock is
1885 present returns a tuple of make_lock, locked, locked_by.
1885 present returns a tuple of make_lock, locked, locked_by.
1886 make_lock can have 3 states None (do nothing) True, make lock
1886 make_lock can have 3 states None (do nothing) True, make lock
1887 False release lock, This value is later propagated to hooks, which
1887 False release lock, This value is later propagated to hooks, which
1888 do the locking. Think about this as signals passed to hooks what to do.
1888 do the locking. Think about this as signals passed to hooks what to do.
1889
1889
1890 """
1890 """
1891 # TODO: johbo: This is part of the business logic and should be moved
1891 # TODO: johbo: This is part of the business logic and should be moved
1892 # into the RepositoryModel.
1892 # into the RepositoryModel.
1893
1893
1894 if action not in ('push', 'pull'):
1894 if action not in ('push', 'pull'):
1895 raise ValueError("Invalid action value: %s" % repr(action))
1895 raise ValueError("Invalid action value: %s" % repr(action))
1896
1896
1897 # defines if locked error should be thrown to user
1897 # defines if locked error should be thrown to user
1898 currently_locked = False
1898 currently_locked = False
1899 # defines if new lock should be made, tri-state
1899 # defines if new lock should be made, tri-state
1900 make_lock = None
1900 make_lock = None
1901 repo = self
1901 repo = self
1902 user = User.get(user_id)
1902 user = User.get(user_id)
1903
1903
1904 lock_info = repo.locked
1904 lock_info = repo.locked
1905
1905
1906 if repo and (repo.enable_locking or not only_when_enabled):
1906 if repo and (repo.enable_locking or not only_when_enabled):
1907 if action == 'push':
1907 if action == 'push':
1908 # check if it's already locked !, if it is compare users
1908 # check if it's already locked !, if it is compare users
1909 locked_by_user_id = lock_info[0]
1909 locked_by_user_id = lock_info[0]
1910 if user.user_id == locked_by_user_id:
1910 if user.user_id == locked_by_user_id:
1911 log.debug(
1911 log.debug(
1912 'Got `push` action from user %s, now unlocking', user)
1912 'Got `push` action from user %s, now unlocking', user)
1913 # unlock if we have push from user who locked
1913 # unlock if we have push from user who locked
1914 make_lock = False
1914 make_lock = False
1915 else:
1915 else:
1916 # we're not the same user who locked, ban with
1916 # we're not the same user who locked, ban with
1917 # code defined in settings (default is 423 HTTP Locked) !
1917 # code defined in settings (default is 423 HTTP Locked) !
1918 log.debug('Repo %s is currently locked by %s', repo, user)
1918 log.debug('Repo %s is currently locked by %s', repo, user)
1919 currently_locked = True
1919 currently_locked = True
1920 elif action == 'pull':
1920 elif action == 'pull':
1921 # [0] user [1] date
1921 # [0] user [1] date
1922 if lock_info[0] and lock_info[1]:
1922 if lock_info[0] and lock_info[1]:
1923 log.debug('Repo %s is currently locked by %s', repo, user)
1923 log.debug('Repo %s is currently locked by %s', repo, user)
1924 currently_locked = True
1924 currently_locked = True
1925 else:
1925 else:
1926 log.debug('Setting lock on repo %s by %s', repo, user)
1926 log.debug('Setting lock on repo %s by %s', repo, user)
1927 make_lock = True
1927 make_lock = True
1928
1928
1929 else:
1929 else:
1930 log.debug('Repository %s do not have locking enabled', repo)
1930 log.debug('Repository %s do not have locking enabled', repo)
1931
1931
1932 log.debug('FINAL locking values make_lock:%s,locked:%s,locked_by:%s',
1932 log.debug('FINAL locking values make_lock:%s,locked:%s,locked_by:%s',
1933 make_lock, currently_locked, lock_info)
1933 make_lock, currently_locked, lock_info)
1934
1934
1935 from rhodecode.lib.auth import HasRepoPermissionAny
1935 from rhodecode.lib.auth import HasRepoPermissionAny
1936 perm_check = HasRepoPermissionAny('repository.write', 'repository.admin')
1936 perm_check = HasRepoPermissionAny('repository.write', 'repository.admin')
1937 if make_lock and not perm_check(repo_name=repo.repo_name, user=user):
1937 if make_lock and not perm_check(repo_name=repo.repo_name, user=user):
1938 # if we don't have at least write permission we cannot make a lock
1938 # if we don't have at least write permission we cannot make a lock
1939 log.debug('lock state reset back to FALSE due to lack '
1939 log.debug('lock state reset back to FALSE due to lack '
1940 'of at least read permission')
1940 'of at least read permission')
1941 make_lock = False
1941 make_lock = False
1942
1942
1943 return make_lock, currently_locked, lock_info
1943 return make_lock, currently_locked, lock_info
1944
1944
1945 @property
1945 @property
1946 def last_db_change(self):
1946 def last_db_change(self):
1947 return self.updated_on
1947 return self.updated_on
1948
1948
1949 @property
1949 @property
1950 def clone_uri_hidden(self):
1950 def clone_uri_hidden(self):
1951 clone_uri = self.clone_uri
1951 clone_uri = self.clone_uri
1952 if clone_uri:
1952 if clone_uri:
1953 import urlobject
1953 import urlobject
1954 url_obj = urlobject.URLObject(cleaned_uri(clone_uri))
1954 url_obj = urlobject.URLObject(cleaned_uri(clone_uri))
1955 if url_obj.password:
1955 if url_obj.password:
1956 clone_uri = url_obj.with_password('*****')
1956 clone_uri = url_obj.with_password('*****')
1957 return clone_uri
1957 return clone_uri
1958
1958
1959 def clone_url(self, **override):
1959 def clone_url(self, **override):
1960 from rhodecode.model.settings import SettingsModel
1960 from rhodecode.model.settings import SettingsModel
1961
1961
1962 uri_tmpl = None
1962 uri_tmpl = None
1963 if 'with_id' in override:
1963 if 'with_id' in override:
1964 uri_tmpl = self.DEFAULT_CLONE_URI_ID
1964 uri_tmpl = self.DEFAULT_CLONE_URI_ID
1965 del override['with_id']
1965 del override['with_id']
1966
1966
1967 if 'uri_tmpl' in override:
1967 if 'uri_tmpl' in override:
1968 uri_tmpl = override['uri_tmpl']
1968 uri_tmpl = override['uri_tmpl']
1969 del override['uri_tmpl']
1969 del override['uri_tmpl']
1970
1970
1971 # we didn't override our tmpl from **overrides
1971 # we didn't override our tmpl from **overrides
1972 if not uri_tmpl:
1972 if not uri_tmpl:
1973 rc_config = SettingsModel().get_all_settings(cache=True)
1973 rc_config = SettingsModel().get_all_settings(cache=True)
1974 uri_tmpl = rc_config.get(
1974 uri_tmpl = rc_config.get(
1975 'rhodecode_clone_uri_tmpl') or self.DEFAULT_CLONE_URI
1975 'rhodecode_clone_uri_tmpl') or self.DEFAULT_CLONE_URI
1976
1976
1977 request = get_current_request()
1977 request = get_current_request()
1978 return get_clone_url(request=request,
1978 return get_clone_url(request=request,
1979 uri_tmpl=uri_tmpl,
1979 uri_tmpl=uri_tmpl,
1980 repo_name=self.repo_name,
1980 repo_name=self.repo_name,
1981 repo_id=self.repo_id, **override)
1981 repo_id=self.repo_id, **override)
1982
1982
1983 def set_state(self, state):
1983 def set_state(self, state):
1984 self.repo_state = state
1984 self.repo_state = state
1985 Session().add(self)
1985 Session().add(self)
1986 #==========================================================================
1986 #==========================================================================
1987 # SCM PROPERTIES
1987 # SCM PROPERTIES
1988 #==========================================================================
1988 #==========================================================================
1989
1989
1990 def get_commit(self, commit_id=None, commit_idx=None, pre_load=None):
1990 def get_commit(self, commit_id=None, commit_idx=None, pre_load=None):
1991 return get_commit_safe(
1991 return get_commit_safe(
1992 self.scm_instance(), commit_id, commit_idx, pre_load=pre_load)
1992 self.scm_instance(), commit_id, commit_idx, pre_load=pre_load)
1993
1993
1994 def get_changeset(self, rev=None, pre_load=None):
1994 def get_changeset(self, rev=None, pre_load=None):
1995 warnings.warn("Use get_commit", DeprecationWarning)
1995 warnings.warn("Use get_commit", DeprecationWarning)
1996 commit_id = None
1996 commit_id = None
1997 commit_idx = None
1997 commit_idx = None
1998 if isinstance(rev, basestring):
1998 if isinstance(rev, basestring):
1999 commit_id = rev
1999 commit_id = rev
2000 else:
2000 else:
2001 commit_idx = rev
2001 commit_idx = rev
2002 return self.get_commit(commit_id=commit_id, commit_idx=commit_idx,
2002 return self.get_commit(commit_id=commit_id, commit_idx=commit_idx,
2003 pre_load=pre_load)
2003 pre_load=pre_load)
2004
2004
2005 def get_landing_commit(self):
2005 def get_landing_commit(self):
2006 """
2006 """
2007 Returns landing commit, or if that doesn't exist returns the tip
2007 Returns landing commit, or if that doesn't exist returns the tip
2008 """
2008 """
2009 _rev_type, _rev = self.landing_rev
2009 _rev_type, _rev = self.landing_rev
2010 commit = self.get_commit(_rev)
2010 commit = self.get_commit(_rev)
2011 if isinstance(commit, EmptyCommit):
2011 if isinstance(commit, EmptyCommit):
2012 return self.get_commit()
2012 return self.get_commit()
2013 return commit
2013 return commit
2014
2014
2015 def update_commit_cache(self, cs_cache=None, config=None):
2015 def update_commit_cache(self, cs_cache=None, config=None):
2016 """
2016 """
2017 Update cache of last changeset for repository, keys should be::
2017 Update cache of last changeset for repository, keys should be::
2018
2018
2019 short_id
2019 short_id
2020 raw_id
2020 raw_id
2021 revision
2021 revision
2022 parents
2022 parents
2023 message
2023 message
2024 date
2024 date
2025 author
2025 author
2026
2026
2027 :param cs_cache:
2027 :param cs_cache:
2028 """
2028 """
2029 from rhodecode.lib.vcs.backends.base import BaseChangeset
2029 from rhodecode.lib.vcs.backends.base import BaseChangeset
2030 if cs_cache is None:
2030 if cs_cache is None:
2031 # use no-cache version here
2031 # use no-cache version here
2032 scm_repo = self.scm_instance(cache=False, config=config)
2032 scm_repo = self.scm_instance(cache=False, config=config)
2033 if scm_repo:
2033 if scm_repo:
2034 cs_cache = scm_repo.get_commit(
2034 cs_cache = scm_repo.get_commit(
2035 pre_load=["author", "date", "message", "parents"])
2035 pre_load=["author", "date", "message", "parents"])
2036 else:
2036 else:
2037 cs_cache = EmptyCommit()
2037 cs_cache = EmptyCommit()
2038
2038
2039 if isinstance(cs_cache, BaseChangeset):
2039 if isinstance(cs_cache, BaseChangeset):
2040 cs_cache = cs_cache.__json__()
2040 cs_cache = cs_cache.__json__()
2041
2041
2042 def is_outdated(new_cs_cache):
2042 def is_outdated(new_cs_cache):
2043 if (new_cs_cache['raw_id'] != self.changeset_cache['raw_id'] or
2043 if (new_cs_cache['raw_id'] != self.changeset_cache['raw_id'] or
2044 new_cs_cache['revision'] != self.changeset_cache['revision']):
2044 new_cs_cache['revision'] != self.changeset_cache['revision']):
2045 return True
2045 return True
2046 return False
2046 return False
2047
2047
2048 # check if we have maybe already latest cached revision
2048 # check if we have maybe already latest cached revision
2049 if is_outdated(cs_cache) or not self.changeset_cache:
2049 if is_outdated(cs_cache) or not self.changeset_cache:
2050 _default = datetime.datetime.fromtimestamp(0)
2050 _default = datetime.datetime.fromtimestamp(0)
2051 last_change = cs_cache.get('date') or _default
2051 last_change = cs_cache.get('date') or _default
2052 log.debug('updated repo %s with new cs cache %s',
2052 log.debug('updated repo %s with new cs cache %s',
2053 self.repo_name, cs_cache)
2053 self.repo_name, cs_cache)
2054 self.updated_on = last_change
2054 self.updated_on = last_change
2055 self.changeset_cache = cs_cache
2055 self.changeset_cache = cs_cache
2056 Session().add(self)
2056 Session().add(self)
2057 Session().commit()
2057 Session().commit()
2058 else:
2058 else:
2059 log.debug('Skipping update_commit_cache for repo:`%s` '
2059 log.debug('Skipping update_commit_cache for repo:`%s` '
2060 'commit already with latest changes', self.repo_name)
2060 'commit already with latest changes', self.repo_name)
2061
2061
2062 @property
2062 @property
2063 def tip(self):
2063 def tip(self):
2064 return self.get_commit('tip')
2064 return self.get_commit('tip')
2065
2065
2066 @property
2066 @property
2067 def author(self):
2067 def author(self):
2068 return self.tip.author
2068 return self.tip.author
2069
2069
2070 @property
2070 @property
2071 def last_change(self):
2071 def last_change(self):
2072 return self.scm_instance().last_change
2072 return self.scm_instance().last_change
2073
2073
2074 def get_comments(self, revisions=None):
2074 def get_comments(self, revisions=None):
2075 """
2075 """
2076 Returns comments for this repository grouped by revisions
2076 Returns comments for this repository grouped by revisions
2077
2077
2078 :param revisions: filter query by revisions only
2078 :param revisions: filter query by revisions only
2079 """
2079 """
2080 cmts = ChangesetComment.query()\
2080 cmts = ChangesetComment.query()\
2081 .filter(ChangesetComment.repo == self)
2081 .filter(ChangesetComment.repo == self)
2082 if revisions:
2082 if revisions:
2083 cmts = cmts.filter(ChangesetComment.revision.in_(revisions))
2083 cmts = cmts.filter(ChangesetComment.revision.in_(revisions))
2084 grouped = collections.defaultdict(list)
2084 grouped = collections.defaultdict(list)
2085 for cmt in cmts.all():
2085 for cmt in cmts.all():
2086 grouped[cmt.revision].append(cmt)
2086 grouped[cmt.revision].append(cmt)
2087 return grouped
2087 return grouped
2088
2088
2089 def statuses(self, revisions=None):
2089 def statuses(self, revisions=None):
2090 """
2090 """
2091 Returns statuses for this repository
2091 Returns statuses for this repository
2092
2092
2093 :param revisions: list of revisions to get statuses for
2093 :param revisions: list of revisions to get statuses for
2094 """
2094 """
2095 statuses = ChangesetStatus.query()\
2095 statuses = ChangesetStatus.query()\
2096 .filter(ChangesetStatus.repo == self)\
2096 .filter(ChangesetStatus.repo == self)\
2097 .filter(ChangesetStatus.version == 0)
2097 .filter(ChangesetStatus.version == 0)
2098
2098
2099 if revisions:
2099 if revisions:
2100 # Try doing the filtering in chunks to avoid hitting limits
2100 # Try doing the filtering in chunks to avoid hitting limits
2101 size = 500
2101 size = 500
2102 status_results = []
2102 status_results = []
2103 for chunk in xrange(0, len(revisions), size):
2103 for chunk in xrange(0, len(revisions), size):
2104 status_results += statuses.filter(
2104 status_results += statuses.filter(
2105 ChangesetStatus.revision.in_(
2105 ChangesetStatus.revision.in_(
2106 revisions[chunk: chunk+size])
2106 revisions[chunk: chunk+size])
2107 ).all()
2107 ).all()
2108 else:
2108 else:
2109 status_results = statuses.all()
2109 status_results = statuses.all()
2110
2110
2111 grouped = {}
2111 grouped = {}
2112
2112
2113 # maybe we have open new pullrequest without a status?
2113 # maybe we have open new pullrequest without a status?
2114 stat = ChangesetStatus.STATUS_UNDER_REVIEW
2114 stat = ChangesetStatus.STATUS_UNDER_REVIEW
2115 status_lbl = ChangesetStatus.get_status_lbl(stat)
2115 status_lbl = ChangesetStatus.get_status_lbl(stat)
2116 for pr in PullRequest.query().filter(PullRequest.source_repo == self).all():
2116 for pr in PullRequest.query().filter(PullRequest.source_repo == self).all():
2117 for rev in pr.revisions:
2117 for rev in pr.revisions:
2118 pr_id = pr.pull_request_id
2118 pr_id = pr.pull_request_id
2119 pr_repo = pr.target_repo.repo_name
2119 pr_repo = pr.target_repo.repo_name
2120 grouped[rev] = [stat, status_lbl, pr_id, pr_repo]
2120 grouped[rev] = [stat, status_lbl, pr_id, pr_repo]
2121
2121
2122 for stat in status_results:
2122 for stat in status_results:
2123 pr_id = pr_repo = None
2123 pr_id = pr_repo = None
2124 if stat.pull_request:
2124 if stat.pull_request:
2125 pr_id = stat.pull_request.pull_request_id
2125 pr_id = stat.pull_request.pull_request_id
2126 pr_repo = stat.pull_request.target_repo.repo_name
2126 pr_repo = stat.pull_request.target_repo.repo_name
2127 grouped[stat.revision] = [str(stat.status), stat.status_lbl,
2127 grouped[stat.revision] = [str(stat.status), stat.status_lbl,
2128 pr_id, pr_repo]
2128 pr_id, pr_repo]
2129 return grouped
2129 return grouped
2130
2130
2131 # ==========================================================================
2131 # ==========================================================================
2132 # SCM CACHE INSTANCE
2132 # SCM CACHE INSTANCE
2133 # ==========================================================================
2133 # ==========================================================================
2134
2134
2135 def scm_instance(self, **kwargs):
2135 def scm_instance(self, **kwargs):
2136 import rhodecode
2136 import rhodecode
2137
2137
2138 # Passing a config will not hit the cache currently only used
2138 # Passing a config will not hit the cache currently only used
2139 # for repo2dbmapper
2139 # for repo2dbmapper
2140 config = kwargs.pop('config', None)
2140 config = kwargs.pop('config', None)
2141 cache = kwargs.pop('cache', None)
2141 cache = kwargs.pop('cache', None)
2142 full_cache = str2bool(rhodecode.CONFIG.get('vcs_full_cache'))
2142 full_cache = str2bool(rhodecode.CONFIG.get('vcs_full_cache'))
2143 # if cache is NOT defined use default global, else we have a full
2143 # if cache is NOT defined use default global, else we have a full
2144 # control over cache behaviour
2144 # control over cache behaviour
2145 if cache is None and full_cache and not config:
2145 if cache is None and full_cache and not config:
2146 return self._get_instance_cached()
2146 return self._get_instance_cached()
2147 return self._get_instance(cache=bool(cache), config=config)
2147 return self._get_instance(cache=bool(cache), config=config)
2148
2148
2149 def _get_instance_cached(self):
2149 def _get_instance_cached(self):
2150 @cache_region('long_term')
2150 @cache_region('long_term')
2151 def _get_repo(cache_key):
2151 def _get_repo(cache_key):
2152 return self._get_instance()
2152 return self._get_instance()
2153
2153
2154 invalidator_context = CacheKey.repo_context_cache(
2154 invalidator_context = CacheKey.repo_context_cache(
2155 _get_repo, self.repo_name, None, thread_scoped=True)
2155 _get_repo, self.repo_name, None, thread_scoped=True)
2156
2156
2157 with invalidator_context as context:
2157 with invalidator_context as context:
2158 context.invalidate()
2158 context.invalidate()
2159 repo = context.compute()
2159 repo = context.compute()
2160
2160
2161 return repo
2161 return repo
2162
2162
2163 def _get_instance(self, cache=True, config=None):
2163 def _get_instance(self, cache=True, config=None):
2164 config = config or self._config
2164 config = config or self._config
2165 custom_wire = {
2165 custom_wire = {
2166 'cache': cache # controls the vcs.remote cache
2166 'cache': cache # controls the vcs.remote cache
2167 }
2167 }
2168 repo = get_vcs_instance(
2168 repo = get_vcs_instance(
2169 repo_path=safe_str(self.repo_full_path),
2169 repo_path=safe_str(self.repo_full_path),
2170 config=config,
2170 config=config,
2171 with_wire=custom_wire,
2171 with_wire=custom_wire,
2172 create=False,
2172 create=False,
2173 _vcs_alias=self.repo_type)
2173 _vcs_alias=self.repo_type)
2174
2174
2175 return repo
2175 return repo
2176
2176
2177 def __json__(self):
2177 def __json__(self):
2178 return {'landing_rev': self.landing_rev}
2178 return {'landing_rev': self.landing_rev}
2179
2179
2180 def get_dict(self):
2180 def get_dict(self):
2181
2181
2182 # Since we transformed `repo_name` to a hybrid property, we need to
2182 # Since we transformed `repo_name` to a hybrid property, we need to
2183 # keep compatibility with the code which uses `repo_name` field.
2183 # keep compatibility with the code which uses `repo_name` field.
2184
2184
2185 result = super(Repository, self).get_dict()
2185 result = super(Repository, self).get_dict()
2186 result['repo_name'] = result.pop('_repo_name', None)
2186 result['repo_name'] = result.pop('_repo_name', None)
2187 return result
2187 return result
2188
2188
2189
2189
2190 class RepoGroup(Base, BaseModel):
2190 class RepoGroup(Base, BaseModel):
2191 __tablename__ = 'groups'
2191 __tablename__ = 'groups'
2192 __table_args__ = (
2192 __table_args__ = (
2193 UniqueConstraint('group_name', 'group_parent_id'),
2193 UniqueConstraint('group_name', 'group_parent_id'),
2194 CheckConstraint('group_id != group_parent_id'),
2194 CheckConstraint('group_id != group_parent_id'),
2195 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2195 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2196 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2196 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2197 )
2197 )
2198 __mapper_args__ = {'order_by': 'group_name'}
2198 __mapper_args__ = {'order_by': 'group_name'}
2199
2199
2200 CHOICES_SEPARATOR = '/' # used to generate select2 choices for nested groups
2200 CHOICES_SEPARATOR = '/' # used to generate select2 choices for nested groups
2201
2201
2202 group_id = Column("group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2202 group_id = Column("group_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2203 group_name = Column("group_name", String(255), nullable=False, unique=True, default=None)
2203 group_name = Column("group_name", String(255), nullable=False, unique=True, default=None)
2204 group_parent_id = Column("group_parent_id", Integer(), ForeignKey('groups.group_id'), nullable=True, unique=None, default=None)
2204 group_parent_id = Column("group_parent_id", Integer(), ForeignKey('groups.group_id'), nullable=True, unique=None, default=None)
2205 group_description = Column("group_description", String(10000), nullable=True, unique=None, default=None)
2205 group_description = Column("group_description", String(10000), nullable=True, unique=None, default=None)
2206 enable_locking = Column("enable_locking", Boolean(), nullable=False, unique=None, default=False)
2206 enable_locking = Column("enable_locking", Boolean(), nullable=False, unique=None, default=False)
2207 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
2207 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=False, default=None)
2208 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
2208 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
2209 personal = Column('personal', Boolean(), nullable=True, unique=None, default=None)
2209 personal = Column('personal', Boolean(), nullable=True, unique=None, default=None)
2210
2210
2211 repo_group_to_perm = relationship('UserRepoGroupToPerm', cascade='all', order_by='UserRepoGroupToPerm.group_to_perm_id')
2211 repo_group_to_perm = relationship('UserRepoGroupToPerm', cascade='all', order_by='UserRepoGroupToPerm.group_to_perm_id')
2212 users_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
2212 users_group_to_perm = relationship('UserGroupRepoGroupToPerm', cascade='all')
2213 parent_group = relationship('RepoGroup', remote_side=group_id)
2213 parent_group = relationship('RepoGroup', remote_side=group_id)
2214 user = relationship('User')
2214 user = relationship('User')
2215 integrations = relationship('Integration',
2215 integrations = relationship('Integration',
2216 cascade="all, delete, delete-orphan")
2216 cascade="all, delete, delete-orphan")
2217
2217
2218 def __init__(self, group_name='', parent_group=None):
2218 def __init__(self, group_name='', parent_group=None):
2219 self.group_name = group_name
2219 self.group_name = group_name
2220 self.parent_group = parent_group
2220 self.parent_group = parent_group
2221
2221
2222 def __unicode__(self):
2222 def __unicode__(self):
2223 return u"<%s('id:%s:%s')>" % (
2223 return u"<%s('id:%s:%s')>" % (
2224 self.__class__.__name__, self.group_id, self.group_name)
2224 self.__class__.__name__, self.group_id, self.group_name)
2225
2225
2226 @hybrid_property
2226 @hybrid_property
2227 def description_safe(self):
2227 def description_safe(self):
2228 from rhodecode.lib import helpers as h
2228 from rhodecode.lib import helpers as h
2229 return h.escape(self.group_description)
2229 return h.escape(self.group_description)
2230
2230
2231 @classmethod
2231 @classmethod
2232 def _generate_choice(cls, repo_group):
2232 def _generate_choice(cls, repo_group):
2233 from webhelpers.html import literal as _literal
2233 from webhelpers.html import literal as _literal
2234 _name = lambda k: _literal(cls.CHOICES_SEPARATOR.join(k))
2234 _name = lambda k: _literal(cls.CHOICES_SEPARATOR.join(k))
2235 return repo_group.group_id, _name(repo_group.full_path_splitted)
2235 return repo_group.group_id, _name(repo_group.full_path_splitted)
2236
2236
2237 @classmethod
2237 @classmethod
2238 def groups_choices(cls, groups=None, show_empty_group=True):
2238 def groups_choices(cls, groups=None, show_empty_group=True):
2239 if not groups:
2239 if not groups:
2240 groups = cls.query().all()
2240 groups = cls.query().all()
2241
2241
2242 repo_groups = []
2242 repo_groups = []
2243 if show_empty_group:
2243 if show_empty_group:
2244 repo_groups = [(-1, u'-- %s --' % _('No parent'))]
2244 repo_groups = [(-1, u'-- %s --' % _('No parent'))]
2245
2245
2246 repo_groups.extend([cls._generate_choice(x) for x in groups])
2246 repo_groups.extend([cls._generate_choice(x) for x in groups])
2247
2247
2248 repo_groups = sorted(
2248 repo_groups = sorted(
2249 repo_groups, key=lambda t: t[1].split(cls.CHOICES_SEPARATOR)[0])
2249 repo_groups, key=lambda t: t[1].split(cls.CHOICES_SEPARATOR)[0])
2250 return repo_groups
2250 return repo_groups
2251
2251
2252 @classmethod
2252 @classmethod
2253 def url_sep(cls):
2253 def url_sep(cls):
2254 return URL_SEP
2254 return URL_SEP
2255
2255
2256 @classmethod
2256 @classmethod
2257 def get_by_group_name(cls, group_name, cache=False, case_insensitive=False):
2257 def get_by_group_name(cls, group_name, cache=False, case_insensitive=False):
2258 if case_insensitive:
2258 if case_insensitive:
2259 gr = cls.query().filter(func.lower(cls.group_name)
2259 gr = cls.query().filter(func.lower(cls.group_name)
2260 == func.lower(group_name))
2260 == func.lower(group_name))
2261 else:
2261 else:
2262 gr = cls.query().filter(cls.group_name == group_name)
2262 gr = cls.query().filter(cls.group_name == group_name)
2263 if cache:
2263 if cache:
2264 name_key = _hash_key(group_name)
2264 name_key = _hash_key(group_name)
2265 gr = gr.options(
2265 gr = gr.options(
2266 FromCache("sql_cache_short", "get_group_%s" % name_key))
2266 FromCache("sql_cache_short", "get_group_%s" % name_key))
2267 return gr.scalar()
2267 return gr.scalar()
2268
2268
2269 @classmethod
2269 @classmethod
2270 def get_user_personal_repo_group(cls, user_id):
2270 def get_user_personal_repo_group(cls, user_id):
2271 user = User.get(user_id)
2271 user = User.get(user_id)
2272 if user.username == User.DEFAULT_USER:
2272 if user.username == User.DEFAULT_USER:
2273 return None
2273 return None
2274
2274
2275 return cls.query()\
2275 return cls.query()\
2276 .filter(cls.personal == true()) \
2276 .filter(cls.personal == true()) \
2277 .filter(cls.user == user).scalar()
2277 .filter(cls.user == user).scalar()
2278
2278
2279 @classmethod
2279 @classmethod
2280 def get_all_repo_groups(cls, user_id=Optional(None), group_id=Optional(None),
2280 def get_all_repo_groups(cls, user_id=Optional(None), group_id=Optional(None),
2281 case_insensitive=True):
2281 case_insensitive=True):
2282 q = RepoGroup.query()
2282 q = RepoGroup.query()
2283
2283
2284 if not isinstance(user_id, Optional):
2284 if not isinstance(user_id, Optional):
2285 q = q.filter(RepoGroup.user_id == user_id)
2285 q = q.filter(RepoGroup.user_id == user_id)
2286
2286
2287 if not isinstance(group_id, Optional):
2287 if not isinstance(group_id, Optional):
2288 q = q.filter(RepoGroup.group_parent_id == group_id)
2288 q = q.filter(RepoGroup.group_parent_id == group_id)
2289
2289
2290 if case_insensitive:
2290 if case_insensitive:
2291 q = q.order_by(func.lower(RepoGroup.group_name))
2291 q = q.order_by(func.lower(RepoGroup.group_name))
2292 else:
2292 else:
2293 q = q.order_by(RepoGroup.group_name)
2293 q = q.order_by(RepoGroup.group_name)
2294 return q.all()
2294 return q.all()
2295
2295
2296 @property
2296 @property
2297 def parents(self):
2297 def parents(self):
2298 parents_recursion_limit = 10
2298 parents_recursion_limit = 10
2299 groups = []
2299 groups = []
2300 if self.parent_group is None:
2300 if self.parent_group is None:
2301 return groups
2301 return groups
2302 cur_gr = self.parent_group
2302 cur_gr = self.parent_group
2303 groups.insert(0, cur_gr)
2303 groups.insert(0, cur_gr)
2304 cnt = 0
2304 cnt = 0
2305 while 1:
2305 while 1:
2306 cnt += 1
2306 cnt += 1
2307 gr = getattr(cur_gr, 'parent_group', None)
2307 gr = getattr(cur_gr, 'parent_group', None)
2308 cur_gr = cur_gr.parent_group
2308 cur_gr = cur_gr.parent_group
2309 if gr is None:
2309 if gr is None:
2310 break
2310 break
2311 if cnt == parents_recursion_limit:
2311 if cnt == parents_recursion_limit:
2312 # this will prevent accidental infinit loops
2312 # this will prevent accidental infinit loops
2313 log.error(('more than %s parents found for group %s, stopping '
2313 log.error(('more than %s parents found for group %s, stopping '
2314 'recursive parent fetching' % (parents_recursion_limit, self)))
2314 'recursive parent fetching' % (parents_recursion_limit, self)))
2315 break
2315 break
2316
2316
2317 groups.insert(0, gr)
2317 groups.insert(0, gr)
2318 return groups
2318 return groups
2319
2319
2320 @property
2320 @property
2321 def children(self):
2321 def children(self):
2322 return RepoGroup.query().filter(RepoGroup.parent_group == self)
2322 return RepoGroup.query().filter(RepoGroup.parent_group == self)
2323
2323
2324 @property
2324 @property
2325 def name(self):
2325 def name(self):
2326 return self.group_name.split(RepoGroup.url_sep())[-1]
2326 return self.group_name.split(RepoGroup.url_sep())[-1]
2327
2327
2328 @property
2328 @property
2329 def full_path(self):
2329 def full_path(self):
2330 return self.group_name
2330 return self.group_name
2331
2331
2332 @property
2332 @property
2333 def full_path_splitted(self):
2333 def full_path_splitted(self):
2334 return self.group_name.split(RepoGroup.url_sep())
2334 return self.group_name.split(RepoGroup.url_sep())
2335
2335
2336 @property
2336 @property
2337 def repositories(self):
2337 def repositories(self):
2338 return Repository.query()\
2338 return Repository.query()\
2339 .filter(Repository.group == self)\
2339 .filter(Repository.group == self)\
2340 .order_by(Repository.repo_name)
2340 .order_by(Repository.repo_name)
2341
2341
2342 @property
2342 @property
2343 def repositories_recursive_count(self):
2343 def repositories_recursive_count(self):
2344 cnt = self.repositories.count()
2344 cnt = self.repositories.count()
2345
2345
2346 def children_count(group):
2346 def children_count(group):
2347 cnt = 0
2347 cnt = 0
2348 for child in group.children:
2348 for child in group.children:
2349 cnt += child.repositories.count()
2349 cnt += child.repositories.count()
2350 cnt += children_count(child)
2350 cnt += children_count(child)
2351 return cnt
2351 return cnt
2352
2352
2353 return cnt + children_count(self)
2353 return cnt + children_count(self)
2354
2354
2355 def _recursive_objects(self, include_repos=True):
2355 def _recursive_objects(self, include_repos=True):
2356 all_ = []
2356 all_ = []
2357
2357
2358 def _get_members(root_gr):
2358 def _get_members(root_gr):
2359 if include_repos:
2359 if include_repos:
2360 for r in root_gr.repositories:
2360 for r in root_gr.repositories:
2361 all_.append(r)
2361 all_.append(r)
2362 childs = root_gr.children.all()
2362 childs = root_gr.children.all()
2363 if childs:
2363 if childs:
2364 for gr in childs:
2364 for gr in childs:
2365 all_.append(gr)
2365 all_.append(gr)
2366 _get_members(gr)
2366 _get_members(gr)
2367
2367
2368 _get_members(self)
2368 _get_members(self)
2369 return [self] + all_
2369 return [self] + all_
2370
2370
2371 def recursive_groups_and_repos(self):
2371 def recursive_groups_and_repos(self):
2372 """
2372 """
2373 Recursive return all groups, with repositories in those groups
2373 Recursive return all groups, with repositories in those groups
2374 """
2374 """
2375 return self._recursive_objects()
2375 return self._recursive_objects()
2376
2376
2377 def recursive_groups(self):
2377 def recursive_groups(self):
2378 """
2378 """
2379 Returns all children groups for this group including children of children
2379 Returns all children groups for this group including children of children
2380 """
2380 """
2381 return self._recursive_objects(include_repos=False)
2381 return self._recursive_objects(include_repos=False)
2382
2382
2383 def get_new_name(self, group_name):
2383 def get_new_name(self, group_name):
2384 """
2384 """
2385 returns new full group name based on parent and new name
2385 returns new full group name based on parent and new name
2386
2386
2387 :param group_name:
2387 :param group_name:
2388 """
2388 """
2389 path_prefix = (self.parent_group.full_path_splitted if
2389 path_prefix = (self.parent_group.full_path_splitted if
2390 self.parent_group else [])
2390 self.parent_group else [])
2391 return RepoGroup.url_sep().join(path_prefix + [group_name])
2391 return RepoGroup.url_sep().join(path_prefix + [group_name])
2392
2392
2393 def permissions(self, with_admins=True, with_owner=True):
2393 def permissions(self, with_admins=True, with_owner=True):
2394 q = UserRepoGroupToPerm.query().filter(UserRepoGroupToPerm.group == self)
2394 q = UserRepoGroupToPerm.query().filter(UserRepoGroupToPerm.group == self)
2395 q = q.options(joinedload(UserRepoGroupToPerm.group),
2395 q = q.options(joinedload(UserRepoGroupToPerm.group),
2396 joinedload(UserRepoGroupToPerm.user),
2396 joinedload(UserRepoGroupToPerm.user),
2397 joinedload(UserRepoGroupToPerm.permission),)
2397 joinedload(UserRepoGroupToPerm.permission),)
2398
2398
2399 # get owners and admins and permissions. We do a trick of re-writing
2399 # get owners and admins and permissions. We do a trick of re-writing
2400 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2400 # objects from sqlalchemy to named-tuples due to sqlalchemy session
2401 # has a global reference and changing one object propagates to all
2401 # has a global reference and changing one object propagates to all
2402 # others. This means if admin is also an owner admin_row that change
2402 # others. This means if admin is also an owner admin_row that change
2403 # would propagate to both objects
2403 # would propagate to both objects
2404 perm_rows = []
2404 perm_rows = []
2405 for _usr in q.all():
2405 for _usr in q.all():
2406 usr = AttributeDict(_usr.user.get_dict())
2406 usr = AttributeDict(_usr.user.get_dict())
2407 usr.permission = _usr.permission.permission_name
2407 usr.permission = _usr.permission.permission_name
2408 perm_rows.append(usr)
2408 perm_rows.append(usr)
2409
2409
2410 # filter the perm rows by 'default' first and then sort them by
2410 # filter the perm rows by 'default' first and then sort them by
2411 # admin,write,read,none permissions sorted again alphabetically in
2411 # admin,write,read,none permissions sorted again alphabetically in
2412 # each group
2412 # each group
2413 perm_rows = sorted(perm_rows, key=display_sort)
2413 perm_rows = sorted(perm_rows, key=display_sort)
2414
2414
2415 _admin_perm = 'group.admin'
2415 _admin_perm = 'group.admin'
2416 owner_row = []
2416 owner_row = []
2417 if with_owner:
2417 if with_owner:
2418 usr = AttributeDict(self.user.get_dict())
2418 usr = AttributeDict(self.user.get_dict())
2419 usr.owner_row = True
2419 usr.owner_row = True
2420 usr.permission = _admin_perm
2420 usr.permission = _admin_perm
2421 owner_row.append(usr)
2421 owner_row.append(usr)
2422
2422
2423 super_admin_rows = []
2423 super_admin_rows = []
2424 if with_admins:
2424 if with_admins:
2425 for usr in User.get_all_super_admins():
2425 for usr in User.get_all_super_admins():
2426 # if this admin is also owner, don't double the record
2426 # if this admin is also owner, don't double the record
2427 if usr.user_id == owner_row[0].user_id:
2427 if usr.user_id == owner_row[0].user_id:
2428 owner_row[0].admin_row = True
2428 owner_row[0].admin_row = True
2429 else:
2429 else:
2430 usr = AttributeDict(usr.get_dict())
2430 usr = AttributeDict(usr.get_dict())
2431 usr.admin_row = True
2431 usr.admin_row = True
2432 usr.permission = _admin_perm
2432 usr.permission = _admin_perm
2433 super_admin_rows.append(usr)
2433 super_admin_rows.append(usr)
2434
2434
2435 return super_admin_rows + owner_row + perm_rows
2435 return super_admin_rows + owner_row + perm_rows
2436
2436
2437 def permission_user_groups(self):
2437 def permission_user_groups(self):
2438 q = UserGroupRepoGroupToPerm.query().filter(UserGroupRepoGroupToPerm.group == self)
2438 q = UserGroupRepoGroupToPerm.query().filter(UserGroupRepoGroupToPerm.group == self)
2439 q = q.options(joinedload(UserGroupRepoGroupToPerm.group),
2439 q = q.options(joinedload(UserGroupRepoGroupToPerm.group),
2440 joinedload(UserGroupRepoGroupToPerm.users_group),
2440 joinedload(UserGroupRepoGroupToPerm.users_group),
2441 joinedload(UserGroupRepoGroupToPerm.permission),)
2441 joinedload(UserGroupRepoGroupToPerm.permission),)
2442
2442
2443 perm_rows = []
2443 perm_rows = []
2444 for _user_group in q.all():
2444 for _user_group in q.all():
2445 usr = AttributeDict(_user_group.users_group.get_dict())
2445 usr = AttributeDict(_user_group.users_group.get_dict())
2446 usr.permission = _user_group.permission.permission_name
2446 usr.permission = _user_group.permission.permission_name
2447 perm_rows.append(usr)
2447 perm_rows.append(usr)
2448
2448
2449 return perm_rows
2449 return perm_rows
2450
2450
2451 def get_api_data(self):
2451 def get_api_data(self):
2452 """
2452 """
2453 Common function for generating api data
2453 Common function for generating api data
2454
2454
2455 """
2455 """
2456 group = self
2456 group = self
2457 data = {
2457 data = {
2458 'group_id': group.group_id,
2458 'group_id': group.group_id,
2459 'group_name': group.group_name,
2459 'group_name': group.group_name,
2460 'group_description': group.description_safe,
2460 'group_description': group.description_safe,
2461 'parent_group': group.parent_group.group_name if group.parent_group else None,
2461 'parent_group': group.parent_group.group_name if group.parent_group else None,
2462 'repositories': [x.repo_name for x in group.repositories],
2462 'repositories': [x.repo_name for x in group.repositories],
2463 'owner': group.user.username,
2463 'owner': group.user.username,
2464 }
2464 }
2465 return data
2465 return data
2466
2466
2467
2467
2468 class Permission(Base, BaseModel):
2468 class Permission(Base, BaseModel):
2469 __tablename__ = 'permissions'
2469 __tablename__ = 'permissions'
2470 __table_args__ = (
2470 __table_args__ = (
2471 Index('p_perm_name_idx', 'permission_name'),
2471 Index('p_perm_name_idx', 'permission_name'),
2472 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2472 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2473 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2473 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2474 )
2474 )
2475 PERMS = [
2475 PERMS = [
2476 ('hg.admin', _('RhodeCode Super Administrator')),
2476 ('hg.admin', _('RhodeCode Super Administrator')),
2477
2477
2478 ('repository.none', _('Repository no access')),
2478 ('repository.none', _('Repository no access')),
2479 ('repository.read', _('Repository read access')),
2479 ('repository.read', _('Repository read access')),
2480 ('repository.write', _('Repository write access')),
2480 ('repository.write', _('Repository write access')),
2481 ('repository.admin', _('Repository admin access')),
2481 ('repository.admin', _('Repository admin access')),
2482
2482
2483 ('group.none', _('Repository group no access')),
2483 ('group.none', _('Repository group no access')),
2484 ('group.read', _('Repository group read access')),
2484 ('group.read', _('Repository group read access')),
2485 ('group.write', _('Repository group write access')),
2485 ('group.write', _('Repository group write access')),
2486 ('group.admin', _('Repository group admin access')),
2486 ('group.admin', _('Repository group admin access')),
2487
2487
2488 ('usergroup.none', _('User group no access')),
2488 ('usergroup.none', _('User group no access')),
2489 ('usergroup.read', _('User group read access')),
2489 ('usergroup.read', _('User group read access')),
2490 ('usergroup.write', _('User group write access')),
2490 ('usergroup.write', _('User group write access')),
2491 ('usergroup.admin', _('User group admin access')),
2491 ('usergroup.admin', _('User group admin access')),
2492
2492
2493 ('hg.repogroup.create.false', _('Repository Group creation disabled')),
2493 ('hg.repogroup.create.false', _('Repository Group creation disabled')),
2494 ('hg.repogroup.create.true', _('Repository Group creation enabled')),
2494 ('hg.repogroup.create.true', _('Repository Group creation enabled')),
2495
2495
2496 ('hg.usergroup.create.false', _('User Group creation disabled')),
2496 ('hg.usergroup.create.false', _('User Group creation disabled')),
2497 ('hg.usergroup.create.true', _('User Group creation enabled')),
2497 ('hg.usergroup.create.true', _('User Group creation enabled')),
2498
2498
2499 ('hg.create.none', _('Repository creation disabled')),
2499 ('hg.create.none', _('Repository creation disabled')),
2500 ('hg.create.repository', _('Repository creation enabled')),
2500 ('hg.create.repository', _('Repository creation enabled')),
2501 ('hg.create.write_on_repogroup.true', _('Repository creation enabled with write permission to a repository group')),
2501 ('hg.create.write_on_repogroup.true', _('Repository creation enabled with write permission to a repository group')),
2502 ('hg.create.write_on_repogroup.false', _('Repository creation disabled with write permission to a repository group')),
2502 ('hg.create.write_on_repogroup.false', _('Repository creation disabled with write permission to a repository group')),
2503
2503
2504 ('hg.fork.none', _('Repository forking disabled')),
2504 ('hg.fork.none', _('Repository forking disabled')),
2505 ('hg.fork.repository', _('Repository forking enabled')),
2505 ('hg.fork.repository', _('Repository forking enabled')),
2506
2506
2507 ('hg.register.none', _('Registration disabled')),
2507 ('hg.register.none', _('Registration disabled')),
2508 ('hg.register.manual_activate', _('User Registration with manual account activation')),
2508 ('hg.register.manual_activate', _('User Registration with manual account activation')),
2509 ('hg.register.auto_activate', _('User Registration with automatic account activation')),
2509 ('hg.register.auto_activate', _('User Registration with automatic account activation')),
2510
2510
2511 ('hg.password_reset.enabled', _('Password reset enabled')),
2511 ('hg.password_reset.enabled', _('Password reset enabled')),
2512 ('hg.password_reset.hidden', _('Password reset hidden')),
2512 ('hg.password_reset.hidden', _('Password reset hidden')),
2513 ('hg.password_reset.disabled', _('Password reset disabled')),
2513 ('hg.password_reset.disabled', _('Password reset disabled')),
2514
2514
2515 ('hg.extern_activate.manual', _('Manual activation of external account')),
2515 ('hg.extern_activate.manual', _('Manual activation of external account')),
2516 ('hg.extern_activate.auto', _('Automatic activation of external account')),
2516 ('hg.extern_activate.auto', _('Automatic activation of external account')),
2517
2517
2518 ('hg.inherit_default_perms.false', _('Inherit object permissions from default user disabled')),
2518 ('hg.inherit_default_perms.false', _('Inherit object permissions from default user disabled')),
2519 ('hg.inherit_default_perms.true', _('Inherit object permissions from default user enabled')),
2519 ('hg.inherit_default_perms.true', _('Inherit object permissions from default user enabled')),
2520 ]
2520 ]
2521
2521
2522 # definition of system default permissions for DEFAULT user
2522 # definition of system default permissions for DEFAULT user
2523 DEFAULT_USER_PERMISSIONS = [
2523 DEFAULT_USER_PERMISSIONS = [
2524 'repository.read',
2524 'repository.read',
2525 'group.read',
2525 'group.read',
2526 'usergroup.read',
2526 'usergroup.read',
2527 'hg.create.repository',
2527 'hg.create.repository',
2528 'hg.repogroup.create.false',
2528 'hg.repogroup.create.false',
2529 'hg.usergroup.create.false',
2529 'hg.usergroup.create.false',
2530 'hg.create.write_on_repogroup.true',
2530 'hg.create.write_on_repogroup.true',
2531 'hg.fork.repository',
2531 'hg.fork.repository',
2532 'hg.register.manual_activate',
2532 'hg.register.manual_activate',
2533 'hg.password_reset.enabled',
2533 'hg.password_reset.enabled',
2534 'hg.extern_activate.auto',
2534 'hg.extern_activate.auto',
2535 'hg.inherit_default_perms.true',
2535 'hg.inherit_default_perms.true',
2536 ]
2536 ]
2537
2537
2538 # defines which permissions are more important higher the more important
2538 # defines which permissions are more important higher the more important
2539 # Weight defines which permissions are more important.
2539 # Weight defines which permissions are more important.
2540 # The higher number the more important.
2540 # The higher number the more important.
2541 PERM_WEIGHTS = {
2541 PERM_WEIGHTS = {
2542 'repository.none': 0,
2542 'repository.none': 0,
2543 'repository.read': 1,
2543 'repository.read': 1,
2544 'repository.write': 3,
2544 'repository.write': 3,
2545 'repository.admin': 4,
2545 'repository.admin': 4,
2546
2546
2547 'group.none': 0,
2547 'group.none': 0,
2548 'group.read': 1,
2548 'group.read': 1,
2549 'group.write': 3,
2549 'group.write': 3,
2550 'group.admin': 4,
2550 'group.admin': 4,
2551
2551
2552 'usergroup.none': 0,
2552 'usergroup.none': 0,
2553 'usergroup.read': 1,
2553 'usergroup.read': 1,
2554 'usergroup.write': 3,
2554 'usergroup.write': 3,
2555 'usergroup.admin': 4,
2555 'usergroup.admin': 4,
2556
2556
2557 'hg.repogroup.create.false': 0,
2557 'hg.repogroup.create.false': 0,
2558 'hg.repogroup.create.true': 1,
2558 'hg.repogroup.create.true': 1,
2559
2559
2560 'hg.usergroup.create.false': 0,
2560 'hg.usergroup.create.false': 0,
2561 'hg.usergroup.create.true': 1,
2561 'hg.usergroup.create.true': 1,
2562
2562
2563 'hg.fork.none': 0,
2563 'hg.fork.none': 0,
2564 'hg.fork.repository': 1,
2564 'hg.fork.repository': 1,
2565 'hg.create.none': 0,
2565 'hg.create.none': 0,
2566 'hg.create.repository': 1
2566 'hg.create.repository': 1
2567 }
2567 }
2568
2568
2569 permission_id = Column("permission_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2569 permission_id = Column("permission_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2570 permission_name = Column("permission_name", String(255), nullable=True, unique=None, default=None)
2570 permission_name = Column("permission_name", String(255), nullable=True, unique=None, default=None)
2571 permission_longname = Column("permission_longname", String(255), nullable=True, unique=None, default=None)
2571 permission_longname = Column("permission_longname", String(255), nullable=True, unique=None, default=None)
2572
2572
2573 def __unicode__(self):
2573 def __unicode__(self):
2574 return u"<%s('%s:%s')>" % (
2574 return u"<%s('%s:%s')>" % (
2575 self.__class__.__name__, self.permission_id, self.permission_name
2575 self.__class__.__name__, self.permission_id, self.permission_name
2576 )
2576 )
2577
2577
2578 @classmethod
2578 @classmethod
2579 def get_by_key(cls, key):
2579 def get_by_key(cls, key):
2580 return cls.query().filter(cls.permission_name == key).scalar()
2580 return cls.query().filter(cls.permission_name == key).scalar()
2581
2581
2582 @classmethod
2582 @classmethod
2583 def get_default_repo_perms(cls, user_id, repo_id=None):
2583 def get_default_repo_perms(cls, user_id, repo_id=None):
2584 q = Session().query(UserRepoToPerm, Repository, Permission)\
2584 q = Session().query(UserRepoToPerm, Repository, Permission)\
2585 .join((Permission, UserRepoToPerm.permission_id == Permission.permission_id))\
2585 .join((Permission, UserRepoToPerm.permission_id == Permission.permission_id))\
2586 .join((Repository, UserRepoToPerm.repository_id == Repository.repo_id))\
2586 .join((Repository, UserRepoToPerm.repository_id == Repository.repo_id))\
2587 .filter(UserRepoToPerm.user_id == user_id)
2587 .filter(UserRepoToPerm.user_id == user_id)
2588 if repo_id:
2588 if repo_id:
2589 q = q.filter(UserRepoToPerm.repository_id == repo_id)
2589 q = q.filter(UserRepoToPerm.repository_id == repo_id)
2590 return q.all()
2590 return q.all()
2591
2591
2592 @classmethod
2592 @classmethod
2593 def get_default_repo_perms_from_user_group(cls, user_id, repo_id=None):
2593 def get_default_repo_perms_from_user_group(cls, user_id, repo_id=None):
2594 q = Session().query(UserGroupRepoToPerm, Repository, Permission)\
2594 q = Session().query(UserGroupRepoToPerm, Repository, Permission)\
2595 .join(
2595 .join(
2596 Permission,
2596 Permission,
2597 UserGroupRepoToPerm.permission_id == Permission.permission_id)\
2597 UserGroupRepoToPerm.permission_id == Permission.permission_id)\
2598 .join(
2598 .join(
2599 Repository,
2599 Repository,
2600 UserGroupRepoToPerm.repository_id == Repository.repo_id)\
2600 UserGroupRepoToPerm.repository_id == Repository.repo_id)\
2601 .join(
2601 .join(
2602 UserGroup,
2602 UserGroup,
2603 UserGroupRepoToPerm.users_group_id ==
2603 UserGroupRepoToPerm.users_group_id ==
2604 UserGroup.users_group_id)\
2604 UserGroup.users_group_id)\
2605 .join(
2605 .join(
2606 UserGroupMember,
2606 UserGroupMember,
2607 UserGroupRepoToPerm.users_group_id ==
2607 UserGroupRepoToPerm.users_group_id ==
2608 UserGroupMember.users_group_id)\
2608 UserGroupMember.users_group_id)\
2609 .filter(
2609 .filter(
2610 UserGroupMember.user_id == user_id,
2610 UserGroupMember.user_id == user_id,
2611 UserGroup.users_group_active == true())
2611 UserGroup.users_group_active == true())
2612 if repo_id:
2612 if repo_id:
2613 q = q.filter(UserGroupRepoToPerm.repository_id == repo_id)
2613 q = q.filter(UserGroupRepoToPerm.repository_id == repo_id)
2614 return q.all()
2614 return q.all()
2615
2615
2616 @classmethod
2616 @classmethod
2617 def get_default_group_perms(cls, user_id, repo_group_id=None):
2617 def get_default_group_perms(cls, user_id, repo_group_id=None):
2618 q = Session().query(UserRepoGroupToPerm, RepoGroup, Permission)\
2618 q = Session().query(UserRepoGroupToPerm, RepoGroup, Permission)\
2619 .join((Permission, UserRepoGroupToPerm.permission_id == Permission.permission_id))\
2619 .join((Permission, UserRepoGroupToPerm.permission_id == Permission.permission_id))\
2620 .join((RepoGroup, UserRepoGroupToPerm.group_id == RepoGroup.group_id))\
2620 .join((RepoGroup, UserRepoGroupToPerm.group_id == RepoGroup.group_id))\
2621 .filter(UserRepoGroupToPerm.user_id == user_id)
2621 .filter(UserRepoGroupToPerm.user_id == user_id)
2622 if repo_group_id:
2622 if repo_group_id:
2623 q = q.filter(UserRepoGroupToPerm.group_id == repo_group_id)
2623 q = q.filter(UserRepoGroupToPerm.group_id == repo_group_id)
2624 return q.all()
2624 return q.all()
2625
2625
2626 @classmethod
2626 @classmethod
2627 def get_default_group_perms_from_user_group(
2627 def get_default_group_perms_from_user_group(
2628 cls, user_id, repo_group_id=None):
2628 cls, user_id, repo_group_id=None):
2629 q = Session().query(UserGroupRepoGroupToPerm, RepoGroup, Permission)\
2629 q = Session().query(UserGroupRepoGroupToPerm, RepoGroup, Permission)\
2630 .join(
2630 .join(
2631 Permission,
2631 Permission,
2632 UserGroupRepoGroupToPerm.permission_id ==
2632 UserGroupRepoGroupToPerm.permission_id ==
2633 Permission.permission_id)\
2633 Permission.permission_id)\
2634 .join(
2634 .join(
2635 RepoGroup,
2635 RepoGroup,
2636 UserGroupRepoGroupToPerm.group_id == RepoGroup.group_id)\
2636 UserGroupRepoGroupToPerm.group_id == RepoGroup.group_id)\
2637 .join(
2637 .join(
2638 UserGroup,
2638 UserGroup,
2639 UserGroupRepoGroupToPerm.users_group_id ==
2639 UserGroupRepoGroupToPerm.users_group_id ==
2640 UserGroup.users_group_id)\
2640 UserGroup.users_group_id)\
2641 .join(
2641 .join(
2642 UserGroupMember,
2642 UserGroupMember,
2643 UserGroupRepoGroupToPerm.users_group_id ==
2643 UserGroupRepoGroupToPerm.users_group_id ==
2644 UserGroupMember.users_group_id)\
2644 UserGroupMember.users_group_id)\
2645 .filter(
2645 .filter(
2646 UserGroupMember.user_id == user_id,
2646 UserGroupMember.user_id == user_id,
2647 UserGroup.users_group_active == true())
2647 UserGroup.users_group_active == true())
2648 if repo_group_id:
2648 if repo_group_id:
2649 q = q.filter(UserGroupRepoGroupToPerm.group_id == repo_group_id)
2649 q = q.filter(UserGroupRepoGroupToPerm.group_id == repo_group_id)
2650 return q.all()
2650 return q.all()
2651
2651
2652 @classmethod
2652 @classmethod
2653 def get_default_user_group_perms(cls, user_id, user_group_id=None):
2653 def get_default_user_group_perms(cls, user_id, user_group_id=None):
2654 q = Session().query(UserUserGroupToPerm, UserGroup, Permission)\
2654 q = Session().query(UserUserGroupToPerm, UserGroup, Permission)\
2655 .join((Permission, UserUserGroupToPerm.permission_id == Permission.permission_id))\
2655 .join((Permission, UserUserGroupToPerm.permission_id == Permission.permission_id))\
2656 .join((UserGroup, UserUserGroupToPerm.user_group_id == UserGroup.users_group_id))\
2656 .join((UserGroup, UserUserGroupToPerm.user_group_id == UserGroup.users_group_id))\
2657 .filter(UserUserGroupToPerm.user_id == user_id)
2657 .filter(UserUserGroupToPerm.user_id == user_id)
2658 if user_group_id:
2658 if user_group_id:
2659 q = q.filter(UserUserGroupToPerm.user_group_id == user_group_id)
2659 q = q.filter(UserUserGroupToPerm.user_group_id == user_group_id)
2660 return q.all()
2660 return q.all()
2661
2661
2662 @classmethod
2662 @classmethod
2663 def get_default_user_group_perms_from_user_group(
2663 def get_default_user_group_perms_from_user_group(
2664 cls, user_id, user_group_id=None):
2664 cls, user_id, user_group_id=None):
2665 TargetUserGroup = aliased(UserGroup, name='target_user_group')
2665 TargetUserGroup = aliased(UserGroup, name='target_user_group')
2666 q = Session().query(UserGroupUserGroupToPerm, UserGroup, Permission)\
2666 q = Session().query(UserGroupUserGroupToPerm, UserGroup, Permission)\
2667 .join(
2667 .join(
2668 Permission,
2668 Permission,
2669 UserGroupUserGroupToPerm.permission_id ==
2669 UserGroupUserGroupToPerm.permission_id ==
2670 Permission.permission_id)\
2670 Permission.permission_id)\
2671 .join(
2671 .join(
2672 TargetUserGroup,
2672 TargetUserGroup,
2673 UserGroupUserGroupToPerm.target_user_group_id ==
2673 UserGroupUserGroupToPerm.target_user_group_id ==
2674 TargetUserGroup.users_group_id)\
2674 TargetUserGroup.users_group_id)\
2675 .join(
2675 .join(
2676 UserGroup,
2676 UserGroup,
2677 UserGroupUserGroupToPerm.user_group_id ==
2677 UserGroupUserGroupToPerm.user_group_id ==
2678 UserGroup.users_group_id)\
2678 UserGroup.users_group_id)\
2679 .join(
2679 .join(
2680 UserGroupMember,
2680 UserGroupMember,
2681 UserGroupUserGroupToPerm.user_group_id ==
2681 UserGroupUserGroupToPerm.user_group_id ==
2682 UserGroupMember.users_group_id)\
2682 UserGroupMember.users_group_id)\
2683 .filter(
2683 .filter(
2684 UserGroupMember.user_id == user_id,
2684 UserGroupMember.user_id == user_id,
2685 UserGroup.users_group_active == true())
2685 UserGroup.users_group_active == true())
2686 if user_group_id:
2686 if user_group_id:
2687 q = q.filter(
2687 q = q.filter(
2688 UserGroupUserGroupToPerm.user_group_id == user_group_id)
2688 UserGroupUserGroupToPerm.user_group_id == user_group_id)
2689
2689
2690 return q.all()
2690 return q.all()
2691
2691
2692
2692
2693 class UserRepoToPerm(Base, BaseModel):
2693 class UserRepoToPerm(Base, BaseModel):
2694 __tablename__ = 'repo_to_perm'
2694 __tablename__ = 'repo_to_perm'
2695 __table_args__ = (
2695 __table_args__ = (
2696 UniqueConstraint('user_id', 'repository_id', 'permission_id'),
2696 UniqueConstraint('user_id', 'repository_id', 'permission_id'),
2697 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2697 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2698 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2698 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2699 )
2699 )
2700 repo_to_perm_id = Column("repo_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2700 repo_to_perm_id = Column("repo_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2701 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2701 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2702 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2702 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2703 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
2703 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
2704
2704
2705 user = relationship('User')
2705 user = relationship('User')
2706 repository = relationship('Repository')
2706 repository = relationship('Repository')
2707 permission = relationship('Permission')
2707 permission = relationship('Permission')
2708
2708
2709 @classmethod
2709 @classmethod
2710 def create(cls, user, repository, permission):
2710 def create(cls, user, repository, permission):
2711 n = cls()
2711 n = cls()
2712 n.user = user
2712 n.user = user
2713 n.repository = repository
2713 n.repository = repository
2714 n.permission = permission
2714 n.permission = permission
2715 Session().add(n)
2715 Session().add(n)
2716 return n
2716 return n
2717
2717
2718 def __unicode__(self):
2718 def __unicode__(self):
2719 return u'<%s => %s >' % (self.user, self.repository)
2719 return u'<%s => %s >' % (self.user, self.repository)
2720
2720
2721
2721
2722 class UserUserGroupToPerm(Base, BaseModel):
2722 class UserUserGroupToPerm(Base, BaseModel):
2723 __tablename__ = 'user_user_group_to_perm'
2723 __tablename__ = 'user_user_group_to_perm'
2724 __table_args__ = (
2724 __table_args__ = (
2725 UniqueConstraint('user_id', 'user_group_id', 'permission_id'),
2725 UniqueConstraint('user_id', 'user_group_id', 'permission_id'),
2726 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2726 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2727 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2727 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2728 )
2728 )
2729 user_user_group_to_perm_id = Column("user_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2729 user_user_group_to_perm_id = Column("user_user_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2730 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2730 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2731 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2731 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2732 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2732 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2733
2733
2734 user = relationship('User')
2734 user = relationship('User')
2735 user_group = relationship('UserGroup')
2735 user_group = relationship('UserGroup')
2736 permission = relationship('Permission')
2736 permission = relationship('Permission')
2737
2737
2738 @classmethod
2738 @classmethod
2739 def create(cls, user, user_group, permission):
2739 def create(cls, user, user_group, permission):
2740 n = cls()
2740 n = cls()
2741 n.user = user
2741 n.user = user
2742 n.user_group = user_group
2742 n.user_group = user_group
2743 n.permission = permission
2743 n.permission = permission
2744 Session().add(n)
2744 Session().add(n)
2745 return n
2745 return n
2746
2746
2747 def __unicode__(self):
2747 def __unicode__(self):
2748 return u'<%s => %s >' % (self.user, self.user_group)
2748 return u'<%s => %s >' % (self.user, self.user_group)
2749
2749
2750
2750
2751 class UserToPerm(Base, BaseModel):
2751 class UserToPerm(Base, BaseModel):
2752 __tablename__ = 'user_to_perm'
2752 __tablename__ = 'user_to_perm'
2753 __table_args__ = (
2753 __table_args__ = (
2754 UniqueConstraint('user_id', 'permission_id'),
2754 UniqueConstraint('user_id', 'permission_id'),
2755 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2755 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2756 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2756 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2757 )
2757 )
2758 user_to_perm_id = Column("user_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2758 user_to_perm_id = Column("user_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2759 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2759 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2760 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2760 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2761
2761
2762 user = relationship('User')
2762 user = relationship('User')
2763 permission = relationship('Permission', lazy='joined')
2763 permission = relationship('Permission', lazy='joined')
2764
2764
2765 def __unicode__(self):
2765 def __unicode__(self):
2766 return u'<%s => %s >' % (self.user, self.permission)
2766 return u'<%s => %s >' % (self.user, self.permission)
2767
2767
2768
2768
2769 class UserGroupRepoToPerm(Base, BaseModel):
2769 class UserGroupRepoToPerm(Base, BaseModel):
2770 __tablename__ = 'users_group_repo_to_perm'
2770 __tablename__ = 'users_group_repo_to_perm'
2771 __table_args__ = (
2771 __table_args__ = (
2772 UniqueConstraint('repository_id', 'users_group_id', 'permission_id'),
2772 UniqueConstraint('repository_id', 'users_group_id', 'permission_id'),
2773 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2773 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2774 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2774 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2775 )
2775 )
2776 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2776 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2777 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2777 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2778 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2778 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2779 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
2779 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=None, default=None)
2780
2780
2781 users_group = relationship('UserGroup')
2781 users_group = relationship('UserGroup')
2782 permission = relationship('Permission')
2782 permission = relationship('Permission')
2783 repository = relationship('Repository')
2783 repository = relationship('Repository')
2784
2784
2785 @classmethod
2785 @classmethod
2786 def create(cls, users_group, repository, permission):
2786 def create(cls, users_group, repository, permission):
2787 n = cls()
2787 n = cls()
2788 n.users_group = users_group
2788 n.users_group = users_group
2789 n.repository = repository
2789 n.repository = repository
2790 n.permission = permission
2790 n.permission = permission
2791 Session().add(n)
2791 Session().add(n)
2792 return n
2792 return n
2793
2793
2794 def __unicode__(self):
2794 def __unicode__(self):
2795 return u'<UserGroupRepoToPerm:%s => %s >' % (self.users_group, self.repository)
2795 return u'<UserGroupRepoToPerm:%s => %s >' % (self.users_group, self.repository)
2796
2796
2797
2797
2798 class UserGroupUserGroupToPerm(Base, BaseModel):
2798 class UserGroupUserGroupToPerm(Base, BaseModel):
2799 __tablename__ = 'user_group_user_group_to_perm'
2799 __tablename__ = 'user_group_user_group_to_perm'
2800 __table_args__ = (
2800 __table_args__ = (
2801 UniqueConstraint('target_user_group_id', 'user_group_id', 'permission_id'),
2801 UniqueConstraint('target_user_group_id', 'user_group_id', 'permission_id'),
2802 CheckConstraint('target_user_group_id != user_group_id'),
2802 CheckConstraint('target_user_group_id != user_group_id'),
2803 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2803 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2804 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2804 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2805 )
2805 )
2806 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)
2806 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)
2807 target_user_group_id = Column("target_user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2807 target_user_group_id = Column("target_user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2808 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2808 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2809 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2809 user_group_id = Column("user_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2810
2810
2811 target_user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id')
2811 target_user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.target_user_group_id==UserGroup.users_group_id')
2812 user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.user_group_id==UserGroup.users_group_id')
2812 user_group = relationship('UserGroup', primaryjoin='UserGroupUserGroupToPerm.user_group_id==UserGroup.users_group_id')
2813 permission = relationship('Permission')
2813 permission = relationship('Permission')
2814
2814
2815 @classmethod
2815 @classmethod
2816 def create(cls, target_user_group, user_group, permission):
2816 def create(cls, target_user_group, user_group, permission):
2817 n = cls()
2817 n = cls()
2818 n.target_user_group = target_user_group
2818 n.target_user_group = target_user_group
2819 n.user_group = user_group
2819 n.user_group = user_group
2820 n.permission = permission
2820 n.permission = permission
2821 Session().add(n)
2821 Session().add(n)
2822 return n
2822 return n
2823
2823
2824 def __unicode__(self):
2824 def __unicode__(self):
2825 return u'<UserGroupUserGroup:%s => %s >' % (self.target_user_group, self.user_group)
2825 return u'<UserGroupUserGroup:%s => %s >' % (self.target_user_group, self.user_group)
2826
2826
2827
2827
2828 class UserGroupToPerm(Base, BaseModel):
2828 class UserGroupToPerm(Base, BaseModel):
2829 __tablename__ = 'users_group_to_perm'
2829 __tablename__ = 'users_group_to_perm'
2830 __table_args__ = (
2830 __table_args__ = (
2831 UniqueConstraint('users_group_id', 'permission_id',),
2831 UniqueConstraint('users_group_id', 'permission_id',),
2832 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2832 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2833 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2833 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2834 )
2834 )
2835 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2835 users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2836 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2836 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2837 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2837 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2838
2838
2839 users_group = relationship('UserGroup')
2839 users_group = relationship('UserGroup')
2840 permission = relationship('Permission')
2840 permission = relationship('Permission')
2841
2841
2842
2842
2843 class UserRepoGroupToPerm(Base, BaseModel):
2843 class UserRepoGroupToPerm(Base, BaseModel):
2844 __tablename__ = 'user_repo_group_to_perm'
2844 __tablename__ = 'user_repo_group_to_perm'
2845 __table_args__ = (
2845 __table_args__ = (
2846 UniqueConstraint('user_id', 'group_id', 'permission_id'),
2846 UniqueConstraint('user_id', 'group_id', 'permission_id'),
2847 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2847 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2848 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2848 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2849 )
2849 )
2850
2850
2851 group_to_perm_id = Column("group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2851 group_to_perm_id = Column("group_to_perm_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2852 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2852 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2853 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
2853 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
2854 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2854 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2855
2855
2856 user = relationship('User')
2856 user = relationship('User')
2857 group = relationship('RepoGroup')
2857 group = relationship('RepoGroup')
2858 permission = relationship('Permission')
2858 permission = relationship('Permission')
2859
2859
2860 @classmethod
2860 @classmethod
2861 def create(cls, user, repository_group, permission):
2861 def create(cls, user, repository_group, permission):
2862 n = cls()
2862 n = cls()
2863 n.user = user
2863 n.user = user
2864 n.group = repository_group
2864 n.group = repository_group
2865 n.permission = permission
2865 n.permission = permission
2866 Session().add(n)
2866 Session().add(n)
2867 return n
2867 return n
2868
2868
2869
2869
2870 class UserGroupRepoGroupToPerm(Base, BaseModel):
2870 class UserGroupRepoGroupToPerm(Base, BaseModel):
2871 __tablename__ = 'users_group_repo_group_to_perm'
2871 __tablename__ = 'users_group_repo_group_to_perm'
2872 __table_args__ = (
2872 __table_args__ = (
2873 UniqueConstraint('users_group_id', 'group_id'),
2873 UniqueConstraint('users_group_id', 'group_id'),
2874 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2874 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2875 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2875 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2876 )
2876 )
2877
2877
2878 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)
2878 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)
2879 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2879 users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable=False, unique=None, default=None)
2880 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
2880 group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable=False, unique=None, default=None)
2881 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2881 permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable=False, unique=None, default=None)
2882
2882
2883 users_group = relationship('UserGroup')
2883 users_group = relationship('UserGroup')
2884 permission = relationship('Permission')
2884 permission = relationship('Permission')
2885 group = relationship('RepoGroup')
2885 group = relationship('RepoGroup')
2886
2886
2887 @classmethod
2887 @classmethod
2888 def create(cls, user_group, repository_group, permission):
2888 def create(cls, user_group, repository_group, permission):
2889 n = cls()
2889 n = cls()
2890 n.users_group = user_group
2890 n.users_group = user_group
2891 n.group = repository_group
2891 n.group = repository_group
2892 n.permission = permission
2892 n.permission = permission
2893 Session().add(n)
2893 Session().add(n)
2894 return n
2894 return n
2895
2895
2896 def __unicode__(self):
2896 def __unicode__(self):
2897 return u'<UserGroupRepoGroupToPerm:%s => %s >' % (self.users_group, self.group)
2897 return u'<UserGroupRepoGroupToPerm:%s => %s >' % (self.users_group, self.group)
2898
2898
2899
2899
2900 class Statistics(Base, BaseModel):
2900 class Statistics(Base, BaseModel):
2901 __tablename__ = 'statistics'
2901 __tablename__ = 'statistics'
2902 __table_args__ = (
2902 __table_args__ = (
2903 UniqueConstraint('repository_id'),
2903 UniqueConstraint('repository_id'),
2904 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2904 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2905 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2905 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2906 )
2906 )
2907 stat_id = Column("stat_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2907 stat_id = Column("stat_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2908 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=True, default=None)
2908 repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=False, unique=True, default=None)
2909 stat_on_revision = Column("stat_on_revision", Integer(), nullable=False)
2909 stat_on_revision = Column("stat_on_revision", Integer(), nullable=False)
2910 commit_activity = Column("commit_activity", LargeBinary(1000000), nullable=False)#JSON data
2910 commit_activity = Column("commit_activity", LargeBinary(1000000), nullable=False)#JSON data
2911 commit_activity_combined = Column("commit_activity_combined", LargeBinary(), nullable=False)#JSON data
2911 commit_activity_combined = Column("commit_activity_combined", LargeBinary(), nullable=False)#JSON data
2912 languages = Column("languages", LargeBinary(1000000), nullable=False)#JSON data
2912 languages = Column("languages", LargeBinary(1000000), nullable=False)#JSON data
2913
2913
2914 repository = relationship('Repository', single_parent=True)
2914 repository = relationship('Repository', single_parent=True)
2915
2915
2916
2916
2917 class UserFollowing(Base, BaseModel):
2917 class UserFollowing(Base, BaseModel):
2918 __tablename__ = 'user_followings'
2918 __tablename__ = 'user_followings'
2919 __table_args__ = (
2919 __table_args__ = (
2920 UniqueConstraint('user_id', 'follows_repository_id'),
2920 UniqueConstraint('user_id', 'follows_repository_id'),
2921 UniqueConstraint('user_id', 'follows_user_id'),
2921 UniqueConstraint('user_id', 'follows_user_id'),
2922 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2922 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2923 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2923 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
2924 )
2924 )
2925
2925
2926 user_following_id = Column("user_following_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2926 user_following_id = Column("user_following_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2927 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2927 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None, default=None)
2928 follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True, unique=None, default=None)
2928 follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable=True, unique=None, default=None)
2929 follows_user_id = Column("follows_user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
2929 follows_user_id = Column("follows_user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=None, default=None)
2930 follows_from = Column('follows_from', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
2930 follows_from = Column('follows_from', DateTime(timezone=False), nullable=True, unique=None, default=datetime.datetime.now)
2931
2931
2932 user = relationship('User', primaryjoin='User.user_id==UserFollowing.user_id')
2932 user = relationship('User', primaryjoin='User.user_id==UserFollowing.user_id')
2933
2933
2934 follows_user = relationship('User', primaryjoin='User.user_id==UserFollowing.follows_user_id')
2934 follows_user = relationship('User', primaryjoin='User.user_id==UserFollowing.follows_user_id')
2935 follows_repository = relationship('Repository', order_by='Repository.repo_name')
2935 follows_repository = relationship('Repository', order_by='Repository.repo_name')
2936
2936
2937 @classmethod
2937 @classmethod
2938 def get_repo_followers(cls, repo_id):
2938 def get_repo_followers(cls, repo_id):
2939 return cls.query().filter(cls.follows_repo_id == repo_id)
2939 return cls.query().filter(cls.follows_repo_id == repo_id)
2940
2940
2941
2941
2942 class CacheKey(Base, BaseModel):
2942 class CacheKey(Base, BaseModel):
2943 __tablename__ = 'cache_invalidation'
2943 __tablename__ = 'cache_invalidation'
2944 __table_args__ = (
2944 __table_args__ = (
2945 UniqueConstraint('cache_key'),
2945 UniqueConstraint('cache_key'),
2946 Index('key_idx', 'cache_key'),
2946 Index('key_idx', 'cache_key'),
2947 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2947 {'extend_existing': True, 'mysql_engine': 'InnoDB',
2948 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2948 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
2949 )
2949 )
2950 CACHE_TYPE_ATOM = 'ATOM'
2950 CACHE_TYPE_ATOM = 'ATOM'
2951 CACHE_TYPE_RSS = 'RSS'
2951 CACHE_TYPE_RSS = 'RSS'
2952 CACHE_TYPE_README = 'README'
2952 CACHE_TYPE_README = 'README'
2953
2953
2954 cache_id = Column("cache_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2954 cache_id = Column("cache_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
2955 cache_key = Column("cache_key", String(255), nullable=True, unique=None, default=None)
2955 cache_key = Column("cache_key", String(255), nullable=True, unique=None, default=None)
2956 cache_args = Column("cache_args", String(255), nullable=True, unique=None, default=None)
2956 cache_args = Column("cache_args", String(255), nullable=True, unique=None, default=None)
2957 cache_active = Column("cache_active", Boolean(), nullable=True, unique=None, default=False)
2957 cache_active = Column("cache_active", Boolean(), nullable=True, unique=None, default=False)
2958
2958
2959 def __init__(self, cache_key, cache_args=''):
2959 def __init__(self, cache_key, cache_args=''):
2960 self.cache_key = cache_key
2960 self.cache_key = cache_key
2961 self.cache_args = cache_args
2961 self.cache_args = cache_args
2962 self.cache_active = False
2962 self.cache_active = False
2963
2963
2964 def __unicode__(self):
2964 def __unicode__(self):
2965 return u"<%s('%s:%s[%s]')>" % (
2965 return u"<%s('%s:%s[%s]')>" % (
2966 self.__class__.__name__,
2966 self.__class__.__name__,
2967 self.cache_id, self.cache_key, self.cache_active)
2967 self.cache_id, self.cache_key, self.cache_active)
2968
2968
2969 def _cache_key_partition(self):
2969 def _cache_key_partition(self):
2970 prefix, repo_name, suffix = self.cache_key.partition(self.cache_args)
2970 prefix, repo_name, suffix = self.cache_key.partition(self.cache_args)
2971 return prefix, repo_name, suffix
2971 return prefix, repo_name, suffix
2972
2972
2973 def get_prefix(self):
2973 def get_prefix(self):
2974 """
2974 """
2975 Try to extract prefix from existing cache key. The key could consist
2975 Try to extract prefix from existing cache key. The key could consist
2976 of prefix, repo_name, suffix
2976 of prefix, repo_name, suffix
2977 """
2977 """
2978 # this returns prefix, repo_name, suffix
2978 # this returns prefix, repo_name, suffix
2979 return self._cache_key_partition()[0]
2979 return self._cache_key_partition()[0]
2980
2980
2981 def get_suffix(self):
2981 def get_suffix(self):
2982 """
2982 """
2983 get suffix that might have been used in _get_cache_key to
2983 get suffix that might have been used in _get_cache_key to
2984 generate self.cache_key. Only used for informational purposes
2984 generate self.cache_key. Only used for informational purposes
2985 in repo_edit.mako.
2985 in repo_edit.mako.
2986 """
2986 """
2987 # prefix, repo_name, suffix
2987 # prefix, repo_name, suffix
2988 return self._cache_key_partition()[2]
2988 return self._cache_key_partition()[2]
2989
2989
2990 @classmethod
2990 @classmethod
2991 def delete_all_cache(cls):
2991 def delete_all_cache(cls):
2992 """
2992 """
2993 Delete all cache keys from database.
2993 Delete all cache keys from database.
2994 Should only be run when all instances are down and all entries
2994 Should only be run when all instances are down and all entries
2995 thus stale.
2995 thus stale.
2996 """
2996 """
2997 cls.query().delete()
2997 cls.query().delete()
2998 Session().commit()
2998 Session().commit()
2999
2999
3000 @classmethod
3000 @classmethod
3001 def get_cache_key(cls, repo_name, cache_type):
3001 def get_cache_key(cls, repo_name, cache_type):
3002 """
3002 """
3003
3003
3004 Generate a cache key for this process of RhodeCode instance.
3004 Generate a cache key for this process of RhodeCode instance.
3005 Prefix most likely will be process id or maybe explicitly set
3005 Prefix most likely will be process id or maybe explicitly set
3006 instance_id from .ini file.
3006 instance_id from .ini file.
3007 """
3007 """
3008 import rhodecode
3008 import rhodecode
3009 prefix = safe_unicode(rhodecode.CONFIG.get('instance_id') or '')
3009 prefix = safe_unicode(rhodecode.CONFIG.get('instance_id') or '')
3010
3010
3011 repo_as_unicode = safe_unicode(repo_name)
3011 repo_as_unicode = safe_unicode(repo_name)
3012 key = u'{}_{}'.format(repo_as_unicode, cache_type) \
3012 key = u'{}_{}'.format(repo_as_unicode, cache_type) \
3013 if cache_type else repo_as_unicode
3013 if cache_type else repo_as_unicode
3014
3014
3015 return u'{}{}'.format(prefix, key)
3015 return u'{}{}'.format(prefix, key)
3016
3016
3017 @classmethod
3017 @classmethod
3018 def set_invalidate(cls, repo_name, delete=False):
3018 def set_invalidate(cls, repo_name, delete=False):
3019 """
3019 """
3020 Mark all caches of a repo as invalid in the database.
3020 Mark all caches of a repo as invalid in the database.
3021 """
3021 """
3022
3022
3023 try:
3023 try:
3024 qry = Session().query(cls).filter(cls.cache_args == repo_name)
3024 qry = Session().query(cls).filter(cls.cache_args == repo_name)
3025 if delete:
3025 if delete:
3026 log.debug('cache objects deleted for repo %s',
3026 log.debug('cache objects deleted for repo %s',
3027 safe_str(repo_name))
3027 safe_str(repo_name))
3028 qry.delete()
3028 qry.delete()
3029 else:
3029 else:
3030 log.debug('cache objects marked as invalid for repo %s',
3030 log.debug('cache objects marked as invalid for repo %s',
3031 safe_str(repo_name))
3031 safe_str(repo_name))
3032 qry.update({"cache_active": False})
3032 qry.update({"cache_active": False})
3033
3033
3034 Session().commit()
3034 Session().commit()
3035 except Exception:
3035 except Exception:
3036 log.exception(
3036 log.exception(
3037 'Cache key invalidation failed for repository %s',
3037 'Cache key invalidation failed for repository %s',
3038 safe_str(repo_name))
3038 safe_str(repo_name))
3039 Session().rollback()
3039 Session().rollback()
3040
3040
3041 @classmethod
3041 @classmethod
3042 def get_active_cache(cls, cache_key):
3042 def get_active_cache(cls, cache_key):
3043 inv_obj = cls.query().filter(cls.cache_key == cache_key).scalar()
3043 inv_obj = cls.query().filter(cls.cache_key == cache_key).scalar()
3044 if inv_obj:
3044 if inv_obj:
3045 return inv_obj
3045 return inv_obj
3046 return None
3046 return None
3047
3047
3048 @classmethod
3048 @classmethod
3049 def repo_context_cache(cls, compute_func, repo_name, cache_type,
3049 def repo_context_cache(cls, compute_func, repo_name, cache_type,
3050 thread_scoped=False):
3050 thread_scoped=False):
3051 """
3051 """
3052 @cache_region('long_term')
3052 @cache_region('long_term')
3053 def _heavy_calculation(cache_key):
3053 def _heavy_calculation(cache_key):
3054 return 'result'
3054 return 'result'
3055
3055
3056 cache_context = CacheKey.repo_context_cache(
3056 cache_context = CacheKey.repo_context_cache(
3057 _heavy_calculation, repo_name, cache_type)
3057 _heavy_calculation, repo_name, cache_type)
3058
3058
3059 with cache_context as context:
3059 with cache_context as context:
3060 context.invalidate()
3060 context.invalidate()
3061 computed = context.compute()
3061 computed = context.compute()
3062
3062
3063 assert computed == 'result'
3063 assert computed == 'result'
3064 """
3064 """
3065 from rhodecode.lib import caches
3065 from rhodecode.lib import caches
3066 return caches.InvalidationContext(
3066 return caches.InvalidationContext(
3067 compute_func, repo_name, cache_type, thread_scoped=thread_scoped)
3067 compute_func, repo_name, cache_type, thread_scoped=thread_scoped)
3068
3068
3069
3069
3070 class ChangesetComment(Base, BaseModel):
3070 class ChangesetComment(Base, BaseModel):
3071 __tablename__ = 'changeset_comments'
3071 __tablename__ = 'changeset_comments'
3072 __table_args__ = (
3072 __table_args__ = (
3073 Index('cc_revision_idx', 'revision'),
3073 Index('cc_revision_idx', 'revision'),
3074 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3074 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3075 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3075 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3076 )
3076 )
3077
3077
3078 COMMENT_OUTDATED = u'comment_outdated'
3078 COMMENT_OUTDATED = u'comment_outdated'
3079 COMMENT_TYPE_NOTE = u'note'
3079 COMMENT_TYPE_NOTE = u'note'
3080 COMMENT_TYPE_TODO = u'todo'
3080 COMMENT_TYPE_TODO = u'todo'
3081 COMMENT_TYPES = [COMMENT_TYPE_NOTE, COMMENT_TYPE_TODO]
3081 COMMENT_TYPES = [COMMENT_TYPE_NOTE, COMMENT_TYPE_TODO]
3082
3082
3083 comment_id = Column('comment_id', Integer(), nullable=False, primary_key=True)
3083 comment_id = Column('comment_id', Integer(), nullable=False, primary_key=True)
3084 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3084 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3085 revision = Column('revision', String(40), nullable=True)
3085 revision = Column('revision', String(40), nullable=True)
3086 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3086 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3087 pull_request_version_id = Column("pull_request_version_id", Integer(), ForeignKey('pull_request_versions.pull_request_version_id'), nullable=True)
3087 pull_request_version_id = Column("pull_request_version_id", Integer(), ForeignKey('pull_request_versions.pull_request_version_id'), nullable=True)
3088 line_no = Column('line_no', Unicode(10), nullable=True)
3088 line_no = Column('line_no', Unicode(10), nullable=True)
3089 hl_lines = Column('hl_lines', Unicode(512), nullable=True)
3089 hl_lines = Column('hl_lines', Unicode(512), nullable=True)
3090 f_path = Column('f_path', Unicode(1000), nullable=True)
3090 f_path = Column('f_path', Unicode(1000), nullable=True)
3091 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
3091 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=False)
3092 text = Column('text', UnicodeText().with_variant(UnicodeText(25000), 'mysql'), nullable=False)
3092 text = Column('text', UnicodeText().with_variant(UnicodeText(25000), 'mysql'), nullable=False)
3093 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3093 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3094 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3094 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3095 renderer = Column('renderer', Unicode(64), nullable=True)
3095 renderer = Column('renderer', Unicode(64), nullable=True)
3096 display_state = Column('display_state', Unicode(128), nullable=True)
3096 display_state = Column('display_state', Unicode(128), nullable=True)
3097
3097
3098 comment_type = Column('comment_type', Unicode(128), nullable=True, default=COMMENT_TYPE_NOTE)
3098 comment_type = Column('comment_type', Unicode(128), nullable=True, default=COMMENT_TYPE_NOTE)
3099 resolved_comment_id = Column('resolved_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'), nullable=True)
3099 resolved_comment_id = Column('resolved_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'), nullable=True)
3100 resolved_comment = relationship('ChangesetComment', remote_side=comment_id, backref='resolved_by')
3100 resolved_comment = relationship('ChangesetComment', remote_side=comment_id, backref='resolved_by')
3101 author = relationship('User', lazy='joined')
3101 author = relationship('User', lazy='joined')
3102 repo = relationship('Repository')
3102 repo = relationship('Repository')
3103 status_change = relationship('ChangesetStatus', cascade="all, delete, delete-orphan", lazy='joined')
3103 status_change = relationship('ChangesetStatus', cascade="all, delete, delete-orphan", lazy='joined')
3104 pull_request = relationship('PullRequest', lazy='joined')
3104 pull_request = relationship('PullRequest', lazy='joined')
3105 pull_request_version = relationship('PullRequestVersion')
3105 pull_request_version = relationship('PullRequestVersion')
3106
3106
3107 @classmethod
3107 @classmethod
3108 def get_users(cls, revision=None, pull_request_id=None):
3108 def get_users(cls, revision=None, pull_request_id=None):
3109 """
3109 """
3110 Returns user associated with this ChangesetComment. ie those
3110 Returns user associated with this ChangesetComment. ie those
3111 who actually commented
3111 who actually commented
3112
3112
3113 :param cls:
3113 :param cls:
3114 :param revision:
3114 :param revision:
3115 """
3115 """
3116 q = Session().query(User)\
3116 q = Session().query(User)\
3117 .join(ChangesetComment.author)
3117 .join(ChangesetComment.author)
3118 if revision:
3118 if revision:
3119 q = q.filter(cls.revision == revision)
3119 q = q.filter(cls.revision == revision)
3120 elif pull_request_id:
3120 elif pull_request_id:
3121 q = q.filter(cls.pull_request_id == pull_request_id)
3121 q = q.filter(cls.pull_request_id == pull_request_id)
3122 return q.all()
3122 return q.all()
3123
3123
3124 @classmethod
3124 @classmethod
3125 def get_index_from_version(cls, pr_version, versions):
3125 def get_index_from_version(cls, pr_version, versions):
3126 num_versions = [x.pull_request_version_id for x in versions]
3126 num_versions = [x.pull_request_version_id for x in versions]
3127 try:
3127 try:
3128 return num_versions.index(pr_version) +1
3128 return num_versions.index(pr_version) +1
3129 except (IndexError, ValueError):
3129 except (IndexError, ValueError):
3130 return
3130 return
3131
3131
3132 @property
3132 @property
3133 def outdated(self):
3133 def outdated(self):
3134 return self.display_state == self.COMMENT_OUTDATED
3134 return self.display_state == self.COMMENT_OUTDATED
3135
3135
3136 def outdated_at_version(self, version):
3136 def outdated_at_version(self, version):
3137 """
3137 """
3138 Checks if comment is outdated for given pull request version
3138 Checks if comment is outdated for given pull request version
3139 """
3139 """
3140 return self.outdated and self.pull_request_version_id != version
3140 return self.outdated and self.pull_request_version_id != version
3141
3141
3142 def older_than_version(self, version):
3142 def older_than_version(self, version):
3143 """
3143 """
3144 Checks if comment is made from previous version than given
3144 Checks if comment is made from previous version than given
3145 """
3145 """
3146 if version is None:
3146 if version is None:
3147 return self.pull_request_version_id is not None
3147 return self.pull_request_version_id is not None
3148
3148
3149 return self.pull_request_version_id < version
3149 return self.pull_request_version_id < version
3150
3150
3151 @property
3151 @property
3152 def resolved(self):
3152 def resolved(self):
3153 return self.resolved_by[0] if self.resolved_by else None
3153 return self.resolved_by[0] if self.resolved_by else None
3154
3154
3155 @property
3155 @property
3156 def is_todo(self):
3156 def is_todo(self):
3157 return self.comment_type == self.COMMENT_TYPE_TODO
3157 return self.comment_type == self.COMMENT_TYPE_TODO
3158
3158
3159 @property
3159 @property
3160 def is_inline(self):
3160 def is_inline(self):
3161 return self.line_no and self.f_path
3161 return self.line_no and self.f_path
3162
3162
3163 def get_index_version(self, versions):
3163 def get_index_version(self, versions):
3164 return self.get_index_from_version(
3164 return self.get_index_from_version(
3165 self.pull_request_version_id, versions)
3165 self.pull_request_version_id, versions)
3166
3166
3167 def __repr__(self):
3167 def __repr__(self):
3168 if self.comment_id:
3168 if self.comment_id:
3169 return '<DB:Comment #%s>' % self.comment_id
3169 return '<DB:Comment #%s>' % self.comment_id
3170 else:
3170 else:
3171 return '<DB:Comment at %#x>' % id(self)
3171 return '<DB:Comment at %#x>' % id(self)
3172
3172
3173 def get_api_data(self):
3173 def get_api_data(self):
3174 comment = self
3174 comment = self
3175 data = {
3175 data = {
3176 'comment_id': comment.comment_id,
3176 'comment_id': comment.comment_id,
3177 'comment_type': comment.comment_type,
3177 'comment_type': comment.comment_type,
3178 'comment_text': comment.text,
3178 'comment_text': comment.text,
3179 'comment_status': comment.status_change,
3179 'comment_status': comment.status_change,
3180 'comment_f_path': comment.f_path,
3180 'comment_f_path': comment.f_path,
3181 'comment_lineno': comment.line_no,
3181 'comment_lineno': comment.line_no,
3182 'comment_author': comment.author,
3182 'comment_author': comment.author,
3183 'comment_created_on': comment.created_on
3183 'comment_created_on': comment.created_on
3184 }
3184 }
3185 return data
3185 return data
3186
3186
3187 def __json__(self):
3187 def __json__(self):
3188 data = dict()
3188 data = dict()
3189 data.update(self.get_api_data())
3189 data.update(self.get_api_data())
3190 return data
3190 return data
3191
3191
3192
3192
3193 class ChangesetStatus(Base, BaseModel):
3193 class ChangesetStatus(Base, BaseModel):
3194 __tablename__ = 'changeset_statuses'
3194 __tablename__ = 'changeset_statuses'
3195 __table_args__ = (
3195 __table_args__ = (
3196 Index('cs_revision_idx', 'revision'),
3196 Index('cs_revision_idx', 'revision'),
3197 Index('cs_version_idx', 'version'),
3197 Index('cs_version_idx', 'version'),
3198 UniqueConstraint('repo_id', 'revision', 'version'),
3198 UniqueConstraint('repo_id', 'revision', 'version'),
3199 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3199 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3200 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3200 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3201 )
3201 )
3202 STATUS_NOT_REVIEWED = DEFAULT = 'not_reviewed'
3202 STATUS_NOT_REVIEWED = DEFAULT = 'not_reviewed'
3203 STATUS_APPROVED = 'approved'
3203 STATUS_APPROVED = 'approved'
3204 STATUS_REJECTED = 'rejected'
3204 STATUS_REJECTED = 'rejected'
3205 STATUS_UNDER_REVIEW = 'under_review'
3205 STATUS_UNDER_REVIEW = 'under_review'
3206
3206
3207 STATUSES = [
3207 STATUSES = [
3208 (STATUS_NOT_REVIEWED, _("Not Reviewed")), # (no icon) and default
3208 (STATUS_NOT_REVIEWED, _("Not Reviewed")), # (no icon) and default
3209 (STATUS_APPROVED, _("Approved")),
3209 (STATUS_APPROVED, _("Approved")),
3210 (STATUS_REJECTED, _("Rejected")),
3210 (STATUS_REJECTED, _("Rejected")),
3211 (STATUS_UNDER_REVIEW, _("Under Review")),
3211 (STATUS_UNDER_REVIEW, _("Under Review")),
3212 ]
3212 ]
3213
3213
3214 changeset_status_id = Column('changeset_status_id', Integer(), nullable=False, primary_key=True)
3214 changeset_status_id = Column('changeset_status_id', Integer(), nullable=False, primary_key=True)
3215 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3215 repo_id = Column('repo_id', Integer(), ForeignKey('repositories.repo_id'), nullable=False)
3216 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None)
3216 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False, unique=None)
3217 revision = Column('revision', String(40), nullable=False)
3217 revision = Column('revision', String(40), nullable=False)
3218 status = Column('status', String(128), nullable=False, default=DEFAULT)
3218 status = Column('status', String(128), nullable=False, default=DEFAULT)
3219 changeset_comment_id = Column('changeset_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'))
3219 changeset_comment_id = Column('changeset_comment_id', Integer(), ForeignKey('changeset_comments.comment_id'))
3220 modified_at = Column('modified_at', DateTime(), nullable=False, default=datetime.datetime.now)
3220 modified_at = Column('modified_at', DateTime(), nullable=False, default=datetime.datetime.now)
3221 version = Column('version', Integer(), nullable=False, default=0)
3221 version = Column('version', Integer(), nullable=False, default=0)
3222 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3222 pull_request_id = Column("pull_request_id", Integer(), ForeignKey('pull_requests.pull_request_id'), nullable=True)
3223
3223
3224 author = relationship('User', lazy='joined')
3224 author = relationship('User', lazy='joined')
3225 repo = relationship('Repository')
3225 repo = relationship('Repository')
3226 comment = relationship('ChangesetComment', lazy='joined')
3226 comment = relationship('ChangesetComment', lazy='joined')
3227 pull_request = relationship('PullRequest', lazy='joined')
3227 pull_request = relationship('PullRequest', lazy='joined')
3228
3228
3229 def __unicode__(self):
3229 def __unicode__(self):
3230 return u"<%s('%s[v%s]:%s')>" % (
3230 return u"<%s('%s[v%s]:%s')>" % (
3231 self.__class__.__name__,
3231 self.__class__.__name__,
3232 self.status, self.version, self.author
3232 self.status, self.version, self.author
3233 )
3233 )
3234
3234
3235 @classmethod
3235 @classmethod
3236 def get_status_lbl(cls, value):
3236 def get_status_lbl(cls, value):
3237 return dict(cls.STATUSES).get(value)
3237 return dict(cls.STATUSES).get(value)
3238
3238
3239 @property
3239 @property
3240 def status_lbl(self):
3240 def status_lbl(self):
3241 return ChangesetStatus.get_status_lbl(self.status)
3241 return ChangesetStatus.get_status_lbl(self.status)
3242
3242
3243 def get_api_data(self):
3243 def get_api_data(self):
3244 status = self
3244 status = self
3245 data = {
3245 data = {
3246 'status_id': status.changeset_status_id,
3246 'status_id': status.changeset_status_id,
3247 'status': status.status,
3247 'status': status.status,
3248 }
3248 }
3249 return data
3249 return data
3250
3250
3251 def __json__(self):
3251 def __json__(self):
3252 data = dict()
3252 data = dict()
3253 data.update(self.get_api_data())
3253 data.update(self.get_api_data())
3254 return data
3254 return data
3255
3255
3256
3256
3257 class _PullRequestBase(BaseModel):
3257 class _PullRequestBase(BaseModel):
3258 """
3258 """
3259 Common attributes of pull request and version entries.
3259 Common attributes of pull request and version entries.
3260 """
3260 """
3261
3261
3262 # .status values
3262 # .status values
3263 STATUS_NEW = u'new'
3263 STATUS_NEW = u'new'
3264 STATUS_OPEN = u'open'
3264 STATUS_OPEN = u'open'
3265 STATUS_CLOSED = u'closed'
3265 STATUS_CLOSED = u'closed'
3266
3266
3267 title = Column('title', Unicode(255), nullable=True)
3267 title = Column('title', Unicode(255), nullable=True)
3268 description = Column(
3268 description = Column(
3269 'description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'),
3269 'description', UnicodeText().with_variant(UnicodeText(10240), 'mysql'),
3270 nullable=True)
3270 nullable=True)
3271 # new/open/closed status of pull request (not approve/reject/etc)
3271 # new/open/closed status of pull request (not approve/reject/etc)
3272 status = Column('status', Unicode(255), nullable=False, default=STATUS_NEW)
3272 status = Column('status', Unicode(255), nullable=False, default=STATUS_NEW)
3273 created_on = Column(
3273 created_on = Column(
3274 'created_on', DateTime(timezone=False), nullable=False,
3274 'created_on', DateTime(timezone=False), nullable=False,
3275 default=datetime.datetime.now)
3275 default=datetime.datetime.now)
3276 updated_on = Column(
3276 updated_on = Column(
3277 'updated_on', DateTime(timezone=False), nullable=False,
3277 'updated_on', DateTime(timezone=False), nullable=False,
3278 default=datetime.datetime.now)
3278 default=datetime.datetime.now)
3279
3279
3280 @declared_attr
3280 @declared_attr
3281 def user_id(cls):
3281 def user_id(cls):
3282 return Column(
3282 return Column(
3283 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
3283 "user_id", Integer(), ForeignKey('users.user_id'), nullable=False,
3284 unique=None)
3284 unique=None)
3285
3285
3286 # 500 revisions max
3286 # 500 revisions max
3287 _revisions = Column(
3287 _revisions = Column(
3288 'revisions', UnicodeText().with_variant(UnicodeText(20500), 'mysql'))
3288 'revisions', UnicodeText().with_variant(UnicodeText(20500), 'mysql'))
3289
3289
3290 @declared_attr
3290 @declared_attr
3291 def source_repo_id(cls):
3291 def source_repo_id(cls):
3292 # TODO: dan: rename column to source_repo_id
3292 # TODO: dan: rename column to source_repo_id
3293 return Column(
3293 return Column(
3294 'org_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3294 'org_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3295 nullable=False)
3295 nullable=False)
3296
3296
3297 source_ref = Column('org_ref', Unicode(255), nullable=False)
3297 source_ref = Column('org_ref', Unicode(255), nullable=False)
3298
3298
3299 @declared_attr
3299 @declared_attr
3300 def target_repo_id(cls):
3300 def target_repo_id(cls):
3301 # TODO: dan: rename column to target_repo_id
3301 # TODO: dan: rename column to target_repo_id
3302 return Column(
3302 return Column(
3303 'other_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3303 'other_repo_id', Integer(), ForeignKey('repositories.repo_id'),
3304 nullable=False)
3304 nullable=False)
3305
3305
3306 target_ref = Column('other_ref', Unicode(255), nullable=False)
3306 target_ref = Column('other_ref', Unicode(255), nullable=False)
3307 _shadow_merge_ref = Column('shadow_merge_ref', Unicode(255), nullable=True)
3307 _shadow_merge_ref = Column('shadow_merge_ref', Unicode(255), nullable=True)
3308
3308
3309 # TODO: dan: rename column to last_merge_source_rev
3309 # TODO: dan: rename column to last_merge_source_rev
3310 _last_merge_source_rev = Column(
3310 _last_merge_source_rev = Column(
3311 'last_merge_org_rev', String(40), nullable=True)
3311 'last_merge_org_rev', String(40), nullable=True)
3312 # TODO: dan: rename column to last_merge_target_rev
3312 # TODO: dan: rename column to last_merge_target_rev
3313 _last_merge_target_rev = Column(
3313 _last_merge_target_rev = Column(
3314 'last_merge_other_rev', String(40), nullable=True)
3314 'last_merge_other_rev', String(40), nullable=True)
3315 _last_merge_status = Column('merge_status', Integer(), nullable=True)
3315 _last_merge_status = Column('merge_status', Integer(), nullable=True)
3316 merge_rev = Column('merge_rev', String(40), nullable=True)
3316 merge_rev = Column('merge_rev', String(40), nullable=True)
3317
3317
3318 reviewer_data = Column(
3318 reviewer_data = Column(
3319 'reviewer_data_json', MutationObj.as_mutable(
3319 'reviewer_data_json', MutationObj.as_mutable(
3320 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3320 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3321
3321
3322 @property
3322 @property
3323 def reviewer_data_json(self):
3323 def reviewer_data_json(self):
3324 return json.dumps(self.reviewer_data)
3324 return json.dumps(self.reviewer_data)
3325
3325
3326 @hybrid_property
3326 @hybrid_property
3327 def description_safe(self):
3327 def description_safe(self):
3328 from rhodecode.lib import helpers as h
3328 from rhodecode.lib import helpers as h
3329 return h.escape(self.description)
3329 return h.escape(self.description)
3330
3330
3331 @hybrid_property
3331 @hybrid_property
3332 def revisions(self):
3332 def revisions(self):
3333 return self._revisions.split(':') if self._revisions else []
3333 return self._revisions.split(':') if self._revisions else []
3334
3334
3335 @revisions.setter
3335 @revisions.setter
3336 def revisions(self, val):
3336 def revisions(self, val):
3337 self._revisions = ':'.join(val)
3337 self._revisions = ':'.join(val)
3338
3338
3339 @declared_attr
3339 @declared_attr
3340 def author(cls):
3340 def author(cls):
3341 return relationship('User', lazy='joined')
3341 return relationship('User', lazy='joined')
3342
3342
3343 @declared_attr
3343 @declared_attr
3344 def source_repo(cls):
3344 def source_repo(cls):
3345 return relationship(
3345 return relationship(
3346 'Repository',
3346 'Repository',
3347 primaryjoin='%s.source_repo_id==Repository.repo_id' % cls.__name__)
3347 primaryjoin='%s.source_repo_id==Repository.repo_id' % cls.__name__)
3348
3348
3349 @property
3349 @property
3350 def source_ref_parts(self):
3350 def source_ref_parts(self):
3351 return self.unicode_to_reference(self.source_ref)
3351 return self.unicode_to_reference(self.source_ref)
3352
3352
3353 @declared_attr
3353 @declared_attr
3354 def target_repo(cls):
3354 def target_repo(cls):
3355 return relationship(
3355 return relationship(
3356 'Repository',
3356 'Repository',
3357 primaryjoin='%s.target_repo_id==Repository.repo_id' % cls.__name__)
3357 primaryjoin='%s.target_repo_id==Repository.repo_id' % cls.__name__)
3358
3358
3359 @property
3359 @property
3360 def target_ref_parts(self):
3360 def target_ref_parts(self):
3361 return self.unicode_to_reference(self.target_ref)
3361 return self.unicode_to_reference(self.target_ref)
3362
3362
3363 @property
3363 @property
3364 def shadow_merge_ref(self):
3364 def shadow_merge_ref(self):
3365 return self.unicode_to_reference(self._shadow_merge_ref)
3365 return self.unicode_to_reference(self._shadow_merge_ref)
3366
3366
3367 @shadow_merge_ref.setter
3367 @shadow_merge_ref.setter
3368 def shadow_merge_ref(self, ref):
3368 def shadow_merge_ref(self, ref):
3369 self._shadow_merge_ref = self.reference_to_unicode(ref)
3369 self._shadow_merge_ref = self.reference_to_unicode(ref)
3370
3370
3371 def unicode_to_reference(self, raw):
3371 def unicode_to_reference(self, raw):
3372 """
3372 """
3373 Convert a unicode (or string) to a reference object.
3373 Convert a unicode (or string) to a reference object.
3374 If unicode evaluates to False it returns None.
3374 If unicode evaluates to False it returns None.
3375 """
3375 """
3376 if raw:
3376 if raw:
3377 refs = raw.split(':')
3377 refs = raw.split(':')
3378 return Reference(*refs)
3378 return Reference(*refs)
3379 else:
3379 else:
3380 return None
3380 return None
3381
3381
3382 def reference_to_unicode(self, ref):
3382 def reference_to_unicode(self, ref):
3383 """
3383 """
3384 Convert a reference object to unicode.
3384 Convert a reference object to unicode.
3385 If reference is None it returns None.
3385 If reference is None it returns None.
3386 """
3386 """
3387 if ref:
3387 if ref:
3388 return u':'.join(ref)
3388 return u':'.join(ref)
3389 else:
3389 else:
3390 return None
3390 return None
3391
3391
3392 def get_api_data(self, with_merge_state=True):
3392 def get_api_data(self, with_merge_state=True):
3393 from rhodecode.model.pull_request import PullRequestModel
3393 from rhodecode.model.pull_request import PullRequestModel
3394
3394
3395 pull_request = self
3395 pull_request = self
3396 if with_merge_state:
3396 if with_merge_state:
3397 merge_status = PullRequestModel().merge_status(pull_request)
3397 merge_status = PullRequestModel().merge_status(pull_request)
3398 merge_state = {
3398 merge_state = {
3399 'status': merge_status[0],
3399 'status': merge_status[0],
3400 'message': safe_unicode(merge_status[1]),
3400 'message': safe_unicode(merge_status[1]),
3401 }
3401 }
3402 else:
3402 else:
3403 merge_state = {'status': 'not_available',
3403 merge_state = {'status': 'not_available',
3404 'message': 'not_available'}
3404 'message': 'not_available'}
3405
3405
3406 merge_data = {
3406 merge_data = {
3407 'clone_url': PullRequestModel().get_shadow_clone_url(pull_request),
3407 'clone_url': PullRequestModel().get_shadow_clone_url(pull_request),
3408 'reference': (
3408 'reference': (
3409 pull_request.shadow_merge_ref._asdict()
3409 pull_request.shadow_merge_ref._asdict()
3410 if pull_request.shadow_merge_ref else None),
3410 if pull_request.shadow_merge_ref else None),
3411 }
3411 }
3412
3412
3413 data = {
3413 data = {
3414 'pull_request_id': pull_request.pull_request_id,
3414 'pull_request_id': pull_request.pull_request_id,
3415 'url': PullRequestModel().get_url(pull_request),
3415 'url': PullRequestModel().get_url(pull_request),
3416 'title': pull_request.title,
3416 'title': pull_request.title,
3417 'description': pull_request.description,
3417 'description': pull_request.description,
3418 'status': pull_request.status,
3418 'status': pull_request.status,
3419 'created_on': pull_request.created_on,
3419 'created_on': pull_request.created_on,
3420 'updated_on': pull_request.updated_on,
3420 'updated_on': pull_request.updated_on,
3421 'commit_ids': pull_request.revisions,
3421 'commit_ids': pull_request.revisions,
3422 'review_status': pull_request.calculated_review_status(),
3422 'review_status': pull_request.calculated_review_status(),
3423 'mergeable': merge_state,
3423 'mergeable': merge_state,
3424 'source': {
3424 'source': {
3425 'clone_url': pull_request.source_repo.clone_url(),
3425 'clone_url': pull_request.source_repo.clone_url(),
3426 'repository': pull_request.source_repo.repo_name,
3426 'repository': pull_request.source_repo.repo_name,
3427 'reference': {
3427 'reference': {
3428 'name': pull_request.source_ref_parts.name,
3428 'name': pull_request.source_ref_parts.name,
3429 'type': pull_request.source_ref_parts.type,
3429 'type': pull_request.source_ref_parts.type,
3430 'commit_id': pull_request.source_ref_parts.commit_id,
3430 'commit_id': pull_request.source_ref_parts.commit_id,
3431 },
3431 },
3432 },
3432 },
3433 'target': {
3433 'target': {
3434 'clone_url': pull_request.target_repo.clone_url(),
3434 'clone_url': pull_request.target_repo.clone_url(),
3435 'repository': pull_request.target_repo.repo_name,
3435 'repository': pull_request.target_repo.repo_name,
3436 'reference': {
3436 'reference': {
3437 'name': pull_request.target_ref_parts.name,
3437 'name': pull_request.target_ref_parts.name,
3438 'type': pull_request.target_ref_parts.type,
3438 'type': pull_request.target_ref_parts.type,
3439 'commit_id': pull_request.target_ref_parts.commit_id,
3439 'commit_id': pull_request.target_ref_parts.commit_id,
3440 },
3440 },
3441 },
3441 },
3442 'merge': merge_data,
3442 'merge': merge_data,
3443 'author': pull_request.author.get_api_data(include_secrets=False,
3443 'author': pull_request.author.get_api_data(include_secrets=False,
3444 details='basic'),
3444 details='basic'),
3445 'reviewers': [
3445 'reviewers': [
3446 {
3446 {
3447 'user': reviewer.get_api_data(include_secrets=False,
3447 'user': reviewer.get_api_data(include_secrets=False,
3448 details='basic'),
3448 details='basic'),
3449 'reasons': reasons,
3449 'reasons': reasons,
3450 'review_status': st[0][1].status if st else 'not_reviewed',
3450 'review_status': st[0][1].status if st else 'not_reviewed',
3451 }
3451 }
3452 for reviewer, reasons, mandatory, st in
3452 for reviewer, reasons, mandatory, st in
3453 pull_request.reviewers_statuses()
3453 pull_request.reviewers_statuses()
3454 ]
3454 ]
3455 }
3455 }
3456
3456
3457 return data
3457 return data
3458
3458
3459
3459
3460 class PullRequest(Base, _PullRequestBase):
3460 class PullRequest(Base, _PullRequestBase):
3461 __tablename__ = 'pull_requests'
3461 __tablename__ = 'pull_requests'
3462 __table_args__ = (
3462 __table_args__ = (
3463 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3463 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3464 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3464 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3465 )
3465 )
3466
3466
3467 pull_request_id = Column(
3467 pull_request_id = Column(
3468 'pull_request_id', Integer(), nullable=False, primary_key=True)
3468 'pull_request_id', Integer(), nullable=False, primary_key=True)
3469
3469
3470 def __repr__(self):
3470 def __repr__(self):
3471 if self.pull_request_id:
3471 if self.pull_request_id:
3472 return '<DB:PullRequest #%s>' % self.pull_request_id
3472 return '<DB:PullRequest #%s>' % self.pull_request_id
3473 else:
3473 else:
3474 return '<DB:PullRequest at %#x>' % id(self)
3474 return '<DB:PullRequest at %#x>' % id(self)
3475
3475
3476 reviewers = relationship('PullRequestReviewers',
3476 reviewers = relationship('PullRequestReviewers',
3477 cascade="all, delete, delete-orphan")
3477 cascade="all, delete, delete-orphan")
3478 statuses = relationship('ChangesetStatus',
3478 statuses = relationship('ChangesetStatus',
3479 cascade="all, delete, delete-orphan")
3479 cascade="all, delete, delete-orphan")
3480 comments = relationship('ChangesetComment',
3480 comments = relationship('ChangesetComment',
3481 cascade="all, delete, delete-orphan")
3481 cascade="all, delete, delete-orphan")
3482 versions = relationship('PullRequestVersion',
3482 versions = relationship('PullRequestVersion',
3483 cascade="all, delete, delete-orphan",
3483 cascade="all, delete, delete-orphan",
3484 lazy='dynamic')
3484 lazy='dynamic')
3485
3485
3486 @classmethod
3486 @classmethod
3487 def get_pr_display_object(cls, pull_request_obj, org_pull_request_obj,
3487 def get_pr_display_object(cls, pull_request_obj, org_pull_request_obj,
3488 internal_methods=None):
3488 internal_methods=None):
3489
3489
3490 class PullRequestDisplay(object):
3490 class PullRequestDisplay(object):
3491 """
3491 """
3492 Special object wrapper for showing PullRequest data via Versions
3492 Special object wrapper for showing PullRequest data via Versions
3493 It mimics PR object as close as possible. This is read only object
3493 It mimics PR object as close as possible. This is read only object
3494 just for display
3494 just for display
3495 """
3495 """
3496
3496
3497 def __init__(self, attrs, internal=None):
3497 def __init__(self, attrs, internal=None):
3498 self.attrs = attrs
3498 self.attrs = attrs
3499 # internal have priority over the given ones via attrs
3499 # internal have priority over the given ones via attrs
3500 self.internal = internal or ['versions']
3500 self.internal = internal or ['versions']
3501
3501
3502 def __getattr__(self, item):
3502 def __getattr__(self, item):
3503 if item in self.internal:
3503 if item in self.internal:
3504 return getattr(self, item)
3504 return getattr(self, item)
3505 try:
3505 try:
3506 return self.attrs[item]
3506 return self.attrs[item]
3507 except KeyError:
3507 except KeyError:
3508 raise AttributeError(
3508 raise AttributeError(
3509 '%s object has no attribute %s' % (self, item))
3509 '%s object has no attribute %s' % (self, item))
3510
3510
3511 def __repr__(self):
3511 def __repr__(self):
3512 return '<DB:PullRequestDisplay #%s>' % self.attrs.get('pull_request_id')
3512 return '<DB:PullRequestDisplay #%s>' % self.attrs.get('pull_request_id')
3513
3513
3514 def versions(self):
3514 def versions(self):
3515 return pull_request_obj.versions.order_by(
3515 return pull_request_obj.versions.order_by(
3516 PullRequestVersion.pull_request_version_id).all()
3516 PullRequestVersion.pull_request_version_id).all()
3517
3517
3518 def is_closed(self):
3518 def is_closed(self):
3519 return pull_request_obj.is_closed()
3519 return pull_request_obj.is_closed()
3520
3520
3521 @property
3521 @property
3522 def pull_request_version_id(self):
3522 def pull_request_version_id(self):
3523 return getattr(pull_request_obj, 'pull_request_version_id', None)
3523 return getattr(pull_request_obj, 'pull_request_version_id', None)
3524
3524
3525 attrs = StrictAttributeDict(pull_request_obj.get_api_data())
3525 attrs = StrictAttributeDict(pull_request_obj.get_api_data())
3526
3526
3527 attrs.author = StrictAttributeDict(
3527 attrs.author = StrictAttributeDict(
3528 pull_request_obj.author.get_api_data())
3528 pull_request_obj.author.get_api_data())
3529 if pull_request_obj.target_repo:
3529 if pull_request_obj.target_repo:
3530 attrs.target_repo = StrictAttributeDict(
3530 attrs.target_repo = StrictAttributeDict(
3531 pull_request_obj.target_repo.get_api_data())
3531 pull_request_obj.target_repo.get_api_data())
3532 attrs.target_repo.clone_url = pull_request_obj.target_repo.clone_url
3532 attrs.target_repo.clone_url = pull_request_obj.target_repo.clone_url
3533
3533
3534 if pull_request_obj.source_repo:
3534 if pull_request_obj.source_repo:
3535 attrs.source_repo = StrictAttributeDict(
3535 attrs.source_repo = StrictAttributeDict(
3536 pull_request_obj.source_repo.get_api_data())
3536 pull_request_obj.source_repo.get_api_data())
3537 attrs.source_repo.clone_url = pull_request_obj.source_repo.clone_url
3537 attrs.source_repo.clone_url = pull_request_obj.source_repo.clone_url
3538
3538
3539 attrs.source_ref_parts = pull_request_obj.source_ref_parts
3539 attrs.source_ref_parts = pull_request_obj.source_ref_parts
3540 attrs.target_ref_parts = pull_request_obj.target_ref_parts
3540 attrs.target_ref_parts = pull_request_obj.target_ref_parts
3541 attrs.revisions = pull_request_obj.revisions
3541 attrs.revisions = pull_request_obj.revisions
3542
3542
3543 attrs.shadow_merge_ref = org_pull_request_obj.shadow_merge_ref
3543 attrs.shadow_merge_ref = org_pull_request_obj.shadow_merge_ref
3544 attrs.reviewer_data = org_pull_request_obj.reviewer_data
3544 attrs.reviewer_data = org_pull_request_obj.reviewer_data
3545 attrs.reviewer_data_json = org_pull_request_obj.reviewer_data_json
3545 attrs.reviewer_data_json = org_pull_request_obj.reviewer_data_json
3546
3546
3547 return PullRequestDisplay(attrs, internal=internal_methods)
3547 return PullRequestDisplay(attrs, internal=internal_methods)
3548
3548
3549 def is_closed(self):
3549 def is_closed(self):
3550 return self.status == self.STATUS_CLOSED
3550 return self.status == self.STATUS_CLOSED
3551
3551
3552 def __json__(self):
3552 def __json__(self):
3553 return {
3553 return {
3554 'revisions': self.revisions,
3554 'revisions': self.revisions,
3555 }
3555 }
3556
3556
3557 def calculated_review_status(self):
3557 def calculated_review_status(self):
3558 from rhodecode.model.changeset_status import ChangesetStatusModel
3558 from rhodecode.model.changeset_status import ChangesetStatusModel
3559 return ChangesetStatusModel().calculated_review_status(self)
3559 return ChangesetStatusModel().calculated_review_status(self)
3560
3560
3561 def reviewers_statuses(self):
3561 def reviewers_statuses(self):
3562 from rhodecode.model.changeset_status import ChangesetStatusModel
3562 from rhodecode.model.changeset_status import ChangesetStatusModel
3563 return ChangesetStatusModel().reviewers_statuses(self)
3563 return ChangesetStatusModel().reviewers_statuses(self)
3564
3564
3565 @property
3565 @property
3566 def workspace_id(self):
3566 def workspace_id(self):
3567 from rhodecode.model.pull_request import PullRequestModel
3567 from rhodecode.model.pull_request import PullRequestModel
3568 return PullRequestModel()._workspace_id(self)
3568 return PullRequestModel()._workspace_id(self)
3569
3569
3570 def get_shadow_repo(self):
3570 def get_shadow_repo(self):
3571 workspace_id = self.workspace_id
3571 workspace_id = self.workspace_id
3572 vcs_obj = self.target_repo.scm_instance()
3572 vcs_obj = self.target_repo.scm_instance()
3573 shadow_repository_path = vcs_obj._get_shadow_repository_path(
3573 shadow_repository_path = vcs_obj._get_shadow_repository_path(
3574 workspace_id)
3574 workspace_id)
3575 return vcs_obj._get_shadow_instance(shadow_repository_path)
3575 return vcs_obj._get_shadow_instance(shadow_repository_path)
3576
3576
3577
3577
3578 class PullRequestVersion(Base, _PullRequestBase):
3578 class PullRequestVersion(Base, _PullRequestBase):
3579 __tablename__ = 'pull_request_versions'
3579 __tablename__ = 'pull_request_versions'
3580 __table_args__ = (
3580 __table_args__ = (
3581 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3581 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3582 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3582 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3583 )
3583 )
3584
3584
3585 pull_request_version_id = Column(
3585 pull_request_version_id = Column(
3586 'pull_request_version_id', Integer(), nullable=False, primary_key=True)
3586 'pull_request_version_id', Integer(), nullable=False, primary_key=True)
3587 pull_request_id = Column(
3587 pull_request_id = Column(
3588 'pull_request_id', Integer(),
3588 'pull_request_id', Integer(),
3589 ForeignKey('pull_requests.pull_request_id'), nullable=False)
3589 ForeignKey('pull_requests.pull_request_id'), nullable=False)
3590 pull_request = relationship('PullRequest')
3590 pull_request = relationship('PullRequest')
3591
3591
3592 def __repr__(self):
3592 def __repr__(self):
3593 if self.pull_request_version_id:
3593 if self.pull_request_version_id:
3594 return '<DB:PullRequestVersion #%s>' % self.pull_request_version_id
3594 return '<DB:PullRequestVersion #%s>' % self.pull_request_version_id
3595 else:
3595 else:
3596 return '<DB:PullRequestVersion at %#x>' % id(self)
3596 return '<DB:PullRequestVersion at %#x>' % id(self)
3597
3597
3598 @property
3598 @property
3599 def reviewers(self):
3599 def reviewers(self):
3600 return self.pull_request.reviewers
3600 return self.pull_request.reviewers
3601
3601
3602 @property
3602 @property
3603 def versions(self):
3603 def versions(self):
3604 return self.pull_request.versions
3604 return self.pull_request.versions
3605
3605
3606 def is_closed(self):
3606 def is_closed(self):
3607 # calculate from original
3607 # calculate from original
3608 return self.pull_request.status == self.STATUS_CLOSED
3608 return self.pull_request.status == self.STATUS_CLOSED
3609
3609
3610 def calculated_review_status(self):
3610 def calculated_review_status(self):
3611 return self.pull_request.calculated_review_status()
3611 return self.pull_request.calculated_review_status()
3612
3612
3613 def reviewers_statuses(self):
3613 def reviewers_statuses(self):
3614 return self.pull_request.reviewers_statuses()
3614 return self.pull_request.reviewers_statuses()
3615
3615
3616
3616
3617 class PullRequestReviewers(Base, BaseModel):
3617 class PullRequestReviewers(Base, BaseModel):
3618 __tablename__ = 'pull_request_reviewers'
3618 __tablename__ = 'pull_request_reviewers'
3619 __table_args__ = (
3619 __table_args__ = (
3620 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3620 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3621 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3621 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3622 )
3622 )
3623
3623
3624 @hybrid_property
3624 @hybrid_property
3625 def reasons(self):
3625 def reasons(self):
3626 if not self._reasons:
3626 if not self._reasons:
3627 return []
3627 return []
3628 return self._reasons
3628 return self._reasons
3629
3629
3630 @reasons.setter
3630 @reasons.setter
3631 def reasons(self, val):
3631 def reasons(self, val):
3632 val = val or []
3632 val = val or []
3633 if any(not isinstance(x, basestring) for x in val):
3633 if any(not isinstance(x, basestring) for x in val):
3634 raise Exception('invalid reasons type, must be list of strings')
3634 raise Exception('invalid reasons type, must be list of strings')
3635 self._reasons = val
3635 self._reasons = val
3636
3636
3637 pull_requests_reviewers_id = Column(
3637 pull_requests_reviewers_id = Column(
3638 'pull_requests_reviewers_id', Integer(), nullable=False,
3638 'pull_requests_reviewers_id', Integer(), nullable=False,
3639 primary_key=True)
3639 primary_key=True)
3640 pull_request_id = Column(
3640 pull_request_id = Column(
3641 "pull_request_id", Integer(),
3641 "pull_request_id", Integer(),
3642 ForeignKey('pull_requests.pull_request_id'), nullable=False)
3642 ForeignKey('pull_requests.pull_request_id'), nullable=False)
3643 user_id = Column(
3643 user_id = Column(
3644 "user_id", Integer(), ForeignKey('users.user_id'), nullable=True)
3644 "user_id", Integer(), ForeignKey('users.user_id'), nullable=True)
3645 _reasons = Column(
3645 _reasons = Column(
3646 'reason', MutationList.as_mutable(
3646 'reason', MutationList.as_mutable(
3647 JsonType('list', dialect_map=dict(mysql=UnicodeText(16384)))))
3647 JsonType('list', dialect_map=dict(mysql=UnicodeText(16384)))))
3648 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3648 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3649 user = relationship('User')
3649 user = relationship('User')
3650 pull_request = relationship('PullRequest')
3650 pull_request = relationship('PullRequest')
3651
3651
3652
3652
3653 class Notification(Base, BaseModel):
3653 class Notification(Base, BaseModel):
3654 __tablename__ = 'notifications'
3654 __tablename__ = 'notifications'
3655 __table_args__ = (
3655 __table_args__ = (
3656 Index('notification_type_idx', 'type'),
3656 Index('notification_type_idx', 'type'),
3657 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3657 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3658 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3658 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
3659 )
3659 )
3660
3660
3661 TYPE_CHANGESET_COMMENT = u'cs_comment'
3661 TYPE_CHANGESET_COMMENT = u'cs_comment'
3662 TYPE_MESSAGE = u'message'
3662 TYPE_MESSAGE = u'message'
3663 TYPE_MENTION = u'mention'
3663 TYPE_MENTION = u'mention'
3664 TYPE_REGISTRATION = u'registration'
3664 TYPE_REGISTRATION = u'registration'
3665 TYPE_PULL_REQUEST = u'pull_request'
3665 TYPE_PULL_REQUEST = u'pull_request'
3666 TYPE_PULL_REQUEST_COMMENT = u'pull_request_comment'
3666 TYPE_PULL_REQUEST_COMMENT = u'pull_request_comment'
3667
3667
3668 notification_id = Column('notification_id', Integer(), nullable=False, primary_key=True)
3668 notification_id = Column('notification_id', Integer(), nullable=False, primary_key=True)
3669 subject = Column('subject', Unicode(512), nullable=True)
3669 subject = Column('subject', Unicode(512), nullable=True)
3670 body = Column('body', UnicodeText().with_variant(UnicodeText(50000), 'mysql'), nullable=True)
3670 body = Column('body', UnicodeText().with_variant(UnicodeText(50000), 'mysql'), nullable=True)
3671 created_by = Column("created_by", Integer(), ForeignKey('users.user_id'), nullable=True)
3671 created_by = Column("created_by", Integer(), ForeignKey('users.user_id'), nullable=True)
3672 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3672 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3673 type_ = Column('type', Unicode(255))
3673 type_ = Column('type', Unicode(255))
3674
3674
3675 created_by_user = relationship('User')
3675 created_by_user = relationship('User')
3676 notifications_to_users = relationship('UserNotification', lazy='joined',
3676 notifications_to_users = relationship('UserNotification', lazy='joined',
3677 cascade="all, delete, delete-orphan")
3677 cascade="all, delete, delete-orphan")
3678
3678
3679 @property
3679 @property
3680 def recipients(self):
3680 def recipients(self):
3681 return [x.user for x in UserNotification.query()\
3681 return [x.user for x in UserNotification.query()\
3682 .filter(UserNotification.notification == self)\
3682 .filter(UserNotification.notification == self)\
3683 .order_by(UserNotification.user_id.asc()).all()]
3683 .order_by(UserNotification.user_id.asc()).all()]
3684
3684
3685 @classmethod
3685 @classmethod
3686 def create(cls, created_by, subject, body, recipients, type_=None):
3686 def create(cls, created_by, subject, body, recipients, type_=None):
3687 if type_ is None:
3687 if type_ is None:
3688 type_ = Notification.TYPE_MESSAGE
3688 type_ = Notification.TYPE_MESSAGE
3689
3689
3690 notification = cls()
3690 notification = cls()
3691 notification.created_by_user = created_by
3691 notification.created_by_user = created_by
3692 notification.subject = subject
3692 notification.subject = subject
3693 notification.body = body
3693 notification.body = body
3694 notification.type_ = type_
3694 notification.type_ = type_
3695 notification.created_on = datetime.datetime.now()
3695 notification.created_on = datetime.datetime.now()
3696
3696
3697 for u in recipients:
3697 for u in recipients:
3698 assoc = UserNotification()
3698 assoc = UserNotification()
3699 assoc.notification = notification
3699 assoc.notification = notification
3700
3700
3701 # if created_by is inside recipients mark his notification
3701 # if created_by is inside recipients mark his notification
3702 # as read
3702 # as read
3703 if u.user_id == created_by.user_id:
3703 if u.user_id == created_by.user_id:
3704 assoc.read = True
3704 assoc.read = True
3705
3705
3706 u.notifications.append(assoc)
3706 u.notifications.append(assoc)
3707 Session().add(notification)
3707 Session().add(notification)
3708
3708
3709 return notification
3709 return notification
3710
3710
3711 @property
3711 @property
3712 def description(self):
3712 def description(self):
3713 from rhodecode.model.notification import NotificationModel
3713 from rhodecode.model.notification import NotificationModel
3714 return NotificationModel().make_description(self)
3714 return NotificationModel().make_description(self)
3715
3715
3716
3716
3717 class UserNotification(Base, BaseModel):
3717 class UserNotification(Base, BaseModel):
3718 __tablename__ = 'user_to_notification'
3718 __tablename__ = 'user_to_notification'
3719 __table_args__ = (
3719 __table_args__ = (
3720 UniqueConstraint('user_id', 'notification_id'),
3720 UniqueConstraint('user_id', 'notification_id'),
3721 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3721 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3722 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3722 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3723 )
3723 )
3724 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
3724 user_id = Column('user_id', Integer(), ForeignKey('users.user_id'), primary_key=True)
3725 notification_id = Column("notification_id", Integer(), ForeignKey('notifications.notification_id'), primary_key=True)
3725 notification_id = Column("notification_id", Integer(), ForeignKey('notifications.notification_id'), primary_key=True)
3726 read = Column('read', Boolean, default=False)
3726 read = Column('read', Boolean, default=False)
3727 sent_on = Column('sent_on', DateTime(timezone=False), nullable=True, unique=None)
3727 sent_on = Column('sent_on', DateTime(timezone=False), nullable=True, unique=None)
3728
3728
3729 user = relationship('User', lazy="joined")
3729 user = relationship('User', lazy="joined")
3730 notification = relationship('Notification', lazy="joined",
3730 notification = relationship('Notification', lazy="joined",
3731 order_by=lambda: Notification.created_on.desc(),)
3731 order_by=lambda: Notification.created_on.desc(),)
3732
3732
3733 def mark_as_read(self):
3733 def mark_as_read(self):
3734 self.read = True
3734 self.read = True
3735 Session().add(self)
3735 Session().add(self)
3736
3736
3737
3737
3738 class Gist(Base, BaseModel):
3738 class Gist(Base, BaseModel):
3739 __tablename__ = 'gists'
3739 __tablename__ = 'gists'
3740 __table_args__ = (
3740 __table_args__ = (
3741 Index('g_gist_access_id_idx', 'gist_access_id'),
3741 Index('g_gist_access_id_idx', 'gist_access_id'),
3742 Index('g_created_on_idx', 'created_on'),
3742 Index('g_created_on_idx', 'created_on'),
3743 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3743 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3744 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3744 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3745 )
3745 )
3746 GIST_PUBLIC = u'public'
3746 GIST_PUBLIC = u'public'
3747 GIST_PRIVATE = u'private'
3747 GIST_PRIVATE = u'private'
3748 DEFAULT_FILENAME = u'gistfile1.txt'
3748 DEFAULT_FILENAME = u'gistfile1.txt'
3749
3749
3750 ACL_LEVEL_PUBLIC = u'acl_public'
3750 ACL_LEVEL_PUBLIC = u'acl_public'
3751 ACL_LEVEL_PRIVATE = u'acl_private'
3751 ACL_LEVEL_PRIVATE = u'acl_private'
3752
3752
3753 gist_id = Column('gist_id', Integer(), primary_key=True)
3753 gist_id = Column('gist_id', Integer(), primary_key=True)
3754 gist_access_id = Column('gist_access_id', Unicode(250))
3754 gist_access_id = Column('gist_access_id', Unicode(250))
3755 gist_description = Column('gist_description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
3755 gist_description = Column('gist_description', UnicodeText().with_variant(UnicodeText(1024), 'mysql'))
3756 gist_owner = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True)
3756 gist_owner = Column('user_id', Integer(), ForeignKey('users.user_id'), nullable=True)
3757 gist_expires = Column('gist_expires', Float(53), nullable=False)
3757 gist_expires = Column('gist_expires', Float(53), nullable=False)
3758 gist_type = Column('gist_type', Unicode(128), nullable=False)
3758 gist_type = Column('gist_type', Unicode(128), nullable=False)
3759 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3759 created_on = Column('created_on', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3760 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3760 modified_at = Column('modified_at', DateTime(timezone=False), nullable=False, default=datetime.datetime.now)
3761 acl_level = Column('acl_level', Unicode(128), nullable=True)
3761 acl_level = Column('acl_level', Unicode(128), nullable=True)
3762
3762
3763 owner = relationship('User')
3763 owner = relationship('User')
3764
3764
3765 def __repr__(self):
3765 def __repr__(self):
3766 return '<Gist:[%s]%s>' % (self.gist_type, self.gist_access_id)
3766 return '<Gist:[%s]%s>' % (self.gist_type, self.gist_access_id)
3767
3767
3768 @hybrid_property
3768 @hybrid_property
3769 def description_safe(self):
3769 def description_safe(self):
3770 from rhodecode.lib import helpers as h
3770 from rhodecode.lib import helpers as h
3771 return h.escape(self.gist_description)
3771 return h.escape(self.gist_description)
3772
3772
3773 @classmethod
3773 @classmethod
3774 def get_or_404(cls, id_, pyramid_exc=False):
3774 def get_or_404(cls, id_, pyramid_exc=False):
3775
3775
3776 if pyramid_exc:
3776 if pyramid_exc:
3777 from pyramid.httpexceptions import HTTPNotFound
3777 from pyramid.httpexceptions import HTTPNotFound
3778 else:
3778 else:
3779 from webob.exc import HTTPNotFound
3779 from webob.exc import HTTPNotFound
3780
3780
3781 res = cls.query().filter(cls.gist_access_id == id_).scalar()
3781 res = cls.query().filter(cls.gist_access_id == id_).scalar()
3782 if not res:
3782 if not res:
3783 raise HTTPNotFound
3783 raise HTTPNotFound
3784 return res
3784 return res
3785
3785
3786 @classmethod
3786 @classmethod
3787 def get_by_access_id(cls, gist_access_id):
3787 def get_by_access_id(cls, gist_access_id):
3788 return cls.query().filter(cls.gist_access_id == gist_access_id).scalar()
3788 return cls.query().filter(cls.gist_access_id == gist_access_id).scalar()
3789
3789
3790 def gist_url(self):
3790 def gist_url(self):
3791 from rhodecode.model.gist import GistModel
3791 from rhodecode.model.gist import GistModel
3792 return GistModel().get_url(self)
3792 return GistModel().get_url(self)
3793
3793
3794 @classmethod
3794 @classmethod
3795 def base_path(cls):
3795 def base_path(cls):
3796 """
3796 """
3797 Returns base path when all gists are stored
3797 Returns base path when all gists are stored
3798
3798
3799 :param cls:
3799 :param cls:
3800 """
3800 """
3801 from rhodecode.model.gist import GIST_STORE_LOC
3801 from rhodecode.model.gist import GIST_STORE_LOC
3802 q = Session().query(RhodeCodeUi)\
3802 q = Session().query(RhodeCodeUi)\
3803 .filter(RhodeCodeUi.ui_key == URL_SEP)
3803 .filter(RhodeCodeUi.ui_key == URL_SEP)
3804 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
3804 q = q.options(FromCache("sql_cache_short", "repository_repo_path"))
3805 return os.path.join(q.one().ui_value, GIST_STORE_LOC)
3805 return os.path.join(q.one().ui_value, GIST_STORE_LOC)
3806
3806
3807 def get_api_data(self):
3807 def get_api_data(self):
3808 """
3808 """
3809 Common function for generating gist related data for API
3809 Common function for generating gist related data for API
3810 """
3810 """
3811 gist = self
3811 gist = self
3812 data = {
3812 data = {
3813 'gist_id': gist.gist_id,
3813 'gist_id': gist.gist_id,
3814 'type': gist.gist_type,
3814 'type': gist.gist_type,
3815 'access_id': gist.gist_access_id,
3815 'access_id': gist.gist_access_id,
3816 'description': gist.gist_description,
3816 'description': gist.gist_description,
3817 'url': gist.gist_url(),
3817 'url': gist.gist_url(),
3818 'expires': gist.gist_expires,
3818 'expires': gist.gist_expires,
3819 'created_on': gist.created_on,
3819 'created_on': gist.created_on,
3820 'modified_at': gist.modified_at,
3820 'modified_at': gist.modified_at,
3821 'content': None,
3821 'content': None,
3822 'acl_level': gist.acl_level,
3822 'acl_level': gist.acl_level,
3823 }
3823 }
3824 return data
3824 return data
3825
3825
3826 def __json__(self):
3826 def __json__(self):
3827 data = dict(
3827 data = dict(
3828 )
3828 )
3829 data.update(self.get_api_data())
3829 data.update(self.get_api_data())
3830 return data
3830 return data
3831 # SCM functions
3831 # SCM functions
3832
3832
3833 def scm_instance(self, **kwargs):
3833 def scm_instance(self, **kwargs):
3834 full_repo_path = os.path.join(self.base_path(), self.gist_access_id)
3834 full_repo_path = os.path.join(self.base_path(), self.gist_access_id)
3835 return get_vcs_instance(
3835 return get_vcs_instance(
3836 repo_path=safe_str(full_repo_path), create=False)
3836 repo_path=safe_str(full_repo_path), create=False)
3837
3837
3838
3838
3839 class ExternalIdentity(Base, BaseModel):
3839 class ExternalIdentity(Base, BaseModel):
3840 __tablename__ = 'external_identities'
3840 __tablename__ = 'external_identities'
3841 __table_args__ = (
3841 __table_args__ = (
3842 Index('local_user_id_idx', 'local_user_id'),
3842 Index('local_user_id_idx', 'local_user_id'),
3843 Index('external_id_idx', 'external_id'),
3843 Index('external_id_idx', 'external_id'),
3844 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3844 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3845 'mysql_charset': 'utf8'})
3845 'mysql_charset': 'utf8'})
3846
3846
3847 external_id = Column('external_id', Unicode(255), default=u'',
3847 external_id = Column('external_id', Unicode(255), default=u'',
3848 primary_key=True)
3848 primary_key=True)
3849 external_username = Column('external_username', Unicode(1024), default=u'')
3849 external_username = Column('external_username', Unicode(1024), default=u'')
3850 local_user_id = Column('local_user_id', Integer(),
3850 local_user_id = Column('local_user_id', Integer(),
3851 ForeignKey('users.user_id'), primary_key=True)
3851 ForeignKey('users.user_id'), primary_key=True)
3852 provider_name = Column('provider_name', Unicode(255), default=u'',
3852 provider_name = Column('provider_name', Unicode(255), default=u'',
3853 primary_key=True)
3853 primary_key=True)
3854 access_token = Column('access_token', String(1024), default=u'')
3854 access_token = Column('access_token', String(1024), default=u'')
3855 alt_token = Column('alt_token', String(1024), default=u'')
3855 alt_token = Column('alt_token', String(1024), default=u'')
3856 token_secret = Column('token_secret', String(1024), default=u'')
3856 token_secret = Column('token_secret', String(1024), default=u'')
3857
3857
3858 @classmethod
3858 @classmethod
3859 def by_external_id_and_provider(cls, external_id, provider_name,
3859 def by_external_id_and_provider(cls, external_id, provider_name,
3860 local_user_id=None):
3860 local_user_id=None):
3861 """
3861 """
3862 Returns ExternalIdentity instance based on search params
3862 Returns ExternalIdentity instance based on search params
3863
3863
3864 :param external_id:
3864 :param external_id:
3865 :param provider_name:
3865 :param provider_name:
3866 :return: ExternalIdentity
3866 :return: ExternalIdentity
3867 """
3867 """
3868 query = cls.query()
3868 query = cls.query()
3869 query = query.filter(cls.external_id == external_id)
3869 query = query.filter(cls.external_id == external_id)
3870 query = query.filter(cls.provider_name == provider_name)
3870 query = query.filter(cls.provider_name == provider_name)
3871 if local_user_id:
3871 if local_user_id:
3872 query = query.filter(cls.local_user_id == local_user_id)
3872 query = query.filter(cls.local_user_id == local_user_id)
3873 return query.first()
3873 return query.first()
3874
3874
3875 @classmethod
3875 @classmethod
3876 def user_by_external_id_and_provider(cls, external_id, provider_name):
3876 def user_by_external_id_and_provider(cls, external_id, provider_name):
3877 """
3877 """
3878 Returns User instance based on search params
3878 Returns User instance based on search params
3879
3879
3880 :param external_id:
3880 :param external_id:
3881 :param provider_name:
3881 :param provider_name:
3882 :return: User
3882 :return: User
3883 """
3883 """
3884 query = User.query()
3884 query = User.query()
3885 query = query.filter(cls.external_id == external_id)
3885 query = query.filter(cls.external_id == external_id)
3886 query = query.filter(cls.provider_name == provider_name)
3886 query = query.filter(cls.provider_name == provider_name)
3887 query = query.filter(User.user_id == cls.local_user_id)
3887 query = query.filter(User.user_id == cls.local_user_id)
3888 return query.first()
3888 return query.first()
3889
3889
3890 @classmethod
3890 @classmethod
3891 def by_local_user_id(cls, local_user_id):
3891 def by_local_user_id(cls, local_user_id):
3892 """
3892 """
3893 Returns all tokens for user
3893 Returns all tokens for user
3894
3894
3895 :param local_user_id:
3895 :param local_user_id:
3896 :return: ExternalIdentity
3896 :return: ExternalIdentity
3897 """
3897 """
3898 query = cls.query()
3898 query = cls.query()
3899 query = query.filter(cls.local_user_id == local_user_id)
3899 query = query.filter(cls.local_user_id == local_user_id)
3900 return query
3900 return query
3901
3901
3902
3902
3903 class Integration(Base, BaseModel):
3903 class Integration(Base, BaseModel):
3904 __tablename__ = 'integrations'
3904 __tablename__ = 'integrations'
3905 __table_args__ = (
3905 __table_args__ = (
3906 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3906 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3907 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3907 'mysql_charset': 'utf8', 'sqlite_autoincrement': True}
3908 )
3908 )
3909
3909
3910 integration_id = Column('integration_id', Integer(), primary_key=True)
3910 integration_id = Column('integration_id', Integer(), primary_key=True)
3911 integration_type = Column('integration_type', String(255))
3911 integration_type = Column('integration_type', String(255))
3912 enabled = Column('enabled', Boolean(), nullable=False)
3912 enabled = Column('enabled', Boolean(), nullable=False)
3913 name = Column('name', String(255), nullable=False)
3913 name = Column('name', String(255), nullable=False)
3914 child_repos_only = Column('child_repos_only', Boolean(), nullable=False,
3914 child_repos_only = Column('child_repos_only', Boolean(), nullable=False,
3915 default=False)
3915 default=False)
3916
3916
3917 settings = Column(
3917 settings = Column(
3918 'settings_json', MutationObj.as_mutable(
3918 'settings_json', MutationObj.as_mutable(
3919 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3919 JsonType(dialect_map=dict(mysql=UnicodeText(16384)))))
3920 repo_id = Column(
3920 repo_id = Column(
3921 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
3921 'repo_id', Integer(), ForeignKey('repositories.repo_id'),
3922 nullable=True, unique=None, default=None)
3922 nullable=True, unique=None, default=None)
3923 repo = relationship('Repository', lazy='joined')
3923 repo = relationship('Repository', lazy='joined')
3924
3924
3925 repo_group_id = Column(
3925 repo_group_id = Column(
3926 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
3926 'repo_group_id', Integer(), ForeignKey('groups.group_id'),
3927 nullable=True, unique=None, default=None)
3927 nullable=True, unique=None, default=None)
3928 repo_group = relationship('RepoGroup', lazy='joined')
3928 repo_group = relationship('RepoGroup', lazy='joined')
3929
3929
3930 @property
3930 @property
3931 def scope(self):
3931 def scope(self):
3932 if self.repo:
3932 if self.repo:
3933 return repr(self.repo)
3933 return repr(self.repo)
3934 if self.repo_group:
3934 if self.repo_group:
3935 if self.child_repos_only:
3935 if self.child_repos_only:
3936 return repr(self.repo_group) + ' (child repos only)'
3936 return repr(self.repo_group) + ' (child repos only)'
3937 else:
3937 else:
3938 return repr(self.repo_group) + ' (recursive)'
3938 return repr(self.repo_group) + ' (recursive)'
3939 if self.child_repos_only:
3939 if self.child_repos_only:
3940 return 'root_repos'
3940 return 'root_repos'
3941 return 'global'
3941 return 'global'
3942
3942
3943 def __repr__(self):
3943 def __repr__(self):
3944 return '<Integration(%r, %r)>' % (self.integration_type, self.scope)
3944 return '<Integration(%r, %r)>' % (self.integration_type, self.scope)
3945
3945
3946
3946
3947 class RepoReviewRuleUser(Base, BaseModel):
3947 class RepoReviewRuleUser(Base, BaseModel):
3948 __tablename__ = 'repo_review_rules_users'
3948 __tablename__ = 'repo_review_rules_users'
3949 __table_args__ = (
3949 __table_args__ = (
3950 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3950 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3951 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3951 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3952 )
3952 )
3953 repo_review_rule_user_id = Column('repo_review_rule_user_id', Integer(), primary_key=True)
3953 repo_review_rule_user_id = Column('repo_review_rule_user_id', Integer(), primary_key=True)
3954 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
3954 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
3955 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False)
3955 user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=False)
3956 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3956 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3957 user = relationship('User')
3957 user = relationship('User')
3958
3958
3959 def rule_data(self):
3959 def rule_data(self):
3960 return {
3960 return {
3961 'mandatory': self.mandatory
3961 'mandatory': self.mandatory
3962 }
3962 }
3963
3963
3964
3964
3965 class RepoReviewRuleUserGroup(Base, BaseModel):
3965 class RepoReviewRuleUserGroup(Base, BaseModel):
3966 __tablename__ = 'repo_review_rules_users_groups'
3966 __tablename__ = 'repo_review_rules_users_groups'
3967 __table_args__ = (
3967 __table_args__ = (
3968 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3968 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3969 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3969 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3970 )
3970 )
3971 repo_review_rule_users_group_id = Column('repo_review_rule_users_group_id', Integer(), primary_key=True)
3971 repo_review_rule_users_group_id = Column('repo_review_rule_users_group_id', Integer(), primary_key=True)
3972 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
3972 repo_review_rule_id = Column("repo_review_rule_id", Integer(), ForeignKey('repo_review_rules.repo_review_rule_id'))
3973 users_group_id = Column("users_group_id", Integer(),ForeignKey('users_groups.users_group_id'), nullable=False)
3973 users_group_id = Column("users_group_id", Integer(),ForeignKey('users_groups.users_group_id'), nullable=False)
3974 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3974 mandatory = Column("mandatory", Boolean(), nullable=False, default=False)
3975 users_group = relationship('UserGroup')
3975 users_group = relationship('UserGroup')
3976
3976
3977 def rule_data(self):
3977 def rule_data(self):
3978 return {
3978 return {
3979 'mandatory': self.mandatory
3979 'mandatory': self.mandatory
3980 }
3980 }
3981
3981
3982
3982
3983 class RepoReviewRule(Base, BaseModel):
3983 class RepoReviewRule(Base, BaseModel):
3984 __tablename__ = 'repo_review_rules'
3984 __tablename__ = 'repo_review_rules'
3985 __table_args__ = (
3985 __table_args__ = (
3986 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3986 {'extend_existing': True, 'mysql_engine': 'InnoDB',
3987 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3987 'mysql_charset': 'utf8', 'sqlite_autoincrement': True,}
3988 )
3988 )
3989
3989
3990 repo_review_rule_id = Column(
3990 repo_review_rule_id = Column(
3991 'repo_review_rule_id', Integer(), primary_key=True)
3991 'repo_review_rule_id', Integer(), primary_key=True)
3992 repo_id = Column(
3992 repo_id = Column(
3993 "repo_id", Integer(), ForeignKey('repositories.repo_id'))
3993 "repo_id", Integer(), ForeignKey('repositories.repo_id'))
3994 repo = relationship('Repository', backref='review_rules')
3994 repo = relationship('Repository', backref='review_rules')
3995
3995
3996 _branch_pattern = Column("branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
3996 _branch_pattern = Column("branch_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
3997 _file_pattern = Column("file_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
3997 _file_pattern = Column("file_pattern", UnicodeText().with_variant(UnicodeText(255), 'mysql'), default=u'*') # glob
3998
3998
3999 use_authors_for_review = Column("use_authors_for_review", Boolean(), nullable=False, default=False)
3999 use_authors_for_review = Column("use_authors_for_review", Boolean(), nullable=False, default=False)
4000 forbid_author_to_review = Column("forbid_author_to_review", Boolean(), nullable=False, default=False)
4000 forbid_author_to_review = Column("forbid_author_to_review", Boolean(), nullable=False, default=False)
4001 forbid_commit_author_to_review = Column("forbid_commit_author_to_review", Boolean(), nullable=False, default=False)
4001 forbid_commit_author_to_review = Column("forbid_commit_author_to_review", Boolean(), nullable=False, default=False)
4002 forbid_adding_reviewers = Column("forbid_adding_reviewers", Boolean(), nullable=False, default=False)
4002 forbid_adding_reviewers = Column("forbid_adding_reviewers", Boolean(), nullable=False, default=False)
4003
4003
4004 rule_users = relationship('RepoReviewRuleUser')
4004 rule_users = relationship('RepoReviewRuleUser')
4005 rule_user_groups = relationship('RepoReviewRuleUserGroup')
4005 rule_user_groups = relationship('RepoReviewRuleUserGroup')
4006
4006
4007 @hybrid_property
4007 @hybrid_property
4008 def branch_pattern(self):
4008 def branch_pattern(self):
4009 return self._branch_pattern or '*'
4009 return self._branch_pattern or '*'
4010
4010
4011 def _validate_glob(self, value):
4011 def _validate_glob(self, value):
4012 re.compile('^' + glob2re(value) + '$')
4012 re.compile('^' + glob2re(value) + '$')
4013
4013
4014 @branch_pattern.setter
4014 @branch_pattern.setter
4015 def branch_pattern(self, value):
4015 def branch_pattern(self, value):
4016 self._validate_glob(value)
4016 self._validate_glob(value)
4017 self._branch_pattern = value or '*'
4017 self._branch_pattern = value or '*'
4018
4018
4019 @hybrid_property
4019 @hybrid_property
4020 def file_pattern(self):
4020 def file_pattern(self):
4021 return self._file_pattern or '*'
4021 return self._file_pattern or '*'
4022
4022
4023 @file_pattern.setter
4023 @file_pattern.setter
4024 def file_pattern(self, value):
4024 def file_pattern(self, value):
4025 self._validate_glob(value)
4025 self._validate_glob(value)
4026 self._file_pattern = value or '*'
4026 self._file_pattern = value or '*'
4027
4027
4028 def matches(self, branch, files_changed):
4028 def matches(self, branch, files_changed):
4029 """
4029 """
4030 Check if this review rule matches a branch/files in a pull request
4030 Check if this review rule matches a branch/files in a pull request
4031
4031
4032 :param branch: branch name for the commit
4032 :param branch: branch name for the commit
4033 :param files_changed: list of file paths changed in the pull request
4033 :param files_changed: list of file paths changed in the pull request
4034 """
4034 """
4035
4035
4036 branch = branch or ''
4036 branch = branch or ''
4037 files_changed = files_changed or []
4037 files_changed = files_changed or []
4038
4038
4039 branch_matches = True
4039 branch_matches = True
4040 if branch:
4040 if branch:
4041 branch_regex = re.compile('^' + glob2re(self.branch_pattern) + '$')
4041 branch_regex = re.compile('^' + glob2re(self.branch_pattern) + '$')
4042 branch_matches = bool(branch_regex.search(branch))
4042 branch_matches = bool(branch_regex.search(branch))
4043
4043
4044 files_matches = True
4044 files_matches = True
4045 if self.file_pattern != '*':
4045 if self.file_pattern != '*':
4046 files_matches = False
4046 files_matches = False
4047 file_regex = re.compile(glob2re(self.file_pattern))
4047 file_regex = re.compile(glob2re(self.file_pattern))
4048 for filename in files_changed:
4048 for filename in files_changed:
4049 if file_regex.search(filename):
4049 if file_regex.search(filename):
4050 files_matches = True
4050 files_matches = True
4051 break
4051 break
4052
4052
4053 return branch_matches and files_matches
4053 return branch_matches and files_matches
4054
4054
4055 @property
4055 @property
4056 def review_users(self):
4056 def review_users(self):
4057 """ Returns the users which this rule applies to """
4057 """ Returns the users which this rule applies to """
4058
4058
4059 users = collections.OrderedDict()
4059 users = collections.OrderedDict()
4060
4060
4061 for rule_user in self.rule_users:
4061 for rule_user in self.rule_users:
4062 if rule_user.user.active:
4062 if rule_user.user.active:
4063 if rule_user.user not in users:
4063 if rule_user.user not in users:
4064 users[rule_user.user.username] = {
4064 users[rule_user.user.username] = {
4065 'user': rule_user.user,
4065 'user': rule_user.user,
4066 'source': 'user',
4066 'source': 'user',
4067 'source_data': {},
4067 'source_data': {},
4068 'data': rule_user.rule_data()
4068 'data': rule_user.rule_data()
4069 }
4069 }
4070
4070
4071 for rule_user_group in self.rule_user_groups:
4071 for rule_user_group in self.rule_user_groups:
4072 source_data = {
4072 source_data = {
4073 'name': rule_user_group.users_group.users_group_name,
4073 'name': rule_user_group.users_group.users_group_name,
4074 'members': len(rule_user_group.users_group.members)
4074 'members': len(rule_user_group.users_group.members)
4075 }
4075 }
4076 for member in rule_user_group.users_group.members:
4076 for member in rule_user_group.users_group.members:
4077 if member.user.active:
4077 if member.user.active:
4078 users[member.user.username] = {
4078 users[member.user.username] = {
4079 'user': member.user,
4079 'user': member.user,
4080 'source': 'user_group',
4080 'source': 'user_group',
4081 'source_data': source_data,
4081 'source_data': source_data,
4082 'data': rule_user_group.rule_data()
4082 'data': rule_user_group.rule_data()
4083 }
4083 }
4084
4084
4085 return users
4085 return users
4086
4086
4087 def __repr__(self):
4087 def __repr__(self):
4088 return '<RepoReviewerRule(id=%r, repo=%r)>' % (
4088 return '<RepoReviewerRule(id=%r, repo=%r)>' % (
4089 self.repo_review_rule_id, self.repo)
4089 self.repo_review_rule_id, self.repo)
4090
4090
4091
4091
4092 class DbMigrateVersion(Base, BaseModel):
4092 class DbMigrateVersion(Base, BaseModel):
4093 __tablename__ = 'db_migrate_version'
4093 __tablename__ = 'db_migrate_version'
4094 __table_args__ = (
4094 __table_args__ = (
4095 {'extend_existing': True, 'mysql_engine': 'InnoDB',
4095 {'extend_existing': True, 'mysql_engine': 'InnoDB',
4096 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
4096 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
4097 )
4097 )
4098 repository_id = Column('repository_id', String(250), primary_key=True)
4098 repository_id = Column('repository_id', String(250), primary_key=True)
4099 repository_path = Column('repository_path', Text)
4099 repository_path = Column('repository_path', Text)
4100 version = Column('version', Integer)
4100 version = Column('version', Integer)
4101
4101
4102
4102
4103 class DbSession(Base, BaseModel):
4103 class DbSession(Base, BaseModel):
4104 __tablename__ = 'db_session'
4104 __tablename__ = 'db_session'
4105 __table_args__ = (
4105 __table_args__ = (
4106 {'extend_existing': True, 'mysql_engine': 'InnoDB',
4106 {'extend_existing': True, 'mysql_engine': 'InnoDB',
4107 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
4107 'mysql_charset': 'utf8', 'sqlite_autoincrement': True},
4108 )
4108 )
4109
4109
4110 def __repr__(self):
4110 def __repr__(self):
4111 return '<DB:DbSession({})>'.format(self.id)
4111 return '<DB:DbSession({})>'.format(self.id)
4112
4112
4113 id = Column('id', Integer())
4113 id = Column('id', Integer())
4114 namespace = Column('namespace', String(255), primary_key=True)
4114 namespace = Column('namespace', String(255), primary_key=True)
4115 accessed = Column('accessed', DateTime, nullable=False)
4115 accessed = Column('accessed', DateTime, nullable=False)
4116 created = Column('created', DateTime, nullable=False)
4116 created = Column('created', DateTime, nullable=False)
4117 data = Column('data', PickleType, nullable=False)
4117 data = Column('data', PickleType, nullable=False)
General Comments 0
You need to be logged in to leave comments. Login now