Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 1 | # |
| 2 | # ex:ts=4:sw=4:sts=4:et |
| 3 | # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- |
| 4 | # |
| 5 | # BitBake Toaster Implementation |
| 6 | # |
| 7 | # Copyright (C) 2016 Intel Corporation |
| 8 | # |
| 9 | # This program is free software; you can redistribute it and/or modify |
| 10 | # it under the terms of the GNU General Public License version 2 as |
| 11 | # published by the Free Software Foundation. |
| 12 | # |
| 13 | # This program is distributed in the hope that it will be useful, |
| 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 16 | # GNU General Public License for more details. |
| 17 | # |
| 18 | # You should have received a copy of the GNU General Public License along |
| 19 | # with this program; if not, write to the Free Software Foundation, Inc., |
| 20 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| 21 | |
| 22 | from django.core.management.base import NoArgsCommand |
| 23 | |
| 24 | from orm.models import LayerSource, Layer, Release, Layer_Version |
| 25 | from orm.models import LayerVersionDependency, Machine, Recipe |
| 26 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 27 | import os |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 28 | import sys |
| 29 | |
| 30 | import json |
| 31 | import logging |
| 32 | import threading |
| 33 | import time |
| 34 | logger = logging.getLogger("toaster") |
| 35 | |
| 36 | DEFAULT_LAYERINDEX_SERVER = "http://layers.openembedded.org/layerindex/api/" |
| 37 | |
| 38 | |
| 39 | class Spinner(threading.Thread): |
| 40 | """ A simple progress spinner to indicate download/parsing is happening""" |
| 41 | def __init__(self, *args, **kwargs): |
| 42 | super(Spinner, self).__init__(*args, **kwargs) |
| 43 | self.setDaemon(True) |
| 44 | self.signal = True |
| 45 | |
| 46 | def run(self): |
| 47 | os.system('setterm -cursor off') |
| 48 | while self.signal: |
| 49 | for char in ["/", "-", "\\", "|"]: |
| 50 | sys.stdout.write("\r" + char) |
| 51 | sys.stdout.flush() |
| 52 | time.sleep(0.25) |
| 53 | os.system('setterm -cursor on') |
| 54 | |
| 55 | def stop(self): |
| 56 | self.signal = False |
| 57 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 58 | |
| 59 | class Command(NoArgsCommand): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 60 | args = "" |
| 61 | help = "Updates locally cached information from a layerindex server" |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 62 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 63 | def mini_progress(self, what, i, total): |
| 64 | i = i + 1 |
| 65 | pec = (float(i)/float(total))*100 |
| 66 | |
| 67 | sys.stdout.write("\rUpdating %s %d%%" % |
| 68 | (what, |
| 69 | pec)) |
| 70 | sys.stdout.flush() |
| 71 | if int(pec) is 100: |
| 72 | sys.stdout.write("\n") |
| 73 | sys.stdout.flush() |
| 74 | |
| 75 | def update(self): |
| 76 | """ |
| 77 | Fetches layer, recipe and machine information from a layerindex |
| 78 | server |
| 79 | """ |
| 80 | os.system('setterm -cursor off') |
| 81 | |
| 82 | self.apiurl = DEFAULT_LAYERINDEX_SERVER |
| 83 | |
| 84 | assert self.apiurl is not None |
| 85 | try: |
| 86 | from urllib.request import urlopen, URLError |
| 87 | from urllib.parse import urlparse |
| 88 | except ImportError: |
| 89 | from urllib2 import urlopen, URLError |
| 90 | from urlparse import urlparse |
| 91 | |
| 92 | proxy_settings = os.environ.get("http_proxy", None) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 93 | |
| 94 | def _get_json_response(apiurl=DEFAULT_LAYERINDEX_SERVER): |
| 95 | http_progress = Spinner() |
| 96 | http_progress.start() |
| 97 | |
| 98 | _parsedurl = urlparse(apiurl) |
| 99 | path = _parsedurl.path |
| 100 | |
| 101 | # logger.debug("Fetching %s", apiurl) |
| 102 | try: |
| 103 | res = urlopen(apiurl) |
| 104 | except URLError as e: |
| 105 | raise Exception("Failed to read %s: %s" % (path, e.reason)) |
| 106 | |
| 107 | parsed = json.loads(res.read().decode('utf-8')) |
| 108 | |
| 109 | http_progress.stop() |
| 110 | return parsed |
| 111 | |
| 112 | # verify we can get the basic api |
| 113 | try: |
| 114 | apilinks = _get_json_response() |
| 115 | except Exception as e: |
| 116 | import traceback |
| 117 | if proxy_settings is not None: |
| 118 | logger.info("EE: Using proxy %s" % proxy_settings) |
| 119 | logger.warning("EE: could not connect to %s, skipping update:" |
| 120 | "%s\n%s" % (self.apiurl, e, traceback.format_exc())) |
| 121 | return |
| 122 | |
| 123 | # update branches; only those that we already have names listed in the |
| 124 | # Releases table |
| 125 | whitelist_branch_names = [rel.branch_name |
| 126 | for rel in Release.objects.all()] |
| 127 | if len(whitelist_branch_names) == 0: |
| 128 | raise Exception("Failed to make list of branches to fetch") |
| 129 | |
| 130 | logger.info("Fetching metadata releases for %s", |
| 131 | " ".join(whitelist_branch_names)) |
| 132 | |
| 133 | branches_info = _get_json_response(apilinks['branches'] + |
| 134 | "?filter=name:%s" |
| 135 | % "OR".join(whitelist_branch_names)) |
| 136 | |
| 137 | # Map the layer index branches to toaster releases |
| 138 | li_branch_id_to_toaster_release = {} |
| 139 | |
| 140 | total = len(branches_info) |
| 141 | for i, branch in enumerate(branches_info): |
| 142 | li_branch_id_to_toaster_release[branch['id']] = \ |
| 143 | Release.objects.get(name=branch['name']) |
| 144 | self.mini_progress("Releases", i, total) |
| 145 | |
| 146 | # keep a track of the layerindex (li) id mappings so that |
| 147 | # layer_versions can be created for these layers later on |
| 148 | li_layer_id_to_toaster_layer_id = {} |
| 149 | |
| 150 | logger.info("Fetching layers") |
| 151 | |
| 152 | layers_info = _get_json_response(apilinks['layerItems']) |
| 153 | |
| 154 | total = len(layers_info) |
| 155 | for i, li in enumerate(layers_info): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 156 | try: |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 157 | l, created = Layer.objects.get_or_create(name=li['name']) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 158 | l.up_date = li['updated'] |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 159 | l.summary = li['summary'] |
| 160 | l.description = li['description'] |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 161 | |
| 162 | if created: |
| 163 | # predefined layers in the fixtures (for example poky.xml) |
| 164 | # always preempt the Layer Index for these values |
| 165 | l.vcs_url = li['vcs_url'] |
| 166 | l.vcs_web_url = li['vcs_web_url'] |
| 167 | l.vcs_web_tree_base_url = li['vcs_web_tree_base_url'] |
| 168 | l.vcs_web_file_base_url = li['vcs_web_file_base_url'] |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 169 | l.save() |
| 170 | except Layer.MultipleObjectsReturned: |
| 171 | logger.info("Skipped %s as we found multiple layers and " |
| 172 | "don't know which to update" % |
| 173 | li['name']) |
| 174 | |
| 175 | li_layer_id_to_toaster_layer_id[li['id']] = l.pk |
| 176 | |
| 177 | self.mini_progress("layers", i, total) |
| 178 | |
| 179 | # update layer_versions |
| 180 | logger.info("Fetching layer versions") |
| 181 | layerbranches_info = _get_json_response( |
| 182 | apilinks['layerBranches'] + "?filter=branch__name:%s" % |
| 183 | "OR".join(whitelist_branch_names)) |
| 184 | |
| 185 | # Map Layer index layer_branch object id to |
| 186 | # layer_version toaster object id |
| 187 | li_layer_branch_id_to_toaster_lv_id = {} |
| 188 | |
| 189 | total = len(layerbranches_info) |
| 190 | for i, lbi in enumerate(layerbranches_info): |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 191 | # release as defined by toaster map to layerindex branch |
| 192 | release = li_branch_id_to_toaster_release[lbi['branch']] |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 193 | |
| 194 | try: |
| 195 | lv, created = Layer_Version.objects.get_or_create( |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 196 | layer=Layer.objects.get( |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 197 | pk=li_layer_id_to_toaster_layer_id[lbi['layer']]), |
| 198 | release=release |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 199 | ) |
| 200 | except KeyError: |
| 201 | logger.warning( |
| 202 | "No such layerindex layer referenced by layerbranch %d" % |
| 203 | lbi['layer']) |
| 204 | continue |
| 205 | |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 206 | if created: |
| 207 | lv.release = li_branch_id_to_toaster_release[lbi['branch']] |
| 208 | lv.up_date = lbi['updated'] |
| 209 | lv.commit = lbi['actual_branch'] |
| 210 | lv.dirpath = lbi['vcs_subdir'] |
| 211 | lv.save() |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 212 | |
| 213 | li_layer_branch_id_to_toaster_lv_id[lbi['id']] =\ |
| 214 | lv.pk |
| 215 | self.mini_progress("layer versions", i, total) |
| 216 | |
| 217 | logger.info("Fetching layer version dependencies") |
| 218 | # update layer dependencies |
| 219 | layerdependencies_info = _get_json_response( |
| 220 | apilinks['layerDependencies'] + |
| 221 | "?filter=layerbranch__branch__name:%s" % |
| 222 | "OR".join(whitelist_branch_names)) |
| 223 | |
| 224 | dependlist = {} |
| 225 | for ldi in layerdependencies_info: |
| 226 | try: |
| 227 | lv = Layer_Version.objects.get( |
| 228 | pk=li_layer_branch_id_to_toaster_lv_id[ldi['layerbranch']]) |
| 229 | except Layer_Version.DoesNotExist as e: |
| 230 | continue |
| 231 | |
| 232 | if lv not in dependlist: |
| 233 | dependlist[lv] = [] |
| 234 | try: |
| 235 | layer_id = li_layer_id_to_toaster_layer_id[ldi['dependency']] |
| 236 | |
| 237 | dependlist[lv].append( |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame^] | 238 | Layer_Version.objects.get(layer__pk=layer_id, |
| 239 | release=lv.release)) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 240 | |
| 241 | except Layer_Version.DoesNotExist: |
| 242 | logger.warning("Cannot find layer version (ls:%s)," |
| 243 | "up_id:%s lv:%s" % |
| 244 | (self, ldi['dependency'], lv)) |
| 245 | |
| 246 | total = len(dependlist) |
| 247 | for i, lv in enumerate(dependlist): |
| 248 | LayerVersionDependency.objects.filter(layer_version=lv).delete() |
| 249 | for lvd in dependlist[lv]: |
| 250 | LayerVersionDependency.objects.get_or_create(layer_version=lv, |
| 251 | depends_on=lvd) |
| 252 | self.mini_progress("Layer version dependencies", i, total) |
| 253 | |
| 254 | # update machines |
| 255 | logger.info("Fetching machine information") |
| 256 | machines_info = _get_json_response( |
| 257 | apilinks['machines'] + "?filter=layerbranch__branch__name:%s" % |
| 258 | "OR".join(whitelist_branch_names)) |
| 259 | |
| 260 | total = len(machines_info) |
| 261 | for i, mi in enumerate(machines_info): |
| 262 | mo, created = Machine.objects.get_or_create( |
| 263 | name=mi['name'], |
| 264 | layer_version=Layer_Version.objects.get( |
| 265 | pk=li_layer_branch_id_to_toaster_lv_id[mi['layerbranch']])) |
| 266 | mo.up_date = mi['updated'] |
| 267 | mo.name = mi['name'] |
| 268 | mo.description = mi['description'] |
| 269 | mo.save() |
| 270 | self.mini_progress("machines", i, total) |
| 271 | |
| 272 | # update recipes; paginate by layer version / layer branch |
| 273 | logger.info("Fetching recipe information") |
| 274 | recipes_info = _get_json_response( |
| 275 | apilinks['recipes'] + "?filter=layerbranch__branch__name:%s" % |
| 276 | "OR".join(whitelist_branch_names)) |
| 277 | |
| 278 | total = len(recipes_info) |
| 279 | for i, ri in enumerate(recipes_info): |
| 280 | try: |
| 281 | lv_id = li_layer_branch_id_to_toaster_lv_id[ri['layerbranch']] |
| 282 | lv = Layer_Version.objects.get(pk=lv_id) |
| 283 | |
| 284 | ro, created = Recipe.objects.get_or_create( |
| 285 | layer_version=lv, |
| 286 | name=ri['pn'] |
| 287 | ) |
| 288 | |
| 289 | ro.layer_version = lv |
| 290 | ro.up_date = ri['updated'] |
| 291 | ro.name = ri['pn'] |
| 292 | ro.version = ri['pv'] |
| 293 | ro.summary = ri['summary'] |
| 294 | ro.description = ri['description'] |
| 295 | ro.section = ri['section'] |
| 296 | ro.license = ri['license'] |
| 297 | ro.homepage = ri['homepage'] |
| 298 | ro.bugtracker = ri['bugtracker'] |
| 299 | ro.file_path = ri['filepath'] + "/" + ri['filename'] |
| 300 | if 'inherits' in ri: |
| 301 | ro.is_image = 'image' in ri['inherits'].split() |
| 302 | else: # workaround for old style layer index |
| 303 | ro.is_image = "-image-" in ri['pn'] |
| 304 | ro.save() |
| 305 | except Exception as e: |
| 306 | logger.warning("Failed saving recipe %s", e) |
| 307 | |
| 308 | self.mini_progress("recipes", i, total) |
| 309 | |
| 310 | os.system('setterm -cursor on') |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 311 | |
| 312 | def handle_noargs(self, **options): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 313 | self.update() |