django-orchestra-test/orchestra/contrib/orchestration/__init__.py

77 lines
2.6 KiB
Python
Raw Normal View History

import collections
2015-04-07 15:14:49 +00:00
import copy
2015-04-05 18:02:36 +00:00
from .backends import ServiceBackend, ServiceController, replace
2015-04-07 15:14:49 +00:00
default_app_config = 'orchestra.contrib.orchestration.apps.OrchestrationConfig'
2015-04-07 15:14:49 +00:00
class Operation():
DELETE = 'delete'
SAVE = 'save'
MONITOR = 'monitor'
EXCEEDED = 'exceeded'
RECOVERY = 'recovery'
def __str__(self):
return '%s.%s(%s)' % (self.backend, self.action, self.instance)
def __hash__(self):
""" set() """
return hash(self.backend) + hash(self.instance) + hash(self.action)
def __eq__(self, operation):
""" set() """
return hash(self) == hash(operation)
2015-05-06 15:32:22 +00:00
def __init__(self, backend, instance, action, routes=None):
2015-04-07 15:14:49 +00:00
self.backend = backend
# instance should maintain any dynamic attribute until backend execution
# deep copy is prefered over copy otherwise objects will share same atributes (queryset cache)
self.instance = copy.deepcopy(instance)
self.action = action
2015-05-06 15:32:22 +00:00
self.routes = routes
2015-04-07 15:14:49 +00:00
@classmethod
2015-05-07 19:00:02 +00:00
def execute(cls, operations, serialize=False, async=None):
2015-04-07 15:14:49 +00:00
from . import manager
scripts, backend_serialize = manager.generate(operations)
return manager.execute(scripts, serialize=(serialize or backend_serialize), async=async)
2015-04-07 15:14:49 +00:00
@classmethod
def create_for_action(cls, instances, action):
if not isinstance(instances, collections.Iterable):
instances = [instances]
operations = []
for instance in instances:
backends = ServiceBackend.get_backends(instance=instance, action=action)
for backend_cls in backends:
operations.append(
cls(backend_cls, instance, action)
)
return operations
@classmethod
def execute_action(cls, instances, action):
""" instances can be an object or an iterable for batch processing """
operations = cls.create_for_action(instances, action)
2015-04-07 15:14:49 +00:00
return cls.execute(operations)
def preload_context(self):
"""
Heuristic: Running get_context will prevent most of related objects do not exist errors
2015-04-07 15:14:49 +00:00
"""
if self.action == self.DELETE:
if hasattr(self.backend, 'get_context'):
self.backend().get_context(self.instance)
2015-04-09 14:32:10 +00:00
def store(self, log):
2015-04-07 15:14:49 +00:00
from .models import BackendOperation
return BackendOperation.objects.create(
log=log,
backend=self.backend.get_name(),
instance=self.instance,
action=self.action,
)