move generic operations to common module

circle/common/operations.py

+from .models import activity_context
+
+from django.core.exceptions import PermissionDenied
+
+
+class Operation(object):
+    """Base class for VM operations.
+    """
+    async_queue = 'localhost.man'
+    required_perms = ()
+
+    def __call__(self, **kwargs):
+        return self.call(**kwargs)
+
+    def __init__(self, subject):
+        """Initialize a new operation bound to the specified subject.
+        """
+        self.subject = subject
+
+    def __unicode__(self):
+        return self.name
+
+    def __prelude(self, kwargs):
+        """This method contains the shared prelude of call and async.
+        """
+        skip_checks = kwargs.setdefault('system', False)
+        user = kwargs.setdefault('user', None)
+        if not skip_checks:
+            self.check_auth(user)
+        self.check_precond()
+        return self.create_activity(user=user)
+
+    def _exec_op(self, activity, user, **kwargs):
+        """Execute the operation inside the specified activity's context.
+        """
+        with activity_context(activity, on_abort=self.on_abort,
+                              on_commit=self.on_commit):
+            return self._operation(activity, user, **kwargs)
+
+    def _operation(self, activity, user, system, **kwargs):
+        """This method is the operation's particular implementation.
+
+        Deriving classes should implement this method.
+        """
+        raise NotImplementedError
+
+    def async(self, **kwargs):
+        """Execute the operation asynchronously.
+
+        Only a quick, preliminary check is ran before creating the associated
+        activity and queuing the job.
+
+        The returned value is the handle for the asynchronous job.
+
+        For more information, check the synchronous call's documentation.
+        """
+        activity = self.__prelude(kwargs)
+        return self.async_operation.apply_async(args=(self.id,
+                                                      self.subject.pk,
+                                                      activity.pk),
+                                                kwargs=kwargs,
+                                                queue=self.async_queue)
+
+    def call(self, **kwargs):
+        """Execute the operation (synchronously).
+
+        Anticipated keyword arguments:
+        * user: The User invoking the operation. If this argument is not
+                present, it'll be provided with a default value of None.
+        * system: Indicates that the operation is invoked by the system, not a
+                  User. If this argument is present and has a value of True,
+                  then authorization checks are skipped.
+        """
+        activity = self.__prelude(kwargs)
+        return self._exec_op(activity=activity, **kwargs)
+
+    def check_precond(self):
+        pass
+
+    def check_auth(self, user):
+        if not user.has_perms(self.required_perms):
+            raise PermissionDenied("%s doesn't have the required permissions."
+                                   % user)
+
+    def create_activity(self, user):
+        raise NotImplementedError
+
+    def on_abort(self, activity, error):
+        """This method is called when the operation aborts (i.e. raises an
+        exception).
+        """
+        pass
+
+    def on_commit(self, activity):
+        """This method is called when the operation executes successfully.
+        """
+        pass
+
+
+operation_registry_name = '_ops'
+
+
+class OperatedMixin(object):
+    def __getattr__(self, name):
+        # NOTE: __getattr__ is only called if the attribute doesn't already
+        # exist in your __dict__
+        cls = self.__class__
+        ops = getattr(cls, operation_registry_name, {})
+        op = ops.get(name)
+        if op:
+            return op(self)
+        else:
+            raise AttributeError("%r object has no attribute %r" %
+                                 (self.__class__.__name__, name))
+
+
+def register_operation(target_cls, op_cls, op_id=None):
+    """Register the specified operation with the target class.
+
+    You can optionally specify an ID to be used for the registration;
+    otherwise, the operation class' 'id' attribute will be used.
+    """
+    if op_id is None:
+        op_id = op_cls.id
+
+    if not issubclass(target_cls, OperatedMixin):
+        raise TypeError("%r is not a subclass of %r" %
+                        (target_cls.__name__, OperatedMixin.__name__))
+
+    if not hasattr(target_cls, operation_registry_name):
+        setattr(target_cls, operation_registry_name, dict())
+
+    getattr(target_cls, operation_registry_name)[op_id] = op_cls

circle/common/tests/test_operations.py

