summaryrefslogtreecommitdiffstats
path: root/ansible-server/src/main
diff options
context:
space:
mode:
authorEnbo Wang <wangenbo@huawei.com>2019-05-01 11:07:21 +0000
committerEnbo Wang <wangenbo@huawei.com>2019-05-02 10:31:55 +0000
commit786db48092901d38ae8710c8534e7c3654e8ece4 (patch)
tree0f7f01f2b18128ea1f89fcf83d202be2f11f6a70 /ansible-server/src/main
parentb0df5e06988b3dc8a5f0463251dd20ef7d1178fb (diff)
Update ansible-server to support playbook of PNF
Use pnf-flag to distinguish PNF and VNF, and use different processing methods respectively. Change-Id: I5de015eb9044b8d66781e1f7bb12f9487e313125 Issue-ID: APPC-1577 Signed-off-by: Enbo Wang <wangenbo@huawei.com>
Diffstat (limited to 'ansible-server/src/main')
-rwxr-xr-xansible-server/src/main/ansible-server/AnsibleModule.py26
-rw-r--r--ansible-server/src/main/ansible-server/BuildPlaybookParams.py147
-rwxr-xr-x[-rw-r--r--]ansible-server/src/main/ansible-server/RestServer.py1626
3 files changed, 1001 insertions, 798 deletions
diff --git a/ansible-server/src/main/ansible-server/AnsibleModule.py b/ansible-server/src/main/ansible-server/AnsibleModule.py
index 180373f4..71d0c059 100755
--- a/ansible-server/src/main/ansible-server/AnsibleModule.py
+++ b/ansible-server/src/main/ansible-server/AnsibleModule.py
@@ -33,40 +33,36 @@ import cherrypy
from cherrypy.lib.httputil import parse_query_string
from cherrypy.lib import auth_basic
-def ansibleSysCall (inventory_path, playbook_path, nodelist, mandatory,
- envparameters, localparameters, timeout, playbookdir):
-
+def ansibleSysCall (inventory_path, playbook_path, nodelist, mandatory, envparameters, localparameters, timeout, playbookdir):
cherrypy.log( "***> in AnsibleModule.ansibleSysCall")
log = []
str_parameters = ''
- if not envparameters == '':
+ if envparameters:
for key in envparameters:
if str_parameters == '':
str_parameters = '"' + str(key) + '=\'' + str(envparameters[key]) + '\''
else:
- #str_parameters += ' ' + str(key) + '=\'' + str(envparameters[key]) + '\''
- str_parameters += ', ' + str(key) + '=\'' + str(envparameters[key]) + '\''
+ str_parameters += ' ' + str(key) + '=\'' + str(envparameters[key]) + '\''
+ # str_parameters += ', ' + str(key) + '=\'' + str(envparameters[key]) + '\''
str_parameters += '"'
if len(str_parameters) > 0:
- cmd = 'export HOME=/root; env; cd ' + playbookdir + ';' +'timeout --signal=KILL ' + str(timeout) + \
+ cmd = 'cd ' + playbookdir + ';' + 'timeout -s KILL -t ' + str(timeout) + \
' ansible-playbook -v --timeout ' + str(timeout) + ' --extra-vars ' + str_parameters + ' -i ' + \
inventory_path + ' ' + playbook_path + ' | tee log.file'
else:
- cmd = 'export HOME=/root; env; cd ' + playbookdir + ';' +'timeout --signal=KILL ' + str(timeout) + \
- ' ansible-playbook -v --timeout ' + str(timeout) + ' -i ' + inventory_path + ' ' + playbook_path +' | tee log.file'
+ cmd = 'cd ' + playbookdir + ';' + 'timeout -s KILL -t ' + str(timeout) + \
+ ' ansible-playbook -v --timeout ' + str(timeout) + ' -i ' + inventory_path + ' ' + playbook_path + ' | tee log.file'
cherrypy.log("CMD: " + cmd)
cherrypy.log("PlayBook Start: " + playbookdir )
- p = subprocess.Popen(cmd, shell=True,
- stdout=subprocess.PIPE,
- stderr=subprocess.STDOUT)
+ p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
#PAP
#p.wait()
- (stdout_value, err) = p.communicate()
+ stdout_value, err = p.communicate()
stdout_value_cleanup = ''
for line in stdout_value:
@@ -78,15 +74,11 @@ def ansibleSysCall (inventory_path, playbook_path, nodelist, mandatory,
returncode = p.returncode
if returncode == 137:
-
cherrypy.log(" ansible-playbook system call timed out")
# ansible-playbook system call timed out
for line in stdout_value: # p.stdout.readlines():
log.append (line)
-
-
else:
-
for line in stdout_value: # p.stdout.readlines():
print line # line,
if ParseFlag and len(line.strip())>0:
diff --git a/ansible-server/src/main/ansible-server/BuildPlaybookParams.py b/ansible-server/src/main/ansible-server/BuildPlaybookParams.py
new file mode 100644
index 00000000..e3a0b4b5
--- /dev/null
+++ b/ansible-server/src/main/ansible-server/BuildPlaybookParams.py
@@ -0,0 +1,147 @@
+'''
+/*-
+* ============LICENSE_START=======================================================
+* ONAP : APPC
+* ================================================================================
+* Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
+* ================================================================================
+* Copyright (C) 2017 Amdocs
+* =============================================================================
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*
+* ============LICENSE_END=========================================================
+*/
+'''
+
+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:
+ local_node_list = "host"
+ local_credentials = "localhost ansible_connection=local"
+
+ f = open(playbook_dir + "/" + target_inv, "w")
+ f.write("[" + local_node_list + "]\n")
+ f.write(local_credentials)
+ 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_flag = True
+ if "[" + node + "]" in data_inventory_orig:
+ if "[" + node + "]" not in data_inventory_target:
+ cherrypy.log("RESET", "[" + node + "]")
+ data_inventory_target["[" + node + "]"] = []
+ else:
+ cherrypy.log("OK", "[" + node + "]")
+ fail_flag = 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_flag = False
+
+ if fail_flag:
+ 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
diff --git a/ansible-server/src/main/ansible-server/RestServer.py b/ansible-server/src/main/ansible-server/RestServer.py
index 65f73986..2b4da335 100644..100755
--- a/ansible-server/src/main/ansible-server/RestServer.py
+++ b/ansible-server/src/main/ansible-server/RestServer.py
@@ -23,9 +23,8 @@
*/
'''
-import time, datetime, json, os, sys, subprocess, re
+import time, datetime, json, os, sys, subprocess
import uuid
-import tarfile
import shutil
import glob
import crypt
@@ -34,12 +33,12 @@ import requests
import cherrypy
from cherrypy.lib.httputil import parse_query_string
-from cherrypy.lib import auth_basic
from multiprocessing import Process, Manager
from AnsibleModule import ansibleSysCall
from BuildHostFile import buildHostsSysCall
+from BuildPlaybookParams import buildInventorySysCall, getPlaybookFile
from os import listdir
from os.path import isfile, join
@@ -47,24 +46,26 @@ from os.path import isfile, join
TestRecord = Manager().dict()
ActiveProcess = {}
+
def validate_password(realm, username, password):
comp = crypt.crypt(password, salt)
if username in userpassdict and userpassdict[username] == comp:
- return True
+ return True
return False
-def sys_call (cmd):
+
+def sys_call(cmd):
p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
output = p.stdout.readlines()
retval = p.wait()
- if len (output) > 0:
+ if len(output) > 0:
for i in range(len(output)):
output[i] = output[i].strip()
return retval, output
-def callback (Id, Result, Output, Log, returncode):
- print "***> in RestServer.callback"
+def callback(Id, Result, Output, Log, returncode):
+ cherrypy.log("***> in RestServer.callback")
if Id in TestRecord:
time_now = datetime.datetime.utcnow()
@@ -79,34 +80,37 @@ def callback (Id, Result, Output, Log, returncode):
Result['StatusMessage'] = "FINISHED"
# Need to update the whole data structure for key=Id otherwise Manager is not updated
- TestRecord[Id] = {'PlaybookName': TestRecord[Id]['PlaybookName'],
- 'Version': TestRecord[Id]['Version'],
- 'NodeList': TestRecord[Id]['NodeList'],
- 'HostGroupList': TestRecord[Id]['HostGroupList'],
- 'HostNameList': TestRecord[Id]['HostNameList'],
- 'Time': TestRecord[Id]['Time'],
- 'Timeout': TestRecord[Id]['Timeout'],
- 'Duration': str(delta_time),
- 'EnvParameters': TestRecord[Id]['EnvParameters'],
- 'LocalParameters': TestRecord[Id]['LocalParameters'],
- 'FileParameters': TestRecord[Id]['FileParameters'],
- 'CallBack': TestRecord[Id]['CallBack'],
- 'Result': Result,
- 'Log': Log,
- 'Output': Output,
- 'Path': TestRecord[Id]['Path'],
- 'Mandatory': TestRecord[Id]['Path']}
-
- if not TestRecord[Id]['CallBack'] == None:
+ TestRecord[Id] = {
+ 'PlaybookName': TestRecord[Id]['PlaybookName'],
+ 'Version': TestRecord[Id]['Version'],
+ 'NodeList': TestRecord[Id]['NodeList'],
+ 'HostGroupList': TestRecord[Id]['HostGroupList'],
+ 'HostNameList': TestRecord[Id]['HostNameList'],
+ 'Time': TestRecord[Id]['Time'],
+ 'Timeout': TestRecord[Id]['Timeout'],
+ 'Duration': str(delta_time),
+ 'EnvParameters': TestRecord[Id]['EnvParameters'],
+ 'LocalParameters': TestRecord[Id]['LocalParameters'],
+ 'FileParameters': TestRecord[Id]['FileParameters'],
+ 'CallBack': TestRecord[Id]['CallBack'],
+ 'Result': Result,
+ 'Log': Log,
+ 'Output': Output,
+ 'Path': TestRecord[Id]['Path'],
+ 'Mandatory': TestRecord[Id]['Path']
+ }
- # Posting results to callback server
+ if TestRecord[Id]['CallBack'] is not None:
- data = {"StatusCode": 200,
- "StatusMessage": "FINISHED",
- "PlaybookName": TestRecord[Id]["PlaybookName"],
- "Version": TestRecord[Id]["Version"],
- "Duration": TestRecord[Id]["Duration"],
- "Results": TestRecord[Id]['Result']['Results']}
+ # Posting results to callback server
+ data = {
+ "StatusCode": 200,
+ "StatusMessage": "FINISHED",
+ "PlaybookName": TestRecord[Id]["PlaybookName"],
+ "Version": TestRecord[Id]["Version"],
+ "Duration": TestRecord[Id]["Duration"],
+ "Results": TestRecord[Id]['Result']['Results']
+ }
cherrypy.log("CALLBACK: TestRecord[Id]['Output']['Output']:", str(TestRecord[Id]['Output']['Output']))
cherrypy.log("CALLBACK: Results:", str(data["Results"]))
@@ -116,35 +120,33 @@ def callback (Id, Result, Output, Log, returncode):
if key in TestRecord[Id]['Output']['Output']:
data["Results"][key]["Output"] = TestRecord[Id]['Output']['Output'][key]
- print " Posting to", TestRecord[Id]['CallBack']
+ cherrypy.log(" Posting to", TestRecord[Id]['CallBack'])
s = requests.Session()
- r = s.post(TestRecord[Id]['CallBack'], data = json.dumps(data),
- headers = {'content-type': 'application/json'})
- print " Response", r.status_code, r.text
+ r = s.post(TestRecord[Id]['CallBack'], data=json.dumps(data),
+ headers={'content-type': 'application/json'})
+ cherrypy.log(" Response", r.status_code, r.text)
-def RunAnsible_Playbook (callback, Id, Inventory, Playbook, NodeList, TestRecord,
- Path, ArchiveFlag):
- print "***> in RestServer.RunAnsible_Playbook"
+def RunAnsible_Playbook(callback, Id, Inventory, Playbook, NodeList, TestRecord, Path, ArchiveFlag, pnf_flag=False):
+ cherrypy.log("***> in RestServer.RunAnsible_Playbook")
# Run test in playbook for given target
- Result = ''
-
- retval, log, returncode = ansibleSysCall (Inventory, Playbook, NodeList,
- TestRecord[Id]['Mandatory'],
- TestRecord[Id]['EnvParameters'],
- TestRecord[Id]['LocalParameters'],
- TestRecord[Id]['Timeout'],
- Path)
-
+ retval, log, returncode = ansibleSysCall(Inventory, Playbook, NodeList,
+ TestRecord[Id]['Mandatory'],
+ TestRecord[Id]['EnvParameters'],
+ TestRecord[Id]['LocalParameters'],
+ TestRecord[Id]['Timeout'],
+ Path)
cherrypy.log("Return code:" + str(returncode))
- cherrypy.log("Return val:" + str(retval))
+ cherrypy.log("Return value:" + str(retval))
Log = ''.join(log)
- #Output = {'Output': {}}
- Output = {}
+ if pnf_flag:
+ Output = {'Output': {}}
+ else:
+ Output = {}
onlyfiles = [f for f in listdir(Path)
if isfile(join(Path, f))]
@@ -152,155 +154,784 @@ def RunAnsible_Playbook (callback, Id, Inventory, Playbook, NodeList, TestRecord
cherrypy.log("Checking for results.txt files: ")
for file in onlyfiles:
if "results.txt" in file:
-# if file.endswith("results.txt"):
+ # if file.endswith("results.txt"):
cherrypy.log("results file: " + file)
f = open(Path + "/" + file, "r")
- resultsData = f.read() # Not to pass vnf instance name
- OutputP = json.loads(resultsData)
- Output['Output'] = OutputP
- cherrypy.log("Output = " + str(Output['Output']))
- #Output['Output'][key] = f.read() # To pass vnf instance name
+ if pnf_flag:
+ key = file.split("_")[0]
+ Output['Output'][key] = f.read()
+ else:
+ resultsData = f.read() # Not to pass vnf instance name
+ OutputP = json.loads(resultsData)
+ Output['Output'] = OutputP
+ cherrypy.log("Output = " + str(Output['Output']))
+ # Output['Output'][key] = f.read() # To pass vnf instance name
f.close()
if Output == {}:
- Output = {'Output': {}}
+ Output = {'Output': {}}
Result = {'Results': {}}
if 'could not be found' in Log:
- Result['Results'] = {"StatusCode": 101,
- "StatusMessage": "PLAYBOOK NOT FOUND"}
- if returncode == 137:
- Result['Results'] = {"StatusCode": 500,
- "StatusMessage": "TERMINATED"}
+ Result['Results'] = {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"}
+ if returncode == 137:
+ Result['Results'] = {"StatusCode": 500, "StatusMessage": "TERMINATED"}
elif TestRecord[Id]['NodeList'] == []:
-
host_index = None
-
if 'TargetNode' in TestRecord[Id]['EnvParameters']:
targetlist = TestRecord[Id]['EnvParameters']['TargetNode'].split(' ')
else:
targetlist = ["localhost"]
for key in retval:
- for i in range (len(targetlist)):
+ for i in range(len(targetlist)):
if key in targetlist[i]:
host_index = i
- if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
- int(retval[key][3]) == 0:
-
+ if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and int(retval[key][3]) == 0:
if host_index:
Result['Results'][targetlist[host_index]] = \
- {"GroupName": 'na', "StatusCode": 200, \
- "StatusMessage": "SUCCESS"}
+ {"GroupName": 'na', "StatusCode": 200, "StatusMessage": "SUCCESS"}
else:
- Result['Results'][key] = \
- {"GroupName": 'na', "StatusCode": 200, \
- "StatusMessage": "SUCCESS"}
+ Result['Results'][key] = {"GroupName": 'na', "StatusCode": 200, "StatusMessage": "SUCCESS"}
elif int(retval[key][2]) > 0:
if host_index:
Result['Results'][targetlist[host_index]] = \
- {"GroupName": 'na', "StatusCode": 400, \
- "StatusMessage": "NOT REACHABLE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
else:
Result['Results'][key] = \
- {"GroupName": 'na', "StatusCode": 400, \
- "StatusMessage": "NOT REACHABLE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
elif int(retval[key][3]) > 0:
if host_index:
Result['Results'][targetlist[host_index]] = \
- {"GroupName": 'na', "StatusCode": 400, \
- "StatusMessage": "FAILURE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "FAILURE"}
else:
Result['Results'][key] = \
- {"GroupName": 'na', "StatusCode": 400, \
- "StatusMessage": "FAILURE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "FAILURE"}
else:
-
for key in retval:
-
if len(TestRecord[Id]['HostNameList']) > 0:
-
host_index = []
- for i in range (len(TestRecord[Id]['HostNameList'])):
+ for i in range(len(TestRecord[Id]['HostNameList'])):
if key in TestRecord[Id]['HostNameList'][i]:
host_index.append(i)
if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
- int(retval[key][3]) == 0:
-
+ int(retval[key][3]) == 0:
if len(host_index) > 0:
Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
- {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
- "StatusCode": 200, "StatusMessage": "SUCCESS"}
-
- for i in range (1, len(host_index)):
- Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
- "," + TestRecord[Id]['HostGroupList'][host_index[i]]
+ {
+ "GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
+ "StatusCode": 200, "StatusMessage": "SUCCESS"
+ }
+
+ for i in range(1, len(host_index)):
+ Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"] += \
+ "," + TestRecord[Id]['HostGroupList'][host_index[i]]
else:
- Result['Results'][key] = \
- {"GroupName": key,
- "StatusCode": 200, "StatusMessage": "SUCCESS"}
+ Result['Results'][key] = {"GroupName": key, "StatusCode": 200, "StatusMessage": "SUCCESS"}
elif int(retval[key][2]) > 0:
-
if len(host_index) > 0:
Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
- {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
- "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
-
- for i in range (1, len(host_index)):
- Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
- "," + TestRecord[Id]['HostGroupList'][host_index[i]]
+ {
+ "GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
+ "StatusCode": 400, "StatusMessage": "NOT REACHABLE"
+ }
+
+ for i in range(1, len(host_index)):
+ Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"] += \
+ "," + TestRecord[Id]['HostGroupList'][host_index[i]]
else:
- Result['Results'][key] = \
- {"GroupName": key,
- "StatusCode": 200, "StatusMessage": "NOT REACHABLE"}
-
+ Result['Results'][key] = \
+ {"GroupName": key, "StatusCode": 200, "StatusMessage": "NOT REACHABLE"}
elif int(retval[key][3]) > 0:
-
if len(host_index) > 0:
Result['Results'][TestRecord[Id]['HostNameList'][host_index[0]]] = \
- {"GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
- "StatusCode": 400, "StatusMessage": "FAILURE"}
-
- for i in range (1, len(host_index)):
- Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"]+=\
- "," + TestRecord[Id]['HostGroupList'][host_index[i]]
+ {
+ "GroupName": TestRecord[Id]['HostGroupList'][host_index[0]],
+ "StatusCode": 400, "StatusMessage": "FAILURE"
+ }
+
+ for i in range(1, len(host_index)):
+ Result['Results'][TestRecord[Id]['HostNameList'][host_index[i]]]["GroupName"] += \
+ "," + TestRecord[Id]['HostGroupList'][host_index[i]]
else:
- Result['Results'][key] = \
- {"GroupName": key,
- "StatusCode": 200, "StatusMessage": "FAILURE"}
+ Result['Results'][key] = \
+ {"GroupName": key, "StatusCode": 200, "StatusMessage": "FAILURE"}
else:
host_index = None
- for i in range (len(TestRecord[Id]['NodeList'])):
+ for i in range(len(TestRecord[Id]['NodeList'])):
if key in TestRecord[Id]['NodeList'][i]:
host_index = i
if int(retval[key][0]) > 0 and int(retval[key][2]) == 0 and \
- int(retval[key][3]) == 0:
+ int(retval[key][3]) == 0:
Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
- {"GroupName": 'na', "StatusCode": 200, \
- "StatusMessage": "SUCCESS"}
+ {"GroupName": 'na', "StatusCode": 200, "StatusMessage": "SUCCESS"}
elif int(retval[key][2]) > 0:
Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
- {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "NOT REACHABLE"}
elif int(retval[key][3]) > 0:
Result['Results'][TestRecord[Id]['NodeList'][host_index]] = \
- {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "FAILURE"}
+ {"GroupName": 'na', "StatusCode": 400, "StatusMessage": "FAILURE"}
+
+ callback(Id, Result, Output, Log, returncode)
+
+
+def store_local_vars(playbook_path, Id):
+ if not os.path.exists(playbook_path + "/vars"):
+ os.mkdir(playbook_path + "/vars")
+
+ if not os.path.isfile(playbook_path + "/vars/defaults.yml"):
+ os.mknod(playbook_path + "/vars/defaults.yml")
+
+ # ##################################################
+ # PAP
+ # write local parameters passed into defaults.yml
+ # PAP
+ local_parms = TestRecord[Id]['LocalParameters']
+ cherrypy.log("LocalParameters: " + str(local_parms))
+
+ f = open(playbook_path + "/vars/defaults.yml", "a")
+ for key, value in local_parms.items():
+ f.write(key + "=" + value + "\n")
+ f.close()
+ # ##################################################
+
+ for key in TestRecord[Id]['LocalParameters']:
+ host_index = []
+ for i in range(len(TestRecord[Id]['HostNameList'])):
+ if key in TestRecord[Id]['HostNameList'][i]:
+ host_index.append(i)
+ if len(host_index) == 0:
+ for i in range(len(TestRecord[Id]['HostGroupList'])):
+ if key in TestRecord[Id]['HostGroupList'][i]:
+ host_index.append(i)
+ if len(host_index) > 0:
+ for i in range(len(host_index)):
+ f = open(playbook_path + "/vars/" + TestRecord[Id]['HostNameList'][host_index[i]] + ".yml", "a")
+ for param in TestRecord[Id]['LocalParameters'][key]:
+ f.write(param + ": " + str(TestRecord[Id]['LocalParameters'][key][param]) + "\n")
+ f.close()
+
+
+def process_pnf_playbook(input_json, Id, EnvParameters, time_now):
+ cherrypy.log("Processing playbook for PNF...")
+
+ PlaybookName = input_json['PlaybookName']
+ version = input_json.get('Version', None)
+
+ if AUTH:
+ cherrypy.log("Request USER : " + cherrypy.request.login)
+ cherrypy.log("Request Decode: ID " + Id)
+ cherrypy.log("Request Decode: EnvParameters " + json.dumps(EnvParameters))
+ cherrypy.log("Request Decode: PlaybookName " + PlaybookName)
+
+ str_uuid = str(uuid.uuid4())
+
+ HomeDir = os.path.dirname(os.path.realpath("~/"))
+
+ PlaybookType = PlaybookName.split(".")[0].split('_')[-1]
+ PlaybookDir = HomeDir + '/' + ANSIBLE_TEMP + "/" + PlaybookName.split(".")[0] + "_" + str_uuid
+ AnsibleInv = PlaybookType + "_" + "inventory"
+ ArchiveFlag = False
+
+ cherrypy.log("Request Decode: PlaybookType " + PlaybookType)
+ cherrypy.log("Request Decode: PlaybookDir " + PlaybookDir)
+ cherrypy.log("Request Decode: AnsibleInv " + AnsibleInv)
+
+ NodeList = input_json.get('NodeList', [])
+ cherrypy.log("Request Decode: NodeList: " + str(NodeList))
+
+ # Create base run directory if it doesn't exist
+ if not os.path.exists(ANSIBLE_TEMP):
+ cherrypy.log("Creating Base Run Directory: " + ANSIBLE_TEMP)
+ os.makedirs(ANSIBLE_TEMP)
+
+ os.mkdir(PlaybookDir)
+
+ # Process inventory file for target
+ hostgrouplist = []
+ hostnamelist = []
+
+ buildInventorySysCall(ANSIBLE_PATH, ANSIBLE_INV, NodeList, PlaybookDir, AnsibleInv, hostgrouplist, hostnamelist)
+
+ version_target = getPlaybookFile(ANSIBLE_PATH, PlaybookName, PlaybookType, PlaybookDir)
+ if not version_target:
+ return {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"}
+
+ if version is None:
+ version = version_target
+
+ if 'Timeout' in input_json:
+ timeout = int(input_json['Timeout'])
+ cherrypy.log("Timeout from API: " + str(timeout))
+ else:
+ timeout = timeout_seconds
+ cherrypy.log("Timeout not passed from API using default: " + str(timeout))
+
+ EnvParam = input_json.get('EnvParameters', {})
+ LocalParam = input_json.get('LocalParameters', {})
+ FileParam = input_json.get('FileParameters', {})
+ callback_flag = input_json.get('CallBack', None)
+
+ # if AnsibleServer is not set to 'na' don't send AnsibleServer in PENDING response.
+ TestRecord[Id] = {
+ 'PlaybookName': PlaybookName,
+ 'Version': version,
+ 'NodeList': NodeList,
+ 'HostGroupList': hostgrouplist,
+ 'HostNameList': hostnamelist,
+ 'Time': time_now,
+ 'Duration': timeout,
+ 'Timeout': timeout,
+ 'EnvParameters': EnvParam,
+ 'LocalParameters': LocalParam,
+ 'FileParameters': FileParam,
+ 'CallBack': callback_flag,
+ 'Result': {
+ "StatusCode": 100,
+ "StatusMessage": 'PENDING',
+ "ExpectedDuration": str(timeout) + "sec"
+ },
+ 'Log': '',
+ 'Output': {},
+ 'Path': PlaybookDir,
+ 'Mandatory': None
+ }
+ if AnsibleServer != 'na':
+ TestRecord[Id]['Result']["AnsibleServer"] = str(AnsibleServer),
+
+ cherrypy.log("Test_Record: " + str(TestRecord[Id]))
+
+ # Write files
+ if TestRecord[Id]['FileParameters']:
+ for key in TestRecord[Id]['FileParameters']:
+ filename = key
+ filecontent = TestRecord[Id]['FileParameters'][key]
+ f = open(PlaybookDir + "/" + filename, "w")
+ f.write(filecontent)
+ f.close()
+
+ playbook_path = PlaybookDir
+
+ # Store local vars
+ store_local_vars(playbook_path, Id)
+
+ # write some info out to files before running
+ if AUTH:
+ f = open(playbook_path + "/User.txt", "a")
+ f.write(cherrypy.request.login)
+ f.close()
+
+ f = open(playbook_path + "/PlaybookName.txt", "a")
+ f.write(PlaybookName)
+ f.close()
+
+ f = open(playbook_path + "/JsonRequest.txt", "w")
+ f.write(json.dumps(input_json, indent=4, sort_keys=True))
+ f.close()
+
+ # Cannot use thread because ansible module uses signals which are only supported in main thread.
+ # So use multiprocess with shared object
+ p = Process(target=RunAnsible_Playbook,
+ args=(callback, Id, PlaybookDir + '/' + AnsibleInv, PlaybookDir + '/' + PlaybookType + '.yml',
+ NodeList, TestRecord, PlaybookDir, ArchiveFlag, True))
+ p.start()
+ ActiveProcess[Id] = p
+ return TestRecord[Id]['Result']
+
+
+def process_vnf_playbook(input_json, Id, EnvParameters, time_now):
+ cherrypy.log("Processing playbook for VNF...")
+
+ PlaybookName = input_json['PlaybookName']
+ VNF_instance = EnvParameters.get('vnf_instance')
+ version = input_json.get('Version', None)
+
+ # GetInventoryNames
+ HaveNodeList = False
+ HaveInventoryNames = False
+ inventory_names = None
+ if 'InventoryNames' in input_json:
+ inventory_names = input_json['InventoryNames']
+ HaveInventoryNames = True
+
+ AnsiblePlaybookFail = True
+
+ str_uuid = str(uuid.uuid4())
+
+ # VnfType = PlaybookName.split("/")[0]
+
+ if AUTH:
+ cherrypy.log("Request USER : " + cherrypy.request.login)
+ cherrypy.log("Request Decode: ID " + Id)
+ # cherrypy.log("Request Decode: VnfType " + VnfType)
+ cherrypy.log("Request Decode: EnvParameters " + json.dumps(EnvParameters))
+
+ # Verify VNF_instance was passed in EnvParameters
+ if VNF_instance is not None:
+ cherrypy.log("Request Decode: VnfInstance " + VNF_instance)
+ else:
+ cherrypy.log("StatusCode: 107, StatusMessage: VNF_instance NOT PROVIDED")
+ return {"StatusCode": 107, "StatusMessage": "VNF_instance NOT PROVIDED"}
+
+ if inventory_names is not None:
+ cherrypy.log("Request Decode: Inventory Names " + inventory_names)
+ else:
+ cherrypy.log("Request Decode: Inventory Names " + "Not provided")
+
+ cherrypy.log("Request Decode: PlaybookName " + PlaybookName)
+
+ PlayBookFunction = PlaybookName.rsplit("/", 2)[1]
+ PlayBookFile = PlayBookFunction + "/site.yml"
+
+ cherrypy.log("Request Decode: PlaybookFunction " + PlayBookFunction)
+ cherrypy.log("Request Decode: PlaybookFile " + PlayBookFile)
+
+ BaseDir = ANSIBLE_PATH + "/" + PlaybookName.rsplit("/", 1)[0]
+ CopyDir = ANSIBLE_PATH + "/" + PlaybookName.rsplit("/", 2)[0]
+ cherrypy.log("Request Decode: Basedir " + BaseDir)
+ cherrypy.log("Request Decode: Copydir " + CopyDir)
+
+ PlaybookDir = ANSIBLE_TEMP + "/" + VNF_instance + "_" + str_uuid + "_" + str(Id)
+ cherrypy.log("Request Decode: PlaybookDir " + PlaybookDir)
+
+ # AnsibleInv is the directory where the host file to be run exists
+ # AnsibleInv = ANSIBLE_PATH + "/" + VnfType + "/latest/ansible/inventory/" + VNF_instance
+ ArchiveFlag = False
+
+ # Create base run directory if it doesn't exist
+ if not os.path.exists(ANSIBLE_TEMP):
+ cherrypy.log("Creating Base Run Directory: " + ANSIBLE_TEMP)
+ os.makedirs(ANSIBLE_TEMP)
+
+ if not os.path.exists(CopyDir):
+ cherrypy.log("Playbook Not Found")
+ return {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"}
+
+ # copy static playbook dir to run dir
+ cherrypy.log("Copying from " + CopyDir + " to " + PlaybookDir)
+ shutil.copytree(CopyDir, PlaybookDir)
+ # cmd="/usr/bin/find " + PlaybookDir + " -exec /usr/bin/touch {} \;"
+ cmd = "/usr/bin/find " + PlaybookDir + " -exec chmod +rx {} \;"
+ sys_call(cmd)
+ cherrypy.log(cmd)
+
+ cherrypy.log("PlaybookDir: " + PlaybookDir)
+ # cherrypy.log("AnsibleInv: " + AnsibleInv)
- callback (Id, Result, Output, Log, returncode)
+ # Process inventory file for target
+ hostgrouplist = []
+ hostnamelist = []
-class TestManager (object):
+ NodeList = input_json.get('NodeList', [])
+ cherrypy.log("NodeList: " + str(NodeList))
+
+ # if NodeList empty
+ if not NodeList:
+ cherrypy.log("*** NodeList - Empty ***")
+ else:
+ HaveNodeList = True
+
+ # ##############################################################################
+ # #### Host file processing ###########################
+ # #### 1. Use file delivered with playbook ###########################
+ # #### 2. If HostNames + NodeList generate and use ###########################
+ # ##############################################################################
+
+ # Verify inventory directory exists
+ path = PlaybookDir + "/inventory/"
+ if not os.path.isdir(path):
+ cherrypy.log("Inventory directory %s does not exist - create it" % path)
+ try:
+ os.mkdir(path)
+ except OSError:
+ cherrypy.log("Creation of the directory %s failed" % path)
+ else:
+ cherrypy.log("Successfully created the directory %s " % path)
+
+ # location of host file - Default
+ HostFile = PlaybookDir + "/inventory/" + VNF_instance + "hosts"
+ cherrypy.log("HostFile: " + HostFile)
+
+ # if NodeList and InventoryNames need to build host file
+ if HaveInventoryNames and HaveNodeList:
+ cherrypy.log("Build host file from NodeList")
+ ret = buildHostsSysCall(input_json, PlaybookDir, inventory_names)
+ if ret < 0:
+ cherrypy.log("Returning Error: Not running Playbook")
+ return {"StatusCode": 105,
+ "StatusMessage": "NodeList: Missing vnfc-type field"}
+
+ # Having been built now copy new file to correct file
+ shutil.copy(PlaybookDir + "/host_file.txt", HostFile)
+ cherrypy.log("Copying Generated host file to: " + HostFile)
+
+ if 'Timeout' in input_json:
+ timeout = int(input_json['Timeout'])
+ cherrypy.log("Timeout from API: " + str(timeout))
+ else:
+ timeout = timeout_seconds
+ cherrypy.log("Timeout not passed from API using default: " + str(timeout))
+
+ EnvParam = input_json.get('EnvParameters', {})
+ LocalParam = input_json.get('LocalParameters', {})
+ FileParam = input_json.get('FileParameters', {})
+ callback_flag = input_json.get('CallBack', None)
+
+ # if AnsibleServer is not set to 'na' don't send AnsibleServer in PENDING response.
+ TestRecord[Id] = {
+ 'PlaybookName': PlaybookName,
+ 'Version': version,
+ 'NodeList': NodeList,
+ 'HostGroupList': hostgrouplist,
+ 'HostNameList': hostnamelist,
+ 'Time': time_now,
+ 'Duration': timeout,
+ 'Timeout': timeout,
+ 'EnvParameters': EnvParam,
+ 'LocalParameters': LocalParam,
+ 'FileParameters': FileParam,
+ 'CallBack': callback_flag,
+ 'Result': {
+ "StatusCode": 100,
+ "StatusMessage": 'PENDING',
+ "ExpectedDuration": str(timeout) + "sec"
+ },
+ 'Log': '',
+ 'Output': {},
+ 'Path': PlaybookDir,
+ 'Mandatory': None
+ }
+ if AnsibleServer != 'na':
+ TestRecord[Id]['Result']["AnsibleServer"] = str(AnsibleServer),
+
+ cherrypy.log("Test_Record: " + str(TestRecord[Id]))
+
+ # Write files
+ if TestRecord[Id]['FileParameters']:
+ for key in TestRecord[Id]['FileParameters']:
+ filename = key
+ filecontent = TestRecord[Id]['FileParameters'][key]
+ f = open(PlaybookDir + "/" + filename, "w")
+ f.write(filecontent)
+ f.close()
+
+ # Process playbook
+ if os.path.exists(ANSIBLE_PATH + '/' + PlaybookName):
+ AnsiblePlaybookFail = False
+
+ if AnsiblePlaybookFail:
+ # if os.path.exists(PlaybookDir):
+ # shutil.rmtree (PlaybookDir)
+ cherrypy.log("AnsiblePlaybookFail")
+ del TestRecord[Id]
+ return {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"}
+ else:
+ # Test EnvParameters
+ playbook_path = PlaybookDir
+
+ # Store local vars
+ store_local_vars(playbook_path, Id)
+
+ # write some info out to files before running
+ if AUTH:
+ f = open(playbook_path + "/User.txt", "a")
+ f.write(cherrypy.request.login)
+ f.close()
+
+ f = open(playbook_path + "/PlaybookName.txt", "a")
+ f.write(PlaybookName)
+ f.close()
+
+ f = open(playbook_path + "/PlaybookExDir.txt", "a")
+ f.write(PlaybookDir + "/" + PlayBookFunction)
+ f.close()
+
+ f = open(playbook_path + "/JsonRequest.txt", "w")
+ f.write(json.dumps(input_json, indent=4, sort_keys=True))
+ f.close()
+
+ # Check that HostFile exists
+ if not os.path.isfile(HostFile):
+ cherrypy.log("Inventory file Not Found: " + HostFile)
+ return {"StatusCode": 101, "StatusMessage": "PLAYBOOK INVENTORY FILE NOT FOUND"}
+
+ # Cannot use thread because ansible module uses signals which are only supported in main thread.
+ # So use multiprocess with shared object
+ p = Process(target=RunAnsible_Playbook,
+ args=(callback, Id, HostFile, PlaybookDir + '/' + PlayBookFile,
+ NodeList, TestRecord, PlaybookDir + "/" + PlayBookFunction, ArchiveFlag))
+ p.start()
+ ActiveProcess[Id] = p
+ return TestRecord[Id]['Result']
+
+
+def handle_post_method(input_json, time_now):
+ cherrypy.log("Payload: " + str(input_json))
+
+ if 'Id' in input_json and 'PlaybookName' in input_json and 'EnvParameters' in input_json:
+ if input_json['Id'] not in TestRecord:
+ # check if Id exists in previous run directory, if so return error
+ Id = input_json['Id']
+ if glob.glob(ANSIBLE_TEMP + '/*_' + input_json['Id']):
+ cherrypy.log("Old directory found for ID: " + Id)
+ return {"StatusCode": 101, "StatusMessage": "TEST ID FILE ALREADY DEFINED"}
+
+ # if required it should be passed as an argument
+ EnvParameters = input_json.get('EnvParameters', {})
+
+ # The lines below are to test multiple EnvParameters being passed
+ # for i in EnvParameters:
+ # cherrypy.log("EnvParameter object: " + i)
+ # cherrypy.log(" EnvParameter Value: " + EnvParameters[ i ])
+
+ pnf_flag = EnvParameters.get("pnf-flag", "")
+ if pnf_flag == "true":
+ return process_pnf_playbook(input_json, Id, EnvParameters, time_now)
+ else:
+ return process_vnf_playbook(input_json, Id, EnvParameters, time_now)
+ else:
+ cherrypy.log("TEST ID ALREADY DEFINED")
+ return {"StatusCode": 101, "StatusMessage": "TEST ID ALREADY DEFINED"}
+ else:
+ return {"StatusCode": 500, "StatusMessage": "JSON OBJECT MUST INCLUDE: ID, PLAYBOOKNAME, EnvParameters"}
+
+
+def handle_get_method(input_data):
+ # Verify we have a Type passed in GET request
+ if 'Type' not in input_data:
+ return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}
+
+ if AUTH:
+ cherrypy.log("Request USER: " + cherrypy.request.login)
+ cherrypy.log("Payload: " + str(input_data) + " Type " + input_data['Type'])
+
+ if 'LogRest' in input_data['Type']:
+ sys.stdout.close()
+ sys.stdout = open("/var/log/RestServer.log", "w")
+
+ # Just a debug to dump any records
+ if 'GetStatus' in input_data['Type']:
+ cherrypy.log("******** Dump Records **********")
+ if TestRecord.items():
+ for id, record in TestRecord.items():
+ cherrypy.log(" Id: " + id)
+ cherrypy.log("Record: " + str(record))
+ else:
+ cherrypy.log(" No Records to dump")
+
+ if 'Id' in input_data and 'Type' in input_data:
+ if not ('GetResult' in input_data['Type'] or 'GetOutputLog' in input_data['Type'] or
+ 'GetTheOutput' in input_data['Type'] or 'GetOutput' in input_data['Type'] or
+ 'GetLog' in input_data['Type']):
+ return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}
+
+ if input_data['Id'] in TestRecord:
+ if 'GetResult' in input_data['Type']:
+ cherrypy.log(" ** GetResult for: " + str(input_data['Id']))
+ if 'StatusMessage' in TestRecord[input_data['Id']]['Result'] and getresults_block:
+ # check if playbook is still running
+ while ActiveProcess[input_data['Id']].is_alive():
+ cherrypy.log("*** Playbook running returning PENDING for " + str(input_data['Id']))
+ # If still running return PENDING response
+ # if AnsibleServer != 'na':
+ # return {"StatusCode": 100, "StatusMessage": 'PENDING', "AnsibleServer": str(AnsibleServer)}
+ # else:
+ # return {"StatusCode": 100, "StatusMessage": 'PENDING'}
+ time.sleep(3)
+
+ # cherrypy.log( "*** Request released " + input_data['Id'])
+
+ cherrypy.log(str(TestRecord[input_data['Id']]['Result']))
+ cherrypy.log("Output: " + str(TestRecord[input_data['Id']]['Output']))
+ cherrypy.log("StatusCode: " + str(TestRecord[input_data['Id']]['Result']['StatusCode']))
+ cherrypy.log("StatusMessage: " + str(TestRecord[input_data['Id']]['Result']['StatusMessage']))
+
+ # out_obj gets returned to GET request
+ if TestRecord[input_data['Id']]['Result']['StatusCode'] == 500:
+ out_obj = TestRecord[input_data['Id']]['Result']['Results']
+ else:
+ out_obj = {
+ "StatusCode": 200,
+ "StatusMessage": "FINISHED",
+ "PlaybookName": TestRecord[input_data['Id']]["PlaybookName"],
+ "Version": TestRecord[input_data['Id']]["Version"],
+ "Duration": TestRecord[input_data['Id']]["Duration"],
+ "Output": TestRecord[input_data['Id']]["Output"]["Output"],
+ "Results": TestRecord[input_data['Id']]['Result']['Results']
+ }
+ if not TestRecord[input_data['Id']]['Output']['Output'] == {}:
+ cherrypy.log("TestRecord has Output:" + str(TestRecord[input_data['Id']]['Output']['Output']))
+ # PAP
+ for key in out_obj["Results"]:
+ cherrypy.log("Output key: " + str(key))
+ if key in TestRecord[input_data['Id']]['Output']['Output']:
+ out_obj["Results"][key]["Output"] = TestRecord[input_data['Id']]['Output']['Output'][key]
+
+ cherrypy.log("***** GET RETURNING RESULTS Back ****")
+ cherrypy.log(str(out_obj))
+ return out_obj
+ elif 'GetStatus' in input_data['Type']:
+ cherrypy.log(" Dump Records")
+ for id, record in TestRecord.items():
+ cherrypy.log(" id: " + id)
+ cherrypy.log(" Record:" + str(record))
+ elif 'GetTheOutput' in input_data['Type'] or 'GetOutput' in input_data['Type']:
+ if TestRecord[input_data['Id']]['Output'] == {} and getresults_block:
+ cherrypy.log("*** Request blocked " + input_data['Id'])
+
+ # while TestRecord[input_data['Id']]['Output'] == {} \
+ # or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
+ while ActiveProcess[input_data['Id']].is_alive():
+ time.sleep(3)
+
+ cherrypy.log("*** Request released " + input_data['Id'])
+
+ cherrypy.log("Output: " + str(TestRecord[input_data['Id']]['Output']))
+ return {"Output": TestRecord[input_data['Id']]['Output']['Output']}
+ elif 'GetOutputLog' in input_data['Type']:
+ cherrypy.log("GetOutputLog: processing.")
+ if glob.glob(ANSIBLE_TEMP + '/*_' + input_data['Id']):
+ id = input_data['Id']
+ cherrypy.log("Old directory found for ID: " + id)
+ run_dir = glob.glob(ANSIBLE_TEMP + '/*_' + input_data['Id'])
+ for dir in run_dir:
+ rdir = dir
+ if os.path.exists(rdir + "/PlaybookExDir.txt"):
+ cherrypy.log("Found PlaybookExDir.txt file")
+ f = open(rdir + '/PlaybookExDir.txt', 'r')
+ playbookexdir = f.readline()
+ rdir = playbookexdir
+ f.close()
+ cherrypy.log("Id: " + id)
+ cherrypy.log("RunDir: " + rdir)
+ if os.path.exists(rdir + "/output.log"):
+ cherrypy.log("Found output.log file")
+ f = open(rdir + '/output.log', 'r')
+ output_log = f.readline()
+ f.close()
+ return output_log
+ else:
+ cherrypy.log("Globglob failed:")
+ return
+
+ else:
+ # GetLog
+ if TestRecord[input_data['Id']]['Log'] == '' and \
+ getresults_block:
+
+ cherrypy.log("*** Request blocked " + input_data['Id'])
+
+ while TestRecord[input_data['Id']]['Log'] == '' \
+ or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
+ time.sleep(5)
+
+ cherrypy.log("*** Request released " + input_data['Id'])
+
+ cherrypy.log("Log:" + str(TestRecord[input_data['Id']]['Log']))
+ return {"Log": TestRecord[input_data['Id']]['Log']}
+ else:
+ # Not in memory check for a file
+ if glob.glob(ANSIBLE_TEMP + '/*_' + input_data['Id']):
+ id = input_data['Id']
+ cherrypy.log("Old directory found for ID: " + id)
+ run_dir = glob.glob(ANSIBLE_TEMP + '/*_' + input_data['Id'])
+ for dir in run_dir:
+ rdir = dir
+ if os.path.exists(rdir + "/PlaybookExDir.txt"):
+ cherrypy.log("Found PlaybookExDir.txt file")
+ f = open(rdir + '/PlaybookExDir.txt', 'r')
+ playbookexdir = f.readline()
+ rdir = playbookexdir
+ f.close()
+ cherrypy.log("Id: " + id)
+ cherrypy.log("RunDir: " + rdir)
+ if 'GetLog' in input_data['Type']:
+ if os.path.exists(rdir + "/output.log"):
+ cherrypy.log("Found output.log file")
+ f = open(rdir + '/output.log', 'r')
+ output_log = f.readline()
+ f.close()
+ return output_log
+ elif 'GetOutputLog' in input_data['Type']:
+ if os.path.exists(rdir + "/output.log"):
+ cherrypy.log("Found output.log file")
+ f = open(rdir + '/output.log', 'r')
+ output_log = f.readline()
+ f.close()
+ return output_log
+ elif 'GetResult' in input_data['Type']:
+ if os.path.exists(rdir + "/PlaybookName.txt"):
+ cherrypy.log("Found PlaybookName.txt file")
+ f = open(rdir + '/PlaybookName.txt', 'r')
+ playbooknametxt = f.readline()
+ f.close()
+ else:
+ playbooknametxt = "NA"
+
+ # Add code to get other items not just output.log from files
+ if os.path.exists(rdir + "/log.file"):
+ cherrypy.log("Found log.file")
+ out_results = "NA:"
+
+ f = open(rdir + '/log.file', 'r')
+ line = f.readline()
+ while line:
+ if "fatal" in line:
+ out_results = out_results + line
+ elif "RECAP" in line:
+ out_results = out_results + line
+ recap_line = f.readline()
+ while recap_line:
+ out_results = out_results + recap_line
+ recap_line = f.readline()
+ line = f.readline()
+ f.close()
+ out_obj = {
+ "StatusCode": 200,
+ "StatusMessage": "FINISHED",
+ "PlaybookName": playbooknametxt,
+ "Version": "Version",
+ "Duration": 200,
+ "Results": out_results
+ }
+ return out_obj
+ else:
+ return {"StatusCode": 500, "StatusMessage": "PLAYBOOK FAILED "}
+
+ return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}
+ else:
+ return {"StatusCode": 500, "StatusMessage": "MALFORMED REQUEST"}
+
+
+def handle_delete_method(input_data):
+ cherrypy.log("***> in RestServer.DELETE")
+ cherrypy.log("Payload: " + str(input_data))
+
+ if input_data['Id'] in TestRecord:
+ if 'PENDING' not in TestRecord[input_data['Id']]['Result']:
+ cherrypy.log(" Path: " + str(TestRecord[input_data['Id']]['Path']))
+ TestRecord.pop(input_data['Id'])
+ if input_data['Id'] in ActiveProcess:
+ ActiveProcess.pop(input_data['Id'])
+ return {"StatusCode": 200, "StatusMessage": "PLAYBOOK EXECUTION RECORDS DELETED"}
+ else:
+ return {"StatusCode": 200, "StatusMessage": "PENDING"}
+ else:
+ return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}
+
+
+class TestManager(object):
@cherrypy.expose
@cherrypy.tools.json_out()
@cherrypy.tools.json_in()
@cherrypy.tools.allow(methods=['POST', 'GET', 'DELETE'])
-
def Dispatch(self, **kwargs):
-
# Let cherrypy error handler deal with malformed requests
# No need for explicit error handler, we use default ones
@@ -310,610 +941,36 @@ class TestManager (object):
# Do cleanup too of ActiveProcess list and old Records - PAP
if TestRecord:
for key in TestRecord.copy():
- cherrypy.log( "LOOKING AT ALL TestRecords: " + str(key))
+ cherrypy.log("LOOKING AT ALL TestRecords: " + str(key))
if key in ActiveProcess:
- if not ActiveProcess[key].is_alive(): # Just to cleanup defunct processes
- cherrypy.log( "Not ActiveProcess for ID: " + str(key))
+ if not ActiveProcess[key].is_alive(): # Just to cleanup defunct processes
+ cherrypy.log("Not ActiveProcess for ID: " + str(key))
delta_time = (time_now - TestRecord[key]['Time']).seconds
- if delta_time > 2*TestRecord[key]['Timeout']:
- cherrypy.log( "DELETED HISTORY for ID: " + str(key))
+ if delta_time > 2 * TestRecord[key]['Timeout']:
+ cherrypy.log("DELETED HISTORY for ID: " + str(key))
if key in ActiveProcess:
- if not ActiveProcess[key].is_alive():
- ActiveProcess.pop (key)
- cherrypy.log( "DELETED ActiveProcess for ID: " + str(key))
- #if os.path.exists(TestRecord[key]['Path']):
- # don't remove run dirrectory
- #shutil.rmtree (TestRecord[key]['Path'])
+ if not ActiveProcess[key].is_alive():
+ ActiveProcess.pop(key)
+ cherrypy.log("DELETED ActiveProcess for ID: " + str(key))
+ # if os.path.exists(TestRecord[key]['Path']):
+ # don't remove run dirrectory
+ # shutil.rmtree (TestRecord[key]['Path'])
del TestRecord[key]
cherrypy.log("RestServer.Dispatch: " + cherrypy.request.method)
-
if 'POST' in cherrypy.request.method:
-
input_json = cherrypy.request.json
- cherrypy.log("Payload: " + str(input_json))
-
- if 'Id' in input_json and 'PlaybookName' in input_json and 'EnvParameters' in input_json:
-
- if True:
-
- if not input_json['Id'] in TestRecord:
- # check if Id exists in previous run dirctory
- # if so retun error
- s_cmd = 'ls ' + ansible_temp + '/*_' + input_json['Id']
- #if subprocess.check_output([s_cmd, ]):
- Id = input_json['Id']
- if glob.glob( ansible_temp + '/*_' + input_json['Id']):
- cherrypy.log("Old directory found for ID: " + Id)
- return {"StatusCode": 101, "StatusMessage": "TEST ID FILE ALREADY DEFINED"}
-
- PlaybookName = input_json['PlaybookName']
- # if required it should be passed as an argument
- EnvParameters = input_json['EnvParameters']
-
- # The lines below are to test multiple EnvParameters being passed
- #for i in EnvParameters:
- # cherrypy.log("EnvParameter object: " + i)
- # cherrypy.log(" EnvParameter Value: " + EnvParameters[ i ])
-
- # Now get things out of EnvParameters
- VNF_instance = None
- VNF_instance = EnvParameters.get('vnf_instance')
-
- # Get Version if present
- version = None
- if 'Version' in input_json:
- version = input_json['Version']
-
- # GetInventoryNames
- HaveNodeList = False
- HaveInventoryNames = False
- inventory_names = None
- if 'InventoryNames' in input_json:
- inventory_names = input_json['InventoryNames']
- HaveInventoryNames = True
-
- #AnsibleInvFail = True
- AnsiblePlaybookFail = True
-
- LocalNodeList = None
-
- str_uuid = str (uuid.uuid4())
-
-
- VnfType= PlaybookName.split("/")[0]
- if auth:
- cherrypy.log( "Request USER : " + cherrypy.request.login)
- cherrypy.log( "Request Decode: ID " + Id)
- cherrypy.log( "Request Decode: VnfType " + VnfType)
- cherrypy.log( "Request Decode: EnvParameters " + json.dumps(EnvParameters))
-
- # Verify VNF_instance was passed in EnvParameters
- if VNF_instance != None:
- cherrypy.log( "Request Decode: VnfInstance " + VNF_instance)
- else:
- cherrypy.log( "StatusCode: 107, StatusMessage: VNF_instance NOT PROVIDED" )
- return {"StatusCode": 107,
- "StatusMessage": "VNF_instance NOT PROVIDED"}
-
- if inventory_names != None:
- cherrypy.log( "Request Decode: Inventory Names " + inventory_names)
- else:
- cherrypy.log( "Request Decode: Inventory Names " + "Not provided")
-
- cherrypy.log( "Request Decode: PlaybookName " + PlaybookName)
- PlayBookFunction = PlaybookName.rsplit("/",2)[1]
- PlayBookFile = PlayBookFunction + "/site.yml"
- cherrypy.log( "Request Decode: PlaybookFunction " + PlayBookFunction)
- cherrypy.log( "Request Decode: Playbook file " + PlayBookFile)
-
- BaseDir = ansible_path + "/" + PlaybookName.rsplit("/",1)[0]
- CopyDir = ansible_path + "/" + PlaybookName.rsplit("/",2)[0]
- cherrypy.log( "Request Decode: Basedir " + BaseDir)
- cherrypy.log( "Request Decode: Copydir " + CopyDir)
-
-
- PlaybookDir = ansible_temp + "/" + \
- VNF_instance + "_" + str_uuid + "_" + str(Id)
-
- # AnsibleInv is the directory where the host file to be run exsists
- AnsibleInv = ansible_path + "/" + VnfType + "/latest/ansible/inventory/" + VNF_instance
- ArchiveFlag = False
-
- # Create base run directory if it doesn't exist
- if not os.path.exists(ansible_temp):
- cherrypy.log( "Creating Base Run Directory: " + ansible_temp)
- os.makedirs(ansible_temp)
-
- if not os.path.exists( CopyDir ):
- cherrypy.log("Playbook Not Found")
- return {"StatusCode": 101,
- "StatusMessage": "PLAYBOOK NOT FOUND"}
-
- # copy static playbook dir to run dir
- cherrypy.log("Copying from " + CopyDir + " to " + PlaybookDir)
- shutil.copytree(CopyDir, PlaybookDir)
- cmd="/usr/bin/find " + PlaybookDir + " -exec /usr/bin/touch {} \;"
- cmd="/usr/bin/find " + PlaybookDir + " -exec chmod +rx {} \;"
- sys_call(cmd)
- cherrypy.log(cmd)
-
- cherrypy.log( "PlaybookDir: " + PlaybookDir)
- cherrypy.log( "AnsibleInv: " + AnsibleInv)
-
- # Process inventory file for target
-
- hostgrouplist = []
- hostnamelist = []
-
- NodeList = []
- if 'NodeList' in input_json:
- NodeList = input_json['NodeList']
-
- cherrypy.log("NodeList: " + str(NodeList));
-
- # if NodeList empty
- if NodeList == []:
- cherrypy.log( "*** NodeList - Empty ***")
-
- else:
- HaveNodeList = True
-
- ###############################################################################
- ##### Host file processing ###########################
- ##### 1. Use file delivered with playbook ###########################
- ##### 2. If HostNames + NodeList generate and use ###########################
- ###############################################################################
-
- #Verify inventory directory exists
- path = PlaybookDir + "/inventory/"
- if not os.path.isdir(path):
- cherrypy.log ("Inventory directory %s does not exist - create it" % path)
- try:
- os.mkdir(path)
- except OSError:
- cherrypy.log ("Creation of the directory %s failed" % path)
- else:
- cherrypy.log ("Successfully created the directory %s " % path)
-
- #location of host file - Default
- HostFile = PlaybookDir + "/inventory/" + VNF_instance + "hosts"
- cherrypy.log("HostFile: " + HostFile)
-
- # if NodeList and InventoryNames need to build host file
- if HaveInventoryNames & HaveNodeList:
- cherrypy.log("Build host file from NodeList")
- ret = buildHostsSysCall (input_json, PlaybookDir, inventory_names)
- if (ret < 0):
- cherrypy.log("Returning Error: Not running Playbook")
- return {"StatusCode": 105,
- "StatusMessage": "NodeList: Missing vnfc-type field"}
-
- # Having been built now copy new file to correct file
- shutil.copy(PlaybookDir + "/host_file.txt", HostFile)
- cherrypy.log("Copying Generated host file to: " + HostFile)
-
- timeout = timeout_seconds
- if 'Timeout' in input_json:
- timeout = int (input_json['Timeout'])
- cherrypy.log("Timeout from API: " + str(timeout))
-
- else:
- cherrypy.log("Timeout not passed from API using default: " + str(timeout))
-
- EnvParam = {}
- if 'EnvParameters' in input_json:
- EnvParam = input_json['EnvParameters']
-
- LocalParam = {}
- if 'LocalParameters' in input_json:
- LocalParam = input_json['LocalParameters']
-
- FileParam = {}
- if 'FileParameters' in input_json:
- FileParam = input_json['FileParameters']
-
- callback_flag = None
- if 'CallBack' in input_json:
- callback_flag = input_json['CallBack']
-
- # if AnsibleServer is not set to 'na' don't send AnsibleServer in PENDING responce.
- if AnsibleServer != 'na':
- TestRecord[Id] = {'PlaybookName': PlaybookName,
- 'Version': version,
- 'NodeList': NodeList,
- 'HostGroupList': hostgrouplist,
- 'HostNameList': hostnamelist,
- 'Time': time_now,
- 'Duration': timeout,
- 'Timeout': timeout,
- 'EnvParameters': EnvParam,
- 'LocalParameters': LocalParam,
- 'FileParameters': FileParam,
- 'CallBack': callback_flag,
- 'Result': {"StatusCode": 100,
- "StatusMessage": 'PENDING',
- "AnsibleServer": str(AnsibleServer),
- "ExpectedDuration": str(timeout) + "sec"},
- 'Log': '',
- 'Output': {},
- 'Path': PlaybookDir,
- 'Mandatory': None}
- else:
- TestRecord[Id] = {'PlaybookName': PlaybookName,
- 'Version': version,
- 'NodeList': NodeList,
- 'HostGroupList': hostgrouplist,
- 'HostNameList': hostnamelist,
- 'Time': time_now,
- 'Duration': timeout,
- 'Timeout': timeout,
- 'EnvParameters': EnvParam,
- 'LocalParameters': LocalParam,
- 'FileParameters': FileParam,
- 'CallBack': callback_flag,
- 'Result': {"StatusCode": 100,
- "StatusMessage": 'PENDING',
- "ExpectedDuration": str(timeout) + "sec"},
- 'Log': '',
- 'Output': {},
- 'Path': PlaybookDir,
- 'Mandatory': None}
-
- cherrypy.log("Test_Record: " + str(TestRecord[Id]))
- # Write files
-
- if not TestRecord[Id]['FileParameters'] == {}:
- for key in TestRecord[Id]['FileParameters']:
- filename = key
- filecontent = TestRecord[Id]['FileParameters'][key]
- f = open(PlaybookDir + "/" + filename, "w")
- f.write(filecontent)
- f.close()
-
-
- # Process playbook
- if os.path.exists( ansible_path + '/' + PlaybookName):
- AnsiblePlaybookFail = False
-
- if AnsiblePlaybookFail:
- #if os.path.exists(PlaybookDir):
- #shutil.rmtree (PlaybookDir)
- del TestRecord[Id]
- return {"StatusCode": 101,
- "StatusMessage": "PLAYBOOK NOT FOUND"}
- else:
-
- # Test EnvParameters
- playbook_path = PlaybookDir
-
- # Store local vars
- if not os.path.exists(playbook_path + "/vars"):
- os.mkdir(playbook_path + "/vars")
- if not os.path.isfile(playbook_path + "/vars/defaults.yml"):
- os.mknod(playbook_path + "/vars/defaults.yml")
-
- ###################################################
- # PAP
- #write local parameters passed into defaults.yml
- # PAP
- f = open(playbook_path + "/vars/defaults.yml","a")
- #for id, record in TestRecord.items():
- print TestRecord[Id]['LocalParameters']
- local_parms = TestRecord[Id]['LocalParameters']
- for key, value in local_parms.items():
- f.write(key +"=" + value + "\n");
- f.close()
- ###################################################
-
- for key in TestRecord[Id]['LocalParameters']:
- host_index = []
- for i in range(len(TestRecord[Id]['HostNameList'])):
- if key in TestRecord[Id]['HostNameList'][i]:
- host_index.append(i)
- if len(host_index) == 0:
- for i in range(len(TestRecord[Id]['HostGroupList'])):
- if key in TestRecord[Id]['HostGroupList'][i]:
- host_index.append(i)
- if len(host_index) > 0:
- for i in range(len(host_index)):
- f = open(playbook_path + "/vars/" +
- TestRecord[Id]['HostNameList'][host_index[i]] +
- ".yml", "a")
- for param in TestRecord[Id]['LocalParameters'][key]:
- f.write(param + ": " +
- str (TestRecord[Id]['LocalParameters'][key][param]) +
- "\n")
- f.close()
-
-
- # write some info out to files before running
- if auth:
- f = open(playbook_path + "/User.txt", "a")
- f.write(cherrypy.request.login)
- f.close()
- f = open(playbook_path + "/PlaybookName.txt", "a")
- f.write(PlaybookName)
- f.close()
- f = open(playbook_path + "/PlaybookExDir.txt", "a")
- f.write(PlaybookDir + "/" + PlayBookFunction)
- f.close()
- f = open(playbook_path + "/JsonRequest.txt", "w")
- #f.write(str(input_json))
- #print( json.dumps(input_json, indent=4, sort_keys=True))
- f.write( json.dumps(input_json, indent=4, sort_keys=True))
- f.close()
-
-
- # Check that HostFile exists
- if not os.path.isfile(HostFile):
- cherrypy.log("Inventory file Not Found: " + HostFile)
- return {"StatusCode": 101,
- "StatusMessage": "PLAYBOOK INVENTORY FILE NOT FOUND"}
-
- # Cannot use thread because ansible module uses
- # signals which are only supported in main thread.
- # So use multiprocess with shared object
- # args = (callback, Id, PlaybookDir + "/" + AnsibleInv,
-
- p = Process(target = RunAnsible_Playbook,
- args = (callback, Id, HostFile,
- PlaybookDir + '/' + PlayBookFile,
- NodeList, TestRecord, PlaybookDir + "/" + PlayBookFunction,
- ArchiveFlag))
- p.start()
- ActiveProcess[Id] = p
- return TestRecord[Id]['Result']
- else:
- cherrypy.log("TEST ID ALREADY DEFINED")
- return {"StatusCode": 101, "StatusMessage": "TEST ID ALREADY DEFINED"}
-
- else:
- return {"StatusCode": 500, "StatusMessage": "REQUEST MUST INCLUDE: NODELIST"}
-
- else:
- return {"StatusCode": 500, "StatusMessage": "JSON OBJECT MUST INCLUDE: ID, PLAYBOOKNAME, EnvParameters"}
-
+ return handle_post_method(input_json, time_now)
elif 'GET' in cherrypy.request.method:
-
- # Lets pause for a second just incase the resquest was just kicked off
+ # Lets pause for a second just in case the request was just kicked off
time.sleep(1)
input_data = parse_query_string(cherrypy.request.query_string)
-
- # Verify we have a Type passed in GET request
- if not ( 'Type' in input_data):
- return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}
-
- if auth:
- cherrypy.log( "Request USER: " + cherrypy.request.login)
- cherrypy.log("Payload: " + str(input_data) + " Type " + input_data['Type'])
-
- if 'LogRest' in input_data['Type']:
- sys.stdout.close()
- sys.stdout = open("/var/log/RestServer.log", "w")
-
- # Just a debug to dump any records
- if 'GetStatus' in input_data['Type']:
- cherrypy.log( "******** Dump Records **********")
- if TestRecord.items():
- for id, record in TestRecord.items():
- cherrypy.log( " Id: " + id)
- cherrypy.log( "Record: " + str(record))
- else:
- cherrypy.log(" No Records to dump")
-
- if 'Id' in input_data and 'Type' in input_data:
- if not ('GetResult' in input_data['Type'] or 'GetOutputLog' in input_data['Type'] or'GetTheOutput' in input_data['Type'] or 'GetLog' in input_data['Type']):
- return {"StatusCode": 500, "StatusMessage": "RESULTS TYPE UNDEFINED"}
- if input_data['Id'] in TestRecord:
-
- if 'GetResult' in input_data['Type']:
-
- cherrypy.log( " ** GetResult for: " + str (input_data['Id']))
-
- if 'StatusMessage' in TestRecord[input_data['Id']]['Result'] and getresults_block:
-
-
- #check if playbook is still running
- while ActiveProcess[input_data['Id']].is_alive():
- cherrypy.log( "*** Playbook running returning PENDING for " + str(input_data['Id']))
- ##
- ## If still running return PENDING response
- ##
- if AnsibleServer != 'na':
- return {"StatusCode": 100,
- "StatusMessage": 'PENDING',
- "AnsibleServer": str(AnsibleServer)}
- else:
- return {"StatusCode": 100,
- "StatusMessage": 'PENDING'}
- #time.sleep(5)
-
- #cherrypy.log( "*** Request released " + input_data['Id'])
-
- cherrypy.log(str( TestRecord[input_data['Id']]['Result']))
- cherrypy.log("Output: " + str( TestRecord[input_data['Id']]['Output']))
- cherrypy.log("StatusCode: " + str( TestRecord[input_data['Id']]['Result']['StatusCode']))
- cherrypy.log("StatusMessage: " + str( TestRecord[input_data['Id']]['Result']['StatusMessage']))
-
- #out_obj gets returned to GET request
- if TestRecord[input_data['Id']]['Result']['StatusCode'] == 500:
- out_obj = TestRecord[input_data['Id']]['Result']['Results']
- else:
- out_obj = {"StatusCode": 200,
- "StatusMessage": "FINISHED",
- "PlaybookName": TestRecord[input_data['Id']]["PlaybookName"],
- "Version": TestRecord[input_data['Id']]["Version"],
- "Duration": TestRecord[input_data['Id']]["Duration"],
- "Output": TestRecord[input_data['Id']]["Output"]["Output"],
- "Results": TestRecord[input_data['Id']]['Result']['Results']}
- if not TestRecord[input_data['Id']]['Output']['Output'] == {}:
- cherrypy.log("TestRecord has Output:" + str(TestRecord[input_data['Id']]['Output']['Output']))
- # PAP
- for key in out_obj["Results"]:
- cherrypy.log("Output key: " + str(key))
- if key in TestRecord[input_data['Id']]['Output']['Output']:
- out_obj["Results"][key]["Output"] = TestRecord[input_data['Id']]['Output']['Output'][key]
-
- cherrypy.log("***** GET RETURNING RESULTS Back ****")
- cherrypy.log(str(out_obj))
- return out_obj
-
- elif 'GetStatus' in input_data['Type']:
- print " Dump Records"
- for id, record in TestRecord,items():
- print " id: " + id
- print " Record:" + str(reecord)
-
- elif 'GetTheOutput' in input_data['Type']:
-
- if TestRecord[input_data['Id']]['Output'] == {} and \
- getresults_block:
-
- cherrypy.log( "*** Request blocked " + input_data['Id'])
-
- while TestRecord[input_data['Id']]['Output'] == {} \
- or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
- time.sleep(5)
-
- cherrypy.log( "*** Request released " + input_data['Id'])
-
- cherrypy.log( "Output: " + str(TestRecord[input_data['Id']]['Output']))
- return {"Output": TestRecord[input_data['Id']]['Output']['Output']}
-
- elif 'GetOutputLog' in input_data['Type']:
- cherrypy.log("GetOutputLog: processing.")
- if glob.glob( ansible_temp + '/*_' + input_data['Id']):
- id = input_data['Id']
- cherrypy.log("Old directory found for ID: " + id)
- run_dir = glob.glob( ansible_temp + '/*_' + input_data['Id'])
- for dir in run_dir:
- rdir=dir
- if os.path.exists (rdir + "/PlaybookExDir.txt"):
- cherrypy.log("Found PlaybookExDir.txt file")
- f = open( rdir + '/PlaybookExDir.txt', 'r')
- playbookexdir = f.readline()
- rdir = playbookexdir
- f.close()
- cherrypy.log("Id: " + id)
- cherrypy.log("RunDir: " + rdir)
- if os.path.exists( rdir + "/output.log"):
- cherrypy.log("Found output.log file")
- f = open( rdir + '/output.log', 'r')
- output_log = f.readline()
- f.close()
- return output_log
- else:
- cherrypy.log("Globglob failed:")
- return
-
- else:
- # GetLog
-
- if TestRecord[input_data['Id']]['Log'] == '' and \
- getresults_block:
-
- cherrypy.log( "*** Request blocked " + input_data['Id'])
-
- while TestRecord[input_data['Id']]['Log'] == '' \
- or 'StatusMessage' in TestRecord[input_data['Id']]['Result']:
- time.sleep(5)
-
- cherrypy.log( "*** Request released " + input_data['Id'])
-
- cherrypy.log( "Log:" + str(TestRecord[input_data['Id']]['Log']))
- return {"Log": TestRecord[input_data['Id']]['Log']}
- else:
- # Not in memory check for a file
- if glob.glob( ansible_temp + '/*_' + input_data['Id']):
- id = input_data['Id']
- cherrypy.log("Old directory found for ID: " + id)
- run_dir = glob.glob( ansible_temp + '/*_' + input_data['Id'])
- for dir in run_dir:
- rdir=dir
- if os.path.exists (rdir + "/PlaybookExDir.txt"):
- cherrypy.log("Found PlaybookExDir.txt file")
- f = open( rdir + '/PlaybookExDir.txt', 'r')
- playbookexdir = f.readline()
- rdir = playbookexdir
- f.close()
- cherrypy.log("Id: " + id)
- cherrypy.log("RunDir: " + rdir)
- if 'GetLog' in input_data['Type']:
- if os.path.exists( rdir + "/output.log"):
- cherrypy.log("Found output.log file")
- f = open( rdir + '/output.log', 'r')
- output_log = f.readline()
- f.close()
- return output_log
- elif 'GetOutputLog' in input_data['Type']:
- if os.path.exists( rdir + "/output.log"):
- cherrypy.log("Found output.log file")
- f = open( rdir + '/output.log', 'r')
- output_log = f.readline()
- f.close()
- return output_log
- elif 'GetResult' in input_data['Type']:
- if os.path.exists (rdir + "/PlaybookName.txt"):
- cherrypy.log("Found PlaybookName.txt file")
- f = open( rdir + '/PlaybookName.txt', 'r')
- playbooknametxt = f.readline()
- f.close()
- else:
- playbooknametxt = "NA"
-
- # Add code to get other items not just output.log from files
- if os.path.exists( rdir + "/log.file"):
- cherrypy.log("Found log.file")
- out_results = "NA:"
- f = open( rdir + '/log.file', 'r')
-
- line = f.readline()
- while line :
- if "fatal" in line:
- out_results = out_results + line
- elif "RECAP" in line:
- out_results = out_results + line
- recap_line = f.readline()
- while recap_line :
- out_results = out_results + recap_line
- recap_line = f.readline()
- line = f.readline()
- f.close()
- out_obj = {"StatusCode": 200,
- "StatusMessage": "FINISHED",
- "PlaybookName": playbooknametxt,
- "Version": "Version",
- "Duration": 200,
- "Results": out_results}
- return out_obj
- else:
- return {"StatusCode": 500, "StatusMessage": "PLAYBOOK FAILED "}
-
-
- return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}
- else:
- return {"StatusCode": 500, "StatusMessage": "MALFORMED REQUEST"}
+ return handle_get_method(input_data)
elif 'DELETE' in cherrypy.request.method:
input_data = parse_query_string(cherrypy.request.query_string)
-
- cherrypy.log( "***> in RestServer.DELETE")
- cherrypy.log("Payload: " + str(input_data))
-
- if input_data['Id'] in TestRecord:
- if not 'PENDING' in TestRecord[input_data['Id']]['Result']:
- cherrypy.log(" Path: " + str(TestRecord[input_data['Id']]['Path']))
- TestRecord.pop (input_data['Id'])
- if input_data['Id'] in ActiveProcess:
- ActiveProcess.pop (input_data['Id'])
-
- return {"StatusCode": 200, "StatusMessage": "PLAYBOOK EXECUTION RECORDS DELETED"}
- else:
- return {"StatusCode": 200, "StatusMessage": "PENDING"}
- else:
- return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"}
+ return handle_delete_method(input_data)
if __name__ == '__main__':
@@ -923,60 +980,66 @@ if __name__ == '__main__':
config_file_path = "RestServer_config"
if not os.path.exists(config_file_path):
- print '[INFO] The config file does not exist'
+ cherrypy.log('[INFO] The config file does not exist')
sys.exit(0)
ip = 'na'
AnsibleServer = 'na'
port = 'na'
tls = False
- auth = False
+ AUTH = False
pub = 'na'
priv = 'na'
+ intermediate = 'na'
timeout_seconds = 'na'
- ansible_path = 'na'
- ansible_temp = 'na'
+ ANSIBLE_PATH = 'na'
+ ANSIBLE_TEMP = 'na'
host = 'na'
- users= 'na'
+ users = 'na'
getresults_block = False
from_files = False
- file = open(config_file_path, 'r')
- for line in file.readlines():
- if '#' not in line:
- if 'ip:' in line:
- ip = line.split(':')[1].strip()
- elif 'AnsibleServer:' in line:
- AnsibleServer = line.split(':')[1].strip()
- elif 'port:' in line:
- port = line.split(':')[1].strip()
- elif 'ksalt:' in line:
- salt = line.split(':')[1].strip()
- elif 'tls:' in line:
- tls = 'YES' in line.split(':')[1].strip().upper()
- elif 'auth:' in line:
- auth = 'YES' in line.split(':')[1].strip().upper()
- if tls and 'priv:' in line:
- priv = line.split(':')[1].strip()
- if tls and 'pub:' in line:
- pub = line.split(':')[1].strip()
- if tls and 'inter_cert:' in line:
- intermediate = line.split(':')[1].strip()
- if 'timeout_seconds' in line:
- timeout_seconds = int (line.split(':')[1].strip())
- if 'ansible_path' in line:
- ansible_path = line.split(':')[1].strip()
- if 'ansible_temp' in line:
- ansible_temp = line.split(':')[1].strip()
- if 'host' in line:
- host = line.split(':')[1].strip()
- if 'users' in line:
- users = line.split(':')[1].strip()
- if 'getresults_block' in line:
- getresults_block = 'YES' in line.split(':')[1].strip().upper()
- if 'from_files' in line:
- from_files = 'YES' in line.split(':')[1].strip().upper()
- file.close()
+ config_file = open(config_file_path, 'r')
+ for config_line in config_file.readlines():
+ if '#' not in config_line:
+ if 'ip:' in config_line:
+ ip = config_line.split(':')[1].strip()
+ elif 'AnsibleServer:' in config_line:
+ AnsibleServer = config_line.split(':')[1].strip()
+ elif 'port:' in config_line:
+ port = config_line.split(':')[1].strip()
+ elif 'ksalt:' in config_line:
+ salt = config_line.split(':')[1].strip()
+ elif 'tls:' in config_line:
+ tls = 'YES' in config_line.split(':')[1].strip().upper()
+ elif 'auth:' in config_line:
+ AUTH = 'YES' in config_line.split(':')[1].strip().upper()
+ if tls and 'priv:' in config_line:
+ priv = config_line.split(':')[1].strip()
+ if tls and 'pub:' in config_line:
+ pub = config_line.split(':')[1].strip()
+ if tls and 'inter_cert:' in config_line:
+ intermediate = config_line.split(':')[1].strip()
+ if 'timeout_seconds' in config_line:
+ timeout_seconds = int(config_line.split(':')[1].strip())
+ if 'ansible_path' in config_line:
+ ANSIBLE_PATH = config_line.split(':')[1].strip()
+ if 'ansible_inv' in config_line:
+ ANSIBLE_INV = config_line.split(':')[1].strip()
+ if not os.path.exists(ANSIBLE_PATH + "/" + ANSIBLE_INV):
+ print '[INFO] The ansible_inv file does not exist'
+ sys.exit(0)
+ if 'ansible_temp' in config_line:
+ ANSIBLE_TEMP = config_line.split(':')[1].strip()
+ if 'host' in config_line:
+ host = config_line.split(':')[1].strip()
+ if 'users' in config_line:
+ users = config_line.split(':')[1].strip()
+ if 'getresults_block' in config_line:
+ getresults_block = 'YES' in config_line.split(':')[1].strip().upper()
+ if 'from_files' in config_line:
+ from_files = 'YES' in config_line.split(':')[1].strip().upper()
+ config_file.close()
# Initialization
@@ -987,8 +1050,8 @@ if __name__ == '__main__':
'server.socket_host': ip,
'server.socket_port': int(port),
'server.protocol_version': 'HTTP/1.1'
- }
}
+ }
if tls:
# Use pythons built-in SSL
@@ -997,42 +1060,41 @@ if __name__ == '__main__':
# Point to certificate files
if not os.path.exists(pub):
- print '[INFO] The public certificate does not exist'
+ cherrypy.log('[INFO] The public certificate does not exist')
sys.exit(0)
if not os.path.exists(priv):
- print '[INFO] The private key does not exist'
+ cherrypy.log('[INFO] The private key does not exist')
sys.exit(0)
if not os.path.exists(intermediate):
- print '[INFO] The intermediate certificate does not exist'
+ cherrypy.log('[INFO] The intermediate certificate does not exist')
sys.exit(0)
-
cherrypy.server.ssl_certificate = pub
cherrypy.server.ssl_certificate_chain = intermediate
cherrypy.server.ssl_private_key = priv
- if auth:
+ if AUTH:
# Read in and build user dictionary
if not os.path.exists(users):
- print '[INFO] The users file does not exist: ' + users
- sys.exit(0)
+ cherrypy.log('[INFO] The users file does not exist: ' + users)
+ sys.exit(0)
userpassdict = {}
user_file = open(users, 'r')
- for line in user_file.readlines():
- if '#' not in line:
- id = line.split(':')[0].strip()
- pw = line.split(':')[1].strip()
- userpassdict[id] = pw
- #print str(userpassdict)
-
- app_conf = {'/':
- {'tools.auth_basic.on': True,
- 'tools.auth_basic.realm': 'earth',
- 'tools.auth_basic.checkpassword': validate_password,
- }
- }
+ for config_line in user_file.readlines():
+ if '#' not in config_line:
+ uid = config_line.split(':')[0].strip()
+ pw = config_line.split(':')[1].strip()
+ userpassdict[uid] = pw
+
+ app_conf = {
+ '/':
+ {'tools.auth_basic.on': True,
+ 'tools.auth_basic.realm': 'earth',
+ 'tools.auth_basic.checkpassword': validate_password
+ }
+ }
application = cherrypy.tree.mount(TestManager(), '/', app_conf)
else:
@@ -1049,9 +1111,11 @@ if __name__ == '__main__':
log.error_file = ""
log.access_file = ""
from logging import handlers
+
applicationLogFileHandler = handlers.RotatingFileHandler(applicationLogName, 'a', 1000000, 5000)
accessLogFileHandler = handlers.RotatingFileHandler(accessLogName, 'a', 1000000, 5000)
import logging
+
applicationLogFileHandler.setLevel(logging.DEBUG)
log.error_log.addHandler(applicationLogFileHandler)
log.access_log.addHandler(accessLogFileHandler)