Package coprs :: Package logic :: Module coprs_logic
[hide private]
[frames] | no frames]

Source Code for Module coprs.logic.coprs_logic

  1  import os 
  2  import time 
  3  import datetime 
  4   
  5  from sqlalchemy import and_ 
  6  from sqlalchemy.sql import func 
  7  from sqlalchemy import asc, desc 
  8  from sqlalchemy.event import listen 
  9  from sqlalchemy.orm.attributes import NEVER_SET 
 10  from sqlalchemy.orm.exc import NoResultFound 
 11  from sqlalchemy.orm.attributes import get_history 
 12   
 13  from copr_common.enums import ActionTypeEnum, BackendResultEnum 
 14  from coprs import db 
 15  from coprs import exceptions 
 16  from coprs import helpers 
 17  from coprs import models 
 18  from coprs.exceptions import MalformedArgumentException, BadRequest 
 19  from coprs.logic import users_logic 
 20  from coprs.whoosheers import CoprWhoosheer 
 21  from coprs.helpers import fix_protocol_for_backend 
 22   
 23  from coprs.logic.actions_logic import ActionsLogic 
 24  from coprs.logic.users_logic import UsersLogic 
25 26 27 -class CoprsLogic(object):
28 """ 29 Used for manipulating Coprs. 30 31 All methods accept user object as a first argument, 32 as this may be needed in future. 33 """ 34 35 @classmethod
36 - def get_all(cls):
37 """ Return all coprs without those which are deleted. """ 38 query = (db.session.query(models.Copr) 39 .join(models.Copr.user) 40 .options(db.contains_eager(models.Copr.user)) 41 .filter(models.Copr.deleted == False)) 42 return query
43 44 @classmethod
45 - def get_by_id(cls, copr_id):
46 return cls.get_all().filter(models.Copr.id == copr_id)
47 48 @classmethod
49 - def attach_build(cls, query):
50 query = (query.outerjoin(models.Copr.builds) 51 .options(db.contains_eager(models.Copr.builds)) 52 .order_by(models.Build.submitted_on.desc())) 53 return query
54 55 @classmethod
56 - def attach_mock_chroots(cls, query):
57 query = (query.outerjoin(*models.Copr.mock_chroots.attr) 58 .options(db.contains_eager(*models.Copr.mock_chroots.attr)) 59 .order_by(models.MockChroot.os_release.asc()) 60 .order_by(models.MockChroot.os_version.asc()) 61 .order_by(models.MockChroot.arch.asc())) 62 return query
63 64 @classmethod
65 - def get_multiple_by_username(cls, username, **kwargs):
66 with_builds = kwargs.get("with_builds", False) 67 with_mock_chroots = kwargs.get("with_mock_chroots", False) 68 69 query = ( 70 cls.get_all() 71 .filter(models.User.username == username) 72 ) 73 74 if with_builds: 75 query = cls.attach_build(query) 76 77 if with_mock_chroots: 78 query = cls.attach_mock_chroots(query) 79 80 return query
81 82 @classmethod
83 - def get_multiple_by_group_id(cls, group_id, **kwargs):
84 with_builds = kwargs.get("with_builds", False) 85 with_mock_chroots = kwargs.get("with_mock_chroots", False) 86 87 query = ( 88 cls.get_all() 89 .filter(models.Copr.group_id == group_id) 90 ) 91 92 if with_builds: 93 query = cls.attach_build(query) 94 95 if with_mock_chroots: 96 query = cls.attach_mock_chroots(query) 97 98 return query
99 100 @classmethod
101 - def get(cls, username, coprname, **kwargs):
102 query = cls.get_multiple_by_username(username, **kwargs) 103 query = query.filter(models.Copr.name == coprname) 104 return query
105 106 @classmethod
107 - def get_by_group_id(cls, group_id, coprname, **kwargs):
108 query = cls.get_multiple_by_group_id(group_id, **kwargs) 109 query = query.filter(models.Copr.name == coprname) 110 return query
111 112 @classmethod
113 - def get_multiple(cls, include_deleted=False, include_unlisted_on_hp=True):
114 query = ( 115 db.session.query(models.Copr) 116 .join(models.Copr.user) 117 .outerjoin(models.Group) 118 .options(db.contains_eager(models.Copr.user)) 119 ) 120 121 if not include_deleted: 122 query = query.filter(models.Copr.deleted.is_(False)) 123 124 if not include_unlisted_on_hp: 125 query = query.filter(models.Copr.unlisted_on_hp.is_(False)) 126 127 return query
128 129 @classmethod
130 - def set_query_order(cls, query, desc=False):
131 if desc: 132 query = query.order_by(models.Copr.id.desc()) 133 else: 134 query = query.order_by(models.Copr.id.asc()) 135 return query
136 137 # user_relation="owned", username=username, with_mock_chroots=False 138 @classmethod
139 - def get_multiple_owned_by_username(cls, username, include_unlisted_on_hp=True):
140 query = cls.get_multiple(include_unlisted_on_hp=include_unlisted_on_hp) 141 return query.filter(models.User.username == username)
142 143 @classmethod
144 - def filter_by_name(cls, query, name):
145 return query.filter(models.Copr.name == name)
146 147 @classmethod
148 - def filter_by_user_name(cls, query, username):
149 # should be already joined with the User table 150 return query.filter(models.User.username == username)
151 152 @classmethod
153 - def filter_by_group_name(cls, query, group_name):
154 # should be already joined with the Group table 155 return query.filter(models.Group.name == group_name)
156 157 @classmethod
158 - def filter_without_group_projects(cls, query):
159 return query.filter(models.Copr.group_id.is_(None))
160 161 @classmethod
162 - def filter_without_ids(cls, query, ids):
163 return query.filter(models.Copr.id.notin_(ids))
164 165 @classmethod
166 - def join_builds(cls, query):
167 return (query.outerjoin(models.Copr.builds) 168 .options(db.contains_eager(models.Copr.builds)) 169 .order_by(models.Build.submitted_on.desc()))
170 171 @classmethod
172 - def join_mock_chroots(cls, query):
173 return (query.outerjoin(*models.Copr.mock_chroots.attr) 174 .options(db.contains_eager(*models.Copr.mock_chroots.attr)) 175 .order_by(models.MockChroot.os_release.asc()) 176 .order_by(models.MockChroot.os_version.asc()) 177 .order_by(models.MockChroot.arch.asc()))
178 179 @classmethod
180 - def get_playground(cls):
181 return cls.get_all().filter(models.Copr.playground == True)
182 183 @classmethod
184 - def set_playground(cls, user, copr):
185 if user.admin: 186 db.session.add(copr) 187 pass 188 else: 189 raise exceptions.InsufficientRightsException( 190 "User is not a system admin")
191 192 @classmethod
193 - def get_multiple_fulltext(cls, search_string):
194 query = (models.Copr.query.order_by(desc(models.Copr.created_on)) 195 .join(models.User) 196 .filter(models.Copr.deleted == False)) 197 if "/" in search_string: # copr search by its full name 198 if search_string[0] == '@': # searching for @group/project 199 group_name = "%{}%".format(search_string.split("/")[0][1:]) 200 project = "%{}%".format(search_string.split("/")[1]) 201 query = query.filter(and_(models.Group.name.ilike(group_name), 202 models.Copr.name.ilike(project), 203 models.Group.id == models.Copr.group_id)) 204 query = query.order_by(asc(func.length(models.Group.name)+func.length(models.Copr.name))) 205 else: # searching for user/project 206 user_name = "%{}%".format(search_string.split("/")[0]) 207 project = "%{}%".format(search_string.split("/")[1]) 208 query = query.filter(and_(models.User.username.ilike(user_name), 209 models.Copr.name.ilike(project), 210 models.User.id == models.Copr.user_id)) 211 query = query.order_by(asc(func.length(models.User.username)+func.length(models.Copr.name))) 212 else: # fulltext search 213 query = query.whooshee_search(search_string, whoosheer=CoprWhoosheer, order_by_relevance=100) 214 return query
215 216 @classmethod
217 - def add(cls, user, name, selected_chroots, repos=None, description=None, 218 instructions=None, check_for_duplicates=False, group=None, persistent=False, 219 auto_prune=True, use_bootstrap_container=False, follow_fedora_branching=False, **kwargs):
220 221 if not user.admin and persistent: 222 raise exceptions.NonAdminCannotCreatePersistentProject() 223 224 if not user.admin and not auto_prune: 225 raise exceptions.NonAdminCannotDisableAutoPrunning() 226 227 # form validation checks for duplicates 228 cls.new(user, name, group, check_for_duplicates=check_for_duplicates) 229 230 copr = models.Copr(name=name, 231 repos=repos or u"", 232 user=user, 233 description=description or u"", 234 instructions=instructions or u"", 235 created_on=int(time.time()), 236 persistent=persistent, 237 auto_prune=auto_prune, 238 use_bootstrap_container=use_bootstrap_container, 239 follow_fedora_branching=follow_fedora_branching, 240 **kwargs) 241 242 243 if group is not None: 244 UsersLogic.raise_if_not_in_group(user, group) 245 copr.group = group 246 247 copr_dir = models.CoprDir( 248 main=True, 249 name=name, 250 copr=copr) 251 252 db.session.add(copr_dir) 253 db.session.add(copr) 254 255 CoprChrootsLogic.new_from_names( 256 copr, selected_chroots) 257 258 db.session.flush() 259 ActionsLogic.send_create_gpg_key(copr) 260 261 return copr
262 263 @classmethod
264 - def new(cls, user, copr_name, group=None, check_for_duplicates=True):
265 if check_for_duplicates: 266 if group is None and cls.exists_for_user(user, copr_name).all(): 267 raise exceptions.DuplicateException( 268 "Copr: '{0}/{1}' already exists".format(user.name, copr_name)) 269 elif group: 270 if cls.exists_for_group(group, copr_name).all(): 271 db.session.rollback() 272 raise exceptions.DuplicateException( 273 "Copr: '@{0}/{1}' already exists".format(group.name, copr_name))
274 275 @classmethod
276 - def update(cls, user, copr):
277 # we should call get_history before other requests, otherwise 278 # the changes would be forgotten 279 if get_history(copr, "name").has_changes(): 280 raise MalformedArgumentException("Change name of the project is forbidden") 281 282 users_logic.UsersLogic.raise_if_cant_update_copr( 283 user, copr, "Only owners and admins may update their projects.") 284 285 if not user.admin and not copr.auto_prune: 286 raise exceptions.NonAdminCannotDisableAutoPrunning() 287 288 db.session.add(copr)
289 290 @classmethod
291 - def delete_unsafe(cls, user, copr):
292 """ 293 Deletes copr without termination of ongoing builds. 294 """ 295 cls.raise_if_cant_delete(user, copr) 296 # TODO: do we want to dump the information somewhere, so that we can 297 # search it in future? 298 cls.raise_if_unfinished_blocking_action( 299 copr, "Can't delete this project," 300 " another operation is in progress: {action}") 301 302 ActionsLogic.send_delete_copr(copr) 303 CoprDirsLogic.delete_all_by_copr(copr) 304 305 copr.deleted = True 306 return copr
307 308 @classmethod
309 - def exists_for_user(cls, user, coprname, incl_deleted=False):
310 existing = (models.Copr.query 311 .order_by(desc(models.Copr.created_on)) 312 .filter(models.Copr.name == coprname) 313 .filter(models.Copr.user_id == user.id)) 314 315 if not incl_deleted: 316 existing = existing.filter(models.Copr.deleted == False) 317 318 return cls.filter_without_group_projects(existing)
319 320 @classmethod
321 - def exists_for_group(cls, group, coprname, incl_deleted=False):
322 existing = (models.Copr.query 323 .order_by(desc(models.Copr.created_on)) 324 .filter(models.Copr.name == coprname) 325 .filter(models.Copr.group_id == group.id)) 326 327 if not incl_deleted: 328 existing = existing.filter(models.Copr.deleted == False) 329 330 return existing
331 332 @classmethod
333 - def unfinished_blocking_actions_for(cls, copr):
334 blocking_actions = [ActionTypeEnum("delete")] 335 336 actions = (models.Action.query 337 .filter(models.Action.object_type == "copr") 338 .filter(models.Action.object_id == copr.id) 339 .filter(models.Action.result == 340 BackendResultEnum("waiting")) 341 .filter(models.Action.action_type.in_(blocking_actions))) 342 343 return actions
344 345 @classmethod
346 - def get_yum_repos(cls, copr, empty=False):
347 repos = {} 348 release_tmpl = "{chroot.os_release}-{chroot.os_version}-{chroot.arch}" 349 build = models.Build.query.filter(models.Build.copr_id == copr.id).first() 350 if build or empty: 351 for chroot in copr.active_chroots: 352 release = release_tmpl.format(chroot=chroot) 353 repos[release] = fix_protocol_for_backend( 354 os.path.join(copr.repo_url, release + '/')) 355 return repos
356 357 @classmethod
358 - def raise_if_unfinished_blocking_action(cls, copr, message):
359 """ 360 Raise ActionInProgressException if given copr has an unfinished 361 action. Return None otherwise. 362 """ 363 364 unfinished_actions = cls.unfinished_blocking_actions_for(copr).all() 365 if unfinished_actions: 366 raise exceptions.ActionInProgressException( 367 message, unfinished_actions[0])
368 369 @classmethod
370 - def raise_if_cant_delete(cls, user, copr):
371 """ 372 Raise InsufficientRightsException if given copr cant be deleted 373 by given user. Return None otherwise. 374 """ 375 if user.admin: 376 return 377 378 if copr.group: 379 return UsersLogic.raise_if_not_in_group(user, copr.group) 380 381 if user == copr.user: 382 return 383 384 raise exceptions.InsufficientRightsException( 385 "Only owners may delete their projects.")
386
387 388 -class CoprPermissionsLogic(object):
389 @classmethod
390 - def get(cls, copr, searched_user):
391 query = (models.CoprPermission.query 392 .filter(models.CoprPermission.copr == copr) 393 .filter(models.CoprPermission.user == searched_user)) 394 395 return query
396 397 @classmethod
398 - def get_for_copr(cls, copr):
399 query = models.CoprPermission.query.filter( 400 models.CoprPermission.copr == copr) 401 402 return query
403 404 @classmethod
405 - def get_admins_for_copr(cls, copr):
406 permissions = cls.get_for_copr(copr) 407 return [copr.user] + [p.user for p in permissions if p.copr_admin == helpers.PermissionEnum("approved")]
408 409 @classmethod
410 - def new(cls, copr_permission):
411 db.session.add(copr_permission)
412 413 @classmethod
414 - def update_permissions(cls, user, copr, copr_permission, 415 new_builder, new_admin):
416 417 users_logic.UsersLogic.raise_if_cant_update_copr( 418 user, copr, "Only owners and admins may update" 419 " their projects permissions.") 420 421 (models.CoprPermission.query 422 .filter(models.CoprPermission.copr_id == copr.id) 423 .filter(models.CoprPermission.user_id == copr_permission.user_id) 424 .update({"copr_builder": new_builder, 425 "copr_admin": new_admin}))
426 427 @classmethod
428 - def update_permissions_by_applier(cls, user, copr, copr_permission, new_builder, new_admin):
429 if copr_permission: 430 # preserve approved permissions if set 431 if (not new_builder or 432 copr_permission.copr_builder != helpers.PermissionEnum("approved")): 433 434 copr_permission.copr_builder = new_builder 435 436 if (not new_admin or 437 copr_permission.copr_admin != helpers.PermissionEnum("approved")): 438 439 copr_permission.copr_admin = new_admin 440 else: 441 perm = models.CoprPermission( 442 user=user, 443 copr=copr, 444 copr_builder=new_builder, 445 copr_admin=new_admin) 446 447 cls.new(perm)
448 449 @classmethod
450 - def delete(cls, copr_permission):
451 db.session.delete(copr_permission)
452 453 @classmethod
454 - def validate_permission(cls, user, copr, permission, state):
455 allowed = ['admin', 'builder'] 456 if permission not in allowed: 457 raise BadRequest( 458 "invalid permission '{0}', allowed {1}".format(permission, 459 '|'.join(allowed))) 460 461 allowed = helpers.PermissionEnum.vals.keys() 462 if state not in allowed: 463 raise BadRequest( 464 "invalid '{0}' permission state '{1}', " 465 "use {2}".format(permission, state, '|'.join(allowed))) 466 467 if user.id == copr.user_id: 468 raise BadRequest("user '{0}' is owner of the '{1}' " 469 "project".format(user.name, copr.full_name))
470 471 @classmethod
472 - def set_permissions(cls, request_user, copr, user, permission, state):
473 users_logic.UsersLogic.raise_if_cant_update_copr( 474 request_user, copr, 475 "only owners and admins may update their projects permissions.") 476 477 cls.validate_permission(user, copr, permission, state) 478 479 perm_o = models.CoprPermission(user_id=user.id, copr_id=copr.id) 480 perm_o = db.session.merge(perm_o) 481 old_state = perm_o.get_permission(permission) 482 483 new_state = helpers.PermissionEnum(state) 484 perm_o.set_permission(permission, new_state) 485 db.session.merge(perm_o) 486 487 return (old_state, new_state) if old_state != new_state else None
488 489 @classmethod
490 - def request_permission(cls, copr, user, permission, req_bool):
491 approved = helpers.PermissionEnum('approved') 492 state = None 493 if req_bool is True: 494 state = 'request' 495 elif req_bool is False: 496 state = 'nothing' 497 else: 498 raise BadRequest("invalid '{0}' permission request '{1}', " 499 "expected True or False".format(permission, 500 req_bool)) 501 502 cls.validate_permission(user, copr, permission, state) 503 perm_o = models.CoprPermission(user_id=user.id, copr_id=copr.id) 504 perm_o = db.session.merge(perm_o) 505 old_state = perm_o.get_permission(permission) 506 if old_state == approved and state == 'request': 507 raise BadRequest("You already are '{0}' in '{1}'".format( 508 permission, copr.full_name)) 509 510 new_state = helpers.PermissionEnum(state) 511 perm_o.set_permission(permission, new_state) 512 513 if old_state != new_state: 514 return (old_state, new_state) 515 return None
516
517 518 -class CoprDirsLogic(object):
519 @classmethod
520 - def get_or_create(cls, copr, dirname, main=False):
521 copr_dir = cls.get_by_copr(copr, dirname).first() 522 523 if copr_dir: 524 return copr_dir 525 526 copr_dir = models.CoprDir( 527 name=dirname, copr=copr, main=main) 528 529 db.session.add(copr_dir) 530 return copr_dir
531 532 @classmethod
533 - def get_by_copr(cls, copr, dirname):
534 return (db.session.query(models.CoprDir) 535 .join(models.Copr) 536 .filter(models.Copr.id==copr.id) 537 .filter(models.CoprDir.name==dirname))
538 539 @classmethod
540 - def get_by_ownername(cls, ownername, dirname):
541 return (db.session.query(models.CoprDir) 542 .filter(models.CoprDir.name==dirname) 543 .filter(models.CoprDir.ownername==ownername))
544 545 @classmethod
546 - def delete(cls, copr_dir):
547 db.session.delete(copr_dir)
548 549 @classmethod
550 - def delete_all_by_copr(cls, copr):
551 for copr_dir in copr.dirs: 552 db.session.delete(copr_dir)
553
554 555 -def on_auto_createrepo_change(target_copr, value_acr, old_value_acr, initiator):
556 """ Emit createrepo action when auto_createrepo re-enabled""" 557 if old_value_acr == NEVER_SET: 558 # created new copr, not interesting 559 return 560 if not old_value_acr and value_acr: 561 # re-enabled 562 ActionsLogic.send_createrepo(target_copr)
563 564 565 listen(models.Copr.auto_createrepo, 'set', on_auto_createrepo_change, 566 active_history=True, retval=False)
567 568 569 -class BranchesLogic(object):
570 @classmethod
571 - def get_or_create(cls, name, session=db.session):
572 item = session.query(models.DistGitBranch).filter_by(name=name).first() 573 if item: 574 return item 575 576 branch = models.DistGitBranch() 577 branch.name = name 578 session.add(branch) 579 return branch
580
581 582 -class CoprChrootsLogic(object):
583 @classmethod
584 - def get_multiple(cls, include_deleted=False):
585 query = models.CoprChroot.query.join(models.Copr) 586 if not include_deleted: 587 query = query.filter(models.Copr.deleted.is_(False)) 588 return query
589 590 @classmethod
591 - def mock_chroots_from_names(cls, names):
592 593 db_chroots = models.MockChroot.query.all() 594 mock_chroots = [] 595 for ch in db_chroots: 596 if ch.name in names: 597 mock_chroots.append(ch) 598 599 return mock_chroots
600 601 @classmethod
602 - def get_by_name(cls, copr, chroot_name, active_only=True):
603 mc = MockChrootsLogic.get_from_name(chroot_name, active_only=active_only).one() 604 query = ( 605 models.CoprChroot.query.join(models.MockChroot) 606 .filter(models.CoprChroot.copr_id == copr.id) 607 .filter(models.MockChroot.id == mc.id) 608 ) 609 return query
610 611 @classmethod
612 - def get_by_name_safe(cls, copr, chroot_name):
613 """ 614 :rtype: models.CoprChroot 615 """ 616 try: 617 return cls.get_by_name(copr, chroot_name).one() 618 except NoResultFound: 619 return None
620 621 @classmethod
622 - def new(cls, mock_chroot):
623 db.session.add(mock_chroot)
624 625 @classmethod
626 - def new_from_names(cls, copr, names):
632 633 @classmethod
634 - def create_chroot(cls, user, copr, mock_chroot, buildroot_pkgs=None, repos=None, comps=None, comps_name=None, 635 with_opts="", without_opts="", 636 delete_after=None, delete_notify=None):
637 """ 638 :type user: models.User 639 :type mock_chroot: models.MockChroot 640 """ 641 if buildroot_pkgs is None: 642 buildroot_pkgs = "" 643 if repos is None: 644 repos = "" 645 UsersLogic.raise_if_cant_update_copr( 646 user, copr, 647 "Only owners and admins may update their projects.") 648 649 chroot = models.CoprChroot(copr=copr, mock_chroot=mock_chroot) 650 cls._update_chroot(buildroot_pkgs, repos, comps, comps_name, chroot, 651 with_opts, without_opts, delete_after, delete_notify) 652 return chroot
653 654 @classmethod
655 - def update_chroot(cls, user, copr_chroot, buildroot_pkgs=None, repos=None, comps=None, comps_name=None, 656 with_opts="", without_opts="", delete_after=None, delete_notify=None):
657 """ 658 :type user: models.User 659 :type copr_chroot: models.CoprChroot 660 """ 661 UsersLogic.raise_if_cant_update_copr( 662 user, copr_chroot.copr, 663 "Only owners and admins may update their projects.") 664 665 cls._update_chroot(buildroot_pkgs, repos, comps, comps_name, 666 copr_chroot, with_opts, without_opts, delete_after, delete_notify) 667 return copr_chroot
668 669 @classmethod
670 - def _update_chroot(cls, buildroot_pkgs, repos, comps, comps_name, 671 copr_chroot, with_opts, without_opts, delete_after, delete_notify):
672 if buildroot_pkgs is not None: 673 copr_chroot.buildroot_pkgs = buildroot_pkgs 674 675 if repos is not None: 676 copr_chroot.repos = repos.replace("\n", " ") 677 678 if with_opts is not None: 679 copr_chroot.with_opts = with_opts 680 681 if without_opts is not None: 682 copr_chroot.without_opts = without_opts 683 684 if comps_name is not None: 685 copr_chroot.update_comps(comps) 686 copr_chroot.comps_name = comps_name 687 ActionsLogic.send_update_comps(copr_chroot) 688 689 if delete_after is not None: 690 copr_chroot.delete_after = delete_after 691 692 if delete_notify is not None: 693 copr_chroot.delete_notify = delete_notify 694 695 db.session.add(copr_chroot)
696 697 @classmethod
698 - def update_from_names(cls, user, copr, names):
699 UsersLogic.raise_if_cant_update_copr( 700 user, copr, 701 "Only owners and admins may update their projects.") 702 current_chroots = copr.mock_chroots 703 new_chroots = cls.mock_chroots_from_names(names) 704 # add non-existing 705 run_createrepo = False 706 for mock_chroot in new_chroots: 707 if mock_chroot not in current_chroots: 708 db.session.add( 709 models.CoprChroot(copr=copr, mock_chroot=mock_chroot)) 710 run_createrepo = True 711 712 if run_createrepo: 713 ActionsLogic.send_createrepo(copr) 714 715 # delete no more present 716 to_remove = [] 717 for mock_chroot in current_chroots: 718 if mock_chroot in new_chroots: 719 continue 720 if not mock_chroot.is_active: 721 continue 722 # can't delete here, it would change current_chroots and break 723 # iteration 724 to_remove.append(mock_chroot) 725 726 for mc in to_remove: 727 copr.mock_chroots.remove(mc)
728 729 @classmethod
730 - def remove_comps(cls, user, copr_chroot):
731 UsersLogic.raise_if_cant_update_copr( 732 user, copr_chroot.copr, 733 "Only owners and admins may update their projects.") 734 735 copr_chroot.comps_name = None 736 copr_chroot.comps_zlib = None 737 ActionsLogic.send_update_comps(copr_chroot) 738 db.session.add(copr_chroot)
739 740 @classmethod
741 - def remove_copr_chroot(cls, user, copr_chroot):
742 """ 743 :param models.CoprChroot chroot: 744 """ 745 UsersLogic.raise_if_cant_update_copr( 746 user, copr_chroot.copr, 747 "Only owners and admins may update their projects.") 748 749 db.session.delete(copr_chroot)
750 751 @classmethod
752 - def filter_outdated(cls, query):
753 return query.filter(models.CoprChroot.delete_after >= datetime.datetime.now())
754 755 @classmethod
756 - def filter_outdated_to_be_deleted(cls, query):
757 return query.filter(models.CoprChroot.delete_after < datetime.datetime.now())
758
759 760 -class MockChrootsLogic(object):
761 @classmethod
762 - def get(cls, os_release, os_version, arch, active_only=False, noarch=False):
772 773 @classmethod
774 - def get_from_name(cls, chroot_name, active_only=False, noarch=False):
775 """ 776 chroot_name should be os-version-architecture, e.g. fedora-rawhide-x86_64 777 the architecture could be optional with noarch=True 778 779 Return MockChroot object for textual representation of chroot 780 """ 781 782 name_tuple = cls.tuple_from_name(chroot_name, noarch=noarch) 783 return cls.get(name_tuple[0], name_tuple[1], name_tuple[2], 784 active_only=active_only, noarch=noarch)
785 786 @classmethod
787 - def get_multiple(cls, active_only=False):
788 query = models.MockChroot.query 789 if active_only: 790 query = query.filter(models.MockChroot.is_active == True) 791 return query
792 793 @classmethod
794 - def add(cls, name):
795 name_tuple = cls.tuple_from_name(name) 796 if cls.get(*name_tuple).first(): 797 raise exceptions.DuplicateException( 798 "Mock chroot with this name already exists.") 799 new_chroot = models.MockChroot(os_release=name_tuple[0], 800 os_version=name_tuple[1], 801 arch=name_tuple[2]) 802 cls.new(new_chroot) 803 return new_chroot
804 805 @classmethod
806 - def active_names(cls):
807 return [ch.name for ch in cls.get_multiple(active_only=True).all()]
808 809 @classmethod
810 - def new(cls, mock_chroot):
811 db.session.add(mock_chroot)
812 813 @classmethod
814 - def edit_by_name(cls, name, is_active):
815 name_tuple = cls.tuple_from_name(name) 816 mock_chroot = cls.get(*name_tuple).first() 817 if not mock_chroot: 818 raise exceptions.NotFoundException( 819 "Mock chroot with this name doesn't exist.") 820 821 mock_chroot.is_active = is_active 822 cls.update(mock_chroot) 823 return mock_chroot
824 825 @classmethod
826 - def update(cls, mock_chroot):
827 db.session.add(mock_chroot)
828 829 @classmethod
830 - def delete_by_name(cls, name):
831 name_tuple = cls.tuple_from_name(name) 832 mock_chroot = cls.get(*name_tuple).first() 833 if not mock_chroot: 834 raise exceptions.NotFoundException( 835 "Mock chroot with this name doesn't exist.") 836 837 cls.delete(mock_chroot)
838 839 @classmethod
840 - def delete(cls, mock_chroot):
841 db.session.delete(mock_chroot)
842 843 @classmethod
844 - def tuple_from_name(cls, name, noarch=False):
845 """ 846 input should be os-version-architecture, e.g. fedora-rawhide-x86_64 847 848 the architecture could be optional with noarch=True 849 850 returns ("os", "version", "arch") or ("os", "version", None) 851 """ 852 split_name = name.rsplit("-", 1) if noarch else name.rsplit("-", 2) 853 854 valid = False 855 if noarch and len(split_name) in [2, 3]: 856 valid = True 857 if not noarch and len(split_name) == 3: 858 valid = True 859 860 if not valid: 861 raise MalformedArgumentException( 862 "Chroot name is not valid") 863 864 if noarch and len(split_name) == 2: 865 split_name.append(None) 866 867 return tuple(split_name)
868 869 @classmethod
870 - def prunerepo_finished(cls, chroots_pruned):
871 for chroot_name in chroots_pruned: 872 chroot = cls.get_from_name(chroot_name).one() 873 if not chroot.is_active: 874 chroot.final_prunerepo_done = True 875 876 db.session.commit() 877 return True
878 879 @classmethod
881 query = models.MockChroot.query 882 chroots = {} 883 for chroot in query: 884 if chroot.is_active or not chroot.final_prunerepo_done: 885 chroots[chroot.name] = chroot.is_active 886 887 return chroots
888
889 890 -class PinnedCoprsLogic(object):
891 892 @classmethod
893 - def get_all(cls):
894 return db.session.query(models.PinnedCoprs).order_by(models.PinnedCoprs.position)
895 896 @classmethod
897 - def get_by_id(cls, pin_id):
898 return cls.get_all().filter(models.PinnedCoprs.id == pin_id)
899 900 @classmethod
901 - def get_by_owner(cls, owner):
902 if isinstance(owner, models.Group): 903 return cls.get_by_group_id(owner.id) 904 return cls.get_by_user_id(owner.id)
905 906 @classmethod
907 - def get_by_user_id(cls, user_id):
909 910 @classmethod
911 - def get_by_group_id(cls, group_id):
913 914 @classmethod
915 - def add(cls, owner, copr_id, position):
916 kwargs = dict(copr_id=copr_id, position=position) 917 kwargs["group_id" if isinstance(owner, models.Group) else "user_id"] = owner.id 918 pin = models.PinnedCoprs(**kwargs) 919 db.session.add(pin)
920 921 @classmethod
922 - def delete_by_owner(cls, owner):
923 query = db.session.query(models.PinnedCoprs) 924 if isinstance(owner, models.Group): 925 return query.filter(models.PinnedCoprs.group_id == owner.id).delete() 926 return query.filter(models.PinnedCoprs.user_id == owner.id).delete()
927