+from mock import MagicMock, patch
+
+from django.test import TestCase
+
+from ..operations import Operation
+
+
+class OperationTestCase(TestCase):
+    def test_activity_created_before_async_job(self):
+        class AbortEx(Exception):
+            pass
+
+        op = Operation(MagicMock())
+        op.activity_code_suffix = 'test'
+        op.id = 'test'
+        op.async_operation = MagicMock(
+            apply_async=MagicMock(side_effect=AbortEx))
+
+        with patch.object(Operation, 'check_precond'):
+            with patch.object(Operation, 'create_activity') as create_act:
+                try:
+                    op.async(system=True)
+                except AbortEx:
+                    self.assertTrue(create_act.called)
+
+    def test_check_precond_called_before_create_activity(self):
+        class AbortEx(Exception):
+            pass
+
+        op = Operation(MagicMock())
+        op.activity_code_suffix = 'test'
+        op.id = 'test'
+        with patch.object(Operation, 'create_activity', side_effect=AbortEx):
+            with patch.object(Operation, 'check_precond') as chk_pre:
+                try:
+                    op.call(system=True)
+                except AbortEx:
+                    self.assertTrue(chk_pre.called)
+
+    def test_auth_check_on_non_system_call(self):
+        op = Operation(MagicMock())
+        op.activity_code_suffix = 'test'
+        op.id = 'test'
+        user = MagicMock()
+        with patch.object(Operation, 'check_auth') as check_auth:
+            with patch.object(Operation, 'check_precond'), \
+                    patch.object(Operation, 'create_activity'), \
+                    patch.object(Operation, '_exec_op'):
+                op.call(user=user)
+            check_auth.assert_called_with(user)
+
+    def test_no_auth_check_on_system_call(self):
+        op = Operation(MagicMock())
+        op.activity_code_suffix = 'test'
+        op.id = 'test'
+        with patch.object(Operation, 'check_auth', side_effect=AssertionError):
+            with patch.object(Operation, 'check_precond'), \
+                    patch.object(Operation, 'create_activity'), \
+                    patch.object(Operation, '_exec_op'):
+                op.call(system=True)

circle/vm/models/instance.py

@@ -20,6 +20,7 @@ from model_utils.models import TimeStampedModel, StatusModel
 from taggit.managers import TaggableManager
 from taggit.managers import TaggableManager
 
 
 from acl.models import AclBase
 from acl.models import AclBase
+from common.operations import OperatedMixin
 from storage.models import Disk
 from storage.models import Disk
 from ..tasks import vm_tasks, agent_tasks
 from ..tasks import vm_tasks, agent_tasks
 from .activity import (ActivityInProgressError, instance_activity,
 from .activity import (ActivityInProgressError, instance_activity,
@@ -160,7 +161,7 @@ class InstanceTemplate(AclBase, VirtualMachineDescModel, TimeStampedModel):
         return ('dashboard.views.template-detail', None, {'pk': self.pk})
         return ('dashboard.views.template-detail', None, {'pk': self.pk})
 
 
 
 
-class Instance(AclBase, VirtualMachineDescModel, StatusModel,
+class Instance(AclBase, VirtualMachineDescModel, StatusModel, OperatedMixin,
                TimeStampedModel):
                TimeStampedModel):
 
 
     """Virtual machine instance.
     """Virtual machine instance.
@@ -216,7 +217,6 @@ class Instance(AclBase, VirtualMachineDescModel, StatusModel,
                                              "destruction."))
                                              "destruction."))
     objects = Manager()
     objects = Manager()
     active = InstanceActiveManager()
     active = InstanceActiveManager()
-    _ops = {}  # operation factory registry
 
 
     class Meta:
     class Meta:
         app_label = 'vm'
         app_label = 'vm'
