storage: refactored create and save_as method to use activites

circle/storage/models.py

@@ -300,10 +300,17 @@ class Disk(AclBase, TimeStampedModel):
                                               queue="localhost.man")
 
     @classmethod
-    def create(cls, **params):
+    def create(cls, instance=None, user=None, **params):
+        """Create disk with activity.
+        """
         datastore = params.pop('datastore', DataStore.objects.get())
         disk = cls(filename=str(uuid.uuid4()), datastore=datastore, **params)
         disk.save()
+        with disk_activity(code_suffix="create",
+                           user=user,
+                           disk=disk):
+            if instance:
+                instance.disks.add(disk)
         return disk
 
     @classmethod
@@ -317,10 +324,7 @@ class Disk(AclBase, TimeStampedModel):
 
         :return: Disk object without a real image, to be .deploy()ed later.
         """
-        disk = cls.create(**kwargs)
-        with disk_activity(code_suffix="create", user=user, disk=disk):
-            if instance:
-                instance.disks.add(disk)
+        disk = Disk.create(instance=None, user=None, **kwargs)
         return disk
 
     @classmethod
@@ -377,9 +381,10 @@ class Disk(AclBase, TimeStampedModel):
             class AbortException(Exception):
                 pass
 
-        with disk_activity(code_suffix='download', disk=disk,
+        with disk_activity(code_suffix='deploy', disk=disk,
                            task_uuid=task_uuid, user=user,
-                           on_abort=__on_abort):
+                           on_abort=__on_abort) as act:
+            with act.sub_activity('downloading_disk'):
                 result = remote_tasks.download.apply_async(
                     kwargs={'url': url, 'parent_id': task_uuid,
                             'disk': disk.get_disk_desc()},
@@ -393,7 +398,6 @@ class Disk(AclBase, TimeStampedModel):
                             AbortableAsyncResult(result.id).abort()
                             raise AbortException("Download aborted by user.")
                 disk.size = size
-            disk.ready = True
                 disk.save()
         return disk
 
@@ -454,15 +458,14 @@ class Disk(AclBase, TimeStampedModel):
 
         disk.save()
         with disk_activity(code_suffix="save_as", disk=self,
-                           user=user, task_uuid=None):
+                           user=user, task_uuid=task_uuid):
+            with disk_activity(code_suffix="deploy", disk=disk,
+                               user=user, task_uuid=task_uuid):
                 queue_name = self.get_remote_queue_name('storage')
                 remote_tasks.merge.apply_async(args=[self.get_disk_desc(),
                                                      disk.get_disk_desc()],
                                                queue=queue_name
                                                ).get()  # Timeout
-            disk.ready = True
-            disk.save()
-
             return disk
 
 
@@ -479,6 +482,15 @@ class DiskActivity(ActivityModel):
         act.save()
         return act
 
+    def __unicode__(self):
+        if self.parent:
+            return '{}({})->{}'.format(self.parent.activity_code,
+                                       self.disk,
+                                       self.activity_code)
+        else:
+            return '{}({})'.format(self.activity_code,
+                                   self.disk)
+
     def create_sub(self, code_suffix, task_uuid=None):
         act = DiskActivity(
             activity_code=self.activity_code + '.' + code_suffix,