vm: add agent tasks

circle/manager/mancelery.py

@@ -9,6 +9,7 @@ celery = Celery('manager', backend='amqp',
                 broker=getenv("AMQP_URI"),
                 broker=getenv("AMQP_URI"),
                 include=['vm.tasks.local_tasks',
                 include=['vm.tasks.local_tasks',
                          'vm.tasks.local_periodic_tasks',
                          'vm.tasks.local_periodic_tasks',
+                         'vm.tasks.local_agent_tasks',
                          'storage.tasks.local_tasks',
                          'storage.tasks.local_tasks',
                          'firewall.tasks.local_tasks'])
                          'firewall.tasks.local_tasks'])
 
 

circle/vm/tasks/agent_tasks.py

+from manager.mancelery import celery
+
+
+@celery.task(name='agent.change_password')
+def change_password(vm, password):
+    pass
+
+
+@celery.task(name='agent.restart_networking')
+def restart_networking(vm):
+    pass
+
+
+@celery.task(name='agent.set_time')
+def set_time(vm, time):
+    pass
+
+
+@celery.task(name='agent.set_hostname')
+def set_hostname(vm, time):
+    pass

circle/vm/tasks/local_agent_tasks.py

+from manager.mancelery import celery
+from vm.tasks.agent_tasks import (restart_networking, change_password,
+                                  set_time, set_hostname)
+import time
+
+
+@celery.task
+def agent_started(vm):
+    from vm.models import Instance, instance_activity
+    instance = Instance.objects.get(id=int(vm.split('-')[-1]))
+
+    with instance_activity(code_suffix='agent', instance=instance) as act:
+        with act.sub_activity('starting'):
+            queue = "%s.agent" % instance.node.host.hostname
+            print queue
+            restart_networking.apply_async(queue=queue,
+                                           args=(vm, ))
+            change_password.apply_async(queue=queue,
+                                        args=(vm, instance.pw))
+            set_time.apply_async(queue=queue,
+                                 args=(vm, time.time()))
+            set_hostname.apply_async(queue=queue,
+                                     args=(vm, instance.primary_host.hostname))
+
+
+@celery.task
+def agent_stopped(vm):
+    from vm.models import Instance, InstanceActivity
+    instance = Instance.objects.get(id=int(vm.split('-')[-1]))
+    qs = InstanceActivity.objects.filter(instance=instance,
+                                         activity_code='vm.Instance.agent')
+    act = qs.latest('id')
+    with act.sub_activity('stopping'):
+        pass
+
+
+@celery.task
+def agent_ok(vm):
+    from vm.models import Instance, InstanceActivity
+    instance = Instance.objects.get(id=int(vm.split('-')[-1]))
+    qs = InstanceActivity.objects.filter(instance=instance,
+                                         activity_code='vm.Instance.agent')
+    act = qs.latest('id')
+    with act.sub_activity('ok'):
+        pass