@@ -254,15 +254,6 @@ class Instance(AclBase, VirtualMachineDescModel, StatusModel,
 
 
             self.instance = instance
             self.instance = instance
 
 
-    def __getattr__(self, name):
-        # NOTE: __getattr__ is only called if the attribute doesn't already
-        # exist in your __dict__
-        if name in self._ops:
-            return self._ops[name](self)
-        else:
-            raise AttributeError("%s object has no attribute '%s'" %
-                                 (self.__class__.__name__, name))
-
     def __unicode__(self):
     def __unicode__(self):
         parts = (self.name, "(" + str(self.id) + ")")
         parts = (self.name, "(" + str(self.id) + ")")
         return " ".join(s for s in parts if s != "")
         return " ".join(s for s in parts if s != "")

circle/vm/operations.py

@@ -8,7 +8,7 @@ from django.utils.translation import ugettext_lazy as _
 
 
 from celery.exceptions import TimeLimitExceeded
 from celery.exceptions import TimeLimitExceeded
 
 
-from common.models import activity_context
+from common.operations import Operation, register_operation
 from storage.models import Disk
 from storage.models import Disk
 from .tasks import vm_tasks
 from .tasks import vm_tasks
 from .tasks.local_tasks import async_instance_operation
 from .tasks.local_tasks import async_instance_operation
@@ -18,78 +18,14 @@ from .models import Instance, InstanceActivity, InstanceTemplate
 logger = getLogger(__name__)
 logger = getLogger(__name__)
 
 
 
 
-class Operation(object):
-    """Base class for VM operations.
-    """
+class InstanceOperation(Operation):
     acl_level = 'owner'
     acl_level = 'owner'
-    async_queue = 'localhost.man'
-    required_perms = ()
-
-    def __call__(self, **kwargs):
-        return self.call(**kwargs)
+    async_operation = async_instance_operation
 
 
     def __init__(self, instance):
     def __init__(self, instance):
-        """Initialize a new operation bound to the specified VM instance.
-        """
+        super(InstanceOperation, self).__init__(subject=instance)
         self.instance = instance
         self.instance = instance
 
 
-    def __unicode__(self):
-        return self.name
-
-    def __prelude(self, kwargs):
-        """This method contains the shared prelude of call and async.
-        """
-        skip_checks = kwargs.setdefault('system', False)
-        user = kwargs.setdefault('user', None)
-        if not skip_checks:
-            self.check_auth(user)
-        self.check_precond()
-        return self.create_activity(user=user)
-
-    def _exec_op(self, activity, user, **kwargs):
-        """Execute the operation inside the specified activity's context.
-        """
-        with activity_context(activity, on_abort=self.on_abort,
-                              on_commit=self.on_commit):
-            return self._operation(activity, user, **kwargs)
-
-    def _operation(self, activity, user, system, **kwargs):
-        """This method is the operation's particular implementation.
-
-        Deriving classes should implement this method.
-        """
-        raise NotImplementedError
-
-    def async(self, **kwargs):
-        """Execute the operation asynchronously.
-
-        Only a quick, preliminary check is ran before creating the associated
-        activity and queuing the job.
-
-        The returned value is the handle for the asynchronous job.
-
-        For more information, check the synchronous call's documentation.
-        """
-        activity = self.__prelude(kwargs)
-        return async_instance_operation.apply_async(args=(self.id,
-                                                          self.instance.pk,
-                                                          activity.pk),
-                                                    kwargs=kwargs,
-                                                    queue=self.async_queue)
-
-    def call(self, **kwargs):
-        """Execute the operation (synchronously).
-
-        Anticipated keyword arguments:
-        * user: The User invoking the operation. If this argument is not
-                present, it'll be provided with a default value of None.
-        * system: Indicates that the operation is invoked by the system, not a
-                  User. If this argument is present and has a value of True,
-                  then authorization checks are skipped.
-        """
-        activity = self.__prelude(kwargs)
-        return self._exec_op(activity=activity, **kwargs)
-
     def check_precond(self):
     def check_precond(self):
         if self.instance.destroyed_at:
         if self.instance.destroyed_at:
             raise self.instance.InstanceDestroyedError(self.instance)
             raise self.instance.InstanceDestroyedError(self.instance)
@@ -99,36 +35,18 @@ class Operation(object):
             raise PermissionDenied("%s doesn't have the required ACL level." %
             raise PermissionDenied("%s doesn't have the required ACL level." %
                                    user)
                                    user)
 
 
-        if not user.has_perms(self.required_perms):
-            raise PermissionDenied("%s doesn't have the required permissions."
-                                   % user)
+        super(InstanceOperation, self).check_auth(user=user)
 
 
     def create_activity(self, user):
     def create_activity(self, user):
         return InstanceActivity.create(code_suffix=self.activity_code_suffix,
         return InstanceActivity.create(code_suffix=self.activity_code_suffix,
                                        instance=self.instance, user=user)
                                        instance=self.instance, user=user)
 
 
-    def on_abort(self, activity, error):
-        """This method is called when the operation aborts (i.e. raises an
-        exception).
-        """
-        pass
-
-    def on_commit(self, activity):
-        """This method is called when the operation executes successfully.
-        """
-        pass
-
-
-def register_operation(op_cls, op_id=None):
-    """Register the specified operation with Instance.
-    """
-    if op_id is None:
-        op_id = op_cls.id
 
 
-    Instance._ops[op_id] = lambda inst: op_cls(inst)
+def register_instance_operation(op_cls, op_id=None):
+    return register_operation(Instance, op_cls, op_id)
 
 
 
 
-class DeployOperation(Operation):
+class DeployOperation(InstanceOperation):
     activity_code_suffix = 'deploy'
     activity_code_suffix = 'deploy'
     id = 'deploy'
     id = 'deploy'
     name = _("deploy")
     name = _("deploy")
@@ -167,10 +85,10 @@ class DeployOperation(Operation):
         self.instance._deploy_vm(activity)
         self.instance._deploy_vm(activity)
 
 
 
 
-register_operation(DeployOperation)
+register_instance_operation(DeployOperation)
 
 
 
 
-class DestroyOperation(Operation):
+class DestroyOperation(InstanceOperation):
     activity_code_suffix = 'destroy'
     activity_code_suffix = 'destroy'
     id = 'destroy'
     id = 'destroy'
     name = _("destroy")
     name = _("destroy")
@@ -205,10 +123,10 @@ class DestroyOperation(Operation):
         self.instance.save()
         self.instance.save()
 
 
 
 
-register_operation(DestroyOperation)
+register_instance_operation(DestroyOperation)
 
 
 
 
-class MigrateOperation(Operation):
+class MigrateOperation(InstanceOperation):
     activity_code_suffix = 'migrate'
     activity_code_suffix = 'migrate'
     id = 'migrate'
     id = 'migrate'
     name = _("migrate")
     name = _("migrate")
@@ -239,10 +157,10 @@ class MigrateOperation(Operation):
                 net.deploy()
                 net.deploy()
 
 
 
 
-register_operation(MigrateOperation)
+register_instance_operation(MigrateOperation)
 
 
 
 
-class RebootOperation(Operation):
+class RebootOperation(InstanceOperation):
     activity_code_suffix = 'reboot'
     activity_code_suffix = 'reboot'
     id = 'reboot'
     id = 'reboot'
     name = _("reboot")
     name = _("reboot")
@@ -254,10 +172,10 @@ class RebootOperation(Operation):
                                     queue=queue_name).get(timeout=timeout)
                                     queue=queue_name).get(timeout=timeout)
 
 
 
 
