[future]ansible运行结果存入数据库中...

apps/ops/ansible_api.py

@@ -3,6 +3,10 @@ from __future__ import unicode_literals
 
 import os
 import json
+import logging
+import ansible.constants as default_config
+
+
 from ansible.executor.task_queue_manager import TaskQueueManager
 from ansible.inventory import Inventory, Host, Group
 from ansible.vars import VariableManager
@@ -10,9 +14,13 @@ from ansible.parsing.dataloader import DataLoader
 from ansible.executor import playbook_executor
 from ansible.utils.display import Display
 from ansible.playbook.play import Play
-import ansible.constants as default_config
 from ansible.plugins.callback import CallbackBase
 
+from models import AnsiblePlay, AnsibleTask, AnsibleHostResult
+
+
+logger = logging.getLogger(__name__)
+
 
 class AnsibleError(StandardError):
     pass
@@ -175,21 +183,28 @@ class CallbackModule(CallbackBase):
         self.output = {}
 
     def _new_play(self, play):
-        return {
-            'play': {
+        """将Play保持到数据里面
+        """
+        ret = {
             'name': play.name,
-                'id': str(play._uuid)
-            },
+            'uuid': str(play._uuid),
             'tasks': []
-
         }
 
+        try:
+            play = AnsiblePlay(name=ret['name'], uuid=ret['uuid'], completed=False)
+            play.save()
+        except Exception as e:
+            logger.error("Save ansible play uuid to database error!, %s" % e.message)
+
+        return ret
+
     def _new_task(self, task):
-        return {
-            'task': {
+        """将Task保持到数据库里,需要和Play进行关联
+        """
+        ret = {
             'name': task.name,
-                'id': str(task._uuid)
-            },
+            'uuid': str(task._uuid),
             'failed': {},
             'unreachable': {},
             'skipped': {},
@@ -197,23 +212,61 @@ class CallbackModule(CallbackBase):
             'success': {}
         }
 
+        try:
+            play = AnsiblePlay.objects.get(uuid=self.__play_uuid)
+            task = AnsibleTask(play=play, uuid=ret['uuid'], name=ret['name'])
+            task.save()
+        except Exception as e:
+            logger.error("Save ansible task uuid to database error!, %s" % e.message)
+
+        return ret
+
+    @property
+    def __task_uuid(self):
+        return self.results[-1]['tasks'][-1]['uuid']
+
+    @property
+    def __play_uuid(self):
+        return self.results[-1]['uuid']
+
+    def save_task_result(self, result):
+        try:
+            task = AnsibleTask.objects.get(uuid=self.__task_uuid)
+            host_result = AnsibleHostResult(task=task, name=result._host)
+            host_result.save()
+        except Exception as e:
+            logger.error("Save Ansible host result to database error!, %s" % e.message)
+
+    @staticmethod
+    def save_no_host_result(task):
+        try:
+            task = AnsibleTask.objects.get(uuid=task._uuid)
+            host_result = AnsibleHostResult(task=task, no_host="no host to run this task")
+            host_result.save()
+        except Exception as e:
+            logger.error("Save Ansible host result to database error!, %s" % e.message)
+
     def v2_runner_on_failed(self, result, ignore_errors=False):
+        self.save_task_result(result)
         host = result._host
         self.results[-1]['tasks'][-1]['failed'][host.name] = result._result
 
     def v2_runner_on_unreachable(self, result):
+        self.save_task_result(result)
         host = result._host
         self.results[-1]['tasks'][-1]['unreachable'][host.name] = result._result
 
     def v2_runner_on_skipped(self, result):
+        self.save_task_result(result)
         host = result._host
         self.results[-1]['tasks'][-1]['skipped'][host.name] = result._result
 
     def v2_runner_on_no_hosts(self, task):
-        self.results[-1]['tasks'][-1]['no_hosts']['name'] = task.name
-        self.results[-1]['tasks'][-1]['no_hosts']['uuid'] = task.uuid
+        self.save_no_host_result(task)
+        self.results[-1]['tasks'][-1]['no_hosts']['msg'] = "no host to run this task"
 
     def v2_runner_on_ok(self, result):
+        self.save_task_result(result)
         host = result._host
         self.results[-1]['tasks'][-1]['success'][host.name] = result._result
 
@@ -224,7 +277,8 @@ class CallbackModule(CallbackBase):
         self.results[-1]['tasks'].append(self._new_task(task))
 
     def v2_playbook_on_stats(self, stats):
-        """Display info about playbook statistics"""
+        """AdHoc模式下这个钩子不会执行
+        """
         hosts = sorted(stats.processed.keys())
 
         summary = {}
@@ -234,6 +288,7 @@ class CallbackModule(CallbackBase):
 
         self.output['plays'] = self.results
         self.output['stats'] = summary
+        print "summary: %s" % summary
 
 
 class PlayBookRunner(InventoryMixin):
@@ -358,8 +413,19 @@ class ADHocRunner(InventoryMixin):
 
         self.play = Play().load(play_data, variable_manager=self.variable_manager, loader=self.loader)
 
+    @staticmethod
+    def update_db_play(result, ext_code):
+        try:
+            play = AnsiblePlay.objects.get(uuid=result[0]['uuid'])
+            play.completed = True
+            play.status_code = ext_code
+            play.save()
+        except Exception as e:
+            print e.message
+            logger.error("Update Ansible Play Status into database error!, %s" % e.message)
+
     def run(self):
-        """执行ADHoc 记录日志，　处理结果
+        """执行ADHoc, 执行完后, 修改AnsiblePlay的状态
         """
         tqm = None
         # TODO:日志和结果分析
@@ -374,13 +440,17 @@ class ADHocRunner(InventoryMixin):
             )
             ext_code = tqm.run(self.play)
             result = json.dumps(self.results_callback.results)
+
+            self.update_db_play(result, ext_code)
+
             return ext_code, result
+
         finally:
             if tqm:
                 tqm.cleanup()
 
 
-if __name__ == "__main__":
+def test_run():
     conf = Config()
     assets = [
         {
@@ -388,7 +458,7 @@ if __name__ == "__main__":
                 "ip": "192.168.1.119",
                 "port": "22",
                 "username": "root",
-                "password": "xxx",
+                "password": "tongfang_test",
                 "key": "asset_private_key",
         },
         {
@@ -408,10 +478,14 @@ if __name__ == "__main__":
             "gather_facts": "no",
             "tasks": [
                 dict(action=dict(module='setup')),
-                dict(action=dict(module='command', args='lsdfd'))
+                dict(action=dict(module='command', args='ls'))
             ]
         }
     hoc = ADHocRunner(conf, play_source, *assets)
     ext_code, result = hoc.run()
     print ext_code
     print result
+
+
+if __name__ == "__main__":
+    test_run()

apps/ops/models.py

@@ -10,25 +10,66 @@ from django.utils.translation import ugettext_lazy as _
 logger = logging.getLogger(__name__)
 
 
-class Play(models.Model):
+class AnsiblePlay(models.Model):
+    uuid = models.CharField(max_length=128, verbose_name=_('UUID'), primary_key=True)
     name = models.CharField(max_length=128, verbose_name=_('Name'))
-    uuid = models.CharField(max_length=128, verbose_name=_('UUID'))
     completed = models.BooleanField(default=False, verbose_name=_('IsCompleted'))
     status_code = models.IntegerField(default=0, verbose_name=_('StatusCode'))
 
     def __unicode__(self):
-        return self.name
+        return "AnsiblePlay: %s<%s>" % (self.name, self.uuid)
 
 
-class Task(models.Model):
-    play = models.ForeignKey(Play, related_name='tasks', blank=True)
-    name = models.CharField(max_length=128, blank=True, blverbose_name=_('Name'))
-    uuid = models.CharField(max_length=128, verbose_name=_('UUID'))
+class AnsibleTask(models.Model):
+    uuid = models.CharField(max_length=128, verbose_name=_('UUID'), primary_key=True)
+    play = models.ForeignKey(AnsiblePlay, related_name='tasks', blank=True)
+    name = models.CharField(max_length=128, blank=True, verbose_name=_('Name'))
 
     def __unicode__(self):
-        return self.clean()
+        return "AnsibleTask: %s<%s>" % (self.name, self.uuid)
 
+    def failed(self):
+        pass
 
-class HostResult(models.Model):
-    task = models.ForeignKey(Task, related_name='host_results', blank=True)
+    def success(self):
+        pass
 
+
+class AnsibleHostResult(models.Model):
+    task = models.ForeignKey(AnsibleTask, related_name='host_results', blank=True)
+    name = models.CharField(max_length=128, blank=True, verbose_name=_('Name'))
+    status = models.BooleanField(blank=True, default=False, verbose_name=_('Status'))
+    success = models.TextField(blank=True, verbose_name=_('Success'))
+    skipped = models.TextField(blank=True, verbose_name=_('Skipped'))
+    failed = models.TextField(blank=True, verbose_name=_('Failed'))
+    unreachable = models.TextField(blank=True, verbose_name=_('Unreachable'))
+    no_host = models.TextField(blank=True, verbose_name=_('NoHost'))
+
+    def __unicode__(self):
+        return "AnsibleHostResult: %s<%s>" % (self.name, str(self.status))
+
+    def is_failed(self):
+        pass
+
+    def result(self):
+        pass
+
+
+class AnsibleSetup(models.Model):
+    task = models.ForeignKey(AnsibleTask, related_name='host_results', blank=True)
+    name = models.CharField(max_length=128, blank=True, verbose_name=_('Name'))
+    status = models.BooleanField(blank=True, default=False, verbose_name=_('Status'))
+    success = models.TextField(blank=True, verbose_name=_('Success'))
+    skipped = models.TextField(blank=True, verbose_name=_('Skipped'))
+    failed = models.TextField(blank=True, verbose_name=_('Failed'))
+    unreachable = models.TextField(blank=True, verbose_name=_('Unreachable'))
+    no_host = models.TextField(blank=True, verbose_name=_('NoHost'))
+
+    def __unicode__(self):
+        return "AnsibleHostResult: %s<%s>" % (self.name, str(self.status))
+
+    def is_failed(self):
+        pass
+
+    def result(self):
+        pass