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