Update ansible-server to support playbook of PNF

[ccsdk/distribution.git] / ansible-server / src / main / ansible-server / BuildPlaybookParams.py

diff --git a/ansible-server/src/main/ansible-server/BuildPlaybookParams.py b/ansible-server/src/main/ansible-server/BuildPlaybookParams.py
new file mode 100644 (file)
index 0000000..1593592
--- /dev/null
+++ b/ansible-server/src/main/ansible-server/BuildPlaybookParams.py
@@ -0,0 +1,122 @@
+
+from os import listdir
+from os.path import isfile, join
+import os.path
+import shutil
+import subprocess
+
+import cherrypy
+
+
+def buildInventorySysCall(ansible_path, ansible_inv, node_list, playbook_dir, target_inv, hostgrouplist, hostnamelist):
+    if not node_list:
+        LocalNodeList = "host"
+        LocalCredentials = "localhost    ansible_connection=local"
+        f = open(playbook_dir + "/" + target_inv, "w")
+        f.write("[" + LocalNodeList + "]\n")
+        f.write(LocalCredentials)
+        f.close()
+    else:
+        # Get credentials from file
+        data_inventory_orig = {}
+        data_inventory_target = {}
+        curr_group = None
+
+        cherrypy.log("***> " + ansible_path + "/" + ansible_inv)
+        f = open(ansible_path + "/" + ansible_inv, "r")
+        for line in f:
+            line = line.rstrip()
+
+            if len(line) > 0:
+                if '#' not in line:
+                    if "[" in line and "]" in line:
+                        data_inventory_orig[line] = []
+                        curr_group = line
+                    else:
+                        data_inventory_orig[curr_group].append(line)
+        f.close()
+
+        for node in node_list:
+            Fail = True
+            if "[" + node + "]" in data_inventory_orig:
+                if not "[" + node + "]" in data_inventory_target:
+                    cherrypy.log("RESET", "[" + node + "]")
+                    data_inventory_target["[" + node + "]"] = []
+                else:
+                    cherrypy.log("OK", "[" + node + "]")
+                Fail = False
+                for cred in data_inventory_orig["[" + node + "]"]:
+                    data_inventory_target["[" + node + "]"].append(cred)
+            else:
+                for key in data_inventory_orig:
+                    if node + " " in " ".join(data_inventory_orig[key]):
+                        if key not in data_inventory_target:
+                            data_inventory_target[key] = []
+                        for cred in data_inventory_orig[key]:
+                            if node + " " in cred:
+                                data_inventory_target[key].append(cred)
+                                Fail = False
+
+            if Fail:
+                data_inventory_target["[" + node + "]"] = \
+                    [node + " ansible_connection=ssh ansible_ssh_user=na ansible_ssh_private_key_file=na"]
+
+        f = open(playbook_dir + "/" + target_inv, "w")
+        for key in data_inventory_target:
+            f.write(key + "\n")
+            for rec in data_inventory_target[key]:
+                hostgrouplist.append(key.replace("[", '').replace("]", ''))
+                hostnamelist.append(rec.split(' ')[0])
+                f.write(rec + "\n")
+        f.close()
+
+
+def getPlaybookFile(ansible_path, playbook_name, playbook_type, playbook_dir):
+    # Get playbooks from files
+
+    version = None
+    target_playbook_name = None
+
+    if '@' in playbook_name:
+        version = playbook_name.split("@")[1]
+        version = version.replace('.yml', '')
+        version = version.replace('.tar.gz', '')
+
+    onlyfiles = [f for f in listdir(ansible_path) if isfile(join(ansible_path, f))]
+
+    version_max = '0.00'
+    version_target = ''
+
+    for filename in onlyfiles:
+        if playbook_type in filename:
+            temp_version = filename.split("@")[1]
+            temp_version = temp_version.replace('.yml', '')
+            temp_version = temp_version.replace('.tar.gz', '')
+            if version_max < temp_version:
+                version_max = temp_version
+
+            if version is not None:
+                if version in playbook_name:
+                    version_target = version
+                    target_playbook_name = filename
+
+    if target_playbook_name is None:
+        for filename in onlyfiles:
+            if playbook_type in filename and version_max in filename:
+                target_playbook_name = filename
+                version_target = version_max
+
+    if target_playbook_name:
+        src = ansible_path + "/" + target_playbook_name
+        if not os.path.exists(src):
+            return ''
+
+        if ".tar.gz" in target_playbook_name:
+            dest = playbook_dir + "/" + playbook_type + ".tar.gz"
+            shutil.copy2(src, dest)
+            subprocess.call(['tar', '-xvzf', dest, "-C", playbook_dir])
+        else:
+            dest = playbook_dir + "/" + playbook_type + ".yml"
+            shutil.copy2(src, dest)
+
+    return version_target