Implement OPNFV workflow

[pharos-tools.git] / dashboard / src / workflow / snapshot_workflow.py

diff --git a/dashboard/src/workflow/snapshot_workflow.py b/dashboard/src/workflow/snapshot_workflow.py
index 0d53ed4..34ac3a5 100644 (file)
--- a/dashboard/src/workflow/snapshot_workflow.py
+++ b/dashboard/src/workflow/snapshot_workflow.py
@@ -8,13 +8,13 @@
 ##############################################################################
 
 
-import datetime
+from django.utils import timezone
 import json
 
 from booking.models import Booking
 from resource_inventory.models import Host, Image
 from workflow.models import WorkflowStep
-from workflow.forms import SnapshotMetaForm, SnapshotHostSelectForm
+from workflow.forms import BasicMetaForm, SnapshotHostSelectForm
 
 
 class Select_Host_Step(WorkflowStep):
@@ -27,7 +27,7 @@ class Select_Host_Step(WorkflowStep):
         context = super(Select_Host_Step, self).get_context()
         context['form'] = SnapshotHostSelectForm()
         booking_hosts = {}
-        now = datetime.datetime.now()
+        now = timezone.now()
         user = self.repo_get(self.repo.SESSION_USER)
         bookings = Booking.objects.filter(start__lt=now, end__gt=now, owner=user)
         for booking in bookings:
@@ -91,14 +91,14 @@ class Image_Meta_Step(WorkflowStep):
         desc = self.repo_get(self.repo.SNAPSHOT_DESC, False)
         form = None
         if name and desc:
-            form = SnapshotMetaForm(initial={"name": name, "description": desc})
+            form = BasicMetaForm(initial={"name": name, "description": desc})
         else:
-            form = SnapshotMetaForm()
+            form = BasicMetaForm()
         context['form'] = form
         return context
 
     def post_render(self, request):
-        form = SnapshotMetaForm(request.POST)
+        form = BasicMetaForm(request.POST)
         if form.is_valid():
             name = form.cleaned_data['name']
             self.repo_put(self.repo.SNAPSHOT_NAME, name)