211 lines
8.5 KiB
Python
211 lines
8.5 KiB
Python
import re
|
|
|
|
from key_request_parser import patterns, keyrequests, helper
|
|
|
|
from enum import Flag, auto
|
|
|
|
|
|
class KROption(Flag):
|
|
VALIDATE_EXISTS = auto()
|
|
VALIDATE_HASDATA = auto()
|
|
RESOLVEKEYREQUETS = auto()
|
|
RESOLVESERVICES = auto()
|
|
|
|
class KRParser:
|
|
patterns=[patterns.Pattern1(), patterns.Pattern2(), patterns.Pattern3()]
|
|
|
|
def normalize(self,x):
|
|
#tmp=x.replace("~","")
|
|
tmp=x.replace("\n","")
|
|
#tmp=tmp.replace("\"/","\"")
|
|
#tmp=tmp.replace("\"/","") -_>was active
|
|
#tmp=tmp.replace("/\"","\"")
|
|
tmp=tmp.replace("/\"","")
|
|
tmp=tmp.replace("\"","")
|
|
tmp=tmp.replace("\t","")
|
|
|
|
tmp=re.sub("([\s]*)\)", ")", tmp)
|
|
|
|
|
|
|
|
tmp=re.sub("\([\s\n\r]*", "(", tmp)
|
|
tmp=re.sub("\,[\s\n\r]*", ",", tmp)
|
|
tmp=re.sub("\)[\s\n\r]*,", "),", tmp)
|
|
tmp=re.sub("in[\s\n\r]*\(", "in(", tmp)
|
|
|
|
return tmp
|
|
|
|
|
|
def applyPatterns(self,subject):
|
|
groups=None
|
|
for p in self.patterns:
|
|
groups=p.parseServicesAndMethods(subject)
|
|
|
|
if len(groups) > 0:
|
|
break
|
|
|
|
return groups
|
|
|
|
|
|
def checkKeyRequetsHasData(self,kr, tfrom, DTAPIURL, DTAPIToken):
|
|
|
|
DTAPIURL = DTAPIURL + "/api/v2/entities"
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': 'Api-Token ' + DTAPIToken
|
|
}
|
|
|
|
for gid, group in enumerate(kr.matchedGroups):
|
|
params={"entitySelector": group["existsQuery"], "from":tfrom["tfrom"], "fields": "fromRelationships"}
|
|
response = helper.get_request(DTAPIURL, headers, params)
|
|
entities = (response.json())['entities']
|
|
|
|
if len(entities) > 0:
|
|
y=0
|
|
for method in kr.keyRequests:
|
|
if method["groupId"] == gid:
|
|
found = [x for x in entities if x[method["comparer"]] == method[method["comparer"]]]
|
|
|
|
if len(found) > 0:
|
|
method["hasData"][tfrom["label"]]=True
|
|
#method["displayName"]=found[0]["displayName"]
|
|
#method["entityId"]=found[0]["entityId"]
|
|
#method["services"]=found[0]["fromRelationships"]["isServiceMethodOfService"]
|
|
|
|
# for idx,o in enumerate(method["services"]):
|
|
# tmpS=[p for p in kr.services if p["entityId"]==o["id"]]
|
|
# if len(tmpS)>0:
|
|
# method["services"][idx]=tmpS[0]
|
|
|
|
else:
|
|
method["hasData"][tfrom["label"]]=False
|
|
|
|
def resolveServices(self,services, DTAPIURL, DTAPIToken):
|
|
#DTAPIURL = DTAPIURL + "/api/v2/entities"
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': 'Api-Token ' + DTAPIToken
|
|
}
|
|
|
|
for gid, service in enumerate(services):
|
|
query="type(SERVICE),entityId("+service["id"]+")"
|
|
params={"entitySelector": query,"from":"now-2y"}
|
|
response = helper.get_request(DTAPIURL, headers, params)
|
|
entities = (response.json())['entities']
|
|
|
|
if len(entities)>0:
|
|
services[gid]=entities[0]
|
|
|
|
def resolveKeyRequests(self,kr, DTAPIURL, DTAPIToken, options):
|
|
DTAPIURL = DTAPIURL + "/api/v2/entities"
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': 'Api-Token ' + DTAPIToken
|
|
}
|
|
|
|
|
|
for gid, k in enumerate(kr.keyRequests):
|
|
try:
|
|
query="type(service_method)"
|
|
group=kr.matchedGroups[k["groupId"]]
|
|
|
|
if len(group["services"])> 0:
|
|
if group["services"][0].startswith("SERVICE-"):
|
|
query+=",fromRelationship.isServiceMethodOfService(type(\"SERVICE\"),entityId(\""+'","'.join(group["services"])+"\"))"
|
|
else:
|
|
query+=",fromRelationship.isServiceMethodOfService(type(\"SERVICE\"),entityName.in(\""+'","'.join(group["services"])+"\"))"
|
|
|
|
if k["comparer"]=="entityId":
|
|
query+=",entityId("+k["entityId"]+")"
|
|
else:
|
|
query+=",entityName.in(\""+k["displayName"]+"\")"
|
|
|
|
params={"entitySelector": query, "from":"now-2y","fields": "fromRelationships"}
|
|
response = helper.get_request(DTAPIURL, headers, params)
|
|
entities = (response.json())['entities']
|
|
|
|
# if len(entities) > 1:
|
|
# kr.keyRequests[gid]['foundCount']=len(entities)
|
|
# print("Multiple keyrequest found: ")
|
|
|
|
if len(entities)> 0:
|
|
kr.keyRequests[gid]["found"]=True
|
|
kr.keyRequests[gid]['foundCount']=len(entities)
|
|
kr.keyRequests[gid]["displayName"]=entities[0]["displayName"]
|
|
kr.keyRequests[gid]["entityId"]=entities[0]["entityId"]
|
|
|
|
if "isServiceMethodOfService" in entities[0]["fromRelationships"]:
|
|
kr.keyRequests[gid]["services"]=entities[0]["fromRelationships"]["isServiceMethodOfService"]
|
|
|
|
if KROption.RESOLVESERVICES in options and len( kr.keyRequests[gid]["services"])>0:
|
|
self.resolveServices(kr.keyRequests[gid]["services"], DTAPIURL, DTAPIToken)
|
|
|
|
except Exception as err:
|
|
kr.keyRequests[gid]["exception"]="resolveKeyRequests failed: "+repr(err)
|
|
|
|
|
|
|
|
#kr.mergeServices(entities)
|
|
|
|
|
|
|
|
|
|
def process(self, kr):
|
|
|
|
for gid, group in enumerate(kr.matchedGroups):
|
|
for method in group["methods"]:
|
|
if method.startswith('SERVICE_METHOD-'):
|
|
tmp={"displayName": None,"comparer": "entityId", "entityId":method, "groupId":gid, "hasData":{}, "services":[], "found":False, "foundCount":0, "exception":""} #"exists":None, 'hasData_1W':None,
|
|
else:
|
|
tmp={"displayName":method,"comparer": "displayName", "entityId":None, "groupId":gid, "hasData":{}, "services":[], "found":False, "foundCount":0, "exception":""} #"exists":None, 'hasData_1W':None,
|
|
|
|
kr.keyRequests.append(tmp)
|
|
|
|
|
|
# if self.options and KROption.VALIDATE_EXISTS in self.options:
|
|
# self.checkKeyRequetsExists(kr,self.DTAPIURL, self.DTAPIToken)
|
|
self.resolveKeyRequests(kr,self.DTAPIURL, self.DTAPIToken, self.options)
|
|
|
|
if KROption.VALIDATE_HASDATA in self.options:
|
|
self.checkKeyRequetsHasData(kr,{"label":"1W", "tfrom":"now-1w"},self.DTAPIURL, self.DTAPIToken)
|
|
self.checkKeyRequetsHasData(kr,{"label":"1M", "tfrom":"now-1M"},self.DTAPIURL, self.DTAPIToken)
|
|
# elif KROption.RESOLVEKEYREQUETS in self.options:
|
|
# self.checkKeyRequetsHasData(kr, {"label":"1W", "tfrom":"now-1w"},self.DTAPIURL, self.DTAPIToken)
|
|
# if KROption.RESOLVESERVICES in self.options:
|
|
# self.resolveServices(kr,self.DTAPIURL, self.DTAPIToken)
|
|
|
|
return kr
|
|
|
|
|
|
def parseBySLO(self, row):
|
|
#normalize
|
|
normFilter=self.normalize(row['filter'])
|
|
normExpresseion=self.normalize(row['metricExpression'])
|
|
|
|
tmp_KR = keyrequests.KR({"sloName":row["name"], "env":row["env"], "metricExpression": normExpresseion, "filter": normFilter, "matchedGroups": None})
|
|
|
|
|
|
#SLO with Filter
|
|
if normFilter.upper().startswith("TYPE(SERVICE_METHOD),"):
|
|
subject=normFilter
|
|
else:
|
|
subject=normExpresseion
|
|
|
|
groups=self.applyPatterns(subject)
|
|
|
|
for g in groups:
|
|
if g["methods"] != None and len(g["methods"]) > 0:
|
|
tmp_KR.matchedGroups.append(g)
|
|
|
|
#self.process(tmp_KR)
|
|
|
|
return self.process(tmp_KR)
|
|
|
|
|
|
def __init__(self, options: KROption , DTAPIURL, DTAPIToken ):
|
|
self.DTAPIURL= DTAPIURL
|
|
self.DTAPIToken=DTAPIToken
|
|
self.options=options |