-register_operation(RebootOperation)
+register_instance_operation(RebootOperation)
 
 
 
 
-class RedeployOperation(Operation):
+class RedeployOperation(InstanceOperation):
     activity_code_suffix = 'redeploy'
     activity_code_suffix = 'redeploy'
     id = 'redeploy'
     id = 'redeploy'
     name = _("redeploy")
     name = _("redeploy")
@@ -286,10 +204,10 @@ class RedeployOperation(Operation):
         self.instance._deploy_vm(activity)
         self.instance._deploy_vm(activity)
 
 
 
 
-register_operation(RedeployOperation)
+register_instance_operation(RedeployOperation)
 
 
 
 
-class ResetOperation(Operation):
+class ResetOperation(InstanceOperation):
     activity_code_suffix = 'reset'
     activity_code_suffix = 'reset'
     id = 'reset'
     id = 'reset'
     name = _("reset")
     name = _("reset")
@@ -301,10 +219,10 @@ class ResetOperation(Operation):
                                    queue=queue_name).get(timeout=timeout)
                                    queue=queue_name).get(timeout=timeout)
 
 
 
 
-register_operation(ResetOperation)
+register_instance_operation(ResetOperation)
 
 
 
 
-class SaveAsTemplateOperation(Operation):
+class SaveAsTemplateOperation(InstanceOperation):
     activity_code_suffix = 'save_as_template'
     activity_code_suffix = 'save_as_template'
     id = 'save_as_template'
     id = 'save_as_template'
     name = _("save as template")
     name = _("save as template")
@@ -358,10 +276,10 @@ class SaveAsTemplateOperation(Operation):
             return tmpl
             return tmpl
 
 
 
 
-register_operation(SaveAsTemplateOperation)
+register_instance_operation(SaveAsTemplateOperation)
 
 
 
 
-class ShutdownOperation(Operation):
+class ShutdownOperation(InstanceOperation):
     activity_code_suffix = 'shutdown'
     activity_code_suffix = 'shutdown'
     id = 'shutdown'
     id = 'shutdown'
     name = _("shutdown")
     name = _("shutdown")
@@ -387,10 +305,10 @@ class ShutdownOperation(Operation):
         self.instance.save()
         self.instance.save()
 
 
 
 
-register_operation(ShutdownOperation)
+register_instance_operation(ShutdownOperation)
 
 
 
 
-class ShutOffOperation(Operation):
+class ShutOffOperation(InstanceOperation):
     activity_code_suffix = 'shut_off'
     activity_code_suffix = 'shut_off'
     id = 'shut_off'
     id = 'shut_off'
     name = _("shut off")
     name = _("shut off")
