X-Git-Url: https://gerrit.onap.org/r/gitweb?p=ccsdk%2Fdistribution.git;a=blobdiff_plain;f=ansible-server%2Fsrc%2Fmain%2Fansible-server%2FRestServer.py;h=483b83224482d46da1f44f65ed0abcf256ef3c0a;hp=d5f85d1a760a57c4992cb8c5072167e33e843f79;hb=ef127a5f590c092e5020d8ecd6cf10eaa8329ddc;hpb=1b3e909f413e81a50cb5f44e187ea3cdd1e32f69 diff --git a/ansible-server/src/main/ansible-server/RestServer.py b/ansible-server/src/main/ansible-server/RestServer.py old mode 100644 new mode 100755 index d5f85d1a..483b8322 --- a/ansible-server/src/main/ansible-server/RestServer.py +++ b/ansible-server/src/main/ansible-server/RestServer.py @@ -1,10 +1,9 @@ ''' -#!/usr/bin/python /*- * ============LICENSE_START======================================================= * ONAP : APPC * ================================================================================ -* Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. +* Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Copyright (C) 2017 Amdocs * ============================================================================= @@ -19,29 +18,27 @@ * 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. -* -* ECOMP is a trademark and service mark of AT&T Intellectual Property. +* * ============LICENSE_END========================================================= */ ''' -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 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 - -import AnsibleSql -from AnsibleSql import readPlaybook, readCredentials +from BuildHostFile import buildHostsSysCall +from BuildPlaybookParams import buildInventorySysCall, getPlaybookFile from os import listdir from os.path import isfile, join @@ -49,18 +46,26 @@ from os.path import isfile, join TestRecord = Manager().dict() ActiveProcess = {} -def sys_call (cmd): + +def validate_password(realm, username, password): + comp = crypt.crypt(password, salt) + if username in userpassdict and userpassdict[username] == comp: + return True + return False + + +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() @@ -75,755 +80,897 @@ 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'], - 'LCM': TestRecord[Id]['LCM'], - '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'] + } + + if TestRecord[Id]['CallBack'] is not None: # 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'] + } - 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"])) if not TestRecord[Id]['Output']['Output'] == {}: for key in data["Results"]: 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]['LCM'], - TestRecord[Id]['Timeout']) + retval, log, returncode = ansibleSysCall(Inventory, Playbook, NodeList, + TestRecord[Id]['Mandatory'], + TestRecord[Id]['EnvParameters'], + TestRecord[Id]['LocalParameters'], + TestRecord[Id]['Timeout'], + Path) - - print " returncode:", returncode - print " retval: ", retval - print " log: ", log + cherrypy.log("Return code:" + str(returncode)) + cherrypy.log("Return value:" + str(retval)) Log = ''.join(log) - Output = {'Output': {}} + if pnf_flag: + Output = {'Output': {}} + else: + Output = {} onlyfiles = [f for f in listdir(Path) if isfile(join(Path, f))] + cherrypy.log("Checking for results.txt files: ") for file in onlyfiles: if "results.txt" in file: + # if file.endswith("results.txt"): + cherrypy.log("results file: " + file) f = open(Path + "/" + file, "r") - key = file.split("_")[0] - Output['Output'][key] = f.read() + 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': {}} + 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) + callback(Id, Result, Output, Log, returncode) -class TestManager (object): - @cherrypy.expose - @cherrypy.tools.json_out() - @cherrypy.tools.json_in() - @cherrypy.tools.allow(methods=['POST', 'GET', 'DELETE']) +def store_local_vars(playbook_path, Id): + if not os.path.exists(playbook_path + "/vars"): + os.mkdir(playbook_path + "/vars") - def Dispatch(self, **kwargs): + if not os.path.isfile(playbook_path + "/vars/defaults.yml"): + os.mknod(playbook_path + "/vars/defaults.yml") - # Let cherrypy error handler deal with malformed requests - # No need for explicit error handler, we use default ones + # ################################################## + # PAP + # write local parameters passed into defaults.yml + # PAP + local_parms = TestRecord[Id]['LocalParameters'] + cherrypy.log("LocalParameters: " + str(local_parms)) - time_now = datetime.datetime.utcnow() + f = open(playbook_path + "/vars/defaults.yml", "a") + for key, value in list(local_parms.items()): + f.write(key + "=" + value + "\n") + f.close() + # ################################################## - # Erase old test results (2x timeout) - if TestRecord: - for key in TestRecord.copy(): - delta_time = (time_now - TestRecord[key]['Time']).seconds - if delta_time > 2*TestRecord[key]['Timeout']: - print "Deleted history for test", key - if os.path.exists(TestRecord[key]['Path']): - shutil.rmtree (TestRecord[key]['Path']) - del TestRecord[key] + 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() - print "***> in RestServer.Dispatch:", cherrypy.request.method - HomeDir = os.path.dirname(os.path.realpath("~/")) +def process_pnf_playbook(input_json, Id, EnvParameters, time_now): + cherrypy.log("Processing playbook for PNF...") - if 'POST' in cherrypy.request.method: + PlaybookName = input_json['PlaybookName'] + version = input_json.get('Version', None) - input_json = cherrypy.request.json - print " Payload: ", input_json + 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) - if 'Id' in input_json and 'PlaybookName' in input_json: + str_uuid = str(uuid.uuid4()) - if True: + HomeDir = os.path.dirname(os.path.realpath("~/")) - if not input_json['Id'] in TestRecord: + PlaybookType = PlaybookName.split(".")[0].split('_')[-1] + PlaybookDir = HomeDir + '/' + ANSIBLE_TEMP + "/" + PlaybookName.split(".")[0] + "_" + str_uuid + AnsibleInv = PlaybookType + "_" + "inventory" + ArchiveFlag = False - Id = input_json['Id'] - PlaybookName = input_json['PlaybookName'] + cherrypy.log("Request Decode: PlaybookType " + PlaybookType) + cherrypy.log("Request Decode: PlaybookDir " + PlaybookDir) + cherrypy.log("Request Decode: AnsibleInv " + AnsibleInv) - version = None - if 'Version' in input_json: - version = input_json['Version'] + NodeList = input_json.get('NodeList', []) + cherrypy.log("Request Decode: NodeList: " + str(NodeList)) - AnsibleInvFail = True - AnsiblePlaybookFail = True + # 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) - MySqlConFail = True - MySqlCause = '' + os.mkdir(PlaybookDir) - LocalNodeList = None + # Process inventory file for target + hostgrouplist = [] + hostnamelist = [] - str_uuid = str (uuid.uuid4()) + buildInventorySysCall(ANSIBLE_PATH, ANSIBLE_INV, NodeList, PlaybookDir, AnsibleInv, hostgrouplist, hostnamelist) - LCM = PlaybookName.split(".")[0].split('_')[-1] - PlaybookDir = HomeDir + "/" + ansible_temp + "/" + \ - PlaybookName.split(".")[0] + "_" + str_uuid - AnsibleInv = LCM + "_" + "inventory" - ArchiveFlag = False + version_target = getPlaybookFile(ANSIBLE_PATH, PlaybookName, PlaybookType, PlaybookDir) + if not version_target: + return {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"} - print " LCM: ", LCM - print " PlaybookDir: ", ansible_temp + PlaybookDir.split(ansible_temp)[1] - print " AnsibleInv: ", AnsibleInv - print " ansible_temp: ", ansible_temp + if version is None: + version = version_target - if not os.path.exists(HomeDir + "/" + ansible_temp): - os.makedirs(HomeDir + "/" + ansible_temp) + 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() - os.mkdir(PlaybookDir) + playbook_path = PlaybookDir - # Process inventory file for target + # Store local vars + store_local_vars(playbook_path, Id) - hostgrouplist = [] - hostnamelist = [] - - NodeList = [] - if 'NodeList' in input_json: - NodeList = input_json['NodeList'] - - print " NodeList: ", NodeList - - if NodeList == []: - # By default set to local host - AnsibleInvFail = False - - LocalNodeList = "host" - LocalCredentials = "localhost ansible_connection=local" - f = open(PlaybookDir + "/" + AnsibleInv, "w") - f.write("[" + LocalNodeList + "]\n") - f.write(LocalCredentials) - f.close() - - else: - - if from_files: - - # Get credentials from file - - data_inventory_orig = {} - data_inventory_target = {} - curr_group = None - - print "***>", 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 NodeList: - Fail = True - if "[" + node + "]" in data_inventory_orig: - if not "[" + node + "]" in data_inventory_target: - - print "RESET", "[" + node + "]" - data_inventory_target["[" + node + "]"] = [] - else: - print "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 not key 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"] - - AnsibleInvFail = False - - f = open(PlaybookDir + "/" + AnsibleInv, "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() - - else: - - # Get credentials from mySQL - - sqlintf = AnsibleSql.mySql (host, user, passwd, - db) - if sqlintf.con: - MySqlConFail = False - errorCode, diag = readCredentials (sqlintf, - NodeList) - - print errorCode, diag - if len (diag) > 0: - f = open(PlaybookDir + "/" + AnsibleInv, - "w") - AnsibleInvFail = False - # [hostgroup, hostname, credentials] - for i in range(len(diag)): - f.write('[' + diag[i][0] + ']' + "\n") - f.write(diag[i][1]+ " " + diag[i][2] + "\n\n") - hostgrouplist.append(diag[i][0]) - hostnamelist.append(diag[i][1]) - f.close() - else: - MySqlConFailCause = sqlintf.error - sqlintf.Close() - - timeout = timeout_seconds - if 'Timeout' in input_json: - timeout = int (input_json['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'] - - TestRecord[Id] = {'PlaybookName': PlaybookName, - 'LCM': LCM, - '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} - - # 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 from_files: - - # Get playbooks from files - - MySqlConFail = False - - version = None - target_PlaybookName = None - - if '@' in PlaybookName: - version = PlaybookName.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 file in onlyfiles: - if LCM in file: - temp_version = file.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 not version == None: - if version in PlaybookName: - version_target = version - target_PlaybookName = file - - if target_PlaybookName == None: - for file in onlyfiles: - if LCM in file and version_max in file: - target_PlaybookName = file - version_target = version_max - - if target_PlaybookName: - AnsiblePlaybookFail = False - readversion = version_target - src = ansible_path + "/" + target_PlaybookName - if ".tar.gz" in target_PlaybookName: - dest = PlaybookDir + "/" + LCM + ".tar.gz" - shutil.copy2(src, dest) - retcode = subprocess.call(['tar', '-xvzf', - dest, "-C", PlaybookDir]) - ArchiveFlag = True - else: - dest = PlaybookDir + "/" + LCM + ".yml" - shutil.copy2(src, dest) - - else: - # Get playbooks from mySQL - - sqlintf = AnsibleSql.mySql (host, user, passwd, db) - if sqlintf.con: - MySqlConFail = False - - name, readversion, AnsiblePlaybookFail, diag = \ - readPlaybook (sqlintf, PlaybookName.split(".")[0], - version) - - if not AnsiblePlaybookFail: - - f = open(PlaybookDir + "/" + LCM + diag[1], "w") - f.write(diag[0]) - f.close() - - if ".tar.gz" in diag[1]: - retcode = subprocess.call(['tar', '-xvzf', - PlaybookDir + "/" + LCM + diag[1], "-C", PlaybookDir]) - f.close() - ArchiveFlag = True - else: - MySqlConFailCause = sqlintf.error - sqlintf.Close() - - if MySqlConFail: - if os.path.exists(PlaybookDir): - shutil.rmtree (PlaybookDir) - del TestRecord[Id] - return {"StatusCode": 101, - "StatusMessage": "CANNOT CONNECT TO MYSQL: " \ - + MySqlConFailCause} - elif AnsiblePlaybookFail: - if os.path.exists(PlaybookDir): - shutil.rmtree (PlaybookDir) - del TestRecord[Id] - return {"StatusCode": 101, - "StatusMessage": "PLAYBOOK NOT FOUND"} - elif AnsibleInvFail: - if os.path.exists(PlaybookDir): - shutil.rmtree (PlaybookDir) - del TestRecord[Id] - return {"StatusCode": 101, - "StatusMessage": "NODE LIST CREDENTIALS NOT FOUND"} - else: - - # Test EnvParameters - playbook_path = None - if ArchiveFlag: - for dName, sdName, fList in os.walk(PlaybookDir): - if LCM+".yml" in fList: - playbook_path = dName - else: - 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") - - 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() - - # Get mandatory parameters from playbook - Mandatory = [] - with open(playbook_path + "/" + LCM + ".yml") as origin_file: - for line in origin_file: - if "Mandatory" in line: - temp = line.split(":")[1].strip().replace(' ', '') - if len(temp) > 0: - Mandatory = temp.split(",") - - TestRecord[Id] = {'PlaybookName': TestRecord[Id]['PlaybookName'], - 'LCM': TestRecord[Id]['LCM'], - 'Version': readversion, - 'NodeList': TestRecord[Id]['NodeList'], - 'HostGroupList': TestRecord[Id]['HostGroupList'], - 'HostNameList': TestRecord[Id]['HostNameList'], - 'Time': TestRecord[Id]['Time'], - 'Timeout': TestRecord[Id]['Timeout'], - 'Duration': TestRecord[Id]['Duration'], - 'EnvParameters': TestRecord[Id]['EnvParameters'], - 'LocalParameters': TestRecord[Id]['LocalParameters'], - 'FileParameters': TestRecord[Id]['FileParameters'], - 'CallBack': TestRecord[Id]['CallBack'], - 'Result': TestRecord[Id]['Result'], - 'Log': TestRecord[Id]['Log'], - 'Output': TestRecord[Id]['Output'], - 'Path': TestRecord[Id]['Path'], - 'Mandatory': Mandatory} - - TestKey = False - - if Mandatory: - for val in Mandatory: - if EnvParam: - if val in EnvParam: - TestKey = True - else: - if LocalParam: - for key in TestRecord[Id]['NodeList']: - if key in LocalParam: - if val in LocalParam[key]: - TestKey = True - else: - if LocalParam: - for key in TestRecord[Id]['NodeList']: - if key in LocalParam: - if val in LocalParam[key]: - TestKey = True - - if not TestKey: - if os.path.exists(PlaybookDir): - shutil.rmtree (PlaybookDir) - del TestRecord[Id] - return {"StatusCode": 101, - "StatusMessage": "MISSING MANDATORY PARAMETER: " + \ - " ".join(str(x) for x in Mandatory)} - - - # 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, - playbook_path + "/" + LCM + ".yml", - NodeList, TestRecord, PlaybookDir, - ArchiveFlag)) - p.start() - ActiveProcess[Id] = p - return TestRecord[Id]['Result'] - else: - return {"StatusCode": 101, "StatusMessage": "TEST ID ALREADY DEFINED"} + # write some info out to files before running + if AUTH: + f = open(playbook_path + "/User.txt", "a") + f.write(cherrypy.request.login) + f.close() - else: - return {"StatusCode": 500, "StatusMessage": "REQUEST MUST INCLUDE: NODELIST"} + f = open(playbook_path + "/PlaybookName.txt", "a") + f.write(PlaybookName) + f.close() - else: - return {"StatusCode": 500, "StatusMessage": "JSON OBJECT MUST INCLUDE: ID, PLAYBOOKNAME"} + f = open(playbook_path + "/JsonRequest.txt", "w") + f.write(json.dumps(input_json, indent=4, sort_keys=True)) + f.close() - elif 'GET' in cherrypy.request.method: + # 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'] - input_data = parse_query_string(cherrypy.request.query_string) - print "***> in RestServer.GET" - print " Payload: ", input_data, input_data['Type'] +def process_vnf_playbook(input_json, Id, EnvParameters, time_now): + cherrypy.log("Processing playbook for VNF...") - if 'Id' in input_data and 'Type' in input_data: - if not ('GetResult' 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: + PlaybookName = input_json['PlaybookName'] + VNF_instance = EnvParameters.get('vnf_instance') + version = input_json.get('Version', None) - if 'GetResult' in input_data['Type']: + # GetInventoryNames + HaveNodeList = False + HaveInventoryNames = False + inventory_names = None + if 'InventoryNames' in input_json: + inventory_names = input_json['InventoryNames'] + HaveInventoryNames = True - print "Result:", TestRecord[input_data['Id']]['Result'] + AnsiblePlaybookFail = True - if 'StatusMessage' in TestRecord[input_data['Id']]['Result'] and getresults_block: + str_uuid = str(uuid.uuid4()) - print "*** Request blocked", input_data['Id'] + # VnfType = PlaybookName.split("/")[0] - while ActiveProcess[input_data['Id']].is_alive(): - time.sleep(5) + 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"} - print "*** Request released ", input_data['Id'] + if inventory_names is not None: + cherrypy.log("Request Decode: Inventory Names " + inventory_names) + else: + cherrypy.log("Request Decode: Inventory Names " + "Not provided") - print TestRecord[input_data['Id']]['Result'] - 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"], - "Results": TestRecord[input_data['Id']]['Result']['Results']} - if not TestRecord[input_data['Id']]['Output']['Output'] == {}: - for key in out_obj["Results"]: - if key in TestRecord[input_data['Id']]['Output']['Output']: - out_obj["Results"][key]["Output"] = TestRecord[input_data['Id']]['Output']['Output'][key] + cherrypy.log("Request Decode: PlaybookName " + PlaybookName) - return out_obj + PlayBookFunction = PlaybookName.rsplit("/", 2)[1] + PlayBookFile = PlayBookFunction + "/site.yml" - elif 'GetOutput' in input_data['Type']: + cherrypy.log("Request Decode: PlaybookFunction " + PlayBookFunction) + cherrypy.log("Request Decode: PlaybookFile " + PlayBookFile) - if TestRecord[input_data['Id']]['Output'] == {} and \ - getresults_block: + 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) - print "*** Request blocked", input_data['Id'] + PlaybookDir = ANSIBLE_TEMP + "/" + VNF_instance + "_" + str_uuid + "_" + str(Id) + cherrypy.log("Request Decode: PlaybookDir " + PlaybookDir) - while TestRecord[input_data['Id']]['Output'] == {} \ - or 'StatusMessage' in TestRecord[input_data['Id']]['Result']: - time.sleep(5) + # AnsibleInv is the directory where the host file to be run exists + # AnsibleInv = ANSIBLE_PATH + "/" + VnfType + "/latest/ansible/inventory/" + VNF_instance + ArchiveFlag = False - print "*** Request released ", input_data['Id'] + # 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) - print "Output:", TestRecord[input_data['Id']]['Output'] - return {"Output": TestRecord[input_data['Id']]['Output']['Output']} - else: - # GetLog + if not os.path.exists(CopyDir): + cherrypy.log("Playbook Not Found") + return {"StatusCode": 101, "StatusMessage": "PLAYBOOK NOT FOUND"} - if TestRecord[input_data['Id']]['Log'] == '' and \ - getresults_block: + # 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) - print "*** Request blocked", input_data['Id'] + cherrypy.log("PlaybookDir: " + PlaybookDir) + # cherrypy.log("AnsibleInv: " + AnsibleInv) - while TestRecord[input_data['Id']]['Log'] == '' \ - or 'StatusMessage' in TestRecord[input_data['Id']]['Result']: - time.sleep(5) + # Process inventory file for target + hostgrouplist = [] + hostnamelist = [] - print "*** Request released ", input_data['Id'] + NodeList = input_json.get('NodeList', []) - print "Log:", TestRecord[input_data['Id']]['Log'] - return {"Log": TestRecord[input_data['Id']]['Log']} - else: - return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"} + 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 {"StatusCode": 500, "StatusMessage": "MALFORMED REQUEST"} - elif 'DELETE' in cherrypy.request.method: - input_data = parse_query_string(cherrypy.request.query_string) + 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"} + - print "***> in RestServer.DELETE" - print " Payload: ", input_data +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 input_data['Id'] in TestRecord: - if not 'PENDING' in TestRecord[input_data['Id']]['Result']: - print " Path:", TestRecord[input_data['Id']]['Path'] - if os.path.exists(TestRecord[input_data['Id']]['Path']): - shutil.rmtree (TestRecord[input_data['Id']]['Path']) - TestRecord.pop (input_data['Id']) - if input_data['Id'] in ActiveProcess: - ActiveProcess.pop (input_data['Id']) + if AUTH: + cherrypy.log("Request USER: " + cherrypy.request.login) + cherrypy.log("Payload: " + str(input_data) + " Type " + input_data['Type']) - return {"StatusCode": 200, "StatusMessage": "PLAYBOOK EXECUTION RECORDS DELETED"} + 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 list(TestRecord.items()): + for id, record in list(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 list(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: - return {"StatusCode": 200, "StatusMessage": "PENDING"} + cherrypy.log("Globglob failed:") + return + else: - return {"StatusCode": 500, "StatusMessage": "TEST ID UNDEFINED"} + # 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 + + time_now = datetime.datetime.utcnow() + + # Erase old test results (2x timeout) + # 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)) + if key in ActiveProcess: + 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 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']) + del TestRecord[key] + + cherrypy.log("RestServer.Dispatch: " + cherrypy.request.method) + + if 'POST' in cherrypy.request.method: + input_json = cherrypy.request.json + return handle_post_method(input_json, time_now) + elif 'GET' in cherrypy.request.method: + # 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) + return handle_get_method(input_data) + elif 'DELETE' in cherrypy.request.method: + input_data = parse_query_string(cherrypy.request.query_string) + return handle_delete_method(input_data) if __name__ == '__main__': @@ -833,81 +980,78 @@ 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' - id = 'na' priv = 'na' - psswd = 'na' + intermediate = 'na' timeout_seconds = 'na' - ansible_path = 'na' - ansible_inv = 'na' - ansible_temp = 'na' + ANSIBLE_PATH = 'na' + ANSIBLE_TEMP = 'na' host = 'na' - user = 'na' - passwd = 'na' - db = '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 'port:' in line: - port = 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 auth and 'id:' in line: - id = line.split(':')[1].strip() - if auth and 'psswd:' in line: - psswd = 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_inv' in line: - ansible_inv = line.split(':')[1].strip() - if not os.path.exists(ansible_path + "/" + ansible_inv): - print '[INFO] The ansible_inv file does not exist' + 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 line: - ansible_temp = line.split(':')[1].strip() - if 'host' in line: - host = line.split(':')[1].strip() - if 'user' in line: - user = line.split(':')[1].strip() - if 'passwd' in line: - passwd = line.split(':')[1].strip() - if 'db' in line: - db = 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() + 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 global_conf = { 'global': { + 'log.screen': True, + 'response.timeout': 5400, 'server.socket_host': ip, 'server.socket_port': int(port), 'server.protocol_version': 'HTTP/1.1' - } } + } if tls: # Use pythons built-in SSL @@ -916,33 +1060,66 @@ 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): + 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: - userpassdict = {id: psswd} - checkpassword = cherrypy.lib.auth_basic.checkpassword_dict(userpassdict) - - app_conf = {'/': - {'tools.auth_basic.on': True, - 'tools.auth_basic.realm': 'earth', - 'tools.auth_basic.checkpassword': checkpassword, - } - } + if AUTH: + # Read in and build user dictionary + if not os.path.exists(users): + cherrypy.log('[INFO] The users file does not exist: ' + users) + sys.exit(0) + userpassdict = {} + user_file = open(users, 'r') + 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 + } + } - cherrypy.tree.mount(TestManager(), '/', app_conf) + application = cherrypy.tree.mount(TestManager(), '/', app_conf) else: - cherrypy.tree.mount(TestManager(), '/') + application = cherrypy.tree.mount(TestManager(), '/') + cherrypy.config.update({ + 'log.access_file': "/var/log/RestServer.access" + }) + accessLogName = "/var/log/RestServer.access" + applicationLogName = "/var/log/RestServer.log" cherrypy.config.update(global_conf) + log = application.log + 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) + # Start server cherrypy.engine.start()