frikky
929 lines
42 KiB
Python
929 lines
42 KiB
Python
import os
|
|
import sys
|
|
import re
|
|
import time
|
|
import json
|
|
import logging
|
|
import requests
|
|
import urllib.parse
|
|
|
|
class AppBase:
|
|
""" The base class for Python-based apps in Shuffle, handles logging and callbacks configurations"""
|
|
__version__ = None
|
|
app_name = None
|
|
|
|
def __init__(self, redis=None, logger=None, console_logger=None):#, docker_client=None):
|
|
self.logger = logger if logger is not None else logging.getLogger("AppBaseLogger")
|
|
self.redis=redis
|
|
self.console_logger = logger if logger is not None else logging.getLogger("AppBaseLogger")
|
|
|
|
# apikey is for the user / org
|
|
# authorization is for the specific workflow
|
|
self.url = os.getenv("CALLBACK_URL", "https://shuffler.io")
|
|
self.action = os.getenv("ACTION", "")
|
|
self.apikey = os.getenv("FUNCTION_APIKEY", "")
|
|
self.authorization = os.getenv("AUTHORIZATION", "")
|
|
self.current_execution_id = os.getenv("EXECUTIONID", "")
|
|
|
|
if len(self.action) == 0:
|
|
print("ACTION env not defined")
|
|
sys.exit(0)
|
|
if len(self.apikey) == 0:
|
|
print("FUNCTION_APIKEY env not defined")
|
|
sys.exit(0)
|
|
if len(self.authorization) == 0:
|
|
print("AUTHORIZATION env not defined")
|
|
sys.exit(0)
|
|
if len(self.current_execution_id) == 0:
|
|
print("EXECUTIONID env not defined")
|
|
sys.exit(0)
|
|
|
|
if isinstance(self.action, str):
|
|
self.action = json.loads(self.action)
|
|
|
|
async def execute_action(self, action):
|
|
# FIXME - add request for the function STARTING here. Use "results stream" or something
|
|
# PAUSED, AWAITING_DATA, PENDING, COMPLETED, ABORTED, EXECUTING, SUCCESS, FAILURE
|
|
|
|
# !!! Let this line stay - its used for some horrible codegeneration / stitching !!! #
|
|
#STARTCOPY
|
|
stream_path = "/api/v1/streams"
|
|
action_result = {
|
|
"action": action,
|
|
"authorization": self.authorization,
|
|
"execution_id": self.current_execution_id,
|
|
"result": "",
|
|
"started_at": int(time.time()),
|
|
"status": "EXECUTING"
|
|
}
|
|
self.logger.info("ACTION RESULT: %s", action_result)
|
|
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"Authorization": "Bearer %s" % self.apikey
|
|
}
|
|
|
|
# Add async logger
|
|
# self.console_logger.handlers[0].stream.set_execution_id()
|
|
#self.logger.info("Before initial stream result")
|
|
try:
|
|
ret = requests.post("%s%s" % (self.url, stream_path), headers=headers, json=action_result)
|
|
self.logger.info("Workflow: %d" % ret.status_code)
|
|
if ret.status_code != 200:
|
|
self.logger.info(ret.text)
|
|
except requests.exceptions.ConnectionError as e:
|
|
print("Connectionerror: %s" % e)
|
|
return
|
|
|
|
# Verify whether there are any parameters with ACTION_RESULT required
|
|
# If found, we get the full results list from backend
|
|
fullexecution = {}
|
|
try:
|
|
tmpdata = {
|
|
"authorization": self.authorization,
|
|
"execution_id": self.current_execution_id
|
|
}
|
|
|
|
self.logger.info("Before FULLEXEC stream result")
|
|
ret = requests.post(
|
|
"%s/api/v1/streams/results" % (self.url),
|
|
headers=headers,
|
|
json=tmpdata
|
|
)
|
|
|
|
if ret.status_code == 200:
|
|
fullexecution = ret.json()
|
|
else:
|
|
self.logger.info("Error: Data: ", ret.json())
|
|
self.logger.info("Error with status code for results. Crashing because ACTION_RESULTS or WORKFLOW_VARIABLE can't be handled. Status: %d" % ret.status_code)
|
|
return
|
|
except requests.exceptions.ConnectionError as e:
|
|
self.logger.info("Connectionerror: %s" % e)
|
|
return
|
|
|
|
self.logger.info("AFTER FULLEXEC stream result")
|
|
|
|
# Gets the value at the paranthesis level you want
|
|
def parse_nested_param(string, level):
|
|
"""
|
|
Generate strings contained in nested (), indexing i = level
|
|
"""
|
|
if len(re.findall("\(", string)) == len(re.findall("\)", string)):
|
|
LeftRightIndex = [x for x in zip(
|
|
[Left.start()+1 for Left in re.finditer('\(', string)],
|
|
reversed([Right.start() for Right in re.finditer('\)', string)]))]
|
|
|
|
elif len(re.findall("\(", string)) > len(re.findall("\)", string)):
|
|
return parse_nested_param(string + ')', level)
|
|
elif len(re.findall("\(", string)) < len(re.findall("\)", string)):
|
|
return parse_nested_param('(' + string, level)
|
|
|
|
else:
|
|
return 'Failed to parse params'
|
|
|
|
try:
|
|
return [string[LeftRightIndex[level][0]:LeftRightIndex[level][1]]]
|
|
except IndexError:
|
|
return [string[LeftRightIndex[level+1][0]:LeftRightIndex[level+1][1]]]
|
|
|
|
# Finds the deepest level paranthesis in a string
|
|
def maxDepth(S):
|
|
current_max = 0
|
|
max = 0
|
|
n = len(S)
|
|
|
|
# Traverse the input string
|
|
for i in range(n):
|
|
if S[i] == '(':
|
|
current_max += 1
|
|
|
|
if current_max > max:
|
|
max = current_max
|
|
elif S[i] == ')':
|
|
if current_max > 0:
|
|
current_max -= 1
|
|
else:
|
|
return -1
|
|
|
|
# finally check for unbalanced string
|
|
if current_max != 0:
|
|
return -1
|
|
|
|
return max-1
|
|
|
|
# Specific type parsing
|
|
def parse_type(data, thistype):
|
|
if data == None:
|
|
return "Empty"
|
|
|
|
if "int" in thistype or "number" in thistype:
|
|
try:
|
|
return int(data)
|
|
except ValueError:
|
|
print("ValueError while casting %s" % data)
|
|
return data
|
|
if "lower" in thistype:
|
|
return data.lower()
|
|
if "upper" in thistype:
|
|
return data.upper()
|
|
if "trim" in thistype:
|
|
return data.strip()
|
|
if "strip" in thistype:
|
|
return data.strip()
|
|
if "split" in thistype:
|
|
return data.split()
|
|
if "len" in thistype or "length" in thistype:
|
|
tmp = ""
|
|
try:
|
|
tmp = json.loads(data)
|
|
except:
|
|
pass
|
|
|
|
if isinstance(tmp, list):
|
|
return str(len(tmp))
|
|
|
|
return str(len(data))
|
|
if "parse" in thistype:
|
|
splitvalues = []
|
|
default_error = """Error. Expected syntax: parse(["hello","test1"],0:1)"""
|
|
if "," in data:
|
|
splitvalues = data.split(",")
|
|
|
|
for item in range(len(splitvalues)):
|
|
splitvalues[item] = splitvalues[item].strip()
|
|
else:
|
|
return default_error
|
|
|
|
lastsplit = []
|
|
if ":" in splitvalues[-1]:
|
|
lastsplit = splitvalues[-1].split(":")
|
|
else:
|
|
try:
|
|
lastsplit = [int(splitvalues[-1])]
|
|
except ValueError:
|
|
return default_error
|
|
|
|
try:
|
|
parsedlist = ",".join(splitvalues[0:-1])
|
|
if len(lastsplit) > 1:
|
|
tmp = json.loads(parsedlist)[int(lastsplit[0]):int(lastsplit[1])]
|
|
else:
|
|
tmp = json.loads(parsedlist)[lastsplit[0]]
|
|
|
|
print(tmp)
|
|
return tmp
|
|
except IndexError as e:
|
|
return default_error
|
|
|
|
# Parses the INNER value and recurses until everything is done
|
|
def parse_wrapper(data):
|
|
try:
|
|
if "(" not in data or ")" not in data:
|
|
return data
|
|
except TypeError:
|
|
return data
|
|
|
|
print("Running %s" % data)
|
|
|
|
# Look for the INNER wrapper first, then move out
|
|
wrappers = ["int", "number", "lower", "upper", "trim", "strip", "split", "parse", "len", "length"]
|
|
found = False
|
|
for wrapper in wrappers:
|
|
if wrapper not in data.lower():
|
|
continue
|
|
|
|
found = True
|
|
break
|
|
|
|
if not found:
|
|
return data
|
|
|
|
# Do stuff here.
|
|
innervalue = parse_nested_param(data, maxDepth(data)-0)
|
|
outervalue = parse_nested_param(data, maxDepth(data)-1)
|
|
print("INNER: ", outervalue)
|
|
print("OUTER: ", outervalue)
|
|
|
|
if outervalue != innervalue:
|
|
#print("Outer: ", outervalue, " inner: ", innervalue)
|
|
for key in range(len(innervalue)):
|
|
# Replace OUTERVALUE[key] with INNERVALUE[key] in data.
|
|
print("Replace %s with %s in %s" % (outervalue[key], innervalue[key], data))
|
|
data = data.replace(outervalue[key], innervalue[key])
|
|
else:
|
|
for thistype in wrappers:
|
|
if thistype.lower() not in data.lower():
|
|
continue
|
|
|
|
parsed_value = parse_type(innervalue[0], thistype.lower())
|
|
return parsed_value
|
|
|
|
print("DATA: %s\n" % data)
|
|
return parse_wrapper(data)
|
|
|
|
def parse_wrapper_start(data):
|
|
newdata = []
|
|
newstring = ""
|
|
record = True
|
|
paranCnt = 0
|
|
for char in data:
|
|
if char == "(":
|
|
paranCnt += 1
|
|
|
|
if not record:
|
|
record = True
|
|
|
|
if record:
|
|
newstring += char
|
|
|
|
if paranCnt == 0 and char == " ":
|
|
newdata.append(newstring)
|
|
newstring = ""
|
|
record = True
|
|
|
|
if char == ")":
|
|
paranCnt -= 1
|
|
|
|
if paranCnt == 0:
|
|
record = False
|
|
|
|
if len(newstring) > 0:
|
|
newdata.append(newstring)
|
|
|
|
print(newdata)
|
|
parsedlist = []
|
|
non_string = False
|
|
for item in newdata:
|
|
ret = parse_wrapper(item)
|
|
if not isinstance(ret, str):
|
|
non_string = True
|
|
|
|
parsedlist.append(ret)
|
|
|
|
if len(parsedlist) > 0 and not non_string:
|
|
return " ".join(parsedlist)
|
|
elif len(parsedlist) == 1 and non_string:
|
|
return parsedlist[0]
|
|
else:
|
|
print("Casting back to string because multi: ", parsedlist)
|
|
newlist = []
|
|
for item in parsedlist:
|
|
try:
|
|
newlist.append(str(item))
|
|
except ValueError:
|
|
newlist.append("parsing_error")
|
|
return " ".join(newlist)
|
|
|
|
# Takes a workflow execution as argument
|
|
# Returns a string if the result is single, or a list if it's a list
|
|
def get_json_value(execution_data, input_data):
|
|
parsersplit = input_data.split(".")
|
|
actionname = parsersplit[0][1:].replace(" ", "_", -1)
|
|
print(f"Actionname: {actionname}")
|
|
|
|
# 1. Find the action
|
|
baseresult = ""
|
|
actionname_lower = actionname.lower()
|
|
try:
|
|
if actionname_lower == "exec":
|
|
baseresult = execution_data["execution_argument"]
|
|
else:
|
|
for result in execution_data["results"]:
|
|
resultlabel = result["action"]["label"].replace(" ", "_", -1).lower()
|
|
if resultlabel.lower() == actionname_lower:
|
|
baseresult = result["result"]
|
|
break
|
|
|
|
print("BEFORE VARIABLES!")
|
|
if len(baseresult) == 0:
|
|
try:
|
|
#print("WF Variables: %s" % execution_data["workflow"]["workflow_variables"])
|
|
for variable in execution_data["workflow"]["workflow_variables"]:
|
|
variablename = variable["name"].replace(" ", "_", -1).lower()
|
|
|
|
if variablename.lower() == actionname_lower:
|
|
baseresult = variable["value"]
|
|
break
|
|
except KeyError as e:
|
|
print("KeyError wf variables: %s" % e)
|
|
pass
|
|
except TypeError as e:
|
|
print("TypeError wf variables: %s" % e)
|
|
pass
|
|
|
|
print("BEFORE EXECUTION VAR")
|
|
if len(baseresult) == 0:
|
|
try:
|
|
#print("Execution Variables: %s" % execution_data["execution_variables"])
|
|
for variable in execution_data["execution_variables"]:
|
|
variablename = variable["name"].replace(" ", "_", -1).lower()
|
|
if variablename.lower() == actionname_lower:
|
|
baseresult = variable["value"]
|
|
break
|
|
except KeyError as e:
|
|
print("KeyError exec variables: %s" % e)
|
|
pass
|
|
except TypeError as e:
|
|
print("TypeError exec variables: %s" % e)
|
|
pass
|
|
|
|
except KeyError as error:
|
|
print(f"KeyError in JSON: {error}")
|
|
|
|
print(f"After first trycatch")
|
|
|
|
# 2. Find the JSON data
|
|
if len(baseresult) == 0:
|
|
return ""
|
|
|
|
if len(parsersplit) == 1:
|
|
return baseresult
|
|
|
|
baseresult = baseresult.replace("\'", "\"")
|
|
basejson = {}
|
|
try:
|
|
basejson = json.loads(baseresult)
|
|
except json.decoder.JSONDecodeError as e:
|
|
return baseresult
|
|
|
|
# This whole thing should be recursive.
|
|
try:
|
|
cnt = 0
|
|
for value in parsersplit[1:]:
|
|
cnt += 1
|
|
|
|
print("VALUE: %s" % value)
|
|
if value == "#":
|
|
# FIXME - not recursive - should go deeper if there are more #
|
|
print("HANDLE RECURSIVE LOOP OF %s" % basejson)
|
|
returnlist = []
|
|
try:
|
|
for innervalue in basejson:
|
|
print("Value: %s" % innervalue[parsersplit[cnt+1]])
|
|
returnlist.append(innervalue[parsersplit[cnt+1]])
|
|
except IndexError as e:
|
|
print("Indexerror inner: %s" % e)
|
|
# Basically means its a normal list, not a crazy one :)
|
|
# Custom format for ${name[0,1,2,...]}$
|
|
indexvalue = "${NO_SPLITTER%s}$" % json.dumps(basejson)
|
|
if len(returnlist) > 0:
|
|
indexvalue = "${NO_SPLITTER%s}$" % json.dumps(returnlist)
|
|
|
|
print("INDEXVAL: ", indexvalue)
|
|
return indexvalue
|
|
except TypeError as e:
|
|
print("TypeError inner: %s" % e)
|
|
|
|
# Example format: ${[]}$
|
|
parseditem = "${%s%s}$" % (parsersplit[cnt+1], json.dumps(returnlist))
|
|
print("PARSED LOOP ITEM: %s" % parseditem)
|
|
return parseditem
|
|
|
|
else:
|
|
print("BEFORE NORMAL VALUE: ", basejson, value)
|
|
if len(value) == 0:
|
|
return basejson
|
|
|
|
if isinstance(basejson[value], str):
|
|
print(f"LOADING STRING '%s' AS JSON" % basejson[value])
|
|
try:
|
|
basejson = json.loads(basejson[value])
|
|
except json.decoder.JSONDecodeError as e:
|
|
print("RETURNING BECAUSE '%s' IS A NORMAL STRING" % basejson[value])
|
|
return basejson[value]
|
|
else:
|
|
basejson = basejson[value]
|
|
|
|
except KeyError as e:
|
|
print("Lower keyerror: %s" % e)
|
|
return "KeyError: Couldn't find key: %s" % e
|
|
|
|
return basejson
|
|
|
|
# Parses parameters sent to it and returns whether it did it successfully with the values found
|
|
def parse_params(action, fullexecution, parameter):
|
|
# Skip if it starts with $?
|
|
jsonparsevalue = "$."
|
|
|
|
# Matches with space in the first part, but not in subsequent parts.
|
|
# JSON / yaml etc shouldn't have spaces in their fields anyway.
|
|
match = ".*?([$]{1}([a-zA-Z0-9 _-]+\.?){1}([a-zA-Z0-9#_-]+\.?){0,})"
|
|
|
|
# Regex to find all the things
|
|
if parameter["variant"] == "STATIC_VALUE":
|
|
data = parameter["value"]
|
|
actualitem = re.findall(match, data, re.MULTILINE)
|
|
#self.logger.debug(f"\n\nHandle static data with JSON: {data}\n\n")
|
|
#self.logger.info("STATIC PARSED: %s" % actualitem)
|
|
if len(actualitem) > 0:
|
|
print("ACTUAL: ", actualitem)
|
|
for replace in actualitem:
|
|
try:
|
|
to_be_replaced = replace[0]
|
|
except IndexError:
|
|
continue
|
|
|
|
value = get_json_value(fullexecution, to_be_replaced)
|
|
if isinstance(value, str):
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, value)
|
|
elif isinstance(value, dict):
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, json.dumps(value))
|
|
else:
|
|
print("Unknown type %s" % type(value))
|
|
try:
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, json.dumps(value))
|
|
except json.decoder.JSONDecodeError as e:
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, value)
|
|
|
|
|
|
if parameter["variant"] == "WORKFLOW_VARIABLE":
|
|
print("Handling workflow variable")
|
|
found = False
|
|
try:
|
|
for item in fullexecution["workflow"]["workflow_variables"]:
|
|
if parameter["action_field"] == item["name"]:
|
|
found = True
|
|
parameter["value"] = item["value"]
|
|
break
|
|
except KeyError as e:
|
|
print("KeyError WF variable 1: %s" % e)
|
|
pass
|
|
except TypeError as e:
|
|
print("TypeError WF variables 1: %s" % e)
|
|
pass
|
|
|
|
if not found:
|
|
try:
|
|
for item in fullexecution["execution_variables"]:
|
|
if parameter["action_field"] == item["name"]:
|
|
parameter["value"] = item["value"]
|
|
break
|
|
except KeyError as e:
|
|
print("KeyError WF variable 2: %s" % e)
|
|
pass
|
|
except TypeError as e:
|
|
print("TypeError WF variables 2: %s" % e)
|
|
pass
|
|
|
|
elif parameter["variant"] == "ACTION_RESULT":
|
|
# FIXME - calculate value based on action_field and $if prominent
|
|
# FIND THE RIGHT LABEL
|
|
# GET THE LABEL'S RESULT
|
|
|
|
tmpvalue = ""
|
|
self.logger.info("ACTION FIELD: %s" % parameter["action_field"])
|
|
|
|
fullname = "$"
|
|
if parameter["action_field"] == "Execution Argument":
|
|
tmpvalue = fullexecution["execution_argument"]
|
|
fullname += "exec"
|
|
else:
|
|
fullname += parameter["action_field"]
|
|
|
|
self.logger.info("PRE Fullname: %s" % fullname)
|
|
|
|
if parameter["value"].startswith(jsonparsevalue):
|
|
fullname += parameter["value"][1:]
|
|
#else:
|
|
# fullname = "$%s" % parameter["action_field"]
|
|
|
|
self.logger.info("Fullname: %s" % fullname)
|
|
actualitem = re.findall(match, fullname, re.MULTILINE)
|
|
self.logger.info("ACTION PARSED: %s" % actualitem)
|
|
if len(actualitem) > 0:
|
|
for replace in actualitem:
|
|
try:
|
|
to_be_replaced = replace[0]
|
|
except IndexError:
|
|
print("Nothing to replace?: " % e)
|
|
continue
|
|
|
|
# This will never be a loop aka multi argument
|
|
parameter["value"] = to_be_replaced
|
|
|
|
value = get_json_value(fullexecution, to_be_replaced)
|
|
if isinstance(value, str):
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, value)
|
|
elif isinstance(value, dict):
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, json.dumps(value))
|
|
else:
|
|
print("Unknown type %s" % type(value))
|
|
try:
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, json.dumps(value))
|
|
except json.decoder.JSONDecodeError as e:
|
|
parameter["value"] = parameter["value"].replace(to_be_replaced, value)
|
|
|
|
return "", parameter["value"]
|
|
|
|
def run_validation(sourcevalue, check, destinationvalue):
|
|
self.logger.info("Checking %s %s %s" % (sourcevalue, check, destinationvalue))
|
|
|
|
if check == "=" or check.lower() == "equals":
|
|
if sourcevalue.lower() == destinationvalue.lower():
|
|
return True
|
|
elif check == "!=" or check.lower() == "does not equal":
|
|
if sourcevalue.lower() != destinationvalue.lower():
|
|
return True
|
|
elif check.lower() == "startswith":
|
|
if sourcevalue.lower().startswith(destinationvalue.lower()):
|
|
return True
|
|
elif check.lower() == "endswith":
|
|
if sourcevalue.lower().endswith(destinationvalue.lower()):
|
|
return True
|
|
elif check.lower() == "contains":
|
|
if destinationvalue.lower() in sourcevalue.lower():
|
|
return True
|
|
elif check.lower() == "larger than":
|
|
try:
|
|
if sourcevalue.isdigit() and destinationvalue.isdigit():
|
|
if int(sourcevalue) > int(destinationvalue):
|
|
return True
|
|
except AttributeError as e:
|
|
self.logger.error("Condition larger than failed with values %s and %s: %s" % (sourcevalue, destinationvalue, e))
|
|
return False
|
|
elif check.lower() == "smaller than":
|
|
try:
|
|
if sourcevalue.isdigit() and destinationvalue.isdigit():
|
|
if int(sourcevalue) < int(destinationvalue):
|
|
return True
|
|
except AttributeError as e:
|
|
self.logger.error("Condition smaller than failed with values %s and %s: %s" % (sourcevalue, destinationvalue, e))
|
|
return False
|
|
else:
|
|
self.logger.info("Condition: can't handle %s yet. Setting to true" % check)
|
|
|
|
return False
|
|
|
|
def check_branch_conditions(action, fullexecution):
|
|
# relevantbranches = workflow.branches where destination = action
|
|
try:
|
|
if fullexecution["workflow"]["branches"] == None or len(fullexecution["workflow"]["branches"]) == 0:
|
|
return True, ""
|
|
except KeyError:
|
|
return True, ""
|
|
|
|
relevantbranches = []
|
|
for branch in fullexecution["workflow"]["branches"]:
|
|
if branch["destination_id"] != action["id"]:
|
|
continue
|
|
|
|
self.logger.info("Relevant branch: %s" % branch)
|
|
|
|
# Remove anything without a condition
|
|
try:
|
|
if (branch["conditions"]) == 0 or branch["conditions"] == None:
|
|
continue
|
|
except KeyError:
|
|
continue
|
|
|
|
self.logger.info("Relevant conditions: %s" % branch["conditions"])
|
|
successful_conditions = []
|
|
failed_conditions = []
|
|
for condition in branch["conditions"]:
|
|
self.logger.info("Getting condition value of %s" % condition)
|
|
|
|
# Parse all values first here
|
|
sourcevalue = condition["source"]["value"]
|
|
check, sourcevalue = parse_params(action, fullexecution, condition["source"])
|
|
if check:
|
|
return False, "Failed condition: %s %s %s because %s" % (sourcevalue, condition["condition"]["value"], destinationvalue, check)
|
|
|
|
|
|
#sourcevalue = sourcevalue.encode("utf-8")
|
|
sourcevalue = parse_wrapper_start(sourcevalue)
|
|
destinationvalue = condition["destination"]["value"]
|
|
|
|
check, destinationvalue = parse_params(action, fullexecution, condition["destination"])
|
|
if check:
|
|
return False, "Failed condition: %s %s %s because %s" % (sourcevalue, condition["condition"]["value"], destinationvalue, check)
|
|
|
|
#destinationvalue = destinationvalue.encode("utf-8")
|
|
destinationvalue = parse_wrapper_start(destinationvalue)
|
|
available_checks = [
|
|
"=",
|
|
"equals",
|
|
"!=",
|
|
"does not equal",
|
|
">",
|
|
"larger than",
|
|
"<",
|
|
"less than",
|
|
">=",
|
|
"<=",
|
|
"startswith",
|
|
"endswith",
|
|
"contains",
|
|
"re",
|
|
"matches regex",
|
|
]
|
|
|
|
# FIXME - what should I do here?
|
|
if not condition["condition"]["value"] in available_checks:
|
|
self.logger.info("Skipping %s %s %s because %s is invalid." % (sourcevalue, condition["condition"]["value"], destinationvalue, condition["condition"]["value"]))
|
|
continue
|
|
|
|
#print(destinationvalue)
|
|
if not run_validation(sourcevalue, condition["condition"]["value"], destinationvalue):
|
|
self.logger.info("Failed condition check for %s %s %s." % (sourcevalue, condition["condition"]["value"], destinationvalue))
|
|
return False, "Failed condition: %s %s %s" % (sourcevalue, condition["condition"]["value"], destinationvalue)
|
|
|
|
|
|
# Make a general parser here, at least to get param["name"] = param["value"] in maparameter[string]string
|
|
#for condition in branch.conditons:
|
|
|
|
return True, ""
|
|
|
|
# Checks whether conditions are met, otherwise set
|
|
branchcheck, tmpresult = check_branch_conditions(action, fullexecution)
|
|
if not branchcheck:
|
|
self.logger.info("Failed one or more branch conditions.")
|
|
action_result["result"] = tmpresult
|
|
action_result["status"] = "FAILURE"
|
|
try:
|
|
ret = requests.post("%s%s" % (self.url, stream_path), headers=headers, json=action_result)
|
|
self.logger.info("Result: %d" % ret.status_code)
|
|
if ret.status_code != 200:
|
|
self.logger.info(ret.text)
|
|
except requests.exceptions.ConnectionError as e:
|
|
self.logger.exception(e)
|
|
|
|
print("\n\nRETURNING BECAUSE A BRANCH FAILED\n\n")
|
|
return
|
|
|
|
# Replace name cus there might be issues
|
|
# Not doing lower() as there might be user-made functions
|
|
actionname = action["name"]
|
|
if " " in actionname:
|
|
actionname.replace(" ", "_", -1)
|
|
#if action.generated:
|
|
# actionname = actionname.lower()
|
|
|
|
# Runs the actual functions
|
|
try:
|
|
func = getattr(self, actionname, None)
|
|
if func == None:
|
|
self.logger.debug("Failed executing %s because func is None." % actionname)
|
|
action_result["status"] = "FAILURE"
|
|
action_result["result"] = "Function %s doesn't exist." % actionname
|
|
elif callable(func):
|
|
try:
|
|
if len(action["parameters"]) < 1:
|
|
result = await func()
|
|
else:
|
|
# Potentially parse JSON here
|
|
# FIXME - add potential authentication as first parameter(s) here
|
|
# params[parameter["name"]] = parameter["value"]
|
|
#print(fullexecution["authentication"]
|
|
# What variables are necessary here tho hmm
|
|
|
|
params = {}
|
|
try:
|
|
for item in action["authentication"]:
|
|
print("AUTH: ", key, value)
|
|
params[item["key"]] = item["value"]
|
|
except KeyError:
|
|
print("No authentication specified!")
|
|
pass
|
|
#action["authentication"]
|
|
|
|
# calltimes is used to handle forloops in the app itself.
|
|
# 2 kinds of loop - one in gui with one app each, and one like this,
|
|
# which is super fast, but has a bad overview (potentially good tho)
|
|
calltimes = 1
|
|
result = ""
|
|
|
|
all_executions = []
|
|
|
|
# Multi_parameter has the data for each. variable
|
|
minlength = 0
|
|
multi_parameters = json.loads(json.dumps(params))
|
|
multiexecution = False
|
|
for parameter in action["parameters"]:
|
|
check, value = parse_params(action, fullexecution, parameter)
|
|
if check:
|
|
raise "Value check error: %s" % Exception(check)
|
|
|
|
# Custom format for ${name[0,1,2,...]}$
|
|
#submatch = "([${]{2}([0-9a-zA-Z_-]+)(\[.*\])[}$]{2})"
|
|
submatch = "([${]{2}([0-9a-zA-Z_-]+)(\[.*\])[}$]{2})"
|
|
actualitem = re.findall(submatch, value, re.MULTILINE)
|
|
print("Multicheck ", actualitem)
|
|
if len(actualitem) > 0:
|
|
multiexecution = True
|
|
|
|
# This is here to handle for loops within variables.. kindof
|
|
# 1. Find the length of the longest array
|
|
# 2. Build an array with the base values based on parameter["value"]
|
|
# 3. Get the n'th value of the generated list from values
|
|
# 4. Execute all n answers
|
|
replacements = {}
|
|
for replace in actualitem:
|
|
try:
|
|
to_be_replaced = replace[0]
|
|
actualitem = replace[2]
|
|
except IndexError:
|
|
continue
|
|
|
|
try:
|
|
itemlist = json.loads(actualitem)
|
|
if len(itemlist) > minlength:
|
|
minlength = len(itemlist)
|
|
except json.decoder.JSONDecodeError as e:
|
|
print("JSON Error: %s in %s" % (e, actualitem))
|
|
|
|
replacements[to_be_replaced] = actualitem
|
|
|
|
# This is a result array for JUST this value..
|
|
# What if there are more?
|
|
resultarray = []
|
|
for i in range(0, minlength):
|
|
tmpitem = json.loads(json.dumps(parameter["value"]))
|
|
for key, value in replacements.items():
|
|
replacement = json.dumps(json.loads(value)[i])
|
|
if replacement.startswith("\"") and replacement.endswith("\""):
|
|
replacement = replacement[1:len(replacement)-1]
|
|
#except json.decoder.JSONDecodeError as e:
|
|
|
|
print("REPLACING %s with %s" % (key, replacement))
|
|
#replacement = parse_wrapper_start(replacement)
|
|
tmpitem = tmpitem.replace(key, replacement, -1)
|
|
|
|
resultarray.append(tmpitem)
|
|
|
|
# With this parameter ready, add it to... a greater list of parameters. Rofl
|
|
multi_parameters[parameter["name"]] = resultarray
|
|
else:
|
|
# Parses things like int(value)
|
|
self.logger.info("Parsing wrapper data for %s" % value)
|
|
value = parse_wrapper_start(value)
|
|
|
|
params[parameter["name"]] = value
|
|
multi_parameters[parameter["name"]] = value
|
|
|
|
# FIXME - this is horrible, but works for now
|
|
#for i in range(calltimes):
|
|
if not multiexecution:
|
|
print("APP_SDK DONE: Starting normal execution of function")
|
|
newres = await func(**params)
|
|
print("NEWRES: ", newres)
|
|
if isinstance(newres, str):
|
|
result += newres
|
|
else:
|
|
try:
|
|
result += str(newres)
|
|
except ValueError:
|
|
result += "Failed autocasting. Can't handle %s type from function. Must be string" % type(newres)
|
|
print("Can't handle type %s value from function" % (type(newres)))
|
|
print("POST NEWRES RESULT: ", result)
|
|
else:
|
|
print("APP_SDK DONE: Starting MULTI execution with", multi_parameters)
|
|
# 1. Use number of executions based on longest array
|
|
# 2. Find the right value from the parsed multi_params
|
|
|
|
results = []
|
|
json_object = False
|
|
for i in range(0, minlength):
|
|
# To be able to use the results as a list:
|
|
baseparams = json.loads(json.dumps(multi_parameters))
|
|
|
|
try:
|
|
for key, value in baseparams.items():
|
|
if isinstance(value, list):
|
|
baseparams[key] = value[i]
|
|
except IndexError as e:
|
|
print("IndexError: %s" % e)
|
|
baseparams[key] = "IndexError: %s" % e
|
|
except KeyError as e:
|
|
print("KeyError: %s" % e)
|
|
baseparams[key] = "KeyError: %s" % e
|
|
|
|
#print("Running with params %s" % baseparams)
|
|
ret = await func(**baseparams)
|
|
ret = ret.replace("\"", "\\\"", -1)
|
|
print("Inner ret parsed: %s" % ret)
|
|
|
|
try:
|
|
results.append(json.loads(ret))
|
|
json_object = True
|
|
except json.decoder.JSONDecodeError as e:
|
|
results.append(ret)
|
|
|
|
# Dump the result as a string of a list
|
|
print("RESULTS: %s" % results)
|
|
if isinstance(results, list):
|
|
print("JSON OBJECT? ", json_object)
|
|
if json_object:
|
|
result = json.dumps(results)
|
|
else:
|
|
result = "[\""+"\", \"".join(results)+"\"]"
|
|
else:
|
|
print("Normal result?")
|
|
result = results
|
|
|
|
print("RESULT: %s" % result)
|
|
action_result["status"] = "SUCCESS"
|
|
action_result["result"] = str(result)
|
|
if action_result["result"] == "":
|
|
action_result["result"] = result
|
|
|
|
self.logger.debug(f"Executed {action['label']}-{action['id']} with result: {result}")
|
|
self.logger.debug(f"Data: %s" % action_result)
|
|
except TypeError as e:
|
|
print("TypeError issue: %s" % e)
|
|
action_result["status"] = "FAILURE"
|
|
action_result["result"] = "TypeError: %s" % str(e)
|
|
else:
|
|
print("Function %s doesn't exist?" % action["name"])
|
|
self.logger.error(f"App {self.__class__.__name__}.{action['name']} is not callable")
|
|
action_result["status"] = "FAILURE"
|
|
action_result["result"] = "Function %s is not callable." % actionname
|
|
|
|
except Exception as e:
|
|
print(f"Failed to execute: {e}")
|
|
self.logger.exception(f"Failed to execute {e}-{action['id']}")
|
|
action_result["status"] = "FAILURE"
|
|
action_result["result"] = "General exception: %s" % e
|
|
|
|
action_result["completed_at"] = int(time.time())
|
|
|
|
# I wonder if this actually works
|
|
self.logger.info("Before last stream result")
|
|
try:
|
|
ret = requests.post("%s%s" % (self.url, stream_path), headers=headers, json=action_result)
|
|
self.logger.info("Result: %d" % ret.status_code)
|
|
if ret.status_code != 200:
|
|
self.logger.info(ret.text)
|
|
except requests.exceptions.ConnectionError as e:
|
|
self.logger.exception(e)
|
|
return
|
|
except TypeError as e:
|
|
self.logger.exception(e)
|
|
action_result["status"] = "FAILURE"
|
|
action_result["result"] = "POST error: %s" % e
|
|
self.logger.info("Before typeerror stream result")
|
|
ret = requests.post("%s%s" % (self.url, stream_path), headers=headers, json=action_result)
|
|
self.logger.info("Result: %d" % ret.status_code)
|
|
if ret.status_code != 200:
|
|
self.logger.info(ret.text)
|
|
|
|
return
|
|
|
|
|
|
#STOPCOPY
|
|
# !!! Let the above line stay - its used for some horrible codegeneration / stitching !!! #
|
|
|
|
@classmethod
|
|
async def run(cls):
|
|
""" Connect to Redis and HTTP session, await actions """
|
|
logging.basicConfig(format="{asctime} - {name} - {levelname}:{message}", style='{')
|
|
logger = logging.getLogger(f"{cls.__name__}")
|
|
logger.setLevel(logging.DEBUG)
|
|
print("Started execution!!")
|
|
|
|
app = cls(redis=None, logger=logger, console_logger=logger)
|
|
|
|
# Authorization for the app/function to control the workflow
|
|
# Function will crash if its wrong, which it probably should.
|
|
|
|
await app.execute_action(app.action)
|