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