92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
|
# Copyright (c) 2022 Intel and/or its affiliates.
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at:
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
from EdgeAction import *
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
class Edge:
|
||
|
|
def __init__(self):
|
||
|
|
self.JSON = None
|
||
|
|
self.Start = None
|
||
|
|
self.End = None
|
||
|
|
self.actionList = []
|
||
|
|
|
||
|
|
def Create(jsonfile):
|
||
|
|
f = open(jsonfile, "r", encoding="utf-8")
|
||
|
|
token = json.load(f)
|
||
|
|
|
||
|
|
if token == None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
if token["type"] != "edge":
|
||
|
|
return None
|
||
|
|
|
||
|
|
edgeList = []
|
||
|
|
|
||
|
|
startNodes = token["start"]
|
||
|
|
endNodes = token["end"]
|
||
|
|
|
||
|
|
if startNodes == None or endNodes == None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
startTokens = startNodes.split(",")
|
||
|
|
endTokens = endNodes.split(",")
|
||
|
|
|
||
|
|
for start in startTokens:
|
||
|
|
for end in endTokens:
|
||
|
|
edge = Edge()
|
||
|
|
|
||
|
|
edge.Start = start
|
||
|
|
edge.End = end
|
||
|
|
|
||
|
|
if "actions" in token:
|
||
|
|
for at in token["actions"]:
|
||
|
|
action = EdgeAction.Create(at)
|
||
|
|
if not action:
|
||
|
|
return None
|
||
|
|
|
||
|
|
edge.actionList.append(action)
|
||
|
|
|
||
|
|
edge.JSON = jsonfile
|
||
|
|
edgeList.append(edge)
|
||
|
|
|
||
|
|
return edgeList
|
||
|
|
|
||
|
|
def Apply(self, first, second):
|
||
|
|
exp = []
|
||
|
|
|
||
|
|
for i in range(len(self.actionList)):
|
||
|
|
act = self.actionList[i]
|
||
|
|
|
||
|
|
if act.FromStartObject == True:
|
||
|
|
exp.append(first.GetValue(act.FromExpression))
|
||
|
|
elif act.FromStartObject == False:
|
||
|
|
exp.append(second.GetValue(act.FromExpression))
|
||
|
|
else:
|
||
|
|
exp.append(act.FromExpression)
|
||
|
|
|
||
|
|
for i in range(len(exp)):
|
||
|
|
act = self.actionList[i]
|
||
|
|
|
||
|
|
if act.ToStartObject:
|
||
|
|
first.SetFieldAuto(act.ToExpression, exp[i])
|
||
|
|
else:
|
||
|
|
second.SetFieldAuto(act.ToExpression, exp[i])
|
||
|
|
|
||
|
|
def Actions(self):
|
||
|
|
return self.actionList
|
||
|
|
|
||
|
|
def Name(self):
|
||
|
|
return self.Start + "_" + self.End
|