From 77131a39d48f87860a42095453a9248f13da8fb6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B9=BF=E5=AE=8F=E4=BC=9F?= <hongweiguang@huangbaoche.com>
Date: Wed, 21 Oct 2015 20:53:07 +0800
Subject: [PATCH] recovery playboo_run

---
 jperm/perm_api.py   |  2 --
 jumpserver/tasks.py | 46 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 jumpserver/tasks.py

diff --git a/jperm/perm_api.py b/jperm/perm_api.py
index 2ded2087c..18892ce94 100644
--- a/jperm/perm_api.py
+++ b/jperm/perm_api.py
@@ -4,8 +4,6 @@ from jasset.models import *
 from jumpserver.api import *
 import uuid
 import re
-from ansible.playbook import PlayBook
-from ansible import callbacks, utils
 from jumpserver.tasks import playbook_run
 
 from jumpserver.models import Setting
diff --git a/jumpserver/tasks.py b/jumpserver/tasks.py
new file mode 100644
index 000000000..54726eb10
--- /dev/null
+++ b/jumpserver/tasks.py
@@ -0,0 +1,46 @@
+# coding: utf-8
+
+from ansible.playbook import PlayBook
+from ansible import callbacks, utils
+
+
+def playbook_run(inventory, playbook, default_user=None, default_port=None, default_pri_key_path=None):
+    stats = callbacks.AggregateStats()
+    playbook_cb = callbacks.PlaybookCallbacks(verbose=utils.VERBOSITY)
+    runner_cb = callbacks.PlaybookRunnerCallbacks(stats, verbose=utils.VERBOSITY)
+    # run the playbook
+    print default_user, default_port, default_pri_key_path, inventory, playbook
+    if default_user and default_port and default_pri_key_path:
+        playbook = PlayBook(host_list=inventory,
+                            playbook=playbook,
+                            forks=5,
+                            remote_user=default_user,
+                            remote_port=default_port,
+                            private_key_file=default_pri_key_path,
+                            callbacks=playbook_cb,
+                            runner_callbacks=runner_cb,
+                            stats=stats,
+                            become=True,
+                            become_user='root')
+    else:
+        playbook = PlayBook(host_list=inventory,
+                            playbook=playbook,
+                            forks=5,
+                            callbacks=playbook_cb,
+                            runner_callbacks=runner_cb,
+                            stats=stats,
+                            become=True,
+                            become_user='root')
+    results = playbook.run()
+    results_r = {'unreachable': [], 'failures': [], 'success': []}
+    for hostname, result in results.items():
+        if result.get('unreachable', 2):
+            results_r['unreachable'].append(hostname)
+            print "%s >>> unreachable" % hostname
+        elif result.get('failures', 2):
+            results_r['failures'].append(hostname)
+            print "%s >>> Failed" % hostname
+        else:
+            results_r['success'].append(hostname)
+            print "%s >>> Success" % hostname
+    return results_r
\ No newline at end of file