diff options
| author |   Parker Berberian <pberberian@iol.unh.edu> | 2020-02-13 14:25:24 -0500 |
|---|---|---|
| committer | Parker Berberian <pberberian@iol.unh.edu> | 2020-02-13 14:25:24 -0500 |
| commit | 8c012f8a9bc64add11920688abcd6981278cb0ea (patch) | |
| tree | 99772d9361784d2724d1665c566c69888ff7d5e9 /src/workflow/snapshot_workflow.py | |
| parent | f5cdab1569b26df0c7ffc3df1529f095116fd13a (diff) | |
Fix Imports
Fixes stale import statements. The dashboard can now come up and we can run our
unit tests
Change-Id: I7189afb2cd37aaa2492de065c236b6aa9a35de5b
Signed-off-by: Parker Berberian <pberberian@iol.unh.edu>
Diffstat (limited to 'src/workflow/snapshot_workflow.py')
| -rw-r--r-- | src/workflow/snapshot_workflow.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/workflow/snapshot_workflow.py b/src/workflow/snapshot_workflow.py index c2f4cd6..c0e2052 100644 --- a/src/workflow/snapshot_workflow.py +++ b/src/workflow/snapshot_workflow.py @@ -12,7 +12,7 @@ from django.utils import timezone import json from booking.models import Booking -from resource_inventory.models import Host, Image +from resource_inventory.models import ResourceQuery, Image from workflow.models import WorkflowStep from workflow.forms import BasicMetaForm, SnapshotHostSelectForm @@ -61,7 +61,7 @@ class Select_Host_Step(WorkflowStep): name = host['name'] booking_id = host['booking'] booking = Booking.objects.get(pk=booking_id) - host = Host.objects.get(bundle=booking.resource, template__resource__name=name) + host = ResourceQuery.get(bundle=booking.resource, template__resource__name=name) models = self.repo_get(self.repo.SNAPSHOT_MODELS, {}) if "host" not in models: models['host'] = host |