Merge branch 'feature-abort-save_as-rebased' into 'master'

Feature: abort save as

circle/storage/models.py

@@ -134,6 +134,7 @@ class Disk(AclBase, TimeStampedModel):
             self.disk = disk
             self.disk = disk
 
 
     class DiskIsNotReady(Exception):
     class DiskIsNotReady(Exception):
+
         """ Exception for operations that need a deployed disk.
         """ Exception for operations that need a deployed disk.
         """
         """
 
 
@@ -394,7 +395,7 @@ class Disk(AclBase, TimeStampedModel):
             args=[self.datastore.path, self.filename],
             args=[self.datastore.path, self.filename],
             queue=queue_name).get(timeout=timeout)
             queue=queue_name).get(timeout=timeout)
 
 
-    def save_as(self, user=None, task_uuid=None, timeout=300):
+    def save_as(self, task=None, user=None, task_uuid=None, timeout=300):
         """Save VM as template.
         """Save VM as template.
 
 
         Based on disk type:
         Based on disk type:
@@ -429,10 +430,18 @@ class Disk(AclBase, TimeStampedModel):
                            type=new_type)
                            type=new_type)
 
 
         queue_name = self.get_remote_queue_name("storage", priority="slow")
         queue_name = self.get_remote_queue_name("storage", priority="slow")
-        storage_tasks.merge.apply_async(args=[self.get_disk_desc(),
-                                              disk.get_disk_desc()],
+        remote = storage_tasks.merge.apply_async(kwargs={
+            "old_json": self.get_disk_desc(),
+            "new_json": disk.get_disk_desc()},
             queue=queue_name
             queue=queue_name
-                                        ).get()  # Timeout
-        disk.is_ready = True
-        disk.save()
+        )  # Timeout
+        while True:
+            try:
+                remote.get(timeout=5)
+                break
+            except TimeoutError:
+                if task is not None and task.is_aborted():
+                    AbortableAsyncResult(remote.id).abort()
+                    disk.destroy()
+                    raise Exception("Save as aborted by use.")
         return disk
         return disk

circle/vm/operations.py

@@ -403,7 +403,7 @@ class SaveAsTemplateOperation(InstanceOperation):
         return "%s v%d" % (name, v)
         return "%s v%d" % (name, v)
 
 
     def on_abort(self, activity, error):
     def on_abort(self, activity, error):
-        if getattr(self, 'disks'):
+        if hasattr(self, 'disks'):
             for disk in self.disks:
             for disk in self.disks:
                 disk.destroy()
                 disk.destroy()
 
 
@@ -445,7 +445,7 @@ class SaveAsTemplateOperation(InstanceOperation):
 
 
         def __try_save_disk(disk):
         def __try_save_disk(disk):
             try:
             try:
-                return disk.save_as()
+                return disk.save_as(task)
             except Disk.WrongDiskTypeError:
             except Disk.WrongDiskTypeError:
                 return disk
                 return disk