@@ -408,10 +326,10 @@ class ShutOffOperation(Operation):
         self.instance.save()
         self.instance.save()
 
 
 
 
-register_operation(ShutOffOperation)
+register_instance_operation(ShutOffOperation)
 
 
 
 
-class SleepOperation(Operation):
+class SleepOperation(InstanceOperation):
     activity_code_suffix = 'sleep'
     activity_code_suffix = 'sleep'
     id = 'sleep'
     id = 'sleep'
     name = _("sleep")
     name = _("sleep")
@@ -447,10 +365,10 @@ class SleepOperation(Operation):
             self.instance.save()
             self.instance.save()
 
 
 
 
-register_operation(SleepOperation)
+register_instance_operation(SleepOperation)
 
 
 
 
-class WakeUpOperation(Operation):
+class WakeUpOperation(InstanceOperation):
     activity_code_suffix = 'wake_up'
     activity_code_suffix = 'wake_up'
     id = 'wake_up'
     id = 'wake_up'
     name = _("wake up")
     name = _("wake up")
@@ -490,4 +408,4 @@ class WakeUpOperation(Operation):
         self.instance.renew(which='both', base_activity=activity)
         self.instance.renew(which='both', base_activity=activity)
 
 
 
 
-register_operation(WakeUpOperation)
+register_instance_operation(WakeUpOperation)

circle/vm/tests/test_operations.py

-from mock import MagicMock, patch
-
 from django.test import TestCase
 from django.test import TestCase
 
 
 from vm.models import Instance
 from vm.models import Instance
 from vm.operations import (
 from vm.operations import (
-    Operation, DeployOperation, DestroyOperation, MigrateOperation,
+    DeployOperation, DestroyOperation, MigrateOperation,
     RebootOperation, RedeployOperation, ResetOperation,
     RebootOperation, RedeployOperation, ResetOperation,
     SaveAsTemplateOperation, ShutdownOperation, ShutOffOperation,
     SaveAsTemplateOperation, ShutdownOperation, ShutOffOperation,
     SleepOperation, WakeUpOperation,
     SleepOperation, WakeUpOperation,
 )
 )
-from vm.tasks.local_tasks import async_instance_operation
-
-
-class OperationTestCase(TestCase):
-    def test_activity_created_before_async_job(self):
-        class AbortEx(Exception):
-            pass
-
-        op = Operation(MagicMock())
-        op.activity_code_suffix = 'test'
-        op.id = 'test'
-        with patch.object(async_instance_operation, 'apply_async',
-                          side_effect=AbortEx):
-            with patch.object(Operation, 'check_precond'):
-                with patch.object(Operation, 'create_activity') as create_act:
-                    try:
-                        op.async(system=True)
-                    except AbortEx:
-                        self.assertTrue(create_act.called)
-
-    def test_check_precond_called_before_create_activity(self):
-        class AbortEx(Exception):
-            pass
-
-        op = Operation(MagicMock())
-        op.activity_code_suffix = 'test'
-        op.id = 'test'
-        with patch.object(Operation, 'create_activity', side_effect=AbortEx):
-            with patch.object(Operation, 'check_precond') as chk_pre:
-                try:
-                    op.call(system=True)
-                except AbortEx:
-                    self.assertTrue(chk_pre.called)
-
-    def test_auth_check_on_non_system_call(self):
-        op = Operation(MagicMock())
-        op.activity_code_suffix = 'test'
-        op.id = 'test'
-        user = MagicMock()
-        with patch.object(Operation, 'check_auth') as check_auth:
-            with patch.object(Operation, 'check_precond'), \
-                    patch.object(Operation, 'create_activity'), \
-                    patch.object(Operation, '_exec_op'):
-                op.call(user=user)
-            check_auth.assert_called_with(user)
-
-    def test_no_auth_check_on_system_call(self):
-        op = Operation(MagicMock())
-        op.activity_code_suffix = 'test'
-        op.id = 'test'
-        with patch.object(Operation, 'check_auth', side_effect=AssertionError):
-            with patch.object(Operation, 'check_precond'), \
-                    patch.object(Operation, 'create_activity'), \
-                    patch.object(Operation, '_exec_op'):
-                op.call(system=True)
 
 
 
 
 class DeployOperationTestCase(TestCase):
 class DeployOperationTestCase(TestCase):