import_otherdistro: fix update recording code

We were missing some import statements here, so clearly I didn't test
this as I thought I had.

Signed-off-by: Paul Eggleton <paul.eggleton@linux.intel.com>
This commit is contained in:
Paul Eggleton 2018-09-18 15:20:32 +12:00
parent 3aa6bf964d
commit 3fd586e8a4

View File

@ -337,6 +337,7 @@ def check_branch_layer(args):
def get_update_obj(args): def get_update_obj(args):
from layerindex.models import Update
updateobj = None updateobj = None
if args.update: if args.update:
updateobj = Update.objects.filter(id=int(args.update)) updateobj = Update.objects.filter(id=int(args.update))
@ -350,7 +351,7 @@ def get_update_obj(args):
def import_pkgspec(args): def import_pkgspec(args):
utils.setup_django() utils.setup_django()
import settings import settings
from layerindex.models import LayerItem, LayerBranch, Recipe, ClassicRecipe, Machine, BBAppend, BBClass from layerindex.models import LayerItem, LayerBranch, Recipe, ClassicRecipe, Machine, BBAppend, BBClass, ComparisonRecipeUpdate
from django.db import transaction from django.db import transaction
ret, layerbranch = check_branch_layer(args) ret, layerbranch = check_branch_layer(args)