Add import disk operation

circle/vm/operations.py

@@ -16,47 +16,46 @@
 # with CIRCLE.  If not, see <http://www.gnu.org/licenses/>.
 # with CIRCLE.  If not, see <http://www.gnu.org/licenses/>.
 
 
 from __future__ import absolute_import, unicode_literals
 from __future__ import absolute_import, unicode_literals
+
+from StringIO import StringIO
 from base64 import encodestring
 from base64 import encodestring
 from hashlib import md5
 from hashlib import md5
 from logging import getLogger
 from logging import getLogger
-import os
-from re import search
 from string import ascii_lowercase
 from string import ascii_lowercase
-from StringIO import StringIO
 from tarfile import TarFile, TarInfo
 from tarfile import TarFile, TarInfo
-import time
 from urlparse import urlsplit
 from urlparse import urlsplit
 
 
+import os
+import time
+from celery.contrib.abortable import AbortableAsyncResult
+from celery.exceptions import TimeLimitExceeded, TimeoutError
+from django.conf import settings
 from django.core.exceptions import PermissionDenied, SuspiciousOperation
 from django.core.exceptions import PermissionDenied, SuspiciousOperation
 from django.core.urlresolvers import reverse
 from django.core.urlresolvers import reverse
+from django.db.models import Q
 from django.utils import timezone
 from django.utils import timezone
 from django.utils.translation import ugettext_lazy as _, ugettext_noop
 from django.utils.translation import ugettext_lazy as _, ugettext_noop
-from django.conf import settings
-from django.db.models import Q
-
+from re import search
 from sizefield.utils import filesizeformat
 from sizefield.utils import filesizeformat
 
 
-from celery.contrib.abortable import AbortableAsyncResult
-from celery.exceptions import TimeLimitExceeded, TimeoutError
-
 from common.models import (
 from common.models import (
     create_readable, humanize_exception, HumanReadableException
     create_readable, humanize_exception, HumanReadableException
 )
 )
 from common.operations import Operation, register_operation, SubOperationMixin
 from common.operations import Operation, register_operation, SubOperationMixin
+from dashboard.store_api import Store, NoStoreException
+from firewall.models import Host
 from manager.scheduler import SchedulerError
 from manager.scheduler import SchedulerError
-from .tasks.local_tasks import (
-    abortable_async_instance_operation, abortable_async_node_operation,
-)
+from monitor.client import Client
+from storage.models import Disk
+from storage.tasks import storage_tasks
 from .models import (
 from .models import (
     Instance, InstanceActivity, InstanceTemplate, Interface, Node,
     Instance, InstanceActivity, InstanceTemplate, Interface, Node,
     NodeActivity, pwgen
     NodeActivity, pwgen
 )
 )
 from .tasks import agent_tasks, vm_tasks
 from .tasks import agent_tasks, vm_tasks
-
-from dashboard.store_api import Store, NoStoreException
-from firewall.models import Host
-from monitor.client import Client
-from storage.tasks import storage_tasks
+from .tasks.local_tasks import (
+    abortable_async_instance_operation, abortable_async_node_operation,
+)
 
 
 logger = getLogger(__name__)
 logger = getLogger(__name__)
 
 
@@ -325,8 +324,6 @@ class DownloadDiskOperation(InstanceOperation):
     async_queue = "localhost.man.slow"
     async_queue = "localhost.man.slow"
 
 
     def _operation(self, user, url, task, activity, name=None):
     def _operation(self, user, url, task, activity, name=None):
-        from storage.models import Disk
-
         disk = Disk.download(url=url, name=name, task=task)
         disk = Disk.download(url=url, name=name, task=task)
         devnums = list(ascii_lowercase)
         devnums = list(ascii_lowercase)
         for d in self.instance.disks.all():
         for d in self.instance.disks.all():
@@ -348,6 +345,31 @@ class DownloadDiskOperation(InstanceOperation):
 
 
 
 
 @register_operation
 @register_operation
+class ImportDiskOperation(InstanceOperation):
+    id = 'import_disk'
+    name = _('import disk')
+    description = _('Import and attach a disk image to the virtual machine '
+                    'from the user store. The disk image has to be in the '
+                    'root directory of the store.')
+    required_perms = ('storage.import_disk',)
+    accept_states = ('STOPPED', 'PENDING', 'RUNNING')
+    async_queue = 'localhost.man.slow'
+
+    def check_auth(self, user):
+        super(ImportDiskOperation, self).check_auth(user)
+        try:
+            Store(user)
+        except NoStoreException:
+            raise PermissionDenied
+
+    def _operation(self, user, name, disk_path):
+        store = Store(user)
+        download_link = store.request_download(disk_path)
+        disk = Disk.import_disk(user, name, download_link)
+        self.instance.disks.add(disk)
+
+
+@register_operation
 class ExportDiskOperation(InstanceOperation):
 class ExportDiskOperation(InstanceOperation):
     id = 'export_disk'
     id = 'export_disk'
     name = _('export disk')
     name = _('export disk')
@@ -365,27 +387,11 @@ class ExportDiskOperation(InstanceOperation):
 
 
     def _operation(self, user, disk, format):
     def _operation(self, user, disk, format):
         store = Store(user)
         store = Store(user)
-        store.new_folder('/export')
-        upload_link = store.request_upload('/export')
+        upload_link = store.request_upload('/')
         disk.export(format, upload_link)
         disk.export(format, upload_link)
 
 
 
 
 @register_operation
 @register_operation
-class ImportDiskOperation(InstanceOperation):
-    id = 'import_disk'
-    name = _('import disk')
-    description = _('Import and attach a disk image to the virtual machine.')
-    abortable = True
-    has_percentage = True
-    required_perms = ('storage.import_disk',)
-    accept_states = ('STOPPED', 'PENDING', 'RUNNING')
-    async_queue = 'localhost.man.slow'
-
-    def _operation(self, activity):
-        pass
-
-
-@register_operation
 class DeployOperation(InstanceOperation):
 class DeployOperation(InstanceOperation):
     id = 'deploy'
     id = 'deploy'
     name = _("deploy")
     name = _("deploy")