diff options
Diffstat (limited to 'bitbake')
-rw-r--r-- | bitbake/lib/toaster/toastergui/api.py | 469 | ||||
-rw-r--r-- | bitbake/lib/toaster/toastergui/urls.py | 18 | ||||
-rwxr-xr-x | bitbake/lib/toaster/toastergui/views.py | 447 |
3 files changed, 482 insertions, 452 deletions
diff --git a/bitbake/lib/toaster/toastergui/api.py b/bitbake/lib/toaster/toastergui/api.py index 09fb02b8fc..be18090daf 100644 --- a/bitbake/lib/toaster/toastergui/api.py +++ b/bitbake/lib/toaster/toastergui/api.py | |||
@@ -16,21 +16,29 @@ | |||
16 | # with this program; if not, write to the Free Software Foundation, Inc., | 16 | # with this program; if not, write to the Free Software Foundation, Inc., |
17 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | 17 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
18 | 18 | ||
19 | # Please run flake8 on this file before sending patches | ||
19 | 20 | ||
20 | # Temporary home for the UI's misc API | ||
21 | import re | 21 | import re |
22 | import logging | ||
22 | 23 | ||
23 | from orm.models import Project, ProjectTarget, Build, Layer_Version | 24 | from orm.models import Project, ProjectTarget, Build, Layer_Version |
24 | from orm.models import LayerVersionDependency, LayerSource, ProjectLayer | 25 | from orm.models import LayerVersionDependency, LayerSource, ProjectLayer |
26 | from orm.models import Recipe, CustomImageRecipe, CustomImagePackage | ||
27 | from orm.models import Layer, Target, Package, Package_Dependency | ||
25 | from bldcontrol.models import BuildRequest | 28 | from bldcontrol.models import BuildRequest |
26 | from bldcontrol import bbcontroller | 29 | from bldcontrol import bbcontroller |
30 | |||
27 | from django.http import HttpResponse, JsonResponse | 31 | from django.http import HttpResponse, JsonResponse |
28 | from django.views.generic import View | 32 | from django.views.generic import View |
29 | from django.core.urlresolvers import reverse | 33 | from django.core.urlresolvers import reverse |
30 | from django.core import serializers | ||
31 | from django.utils import timezone | 34 | from django.utils import timezone |
32 | from django.template.defaultfilters import date | 35 | from django.db.models import Q, F |
36 | from django.db import Error | ||
33 | from toastergui.templatetags.projecttags import json, sectohms, get_tasks | 37 | from toastergui.templatetags.projecttags import json, sectohms, get_tasks |
38 | from toastergui.templatetags.projecttags import filtered_filesizeformat | ||
39 | |||
40 | logger = logging.getLogger("toaster") | ||
41 | |||
34 | 42 | ||
35 | def error_response(error): | 43 | def error_response(error): |
36 | return JsonResponse({"error": error}) | 44 | return JsonResponse({"error": error}) |
@@ -216,6 +224,7 @@ class XhrLayer(View): | |||
216 | "redirect": reverse('project', args=(kwargs['pid'],)) | 224 | "redirect": reverse('project', args=(kwargs['pid'],)) |
217 | }) | 225 | }) |
218 | 226 | ||
227 | |||
219 | class MostRecentBuildsView(View): | 228 | class MostRecentBuildsView(View): |
220 | def _was_yesterday_or_earlier(self, completed_on): | 229 | def _was_yesterday_or_earlier(self, completed_on): |
221 | now = timezone.now() | 230 | now = timezone.now() |
@@ -230,13 +239,11 @@ class MostRecentBuildsView(View): | |||
230 | """ | 239 | """ |
231 | Returns a list of builds in JSON format. | 240 | Returns a list of builds in JSON format. |
232 | """ | 241 | """ |
233 | mrb_type = 'all' | ||
234 | project = None | 242 | project = None |
235 | 243 | ||
236 | project_id = request.GET.get('project_id', None) | 244 | project_id = request.GET.get('project_id', None) |
237 | if project_id: | 245 | if project_id: |
238 | try: | 246 | try: |
239 | mrb_type = 'project' | ||
240 | project = Project.objects.get(pk=project_id) | 247 | project = Project.objects.get(pk=project_id) |
241 | except: | 248 | except: |
242 | # if project lookup fails, assume no project | 249 | # if project lookup fails, assume no project |
@@ -245,9 +252,6 @@ class MostRecentBuildsView(View): | |||
245 | recent_build_objs = Build.get_recent(project) | 252 | recent_build_objs = Build.get_recent(project) |
246 | recent_builds = [] | 253 | recent_builds = [] |
247 | 254 | ||
248 | # for timezone conversion | ||
249 | tz = timezone.get_current_timezone() | ||
250 | |||
251 | for build_obj in recent_build_objs: | 255 | for build_obj in recent_build_objs: |
252 | dashboard_url = reverse('builddashboard', args=(build_obj.pk,)) | 256 | dashboard_url = reverse('builddashboard', args=(build_obj.pk,)) |
253 | buildtime_url = reverse('buildtime', args=(build_obj.pk,)) | 257 | buildtime_url = reverse('buildtime', args=(build_obj.pk,)) |
@@ -266,7 +270,8 @@ class MostRecentBuildsView(View): | |||
266 | build['buildrequest_id'] = buildrequest_id | 270 | build['buildrequest_id'] = buildrequest_id |
267 | 271 | ||
268 | build['recipes_parsed_percentage'] = \ | 272 | build['recipes_parsed_percentage'] = \ |
269 | int((build_obj.recipes_parsed / build_obj.recipes_to_parse) * 100) | 273 | int((build_obj.recipes_parsed / |
274 | build_obj.recipes_to_parse) * 100) | ||
270 | 275 | ||
271 | tasks_complete_percentage = 0 | 276 | tasks_complete_percentage = 0 |
272 | if build_obj.outcome in (Build.SUCCEEDED, Build.FAILED): | 277 | if build_obj.outcome in (Build.SUCCEEDED, Build.FAILED): |
@@ -300,7 +305,8 @@ class MostRecentBuildsView(View): | |||
300 | completed_on_template = '%H:%M' | 305 | completed_on_template = '%H:%M' |
301 | if self._was_yesterday_or_earlier(completed_on): | 306 | if self._was_yesterday_or_earlier(completed_on): |
302 | completed_on_template = '%d/%m/%Y ' + completed_on_template | 307 | completed_on_template = '%d/%m/%Y ' + completed_on_template |
303 | build['completed_on'] = completed_on.strftime(completed_on_template) | 308 | build['completed_on'] = completed_on.strftime( |
309 | completed_on_template) | ||
304 | 310 | ||
305 | targets = [] | 311 | targets = [] |
306 | target_objs = build_obj.get_sorted_target_list() | 312 | target_objs = build_obj.get_sorted_target_list() |
@@ -323,3 +329,446 @@ class MostRecentBuildsView(View): | |||
323 | recent_builds.append(build) | 329 | recent_builds.append(build) |
324 | 330 | ||
325 | return JsonResponse(recent_builds, safe=False) | 331 | return JsonResponse(recent_builds, safe=False) |
332 | |||
333 | |||
334 | class XhrCustomRecipe(View): | ||
335 | """ Create a custom image recipe """ | ||
336 | |||
337 | def post(self, request, *args, **kwargs): | ||
338 | """ | ||
339 | Custom image recipe REST API | ||
340 | |||
341 | Entry point: /xhr_customrecipe/ | ||
342 | Method: POST | ||
343 | |||
344 | Args: | ||
345 | name: name of custom recipe to create | ||
346 | project: target project id of orm.models.Project | ||
347 | base: base recipe id of orm.models.Recipe | ||
348 | |||
349 | Returns: | ||
350 | {"error": "ok", | ||
351 | "url": <url of the created recipe>} | ||
352 | or | ||
353 | {"error": <error message>} | ||
354 | """ | ||
355 | # check if request has all required parameters | ||
356 | for param in ('name', 'project', 'base'): | ||
357 | if param not in request.POST: | ||
358 | return error_response("Missing parameter '%s'" % param) | ||
359 | |||
360 | # get project and baserecipe objects | ||
361 | params = {} | ||
362 | for name, model in [("project", Project), | ||
363 | ("base", Recipe)]: | ||
364 | value = request.POST[name] | ||
365 | try: | ||
366 | params[name] = model.objects.get(id=value) | ||
367 | except model.DoesNotExist: | ||
368 | return error_response("Invalid %s id %s" % (name, value)) | ||
369 | |||
370 | # create custom recipe | ||
371 | try: | ||
372 | |||
373 | # Only allowed chars in name are a-z, 0-9 and - | ||
374 | if re.search(r'[^a-z|0-9|-]', request.POST["name"]): | ||
375 | return error_response("invalid-name") | ||
376 | |||
377 | custom_images = CustomImageRecipe.objects.all() | ||
378 | |||
379 | # Are there any recipes with this name already in our project? | ||
380 | existing_image_recipes_in_project = custom_images.filter( | ||
381 | name=request.POST["name"], project=params["project"]) | ||
382 | |||
383 | if existing_image_recipes_in_project.count() > 0: | ||
384 | return error_response("image-already-exists") | ||
385 | |||
386 | # Are there any recipes with this name which aren't custom | ||
387 | # image recipes? | ||
388 | custom_image_ids = custom_images.values_list('id', flat=True) | ||
389 | existing_non_image_recipes = Recipe.objects.filter( | ||
390 | Q(name=request.POST["name"]) & ~Q(pk__in=custom_image_ids) | ||
391 | ) | ||
392 | |||
393 | if existing_non_image_recipes.count() > 0: | ||
394 | return error_response("recipe-already-exists") | ||
395 | |||
396 | # create layer 'Custom layer' and verion if needed | ||
397 | layer = Layer.objects.get_or_create( | ||
398 | name=CustomImageRecipe.LAYER_NAME, | ||
399 | summary="Layer for custom recipes", | ||
400 | vcs_url="file:///toaster_created_layer")[0] | ||
401 | |||
402 | # Check if we have a layer version already | ||
403 | # We don't use get_or_create here because the dirpath will change | ||
404 | # and is a required field | ||
405 | lver = Layer_Version.objects.filter(Q(project=params['project']) & | ||
406 | Q(layer=layer) & | ||
407 | Q(build=None)).last() | ||
408 | if lver is None: | ||
409 | lver, created = Layer_Version.objects.get_or_create( | ||
410 | project=params['project'], | ||
411 | layer=layer, | ||
412 | dirpath="toaster_created_layer") | ||
413 | |||
414 | # Add a dependency on our layer to the base recipe's layer | ||
415 | LayerVersionDependency.objects.get_or_create( | ||
416 | layer_version=lver, | ||
417 | depends_on=params["base"].layer_version) | ||
418 | |||
419 | # Add it to our current project if needed | ||
420 | ProjectLayer.objects.get_or_create(project=params['project'], | ||
421 | layercommit=lver, | ||
422 | optional=False) | ||
423 | |||
424 | # Create the actual recipe | ||
425 | recipe, created = CustomImageRecipe.objects.get_or_create( | ||
426 | name=request.POST["name"], | ||
427 | base_recipe=params["base"], | ||
428 | project=params["project"], | ||
429 | layer_version=lver, | ||
430 | is_image=True) | ||
431 | |||
432 | # If we created the object then setup these fields. They may get | ||
433 | # overwritten later on and cause the get_or_create to create a | ||
434 | # duplicate if they've changed. | ||
435 | if created: | ||
436 | recipe.file_path = request.POST["name"] | ||
437 | recipe.license = "MIT" | ||
438 | recipe.version = "0.1" | ||
439 | recipe.save() | ||
440 | |||
441 | except Error as err: | ||
442 | return error_response("Can't create custom recipe: %s" % err) | ||
443 | |||
444 | # Find the package list from the last build of this recipe/target | ||
445 | target = Target.objects.filter(Q(build__outcome=Build.SUCCEEDED) & | ||
446 | Q(build__project=params['project']) & | ||
447 | (Q(target=params['base'].name) | | ||
448 | Q(target=recipe.name))).last() | ||
449 | if target: | ||
450 | # Copy in every package | ||
451 | # We don't want these packages to be linked to anything because | ||
452 | # that underlying data may change e.g. delete a build | ||
453 | for tpackage in target.target_installed_package_set.all(): | ||
454 | try: | ||
455 | built_package = tpackage.package | ||
456 | # The package had no recipe information so is a ghost | ||
457 | # package skip it | ||
458 | if built_package.recipe is None: | ||
459 | continue | ||
460 | |||
461 | config_package = CustomImagePackage.objects.get( | ||
462 | name=built_package.name) | ||
463 | |||
464 | recipe.includes_set.add(config_package) | ||
465 | except Exception as e: | ||
466 | logger.warning("Error adding package %s %s" % | ||
467 | (tpackage.package.name, e)) | ||
468 | pass | ||
469 | |||
470 | return JsonResponse( | ||
471 | {"error": "ok", | ||
472 | "packages": recipe.get_all_packages().count(), | ||
473 | "url": reverse('customrecipe', args=(params['project'].pk, | ||
474 | recipe.id))}) | ||
475 | |||
476 | |||
477 | class XhrCustomRecipeId(View): | ||
478 | """ | ||
479 | Set of ReST API processors working with recipe id. | ||
480 | |||
481 | Entry point: /xhr_customrecipe/<recipe_id> | ||
482 | |||
483 | Methods: | ||
484 | GET - Get details of custom image recipe | ||
485 | DELETE - Delete custom image recipe | ||
486 | |||
487 | Returns: | ||
488 | GET: | ||
489 | {"error": "ok", | ||
490 | "info": dictionary of field name -> value pairs | ||
491 | of the CustomImageRecipe model} | ||
492 | DELETE: | ||
493 | {"error": "ok"} | ||
494 | or | ||
495 | {"error": <error message>} | ||
496 | """ | ||
497 | @staticmethod | ||
498 | def _get_ci_recipe(recipe_id): | ||
499 | """ Get Custom Image recipe or return an error response""" | ||
500 | try: | ||
501 | custom_recipe = \ | ||
502 | CustomImageRecipe.objects.get(pk=recipe_id) | ||
503 | return custom_recipe, None | ||
504 | |||
505 | except CustomImageRecipe.DoesNotExist: | ||
506 | return None, error_response("Custom recipe with id=%s " | ||
507 | "not found" % recipe_id) | ||
508 | |||
509 | def get(self, request, *args, **kwargs): | ||
510 | custom_recipe, error = self._get_ci_recipe(kwargs['recipe_id']) | ||
511 | if error: | ||
512 | return error | ||
513 | |||
514 | if request.method == 'GET': | ||
515 | info = {"id": custom_recipe.id, | ||
516 | "name": custom_recipe.name, | ||
517 | "base_recipe_id": custom_recipe.base_recipe.id, | ||
518 | "project_id": custom_recipe.project.id} | ||
519 | |||
520 | return JsonResponse({"error": "ok", "info": info}) | ||
521 | |||
522 | def delete(self, request, *args, **kwargs): | ||
523 | custom_recipe, error = self._get_ci_recipe(kwargs['recipe_id']) | ||
524 | if error: | ||
525 | return error | ||
526 | |||
527 | custom_recipe.delete() | ||
528 | return JsonResponse({"error": "ok"}) | ||
529 | |||
530 | |||
531 | class XhrCustomRecipePackages(View): | ||
532 | """ | ||
533 | ReST API to add/remove packages to/from custom recipe. | ||
534 | |||
535 | Entry point: /xhr_customrecipe/<recipe_id>/packages/<package_id> | ||
536 | Methods: | ||
537 | PUT - Add package to the recipe | ||
538 | DELETE - Delete package from the recipe | ||
539 | GET - Get package information | ||
540 | |||
541 | Returns: | ||
542 | {"error": "ok"} | ||
543 | or | ||
544 | {"error": <error message>} | ||
545 | """ | ||
546 | @staticmethod | ||
547 | def _get_package(package_id): | ||
548 | try: | ||
549 | package = CustomImagePackage.objects.get(pk=package_id) | ||
550 | return package, None | ||
551 | except Package.DoesNotExist: | ||
552 | return None, error_response("Package with id=%s " | ||
553 | "not found" % package_id) | ||
554 | |||
555 | def _traverse_dependents(self, next_package_id, | ||
556 | rev_deps, all_current_packages, tree_level=0): | ||
557 | """ | ||
558 | Recurse through reverse dependency tree for next_package_id. | ||
559 | Limit the reverse dependency search to packages not already scanned, | ||
560 | that is, not already in rev_deps. | ||
561 | Limit the scan to a depth (tree_level) not exceeding the count of | ||
562 | all packages in the custom image, and if that depth is exceeded | ||
563 | return False, pop out of the recursion, and write a warning | ||
564 | to the log, but this is unlikely, suggesting a dependency loop | ||
565 | not caught by bitbake. | ||
566 | On return, the input/output arg rev_deps is appended with queryset | ||
567 | dictionary elements, annotated for use in the customimage template. | ||
568 | The list has unsorted, but unique elements. | ||
569 | """ | ||
570 | max_dependency_tree_depth = all_current_packages.count() | ||
571 | if tree_level >= max_dependency_tree_depth: | ||
572 | logger.warning( | ||
573 | "The number of reverse dependencies " | ||
574 | "for this package exceeds " + max_dependency_tree_depth + | ||
575 | " and the remaining reverse dependencies will not be removed") | ||
576 | return True | ||
577 | |||
578 | package = CustomImagePackage.objects.get(id=next_package_id) | ||
579 | dependents = \ | ||
580 | package.package_dependencies_target.annotate( | ||
581 | name=F('package__name'), | ||
582 | pk=F('package__pk'), | ||
583 | size=F('package__size'), | ||
584 | ).values("name", "pk", "size").exclude( | ||
585 | ~Q(pk__in=all_current_packages) | ||
586 | ) | ||
587 | |||
588 | for pkg in dependents: | ||
589 | if pkg in rev_deps: | ||
590 | # already seen, skip dependent search | ||
591 | continue | ||
592 | |||
593 | rev_deps.append(pkg) | ||
594 | if (self._traverse_dependents(pkg["pk"], rev_deps, | ||
595 | all_current_packages, | ||
596 | tree_level+1)): | ||
597 | return True | ||
598 | |||
599 | return False | ||
600 | |||
601 | def _get_all_dependents(self, package_id, all_current_packages): | ||
602 | """ | ||
603 | Returns sorted list of recursive reverse dependencies for package_id, | ||
604 | as a list of dictionary items, by recursing through dependency | ||
605 | relationships. | ||
606 | """ | ||
607 | rev_deps = [] | ||
608 | self._traverse_dependents(package_id, rev_deps, all_current_packages) | ||
609 | rev_deps = sorted(rev_deps, key=lambda x: x["name"]) | ||
610 | return rev_deps | ||
611 | |||
612 | def get(self, request, *args, **kwargs): | ||
613 | recipe, error = XhrCustomRecipeId._get_ci_recipe( | ||
614 | kwargs['recipe_id']) | ||
615 | if error: | ||
616 | return error | ||
617 | |||
618 | # If no package_id then list all the current packages | ||
619 | if not kwargs['package_id']: | ||
620 | total_size = 0 | ||
621 | packages = recipe.get_all_packages().values("id", | ||
622 | "name", | ||
623 | "version", | ||
624 | "size") | ||
625 | for package in packages: | ||
626 | package['size_formatted'] = \ | ||
627 | filtered_filesizeformat(package['size']) | ||
628 | total_size += package['size'] | ||
629 | |||
630 | return JsonResponse({"error": "ok", | ||
631 | "packages": list(packages), | ||
632 | "total": len(packages), | ||
633 | "total_size": total_size, | ||
634 | "total_size_formatted": | ||
635 | filtered_filesizeformat(total_size)}) | ||
636 | else: | ||
637 | package, error = XhrCustomRecipePackages._get_package( | ||
638 | kwargs['package_id']) | ||
639 | if error: | ||
640 | return error | ||
641 | |||
642 | all_current_packages = recipe.get_all_packages() | ||
643 | |||
644 | # Dependencies for package which aren't satisfied by the | ||
645 | # current packages in the custom image recipe | ||
646 | deps = package.package_dependencies_source.for_target_or_none( | ||
647 | recipe.name)['packages'].annotate( | ||
648 | name=F('depends_on__name'), | ||
649 | pk=F('depends_on__pk'), | ||
650 | size=F('depends_on__size'), | ||
651 | ).values("name", "pk", "size").filter( | ||
652 | # There are two depends types we don't know why | ||
653 | (Q(dep_type=Package_Dependency.TYPE_TRDEPENDS) | | ||
654 | Q(dep_type=Package_Dependency.TYPE_RDEPENDS)) & | ||
655 | ~Q(pk__in=all_current_packages) | ||
656 | ) | ||
657 | |||
658 | # Reverse dependencies which are needed by packages that are | ||
659 | # in the image. Recursive search providing all dependents, | ||
660 | # not just immediate dependents. | ||
661 | reverse_deps = self._get_all_dependents(kwargs['package_id'], | ||
662 | all_current_packages) | ||
663 | total_size_deps = 0 | ||
664 | total_size_reverse_deps = 0 | ||
665 | |||
666 | for dep in deps: | ||
667 | dep['size_formatted'] = \ | ||
668 | filtered_filesizeformat(dep['size']) | ||
669 | total_size_deps += dep['size'] | ||
670 | |||
671 | for dep in reverse_deps: | ||
672 | dep['size_formatted'] = \ | ||
673 | filtered_filesizeformat(dep['size']) | ||
674 | total_size_reverse_deps += dep['size'] | ||
675 | |||
676 | return JsonResponse( | ||
677 | {"error": "ok", | ||
678 | "id": package.pk, | ||
679 | "name": package.name, | ||
680 | "version": package.version, | ||
681 | "unsatisfied_dependencies": list(deps), | ||
682 | "unsatisfied_dependencies_size": total_size_deps, | ||
683 | "unsatisfied_dependencies_size_formatted": | ||
684 | filtered_filesizeformat(total_size_deps), | ||
685 | "reverse_dependencies": list(reverse_deps), | ||
686 | "reverse_dependencies_size": total_size_reverse_deps, | ||
687 | "reverse_dependencies_size_formatted": | ||
688 | filtered_filesizeformat(total_size_reverse_deps)}) | ||
689 | |||
690 | def put(self, request, *args, **kwargs): | ||
691 | recipe, error = XhrCustomRecipeId._get_ci_recipe(kwargs['recipe_id']) | ||
692 | package, error = self._get_package(kwargs['package_id']) | ||
693 | if error: | ||
694 | return error | ||
695 | |||
696 | included_packages = recipe.includes_set.values_list('pk', | ||
697 | flat=True) | ||
698 | |||
699 | # If we're adding back a package which used to be included in this | ||
700 | # image all we need to do is remove it from the excludes | ||
701 | if package.pk in included_packages: | ||
702 | try: | ||
703 | recipe.excludes_set.remove(package) | ||
704 | return {"error": "ok"} | ||
705 | except Package.DoesNotExist: | ||
706 | return error_response("Package %s not found in excludes" | ||
707 | " but was in included list" % | ||
708 | package.name) | ||
709 | |||
710 | else: | ||
711 | recipe.appends_set.add(package) | ||
712 | # Make sure that package is not in the excludes set | ||
713 | try: | ||
714 | recipe.excludes_set.remove(package) | ||
715 | except: | ||
716 | pass | ||
717 | # Add the dependencies we think will be added to the recipe | ||
718 | # as a result of appending this package. | ||
719 | # TODO this should recurse down the entire deps tree | ||
720 | for dep in package.package_dependencies_source.all_depends(): | ||
721 | try: | ||
722 | cust_package = CustomImagePackage.objects.get( | ||
723 | name=dep.depends_on.name) | ||
724 | |||
725 | recipe.includes_set.add(cust_package) | ||
726 | try: | ||
727 | # When adding the pre-requisite package, make | ||
728 | # sure it's not in the excluded list from a | ||
729 | # prior removal. | ||
730 | recipe.excludes_set.remove(cust_package) | ||
731 | except package.DoesNotExist: | ||
732 | # Don't care if the package had never been excluded | ||
733 | pass | ||
734 | except: | ||
735 | logger.warning("Could not add package's suggested" | ||
736 | "dependencies to the list") | ||
737 | return JsonResponse({"error": "ok"}) | ||
738 | |||
739 | def delete(self, request, *args, **kwargs): | ||
740 | recipe, error = XhrCustomRecipeId._get_ci_recipe(kwargs['recipe_id']) | ||
741 | package, error = self._get_package(kwargs['package_id']) | ||
742 | if error: | ||
743 | return error | ||
744 | |||
745 | try: | ||
746 | included_packages = recipe.includes_set.values_list('pk', | ||
747 | flat=True) | ||
748 | # If we're deleting a package which is included we need to | ||
749 | # Add it to the excludes list. | ||
750 | if package.pk in included_packages: | ||
751 | recipe.excludes_set.add(package) | ||
752 | else: | ||
753 | recipe.appends_set.remove(package) | ||
754 | all_current_packages = recipe.get_all_packages() | ||
755 | |||
756 | reverse_deps_dictlist = self._get_all_dependents( | ||
757 | package.pk, | ||
758 | all_current_packages) | ||
759 | |||
760 | ids = [entry['pk'] for entry in reverse_deps_dictlist] | ||
761 | reverse_deps = CustomImagePackage.objects.filter(id__in=ids) | ||
762 | for r in reverse_deps: | ||
763 | try: | ||
764 | if r.id in included_packages: | ||
765 | recipe.excludes_set.add(r) | ||
766 | else: | ||
767 | recipe.appends_set.remove(r) | ||
768 | except: | ||
769 | pass | ||
770 | |||
771 | return JsonResponse({"error": "ok"}) | ||
772 | except CustomImageRecipe.DoesNotExist: | ||
773 | return error_response("Tried to remove package that wasn't" | ||
774 | " present") | ||
diff --git a/bitbake/lib/toaster/toastergui/urls.py b/bitbake/lib/toaster/toastergui/urls.py index 9892d2ab93..9509cd5928 100644 --- a/bitbake/lib/toaster/toastergui/urls.py +++ b/bitbake/lib/toaster/toastergui/urls.py | |||
@@ -199,19 +199,25 @@ urlpatterns = patterns('toastergui.views', | |||
199 | url(r'^js-unit-tests/$', 'jsunittests', name='js-unit-tests'), | 199 | url(r'^js-unit-tests/$', 'jsunittests', name='js-unit-tests'), |
200 | 200 | ||
201 | # image customisation functionality | 201 | # image customisation functionality |
202 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)/packages/(?P<package_id>\d+|)$', | 202 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)' |
203 | 'xhr_customrecipe_packages', name='xhr_customrecipe_packages'), | 203 | '/packages/(?P<package_id>\d+|)$', |
204 | api.XhrCustomRecipePackages.as_view(), | ||
205 | name='xhr_customrecipe_packages'), | ||
204 | 206 | ||
205 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)/packages/$', | 207 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)/packages/$', |
206 | 'xhr_customrecipe_packages', name='xhr_customrecipe_packages'), | 208 | api.XhrCustomRecipePackages.as_view(), |
209 | name='xhr_customrecipe_packages'), | ||
207 | 210 | ||
208 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)$', 'xhr_customrecipe_id', | 211 | url(r'^xhr_customrecipe/(?P<recipe_id>\d+)$', |
212 | api.XhrCustomRecipeId.as_view(), | ||
209 | name='xhr_customrecipe_id'), | 213 | name='xhr_customrecipe_id'), |
210 | url(r'^xhr_customrecipe/', 'xhr_customrecipe', | 214 | |
215 | url(r'^xhr_customrecipe/', | ||
216 | api.XhrCustomRecipe.as_view(), | ||
211 | name='xhr_customrecipe'), | 217 | name='xhr_customrecipe'), |
212 | 218 | ||
213 | url(r'^xhr_buildrequest/project/(?P<pid>\d+)$', | 219 | url(r'^xhr_buildrequest/project/(?P<pid>\d+)$', |
214 | api.XhrBuildRequest.as_view(), | 220 | api.XhrBuildRequest.as_view(), |
215 | name='xhr_buildrequest'), | 221 | name='xhr_buildrequest'), |
216 | 222 | ||
217 | url(r'^mostrecentbuilds$', api.MostRecentBuildsView.as_view(), | 223 | url(r'^mostrecentbuilds$', api.MostRecentBuildsView.as_view(), |
diff --git a/bitbake/lib/toaster/toastergui/views.py b/bitbake/lib/toaster/toastergui/views.py index 940ea255fd..365a1e88ff 100755 --- a/bitbake/lib/toaster/toastergui/views.py +++ b/bitbake/lib/toaster/toastergui/views.py | |||
@@ -19,43 +19,37 @@ | |||
19 | # with this program; if not, write to the Free Software Foundation, Inc., | 19 | # with this program; if not, write to the Free Software Foundation, Inc., |
20 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | 20 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
21 | 21 | ||
22 | # pylint: disable=method-hidden | ||
23 | # Gives E:848, 4: An attribute defined in json.encoder line 162 hides this method (method-hidden) | ||
24 | # which is an invalid warning | ||
25 | 22 | ||
26 | import operator,re | 23 | import re |
27 | 24 | ||
28 | from django.db.models import F, Q, Sum, Count, Max | 25 | from django.db.models import F, Q, Sum |
29 | from django.db import IntegrityError, Error | 26 | from django.db import IntegrityError |
30 | from django.shortcuts import render, redirect, get_object_or_404 | 27 | from django.shortcuts import render, redirect, get_object_or_404 |
31 | from orm.models import Build, Target, Task, Layer, Layer_Version, Recipe, LogMessage, Variable | 28 | from orm.models import Build, Target, Task, Layer, Layer_Version, Recipe |
32 | from orm.models import Task_Dependency, Recipe_Dependency, Package, Package_File, Package_Dependency | 29 | from orm.models import LogMessage, Variable, Package_Dependency, Package |
33 | from orm.models import Target_Installed_Package, Target_File, Target_Image_File, CustomImagePackage | 30 | from orm.models import Task_Dependency, Package_File |
34 | from orm.models import TargetKernelFile, TargetSDKFile | 31 | from orm.models import Target_Installed_Package, Target_File |
32 | from orm.models import TargetKernelFile, TargetSDKFile, Target_Image_File | ||
35 | from orm.models import BitbakeVersion, CustomImageRecipe | 33 | from orm.models import BitbakeVersion, CustomImageRecipe |
36 | from bldcontrol import bbcontroller | 34 | |
37 | from django.views.decorators.cache import cache_control | ||
38 | from django.core.urlresolvers import reverse, resolve | 35 | from django.core.urlresolvers import reverse, resolve |
39 | from django.core.exceptions import MultipleObjectsReturned, ObjectDoesNotExist | 36 | from django.core.exceptions import MultipleObjectsReturned, ObjectDoesNotExist |
40 | from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger | 37 | from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger |
41 | from django.http import HttpResponseBadRequest, HttpResponseNotFound | 38 | from django.http import HttpResponseNotFound |
42 | from django.utils import timezone | 39 | from django.utils import timezone |
43 | from django.utils.html import escape | ||
44 | from datetime import timedelta, datetime | 40 | from datetime import timedelta, datetime |
45 | from django.utils import formats | ||
46 | from toastergui.templatetags.projecttags import json as jsonfilter | 41 | from toastergui.templatetags.projecttags import json as jsonfilter |
47 | from decimal import Decimal | 42 | from decimal import Decimal |
48 | import json | 43 | import json |
49 | import os | 44 | import os |
50 | from os.path import dirname | 45 | from os.path import dirname |
51 | from functools import wraps | ||
52 | import itertools | ||
53 | import mimetypes | 46 | import mimetypes |
54 | 47 | ||
55 | import logging | 48 | import logging |
56 | 49 | ||
57 | logger = logging.getLogger("toaster") | 50 | logger = logging.getLogger("toaster") |
58 | 51 | ||
52 | |||
59 | class MimeTypeFinder(object): | 53 | class MimeTypeFinder(object): |
60 | # setting this to False enables additional non-standard mimetypes | 54 | # setting this to False enables additional non-standard mimetypes |
61 | # to be included in the guess | 55 | # to be included in the guess |
@@ -1498,18 +1492,6 @@ if True: | |||
1498 | 1492 | ||
1499 | return context | 1493 | return context |
1500 | 1494 | ||
1501 | def xhr_response(fun): | ||
1502 | """ | ||
1503 | Decorator for REST methods. | ||
1504 | calls jsonfilter on the returned dictionary and returns result | ||
1505 | as HttpResponse object of content_type application/json | ||
1506 | """ | ||
1507 | @wraps(fun) | ||
1508 | def wrapper(*args, **kwds): | ||
1509 | return HttpResponse(jsonfilter(fun(*args, **kwds)), | ||
1510 | content_type="application/json") | ||
1511 | return wrapper | ||
1512 | |||
1513 | def jsunittests(request): | 1495 | def jsunittests(request): |
1514 | """ Provides a page for the js unit tests """ | 1496 | """ Provides a page for the js unit tests """ |
1515 | bbv = BitbakeVersion.objects.filter(branch="master").first() | 1497 | bbv = BitbakeVersion.objects.filter(branch="master").first() |
@@ -1767,187 +1749,6 @@ if True: | |||
1767 | 1749 | ||
1768 | return HttpResponse(jsonfilter(json_response), content_type = "application/json") | 1750 | return HttpResponse(jsonfilter(json_response), content_type = "application/json") |
1769 | 1751 | ||
1770 | @xhr_response | ||
1771 | def xhr_customrecipe(request): | ||
1772 | """ | ||
1773 | Custom image recipe REST API | ||
1774 | |||
1775 | Entry point: /xhr_customrecipe/ | ||
1776 | Method: POST | ||
1777 | |||
1778 | Args: | ||
1779 | name: name of custom recipe to create | ||
1780 | project: target project id of orm.models.Project | ||
1781 | base: base recipe id of orm.models.Recipe | ||
1782 | |||
1783 | Returns: | ||
1784 | {"error": "ok", | ||
1785 | "url": <url of the created recipe>} | ||
1786 | or | ||
1787 | {"error": <error message>} | ||
1788 | """ | ||
1789 | # check if request has all required parameters | ||
1790 | for param in ('name', 'project', 'base'): | ||
1791 | if param not in request.POST: | ||
1792 | return {"error": "Missing parameter '%s'" % param} | ||
1793 | |||
1794 | # get project and baserecipe objects | ||
1795 | params = {} | ||
1796 | for name, model in [("project", Project), | ||
1797 | ("base", Recipe)]: | ||
1798 | value = request.POST[name] | ||
1799 | try: | ||
1800 | params[name] = model.objects.get(id=value) | ||
1801 | except model.DoesNotExist: | ||
1802 | return {"error": "Invalid %s id %s" % (name, value)} | ||
1803 | |||
1804 | # create custom recipe | ||
1805 | try: | ||
1806 | |||
1807 | # Only allowed chars in name are a-z, 0-9 and - | ||
1808 | if re.search(r'[^a-z|0-9|-]', request.POST["name"]): | ||
1809 | return {"error": "invalid-name"} | ||
1810 | |||
1811 | custom_images = CustomImageRecipe.objects.all() | ||
1812 | |||
1813 | # Are there any recipes with this name already in our project? | ||
1814 | existing_image_recipes_in_project = custom_images.filter( | ||
1815 | name=request.POST["name"], project=params["project"]) | ||
1816 | |||
1817 | if existing_image_recipes_in_project.count() > 0: | ||
1818 | return {"error": "image-already-exists"} | ||
1819 | |||
1820 | # Are there any recipes with this name which aren't custom | ||
1821 | # image recipes? | ||
1822 | custom_image_ids = custom_images.values_list('id', flat=True) | ||
1823 | existing_non_image_recipes = Recipe.objects.filter( | ||
1824 | Q(name=request.POST["name"]) & ~Q(pk__in=custom_image_ids) | ||
1825 | ) | ||
1826 | |||
1827 | if existing_non_image_recipes.count() > 0: | ||
1828 | return {"error": "recipe-already-exists"} | ||
1829 | |||
1830 | # create layer 'Custom layer' and verion if needed | ||
1831 | layer = Layer.objects.get_or_create( | ||
1832 | name=CustomImageRecipe.LAYER_NAME, | ||
1833 | summary="Layer for custom recipes", | ||
1834 | vcs_url="file:///toaster_created_layer")[0] | ||
1835 | |||
1836 | # Check if we have a layer version already | ||
1837 | # We don't use get_or_create here because the dirpath will change | ||
1838 | # and is a required field | ||
1839 | lver = Layer_Version.objects.filter(Q(project=params['project']) & | ||
1840 | Q(layer=layer) & | ||
1841 | Q(build=None)).last() | ||
1842 | if lver == None: | ||
1843 | lver, created = Layer_Version.objects.get_or_create( | ||
1844 | project=params['project'], | ||
1845 | layer=layer, | ||
1846 | dirpath="toaster_created_layer") | ||
1847 | |||
1848 | # Add a dependency on our layer to the base recipe's layer | ||
1849 | LayerVersionDependency.objects.get_or_create( | ||
1850 | layer_version=lver, | ||
1851 | depends_on=params["base"].layer_version) | ||
1852 | |||
1853 | # Add it to our current project if needed | ||
1854 | ProjectLayer.objects.get_or_create(project=params['project'], | ||
1855 | layercommit=lver, | ||
1856 | optional=False) | ||
1857 | |||
1858 | # Create the actual recipe | ||
1859 | recipe, created = CustomImageRecipe.objects.get_or_create( | ||
1860 | name=request.POST["name"], | ||
1861 | base_recipe=params["base"], | ||
1862 | project=params["project"], | ||
1863 | layer_version=lver, | ||
1864 | is_image=True) | ||
1865 | |||
1866 | # If we created the object then setup these fields. They may get | ||
1867 | # overwritten later on and cause the get_or_create to create a | ||
1868 | # duplicate if they've changed. | ||
1869 | if created: | ||
1870 | recipe.file_path = request.POST["name"] | ||
1871 | recipe.license = "MIT" | ||
1872 | recipe.version = "0.1" | ||
1873 | recipe.save() | ||
1874 | |||
1875 | except Error as err: | ||
1876 | return {"error": "Can't create custom recipe: %s" % err} | ||
1877 | |||
1878 | # Find the package list from the last build of this recipe/target | ||
1879 | target = Target.objects.filter(Q(build__outcome=Build.SUCCEEDED) & | ||
1880 | Q(build__project=params['project']) & | ||
1881 | (Q(target=params['base'].name) | | ||
1882 | Q(target=recipe.name))).last() | ||
1883 | if target: | ||
1884 | # Copy in every package | ||
1885 | # We don't want these packages to be linked to anything because | ||
1886 | # that underlying data may change e.g. delete a build | ||
1887 | for tpackage in target.target_installed_package_set.all(): | ||
1888 | try: | ||
1889 | built_package = tpackage.package | ||
1890 | # The package had no recipe information so is a ghost | ||
1891 | # package skip it | ||
1892 | if built_package.recipe == None: | ||
1893 | continue; | ||
1894 | |||
1895 | config_package = CustomImagePackage.objects.get( | ||
1896 | name=built_package.name) | ||
1897 | |||
1898 | recipe.includes_set.add(config_package) | ||
1899 | except Exception as e: | ||
1900 | logger.warning("Error adding package %s %s" % | ||
1901 | (tpackage.package.name, e)) | ||
1902 | pass | ||
1903 | |||
1904 | return {"error": "ok", | ||
1905 | "packages" : recipe.get_all_packages().count(), | ||
1906 | "url": reverse('customrecipe', args=(params['project'].pk, | ||
1907 | recipe.id))} | ||
1908 | |||
1909 | @xhr_response | ||
1910 | def xhr_customrecipe_id(request, recipe_id): | ||
1911 | """ | ||
1912 | Set of ReST API processors working with recipe id. | ||
1913 | |||
1914 | Entry point: /xhr_customrecipe/<recipe_id> | ||
1915 | |||
1916 | Methods: | ||
1917 | GET - Get details of custom image recipe | ||
1918 | DELETE - Delete custom image recipe | ||
1919 | |||
1920 | Returns: | ||
1921 | GET: | ||
1922 | {"error": "ok", | ||
1923 | "info": dictionary of field name -> value pairs | ||
1924 | of the CustomImageRecipe model} | ||
1925 | DELETE: | ||
1926 | {"error": "ok"} | ||
1927 | or | ||
1928 | {"error": <error message>} | ||
1929 | """ | ||
1930 | try: | ||
1931 | custom_recipe = CustomImageRecipe.objects.get(id=recipe_id) | ||
1932 | except CustomImageRecipe.DoesNotExist: | ||
1933 | return {"error": "Custom recipe with id=%s " | ||
1934 | "not found" % recipe_id} | ||
1935 | |||
1936 | if request.method == 'GET': | ||
1937 | info = {"id" : custom_recipe.id, | ||
1938 | "name" : custom_recipe.name, | ||
1939 | "base_recipe_id": custom_recipe.base_recipe.id, | ||
1940 | "project_id": custom_recipe.project.id, | ||
1941 | } | ||
1942 | |||
1943 | return {"error": "ok", "info": info} | ||
1944 | |||
1945 | elif request.method == 'DELETE': | ||
1946 | custom_recipe.delete() | ||
1947 | return {"error": "ok"} | ||
1948 | else: | ||
1949 | return {"error": "Method %s is not supported" % request.method} | ||
1950 | |||
1951 | def customrecipe_download(request, pid, recipe_id): | 1752 | def customrecipe_download(request, pid, recipe_id): |
1952 | recipe = get_object_or_404(CustomImageRecipe, pk=recipe_id) | 1753 | recipe = get_object_or_404(CustomImageRecipe, pk=recipe_id) |
1953 | 1754 | ||
@@ -1960,232 +1761,6 @@ if True: | |||
1960 | 1761 | ||
1961 | return response | 1762 | return response |
1962 | 1763 | ||
1963 | def _traverse_dependents(next_package_id, rev_deps, all_current_packages, tree_level=0): | ||
1964 | """ | ||
1965 | Recurse through reverse dependency tree for next_package_id. | ||
1966 | Limit the reverse dependency search to packages not already scanned, | ||
1967 | that is, not already in rev_deps. | ||
1968 | Limit the scan to a depth (tree_level) not exceeding the count of | ||
1969 | all packages in the custom image, and if that depth is exceeded | ||
1970 | return False, pop out of the recursion, and write a warning | ||
1971 | to the log, but this is unlikely, suggesting a dependency loop | ||
1972 | not caught by bitbake. | ||
1973 | On return, the input/output arg rev_deps is appended with queryset | ||
1974 | dictionary elements, annotated for use in the customimage template. | ||
1975 | The list has unsorted, but unique elements. | ||
1976 | """ | ||
1977 | max_dependency_tree_depth = all_current_packages.count() | ||
1978 | if tree_level >= max_dependency_tree_depth: | ||
1979 | logger.warning( | ||
1980 | "The number of reverse dependencies " | ||
1981 | "for this package exceeds " + max_dependency_tree_depth + | ||
1982 | " and the remaining reverse dependencies will not be removed") | ||
1983 | return True | ||
1984 | |||
1985 | package = CustomImagePackage.objects.get(id=next_package_id) | ||
1986 | dependents = \ | ||
1987 | package.package_dependencies_target.annotate( | ||
1988 | name=F('package__name'), | ||
1989 | pk=F('package__pk'), | ||
1990 | size=F('package__size'), | ||
1991 | ).values("name", "pk", "size").exclude( | ||
1992 | ~Q(pk__in=all_current_packages) | ||
1993 | ) | ||
1994 | |||
1995 | for pkg in dependents: | ||
1996 | if pkg in rev_deps: | ||
1997 | # already seen, skip dependent search | ||
1998 | continue | ||
1999 | |||
2000 | rev_deps.append(pkg) | ||
2001 | if (_traverse_dependents( | ||
2002 | pkg["pk"], rev_deps, all_current_packages, tree_level+1)): | ||
2003 | return True | ||
2004 | |||
2005 | return False | ||
2006 | |||
2007 | def _get_all_dependents(package_id, all_current_packages): | ||
2008 | """ | ||
2009 | Returns sorted list of recursive reverse dependencies for package_id, | ||
2010 | as a list of dictionary items, by recursing through dependency | ||
2011 | relationships. | ||
2012 | """ | ||
2013 | rev_deps = [] | ||
2014 | _traverse_dependents(package_id, rev_deps, all_current_packages) | ||
2015 | rev_deps = sorted(rev_deps, key=lambda x: x["name"]) | ||
2016 | return rev_deps | ||
2017 | |||
2018 | @xhr_response | ||
2019 | def xhr_customrecipe_packages(request, recipe_id, package_id): | ||
2020 | """ | ||
2021 | ReST API to add/remove packages to/from custom recipe. | ||
2022 | |||
2023 | Entry point: /xhr_customrecipe/<recipe_id>/packages/<package_id> | ||
2024 | |||
2025 | Methods: | ||
2026 | PUT - Add package to the recipe | ||
2027 | DELETE - Delete package from the recipe | ||
2028 | GET - Get package information | ||
2029 | |||
2030 | Returns: | ||
2031 | {"error": "ok"} | ||
2032 | or | ||
2033 | {"error": <error message>} | ||
2034 | """ | ||
2035 | try: | ||
2036 | recipe = CustomImageRecipe.objects.get(id=recipe_id) | ||
2037 | except CustomImageRecipe.DoesNotExist: | ||
2038 | return {"error": "Custom recipe with id=%s " | ||
2039 | "not found" % recipe_id} | ||
2040 | |||
2041 | if package_id: | ||
2042 | try: | ||
2043 | package = CustomImagePackage.objects.get(id=package_id) | ||
2044 | except Package.DoesNotExist: | ||
2045 | return {"error": "Package with id=%s " | ||
2046 | "not found" % package_id} | ||
2047 | |||
2048 | if request.method == 'GET': | ||
2049 | # If no package_id then list the current packages | ||
2050 | if not package_id: | ||
2051 | total_size = 0 | ||
2052 | packages = recipe.get_all_packages().values("id", | ||
2053 | "name", | ||
2054 | "version", | ||
2055 | "size") | ||
2056 | for package in packages: | ||
2057 | package['size_formatted'] = \ | ||
2058 | filtered_filesizeformat(package['size']) | ||
2059 | total_size += package['size'] | ||
2060 | |||
2061 | return {"error": "ok", | ||
2062 | "packages" : list(packages), | ||
2063 | "total" : len(packages), | ||
2064 | "total_size" : total_size, | ||
2065 | "total_size_formatted" : | ||
2066 | filtered_filesizeformat(total_size) | ||
2067 | } | ||
2068 | else: | ||
2069 | all_current_packages = recipe.get_all_packages() | ||
2070 | |||
2071 | # Dependencies for package which aren't satisfied by the | ||
2072 | # current packages in the custom image recipe | ||
2073 | deps =\ | ||
2074 | package.package_dependencies_source.for_target_or_none( | ||
2075 | recipe.name)['packages'].annotate( | ||
2076 | name=F('depends_on__name'), | ||
2077 | pk=F('depends_on__pk'), | ||
2078 | size=F('depends_on__size'), | ||
2079 | ).values("name", "pk", "size").filter( | ||
2080 | # There are two depends types we don't know why | ||
2081 | (Q(dep_type=Package_Dependency.TYPE_TRDEPENDS) | | ||
2082 | Q(dep_type=Package_Dependency.TYPE_RDEPENDS)) & | ||
2083 | ~Q(pk__in=all_current_packages) | ||
2084 | ) | ||
2085 | |||
2086 | # Reverse dependencies which are needed by packages that are | ||
2087 | # in the image. Recursive search providing all dependents, | ||
2088 | # not just immediate dependents. | ||
2089 | reverse_deps = _get_all_dependents(package_id, all_current_packages) | ||
2090 | total_size_deps = 0 | ||
2091 | total_size_reverse_deps = 0 | ||
2092 | |||
2093 | for dep in deps: | ||
2094 | dep['size_formatted'] = \ | ||
2095 | filtered_filesizeformat(dep['size']) | ||
2096 | total_size_deps += dep['size'] | ||
2097 | |||
2098 | for dep in reverse_deps: | ||
2099 | dep['size_formatted'] = \ | ||
2100 | filtered_filesizeformat(dep['size']) | ||
2101 | total_size_reverse_deps += dep['size'] | ||
2102 | |||
2103 | |||
2104 | return {"error": "ok", | ||
2105 | "id": package.pk, | ||
2106 | "name": package.name, | ||
2107 | "version": package.version, | ||
2108 | "unsatisfied_dependencies": list(deps), | ||
2109 | "unsatisfied_dependencies_size": total_size_deps, | ||
2110 | "unsatisfied_dependencies_size_formatted": | ||
2111 | filtered_filesizeformat(total_size_deps), | ||
2112 | "reverse_dependencies": list(reverse_deps), | ||
2113 | "reverse_dependencies_size": total_size_reverse_deps, | ||
2114 | "reverse_dependencies_size_formatted": | ||
2115 | filtered_filesizeformat(total_size_reverse_deps)} | ||
2116 | |||
2117 | included_packages = recipe.includes_set.values_list('pk', flat=True) | ||
2118 | |||
2119 | if request.method == 'PUT': | ||
2120 | # If we're adding back a package which used to be included in this | ||
2121 | # image all we need to do is remove it from the excludes | ||
2122 | if package.pk in included_packages: | ||
2123 | try: | ||
2124 | recipe.excludes_set.remove(package) | ||
2125 | return {"error": "ok"} | ||
2126 | except Package.DoesNotExist: | ||
2127 | return {"error": | ||
2128 | "Package %s not found in excludes but was in " | ||
2129 | "included list" % package.name} | ||
2130 | |||
2131 | else: | ||
2132 | recipe.appends_set.add(package) | ||
2133 | # Make sure that package is not in the excludes set | ||
2134 | try: | ||
2135 | recipe.excludes_set.remove(package) | ||
2136 | except: | ||
2137 | pass | ||
2138 | # Add the dependencies we think will be added to the recipe | ||
2139 | # as a result of appending this package. | ||
2140 | # TODO this should recurse down the entire deps tree | ||
2141 | for dep in package.package_dependencies_source.all_depends(): | ||
2142 | try: | ||
2143 | cust_package = CustomImagePackage.objects.get( | ||
2144 | name=dep.depends_on.name) | ||
2145 | |||
2146 | recipe.includes_set.add(cust_package) | ||
2147 | try: | ||
2148 | # When adding the pre-requisite package, make | ||
2149 | # sure it's not in the excluded list from a | ||
2150 | # prior removal. | ||
2151 | recipe.excludes_set.remove(cust_package) | ||
2152 | except Package.DoesNotExist: | ||
2153 | # Don't care if the package had never been excluded | ||
2154 | pass | ||
2155 | except: | ||
2156 | logger.warning("Could not add package's suggested" | ||
2157 | "dependencies to the list") | ||
2158 | |||
2159 | return {"error": "ok"} | ||
2160 | |||
2161 | elif request.method == 'DELETE': | ||
2162 | try: | ||
2163 | # If we're deleting a package which is included we need to | ||
2164 | # Add it to the excludes list. | ||
2165 | if package.pk in included_packages: | ||
2166 | recipe.excludes_set.add(package) | ||
2167 | else: | ||
2168 | recipe.appends_set.remove(package) | ||
2169 | all_current_packages = recipe.get_all_packages() | ||
2170 | reverse_deps_dictlist = _get_all_dependents(package.pk, all_current_packages) | ||
2171 | ids = [entry['pk'] for entry in reverse_deps_dictlist] | ||
2172 | reverse_deps = CustomImagePackage.objects.filter(id__in=ids) | ||
2173 | for r in reverse_deps: | ||
2174 | try: | ||
2175 | if r.id in included_packages: | ||
2176 | recipe.excludes_set.add(r) | ||
2177 | else: | ||
2178 | recipe.appends_set.remove(r) | ||
2179 | except: | ||
2180 | pass | ||
2181 | |||
2182 | return {"error": "ok"} | ||
2183 | except CustomImageRecipe.DoesNotExist: | ||
2184 | return {"error": "Tried to remove package that wasn't present"} | ||
2185 | |||
2186 | else: | ||
2187 | return {"error": "Method %s is not supported" % request.method} | ||
2188 | |||
2189 | def importlayer(request, pid): | 1764 | def importlayer(request, pid): |
2190 | template = "importlayer.html" | 1765 | template = "importlayer.html" |
2191 | context = { | 1766 | context = { |