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

Source Code for Module coprs.logic.modules_logic

  1  import os 
  2  import time 
  3  import base64 
  4  import json 
  5  import requests 
  6  from collections import defaultdict 
  7  from sqlalchemy import and_ 
  8  from datetime import datetime 
  9  from coprs import models 
 10  from coprs import db 
 11  from coprs import exceptions 
 12  from coprs.logic import builds_logic 
 13  from wtforms import ValidationError 
 14   
 15  import gi 
 16  gi.require_version('Modulemd', '1.0') 
 17  from gi.repository import Modulemd 
18 19 20 -class ModulesLogic(object):
21 @classmethod
22 - def get(cls, module_id):
23 """ 24 Return single module identified by `module_id` 25 """ 26 return models.Module.query.filter(models.Module.id == module_id)
27 28 @classmethod
29 - def get_by_nsv(cls, copr, name, stream, version):
35 36 @classmethod
37 - def get_by_nsv_str(cls, copr, nsv):
38 name, stream, version = nsv.split("-") 39 return cls.get_by_nsv(copr, name, stream, version)
40 41 @classmethod
42 - def get_multiple(cls):
43 return models.Module.query.order_by(models.Module.id.desc())
44 45 @classmethod
46 - def get_multiple_by_copr(cls, copr):
48 49 @classmethod
50 - def yaml2modulemd(cls, yaml):
51 mmd = Modulemd.ModuleStream() 52 mmd.import_from_string(yaml) 53 return mmd
54 55 @classmethod
56 - def from_modulemd(cls, mmd):
57 yaml_b64 = base64.b64encode(mmd.dumps().encode("utf-8")).decode("utf-8") 58 return models.Module(name=mmd.get_name(), stream=mmd.get_stream(), version=mmd.get_version(), 59 summary=mmd.get_summary(), description=mmd.get_description(), yaml_b64=yaml_b64)
60 61 @classmethod
62 - def validate(cls, mmd):
63 if not all([mmd.get_name(), mmd.get_stream(), mmd.get_version()]): 64 raise ValidationError("Module should contain name, stream and version")
65 66 @classmethod
67 - def add(cls, user, copr, module):
68 if not user.can_build_in(copr): 69 raise exceptions.InsufficientRightsException("You don't have permissions to build in this copr.") 70 71 module.copr_id = copr.id 72 module.copr = copr 73 module.created_on = time.time() 74 75 db.session.add(module) 76 return module
77 78 @classmethod
79 - def set_defaults_for_optional_params(cls, mmd, filename=None):
80 mmd.set_name(mmd.get_name() or str(os.path.splitext(filename)[0])) 81 mmd.set_stream(mmd.get_stream() or "master") 82 mmd.set_version(mmd.get_version() or int(datetime.now().strftime("%Y%m%d%H%M%S")))
83
84 85 -class ModuleBuildFacade(object):
86 - def __init__(self, user, copr, yaml, filename=None):
87 self.user = user 88 self.copr = copr 89 self.yaml = yaml 90 self.filename = filename 91 92 self.modulemd = ModulesLogic.yaml2modulemd(yaml) 93 ModulesLogic.set_defaults_for_optional_params(self.modulemd, filename=filename) 94 ModulesLogic.validate(self.modulemd)
95
96 - def submit_build(self):
97 module = ModulesLogic.add(self.user, self.copr, ModulesLogic.from_modulemd(self.modulemd)) 98 self.add_builds(self.modulemd.get_rpm_components(), module) 99 return module
100 101 @classmethod
102 - def get_build_batches(cls, rpms):
103 """ 104 Determines Which component should be built in which batch. Returns an ordered list of grouped components, 105 first group of components should be built as a first batch, second as second and so on. 106 Particular components groups are represented by dicts and can by built in a random order within the batch. 107 :return: list of lists 108 """ 109 batches = defaultdict(dict) 110 for pkgname, rpm in rpms.items(): 111 batches[rpm.get_buildorder()][pkgname] = rpm 112 return [batches[number] for number in sorted(batches.keys())]
113
114 - def add_builds(self, rpms, module):
115 for group in self.get_build_batches(rpms): 116 batch = models.Batch() 117 db.session.add(batch) 118 for pkgname, rpm in group.items(): 119 clone_url = self.get_clone_url(pkgname, rpm) 120 build = builds_logic.BuildsLogic.create_new_from_scm(self.user, self.copr, scm_type="git", 121 clone_url=clone_url, committish=rpm.get_ref()) 122 build.batch = batch 123 build.batch_id = batch.id 124 build.module_id = module.id 125 db.session.add(build)
126
127 - def get_clone_url(self, pkgname, rpm):
128 if rpm.get_repository(): 129 return rpm.get_repository() 130 return self.default_distgit.format(pkgname=pkgname)
131 132 @property
133 - def default_distgit(self):
134 # @TODO move to better place 135 return "https://src.fedoraproject.org/rpms/{pkgname}"
136
137 138 -class ModulemdGenerator(object):
139 - def __init__(self, name="", stream="", version=0, summary="", config=None):
140 self.config = config 141 licenses = Modulemd.SimpleSet() 142 licenses.add("unknown") 143 self.mmd = Modulemd.ModuleStream(mdversion=1, name=name, stream=stream, version=version, summary=summary, 144 description="", content_licenses=licenses, module_licenses=licenses)
145 146 @property
147 - def nsv(self):
148 return "{}-{}-{}".format(self.mmd.get_name(), self.mmd.get_stream(), self.mmd.get_version())
149
150 - def add_api(self, packages):
151 mmd_set = Modulemd.SimpleSet() 152 for package in packages: 153 mmd_set.add(str(package)) 154 self.mmd.set_rpm_api(mmd_set)
155
156 - def add_filter(self, packages):
157 mmd_set = Modulemd.SimpleSet() 158 for package in packages: 159 mmd_set.add(str(package)) 160 self.mmd.set_rpm_filter(mmd_set)
161
162 - def add_profiles(self, profiles):
163 for i, values in profiles: 164 name, packages = values 165 profile = Modulemd.Profile(name=name) 166 for package in packages: 167 profile.add_rpm(str(package)) 168 self.mmd.add_profile(profile)
169
170 - def add_components(self, packages, filter_packages, builds):
171 build_ids = sorted(list(set([int(id) for p, id in zip(packages, builds) 172 if p in filter_packages]))) 173 for package in filter_packages: 174 build_id = builds[packages.index(package)] 175 build = builds_logic.BuildsLogic.get_by_id(build_id).first() 176 build_chroot = self._build_chroot(build) 177 buildorder = build_ids.index(int(build.id)) 178 rationale = "User selected the package as a part of the module" 179 self.add_component(package, build, build_chroot, rationale, buildorder)
180
181 - def _build_chroot(self, build):
182 chroot = None 183 for chroot in build.build_chroots: 184 if chroot.name == "custom-1-x86_64": 185 break 186 return chroot
187
188 - def add_component(self, package_name, build, chroot, rationale, buildorder=1):
189 ref = str(chroot.git_hash) if chroot else "" 190 distgit_url = self.config["DIST_GIT_URL"].replace("/cgit", "/git") 191 url = os.path.join(distgit_url, build.copr.full_name, "{}.git".format(build.package.name)) 192 component = Modulemd.ComponentRpm(name=str(package_name), rationale=rationale, 193 repository=url, ref=ref, buildorder=1) 194 self.mmd.add_rpm_component(component)
195
196 - def generate(self):
197 return self.mmd.dumps()
198
199 200 -class ModuleProvider(object):
201 - def __init__(self, filename, yaml):
202 self.filename = filename 203 self.yaml = yaml
204 205 @classmethod
206 - def from_input(cls, obj):
207 if hasattr(obj, "read"): 208 return cls.from_file(obj) 209 return cls.from_url(obj)
210 211 @classmethod
212 - def from_file(cls, ref):
213 return cls(ref.filename, ref.read())
214 215 @classmethod
216 - def from_url(cls, url):
217 if not url.endswith(".yaml"): 218 raise ValidationError("This URL doesn't point to a .yaml file") 219 220 request = requests.get(url) 221 if request.status_code != 200: 222 raise requests.RequestException("This URL seems to be wrong") 223 return cls(os.path.basename(url), request.text)
224