def setupGraph(self):
graph = self.ui.graph
qgv = QGraphViz(
# show_subgraphs=show_subgraphs,
# node_selected_callback=node_selected,
# edge_selected_callback=edge_selected,
# node_invoked_callback=node_invoked,
# edge_invoked_callback=edge_invoked,
)
qgv.new(Dot(Graph("Main_Graph")))
n1 = qgv.addNode(qgv.engine.graph, "Node1", label="N1")
n2 = qgv.addNode(qgv.engine.graph, "Node2", label="N2")
n3 = qgv.addNode(qgv.engine.graph, "Node3", label="N3")
n4 = qgv.addNode(qgv.engine.graph, "Node4", label="N4")
n5 = qgv.addNode(qgv.engine.graph, "Node5", label="N5")
n6 = qgv.addNode(qgv.engine.graph, "Node6", label="N6")
qgv.addEdge(n1, n2, {})
qgv.addEdge(n3, n2, {})
qgv.addEdge(n2, n4, {"width": 2})
qgv.addEdge(n4, n5, {"width": 4})
qgv.addEdge(n4, n6, {"width": 5, "color": "red"})
qgv.addEdge(n3, n6, {"width": 2})
# Build the graph (the layout engine organizes where the nodes and connections are)
qgv.build()
qgv.save("test.gv")
graph.layout().addWidget(qgv)
def Graphics(self, x, y):
item = QGraphViz(
window,
auto_freeze=True,
hilight_Nodes=True,
)
item.setStyleSheet("background-color:white;")
item.new(
Dot(Graph("Main_Graph"), font=QFont("Arial", 12), margins=[40, 40]))
item.move(x, y)
item.resize(1150, 500)
self.widgets.append(item)
return item
def __init__(self):
super(MainWindow, self).__init__()
loadUi('../ui/MainWindow.ui', self)
self.FilterConfigButton = self.findChild(QPushButton, 'filterButton')
self.FilterConfigButton.clicked.connect(self.openFilterConfig)
self.settingsConfig = self.findChild(QAction, 'actionSettings')
self.settingsConfig.triggered.connect(self.openSettings)
#Search button functionality ##need to add keyword search still
self.searchSearchButton = self.findChild(QPushButton,
'searchSearchButton')
self.searchSearchButton.clicked.connect(self.openFilterConfig)
# Enter press on qlineedit search tab triggers the search button
self.searchLineEdit = self.findChild(QLineEdit, 'lineEdit_2')
self.searchLineEdit.returnPressed.connect(
self.searchSearchButton.click)
#Search button on graph tab functionality ##need keyword functionality added
self.graphSearchButton = self.findChild(QPushButton,
'graphSearchButton_2')
# Enter press on qlineedit graph tab triggers search button
self.graphSearchEdit = self.findChild(QLineEdit, 'graphLineEdit')
self.graphSearchEdit.returnPressed.connect(
self.graphSearchButton.click)
#Exit menu option functionality
self.CloseMenuSelect = self.findChild(QAction, 'actionClose_Exit')
self.CloseMenuSelect.setShortcut('Ctrl+Q')
self.CloseMenuSelect.triggered.connect(qApp.quit)
#Export menu option functionality
self.exportConfig = self.findChild(QAction, 'actionExport')
self.exportConfig.setShortcut('Ctrl+E')
self.exportConfig.triggered.connect(self.openExportConfig)
#VectorDBConfig linked to menu option
self.versionControl = self.findChild(QAction, 'actionVersion_Control')
self.versionControl.setShortcut('Ctrl+S')
self.versionControl.triggered.connect(self.openVectDBConfig)
self.actionLogFile = self.findChild(QAction, 'actionFileConfig')
self.actionLogFile.setShortcut('Ctrl+F')
self.actionLogFile.triggered.connect(self.openLogConfig)
self.GraphTable = self.findChild(QTableWidget, 'tableWidget')
self.graphArea = self.findChild(QWidget, 'graphArea')
self.graphArea.setLayout(QVBoxLayout())
self.GraphTable.setItem(0, 0, QTableWidgetItem("Cell (1,1)"))
# Events
def node_selected(node):
if qgv.manipulation_mode == QGraphVizManipulationMode.Node_remove_Mode:
print("Node {} removed".format(node))
else:
print("Node selected {}".format(node))
def edge_selected(edge):
if qgv.manipulation_mode == QGraphVizManipulationMode.Edge_remove_Mode:
print("Edge {} removed".format(edge))
else:
print("Edge selected {}".format(edge))
def node_invoked(node):
print("Node double clicked")
def edge_invoked(node):
print("Edge double clicked")
def node_removed(node):
print("Node removed")
def edge_removed(node):
print("Edge removed")
# Create QGraphViz widget
show_subgraphs = True
qgv = QGraphViz(show_subgraphs=show_subgraphs,
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True)
qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
qgv.new(Dot(Graph("Main_Graph"), show_subgraphs=show_subgraphs))
# Adding nodes with an image as its shape
icon_path = os.path.dirname(os.path.abspath(__file__)) + r"\dbicon.png"
# Build the graph (the layout engine organizes where the nodes and connections are)
qgv.build()
# Save it to a file to be loaded by Graphviz if needed
qgv.save("test.gv")
# Add the QGraphViz object to the layout
self.graphArea.layout().addWidget(qgv)
# Add a horizontal layout (a pannel to select what to do)
hpanel = QHBoxLayout()
self.graphArea.layout().addLayout(hpanel)
# Add few buttons to the panel
def manipulate():
qgv.manipulation_mode = QGraphVizManipulationMode.Nodes_Move_Mode
def save():
fname = QFileDialog.getSaveFileName(qgv, "Save", "", "*.json")
if (fname[0] != ""):
qgv.saveAsJson(fname[0])
def new():
qgv.engine.graph = Graph("MainGraph")
qgv.build()
qgv.repaint()
def load():
fname = QFileDialog.getOpenFileName(qgv, "Open", "", "*.json")
if (fname[0] != ""):
qgv.loadAJson(fname[0])
def add_node():
dlg = QDialog()
dlg.ok = False
dlg.node_name = ""
dlg.node_label = ""
dlg.node_type = "None"
# Layouts
main_layout = QVBoxLayout()
l = QFormLayout()
buttons_layout = QHBoxLayout()
main_layout.addLayout(l)
main_layout.addLayout(buttons_layout)
dlg.setLayout(main_layout)
leNodeName = QLineEdit()
leNodeLabel = QLineEdit()
cbxNodeType = QComboBox()
leImagePath = QLineEdit()
pbOK = QPushButton()
pbCancel = QPushButton()
cbxNodeType.addItems(["None", "circle", "box"])
pbOK.setText("&OK")
pbCancel.setText("&Cancel")
l.setWidget(0, QFormLayout.LabelRole, QLabel("Node Name"))
l.setWidget(0, QFormLayout.FieldRole, leNodeName)
l.setWidget(1, QFormLayout.LabelRole, QLabel("Node Label"))
l.setWidget(1, QFormLayout.FieldRole, leNodeLabel)
l.setWidget(2, QFormLayout.LabelRole, QLabel("Node Type"))
l.setWidget(2, QFormLayout.FieldRole, cbxNodeType)
l.setWidget(3, QFormLayout.LabelRole, QLabel("Node Image"))
l.setWidget(3, QFormLayout.FieldRole, leImagePath)
def ok():
dlg.OK = True
dlg.node_name = leNodeName.text()
dlg.node_label = leNodeLabel.text()
if (leImagePath.text()):
dlg.node_type = leImagePath.text()
else:
dlg.node_type = cbxNodeType.currentText()
dlg.close()
def cancel():
dlg.OK = False
dlg.close()
pbOK.clicked.connect(ok)
pbCancel.clicked.connect(cancel)
buttons_layout.addWidget(pbOK)
buttons_layout.addWidget(pbCancel)
dlg.exec_()
# node_name, okPressed = QInputDialog.getText(wi, "Node name","Node name:", QLineEdit.Normal, "")
if dlg.OK and dlg.node_name != '':
qgv.addNode(qgv.engine.graph,
dlg.node_name,
label=dlg.node_label,
shape=dlg.node_type)
qgv.build()
def add_node(node_id, node_name, node_timestamp, node_description,
log_entry_reference, log_creator, event_type, icon_type,
source, visibility):
qgv.addNode(qgv.engine.graph,
node_name,
label=node_description,
shape='icons/blue_circle.png')
def remove_node():
qgv.manipulation_mode = QGraphVizManipulationMode.Node_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemoveNode.setChecked(True)
def remove_edge():
qgv.manipulation_mode = QGraphVizManipulationMode.Edge_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemoveEdge.setChecked(True)
def add_edge():
qgv.manipulation_mode = QGraphVizManipulationMode.Edges_Connect_Mode
for btn in buttons_list:
btn.setChecked(False)
btnAddEdge.setChecked(True)
# Add buttons
btnNew = QPushButton("New")
btnNew.clicked.connect(new)
btnOpen = QPushButton("Open")
btnOpen.clicked.connect(load)
btnSave = QPushButton("Save")
btnSave.clicked.connect(save)
hpanel.addWidget(btnNew)
hpanel.addWidget(btnOpen)
hpanel.addWidget(btnSave)
buttons_list = []
btnManip = QPushButton("Manipulate")
btnManip.setCheckable(True)
btnManip.setChecked(True)
btnManip.clicked.connect(manipulate)
hpanel.addWidget(btnManip)
buttons_list.append(btnManip)
btnAddNode = QPushButton("Add Node")
btnAddNode.clicked.connect(add_node)
hpanel.addWidget(btnAddNode)
buttons_list.append(btnAddNode)
btnRemoveNode = QPushButton("Remove Node")
btnRemoveNode.setCheckable(True)
btnRemoveNode.clicked.connect(remove_node)
hpanel.addWidget(btnRemoveNode)
buttons_list.append(btnRemoveNode)
btnAddEdge = QPushButton("Add Edge")
btnAddEdge.setCheckable(True)
btnAddEdge.clicked.connect(add_edge)
hpanel.addWidget(btnAddEdge)
buttons_list.append(btnAddEdge)
btnRemoveEdge = QPushButton("Remove Edge")
btnRemoveEdge.setCheckable(True)
btnRemoveEdge.clicked.connect(remove_edge)
hpanel.addWidget(btnRemoveEdge)
buttons_list.append(btnRemoveEdge)
#drop down menus vector collumn search table
self.searchSearchTableWidget = self.findChild(QTableWidget,
'tableWidget_2')
i = 0
while i < 10:
combo = QComboBox()
combo.addItems([' ', '1', '2', '3'])
self.searchSearchTableWidget.setCellWidget(i, 3, combo)
i += 1
self.showMaximized()
qgv = QGraphViz(
show_subgraphs=show_subgraphs,
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True
)
qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
qgv.new(Dot(Graph("Main_Graph"), show_subgraphs=show_subgraphs))
# Adding nodes with an image as its shape
icon_path = os.path.dirname(os.path.abspath(__file__)) + r"\dbicon.png"
# Build the graph (the layout engine organizes where the nodes and connections are)
qgv.build()
# Save it to a file to be loaded by Graphviz if needed
qgv.save("test.gv")
# Create a Main window
w = QMainWindow()
w.setWindowTitle('Graph')
# Create a central widget to handle the QGraphViz object
wi = QWidget()
wi.setLayout(QVBoxLayout())
w.setCentralWidget(wi)
class GraphGenerator(object):
def __init__(self, name):
self.name = name
self.avlbIndex = 0
self.nodes = dict()
self.qgv = QGraphViz(node_invoked_callback=self.nodeInvoked)
self.setup()
def setup(self):
self.qgv.new(Dot(Graph(self.name)))
# Leaving this here for examples
# n1 = self.qgv.addNode(self.qgv.engine.graph, "Node1", label="N1")
# n2 = self.qgv.addNode(self.qgv.engine.graph, "Node2", label="N2")
# n3 = self.qgv.addNode(self.qgv.engine.graph, "Node3", label="N3")
# n4 = self.qgv.addNode(self.qgv.engine.graph, "Node4", label="N4")
# n5 = self.qgv.addNode(self.qgv.engine.graph, "Node5", label="N5")
# n6 = self.qgv.addNode(self.qgv.engine.graph, "Node6", label="N6")
# self.qgv.addEdge(n1, n2, {})
# self.qgv.addEdge(n3, n2, {})
# self.qgv.addEdge(n2, n4, {"width":2})
# self.qgv.addEdge(n4, n5, {"width":4})
# self.qgv.addEdge(n4, n6, {"width":5,"color":"red"})
# self.qgv.addEdge(n3, n6, {"width":2})
# self.build()
def addNode(self, **kwargs):
if kwargs['name']:
name = kwargs['name']
else:
name = "Node" + str(self.avlbIndex)
if kwargs['label']:
label = kwargs['label']
else:
label = "N" + str(self.avlbIndex)
if kwargs['shape']:
shape = kwargs['shape']
else:
shape = "square"
node = self.qgv.addNode(self.qgv.engine.graph,
name,
label=label,
shape=shape)
self.nodes[node] = []
def addEdge(self, source, destination):
src = None
for node in self.nodes.keys():
if node.name == source:
src = node
break
if src == None:
raise Exception("No such node")
dest = None
for node in self.nodes.keys():
if node.name == destination:
dest = node
if dest == None:
raise Exception("No such node")
self.nodes[src].append(dest)
self.qgv.addEdge(src, dest, {})
def build(self):
self.qgv.build()
def save(self):
filename = self.name + ".gv"
self.qgv.save(filename)
def getGraph(self):
return self.qgv
def getNodes(self):
return self.nodes.keys()
def nodeInvoked(self, node):
print(node)
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
loadUi('../ui/MainWindow.ui', self)
self.myDb = DBHandler()
self.updateTableFlag = True
self.FilterConfigButton = self.findChild(QPushButton, 'filterButton')
self.FilterConfigButton.clicked.connect(self.openFilterConfig)
self.settingsConfig = self.findChild(QAction, 'actionSettings')
self.settingsConfig.triggered.connect(self.openSettings)
#Search button functionality ##need to add keyword search still
self.searchSearchButton = self.findChild(QPushButton,
'searchSearchButton')
self.searchSearchButton.clicked.connect(self.openFilterConfig)
# Enter press on qlineedit search tab triggers the search button
self.searchLineEdit = self.findChild(QLineEdit, 'lineEdit_2')
self.searchLineEdit.returnPressed.connect(
self.searchSearchButton.click)
# Enter press on qlineedit graph tab triggers search button
self.graphSearchEdit = self.findChild(QLineEdit, 'graphLineEdit')
#Exit menu option functionality
self.CloseMenuSelect = self.findChild(QAction, 'actionClose_Exit')
self.CloseMenuSelect.setShortcut('Ctrl+Q')
self.CloseMenuSelect.triggered.connect(qApp.quit)
#Export menu option functionality
self.exportConfig = self.findChild(QAction, 'actionExport')
self.exportConfig.setShortcut('Ctrl+E')
self.exportConfig.triggered.connect(self.openExportConfig)
#VectorDBConfig linked to menu option
self.versionControl = self.findChild(QAction, 'actionVersion_Control')
self.versionControl.setShortcut('Ctrl+S')
self.versionControl.triggered.connect(self.openVectDBConfig)
self.GraphTable = self.findChild(QTableWidget, 'tableWidget')
self.graphArea = self.findChild(QWidget, 'graphArea')
self.graphArea.setLayout(QVBoxLayout())
self.graphSearchButton = self.findChild(QPushButton,
'graphSearchButton_2')
# self.graphSearchButton.clicked.connect()
self.currentVectorMenu = self.findChild(QComboBox, 'VectorMenu')
self.currentVectorLabel = self.findChild(QLabel, 'CurrentVector')
self.currentVectorMenu.addItems(self.myDb.get_vector_names())
self.currentVectorLabel.setText(self.currentVectorMenu.currentText())
self.currentVectorMenu.activated.connect(self.vectorSelected)
self.GraphTable.cellClicked.connect(self.entryFieldSelected)
self.GraphTable.cellChanged.connect(self.updateEntryFromTable)
# Events
def node_selected(node):
if self.qgv.manipulation_mode == QGraphVizManipulationMode.Node_remove_Mode:
print("Node {} removed".format(node))
self.saveGraph()
# self.updateTableFromGraph()
else:
print("Node selected {}".format(node))
def edge_selected(edge):
if self.qgv.manipulation_mode == QGraphVizManipulationMode.Edge_remove_Mode:
print("Edge {} removed".format(edge))
else:
print("Edge selected {}".format(edge))
def node_invoked(node):
print("Node double clicked")
def edge_invoked(node):
print("Edge double clicked")
def node_removed(node):
print("Node removed")
def edge_removed(node):
print("Edge removed")
# Create QGraphViz widget
show_subgraphs = True
self.qgv = QGraphViz(show_subgraphs=show_subgraphs,
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True)
self.qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
self.qgv.new(Dot(Graph("Main_Graph"), show_subgraphs=show_subgraphs))
# Adding nodes with an image as its shape
icon_path = os.path.dirname(os.path.abspath(__file__)) + r"\dbicon.png"
# Build the graph (the layout engine organizes where the nodes and connections are)
self.qgv.build()
# Save it to a file to be loaded by Graphviz if needed
self.qgv.save("test.gv")
# Add the QGraphViz object to the layout
self.graphArea.layout().addWidget(self.qgv)
# Add a horizontal layout (a pannel to select what to do)
self.hpanel = QHBoxLayout()
self.graphArea.layout().addLayout(self.hpanel)
# Add buttons to the panel
def save():
fname = QFileDialog.getSaveFileName(self.qgv, "Save", "", "*.json")
if (fname[0] != ""):
self.qgv.saveAsJson(fname[0])
def new():
self.qgv.engine.graph = Graph("MainGraph")
self.qgv.build()
self.qgv.repaint()
def load():
fname = QFileDialog.getOpenFileName(self.qgv, "Open", "", "*.json")
if (fname[0] != ""):
self.qgv.loadAJson(fname[0])
def add_node():
dlg = QDialog()
dlg.ok = False
dlg.node_name = ""
dlg.node_label = ""
dlg.node_color = ""
dlg.node_type = "None"
# Layouts
main_layout = QVBoxLayout()
l = QFormLayout()
buttons_layout = QHBoxLayout()
main_layout.addLayout(l)
main_layout.addLayout(buttons_layout)
dlg.setLayout(main_layout)
leNodeName = QLineEdit()
leNodeLabel = QLineEdit()
cbxNodeType = QComboBox()
leImagePath = QLineEdit()
leNodeColor = QLineEdit()
pbOK = QPushButton()
pbCancel = QPushButton()
cbxNodeType.addItems(["None", "circle", "box"])
pbOK.setText("&OK")
pbCancel.setText("&Cancel")
l.setWidget(0, QFormLayout.LabelRole, QLabel("Node Name"))
l.setWidget(0, QFormLayout.FieldRole, leNodeName)
l.setWidget(1, QFormLayout.LabelRole, QLabel("Node Label"))
l.setWidget(1, QFormLayout.FieldRole, leNodeLabel)
l.setWidget(2, QFormLayout.LabelRole, QLabel("Node Type"))
l.setWidget(2, QFormLayout.FieldRole, cbxNodeType)
l.setWidget(3, QFormLayout.LabelRole, QLabel("Node Image"))
l.setWidget(3, QFormLayout.FieldRole, leImagePath)
l.setWidget(4, QFormLayout.LabelRole, QLabel("Node Color"))
l.setWidget(4, QFormLayout.FieldRole, leNodeColor)
def ok():
dlg.OK = True
dlg.node_name = leNodeName.text()
dlg.node_label = leNodeLabel.text()
if (leImagePath.text()):
dlg.node_type = leImagePath.text()
else:
dlg.node_type = cbxNodeType.currentText()
dlg.node_color = leNodeColor.text()
dlg.close()
def cancel():
dlg.OK = False
dlg.close()
pbOK.clicked.connect(ok)
pbCancel.clicked.connect(cancel)
buttons_layout.addWidget(pbOK)
buttons_layout.addWidget(pbCancel)
dlg.exec_()
# node_name, okPressed = QInputDialog.getText(wi, "Node name","Node name:", QLineEdit.Normal, "")
if dlg.OK and dlg.node_name != '':
self.qgv.addNode(self.qgv.engine.graph,
dlg.node_name,
label=dlg.node_label,
shape=dlg.node_type,
color=dlg.node_color)
self.qgv.build()
def remove_node():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Node_remove_Mode
for btn in self.buttons_list:
btn.setChecked(False)
self.btnRemoveNode.setChecked(True)
def remove_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edge_remove_Mode
for btn in self.buttons_list:
btn.setChecked(False)
self.btnRemoveEdge.setChecked(True)
def add_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edges_Connect_Mode
for btn in self.buttons_list:
btn.setChecked(False)
self.btnAddEdge.setChecked(True)
# Add buttons
self.btnNew = QPushButton("New")
self.btnNew.clicked.connect(new)
self.btnOpen = QPushButton("Open")
self.btnOpen.clicked.connect(load)
self.btnSave = QPushButton("Save")
self.btnSave.clicked.connect(save)
self.hpanel.addWidget(self.btnNew)
self.hpanel.addWidget(self.btnOpen)
self.hpanel.addWidget(self.btnSave)
self.buttons_list = []
self.btnAddNode = QPushButton("Add Node")
self.btnAddNode.clicked.connect(add_node)
self.hpanel.addWidget(self.btnAddNode)
self.buttons_list.append(self.btnAddNode)
self.btnRemoveNode = QPushButton("Remove Node")
self.btnRemoveNode.setCheckable(True)
self.btnRemoveNode.clicked.connect(remove_node)
self.hpanel.addWidget(self.btnRemoveNode)
self.buttons_list.append(self.btnRemoveNode)
self.btnAddEdge = QPushButton("Add Edge")
self.btnAddEdge.setCheckable(True)
self.btnAddEdge.clicked.connect(add_edge)
self.hpanel.addWidget(self.btnAddEdge)
self.buttons_list.append(self.btnAddEdge)
self.btnRemoveEdge = QPushButton("Remove Edge")
self.btnRemoveEdge.setCheckable(True)
self.btnRemoveEdge.clicked.connect(remove_edge)
self.hpanel.addWidget(self.btnRemoveEdge)
self.buttons_list.append(self.btnRemoveEdge)
#icon_path = os.path.dirname(os.path.abspath(__file__)) + r"\Resouces\IconDir,100,100"
# n9 = qgv.addNode(qgv.engine.graph, "Node9", label="N9", shape=icon_path)
#drop down menus vector collumn search table
self.SearchTable = self.findChild(QTableWidget, 'tableWidget_2')
self.showMaximized()
# DBHandler.create_vector_entry('../Resources/LocalGraphs/VECTOR_3.json')
self.updateViews()
def updateViews(self):
self.generateModels()
self.populateTable()
self.populateGraph()
self.populateSearchTable()
def openVectDBConfig(self):
self.window = VectorDBConfig()
self.window.show()
def openExportConfig(self):
self.window = ExportConfig()
self.window.show()
def openFilterConfig(self):
self.window = FilterConfig()
self.window.show()
def openSettings(self):
self.window = setting_view.SettingsWindow()
self.window.show()
def vectorSelected(self, index):
self.currentVectorLabel.setText(self.currentVectorMenu.currentText())
self.updateViews()
def populateSearchTable(self):
self.SearchTable.setRowCount(0)
entriesList = self.myDb.get_all_log_entries()
i = 0
for entry in entriesList:
self.SearchTable.insertRow(i)
self.SearchTable.setItem(i, 0, QTableWidgetItem(str(entry['id'])))
self.SearchTable.setItem(
i, 1, QTableWidgetItem(str(entry['time_stamp'])))
self.SearchTable.setItem(i, 2,
QTableWidgetItem(str(entry['content'])))
self.SearchTable.setItem(i, 3,
QTableWidgetItem(str(entry['source'])))
self.SearchTable.setItem(i, 4,
QTableWidgetItem(str(entry['host'])))
self.SearchTable.setItem(i, 5,
QTableWidgetItem('No vector assigned'))
i += 1
def populateTable(self):
self.updateTableFlag = False
self.GraphTable.setRowCount(0)
vectorList = self.myDb.get_all_vectors_raw()
i = 0
for vector in vectorList:
if vector['name'] == self.currentVectorMenu.currentText():
for entry in vector['entries']:
self.GraphTable.insertRow(i)
self.GraphTable.setItem(i, 0,
QTableWidgetItem(str(entry['id'])))
self.GraphTable.setItem(
i, 1, QTableWidgetItem(str(entry['name'])))
self.GraphTable.setItem(
i, 2, QTableWidgetItem(str(entry['timestamp'])))
self.GraphTable.setItem(
i, 3, QTableWidgetItem(entry['description']))
self.GraphTable.setItem(
i, 4, QTableWidgetItem(str(entry['reference'])))
self.GraphTable.setItem(
i, 5, QTableWidgetItem(str(entry['source'])))
i += 1
self.updateTableFlag = True
def populateGraph(self):
vectorList = self.myDb.get_all_vectors_raw()
for vector in vectorList:
if vector['name'] == self.currentVectorMenu.currentText():
self.qgv.loadAJson("../Resouces/LocalGraphs/" +
vector['name'] + ".json")
def generateModels(self):
vectorList = self.myDb.get_all_vectors_raw()
for vector in vectorList:
GraphOperations.graph_from_raw_vector(vector)
def updateEntryFromTable(self, row, col):
# This prevents table from updating vector db entries when clearing table and populating
if self.updateTableFlag:
print(row, col)
def entryFieldSelected(self, row, col):
if self.GraphTable.item(row, col):
print(self.GraphTable.item(row, col).text())
if self.GraphTable.item(row, 0):
print(self.GraphTable.item(row, 0).text())
#make copy
#delete old
#upload copy
# def refreshView(self): # self.populateTable()
# self.populateGraph()
#
# #compares graph to table elements
# def updateTableFromGraph(self):
# client = MongoClient(port=27017)
# db = client.business
# cursor = db.Vectors.find({})
# for vector in cursor:
# if vector['name'] == self.currentVectorMenu.currentText():
# f = open("../Resouces/LocalGraphs/" + vector['name'] + ".json", "r")
# tempFile = f.read()
# tempJson = json.loads(tempFile)
# f.close
# tempSet1 = []
# tempSet2 = []
# for each in vector['entries']:
# tempSet1.append(str(each['number']))
# for each in tempJson['nodes']:
# tempSet2.append(str(each['name']))
# listDifference = [item for item in tempSet1 if item not in tempSet2]
#
# tempVector = vector
# self.deleteThis(listDifference)
# if '_id' in tempVector:
# del tempVector['_id']
# i = 0
# for entry in tempVector['entries']:
# if int(listDifference) == entry['number']:
# del tempVector['entries'][i]
# i+=1
# db.Vectors.insert_one(tempVector)
#
# def deleteThis(self, num):
# client = MongoClient(port=27017)
# db = client.business
# db.Vectors.delete_one({'entries.number': int(num)})
#
def saveGraph(self):
self.qgv.saveAsJson("../Resouces/LocalGraphs/" +
self.currentVectorMenu.currentText() + ".json")
qgv = QGraphViz(
show_subgraphs=show_subgraphs,
auto_freeze=True, # show autofreeze capability
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True)
qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
qgv.new(
Dot(Graph("Main_Graph"),
show_subgraphs=show_subgraphs,
font=QFont("Arial", 12),
margins=[20, 20]))
# Define sone graph
n1 = qgv.addNode(qgv.engine.graph, "Node1", label="N1", fillcolor="red")
n2 = qgv.addNode(qgv.engine.graph,
"Node2",
label="N2",
fillcolor="blue:white:red")
n3 = qgv.addNode(qgv.engine.graph,
"Node3",
label="N3",
shape="diamond",
fillcolor="orange")
n4 = qgv.addNode(qgv.engine.graph,
"Node4",
class NodeEditorWindow(QWidget):
vectors: dict
nodes: dict
relationships: dict
export_signal = pyqtSignal()
node_added = pyqtSignal()
node_deleted = pyqtSignal()
def __init__(self, parent=None):
super().__init__(parent)
self.vectors = {}
self.nodes = {}
self.relationships = {}
self.graph_nodes = []
self.scene_width = 64000
self.scene_height = 64000
self.setMouseTracking(True)
self.initUI()
self.export_window = ExportConfigurationWindow()
self.export_signal.connect(self.export_image)
self.ok = True
self.node_count = 0
def initUI(self):
self.setGeometry(200, 200, 800, 600)
self.layout = QHBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
# Create graphics scene
# Creating the tablular view
self.table = QTableWidget(0, 10, self)
# self.table.setFixedWidth(850)
self.table.setHorizontalHeaderItem(0, QTableWidgetItem(QIcon('icons/up_arrow.png'), "Node ID"))
self.table.setHorizontalHeaderItem(1, QTableWidgetItem(QIcon('icons/up_arrow.png'), "Node Name"))
self.table.setHorizontalHeaderItem(2, QTableWidgetItem(QIcon('icons/up_arrow.png'), "Node Timestamp"))
self.table.setHorizontalHeaderItem(3, QTableWidgetItem(QIcon('icons/up_arrow.png'), "Node Description"))
self.table.setHorizontalHeaderItem(4, QTableWidgetItem(QIcon('icons/up_arrow.png'), "Log Entry Reference"))
self.table.setHorizontalHeaderItem(5, QTableWidgetItem(QIcon('icons/up_arrow.png'), 'Log Creator'))
self.table.setHorizontalHeaderItem(6, QTableWidgetItem('Event Type'))
self.table.setHorizontalHeaderItem(7, QTableWidgetItem(QIcon('icon/unchecked.png'), "Icon Type"))
self.table.setHorizontalHeaderItem(8, QTableWidgetItem(QIcon('icons/up_arrow.png'), 'Source'))
self.table.setHorizontalHeaderItem(9, QTableWidgetItem('Node Visibility'))
self.header = self.table.horizontalHeader()
for i in range(self.table.columnCount()):
self.header.setSectionResizeMode(i, QHeaderView.ResizeToContents)
# Creating the Graphical View
self.layout.addWidget(self.table, 1)
self.setWindowTitle('Node Editor')
# Events
def node_selected(node):
if self.qgv.manipulation_mode == QGraphVizManipulationMode.Node_remove_Mode:
print("Node {} removed".format(node))
if self.qgv.manipulation_mode == QGraphVizManipulationMode.Nodes_Move_Mode:
pass
else:
print("Node selected {}".format(node))
def edge_selected(edge):
if self.qgv.manipulation_mode == QGraphVizManipulationMode.Edge_remove_Mode:
print("Edge {} removed".format(edge))
else:
print("Edge selected {}".format(edge))
def node_invoked(node):
print("Node double clicked")
def edge_invoked(node):
print("Edge double clicked")
def node_removed(node):
print("Node removed")
def edge_removed(node):
print("Edge removed")
# Create QGraphViz widget
show_subgraphs = True
self.qgv = QGraphViz(
show_subgraphs=show_subgraphs,
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True
)
self.qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
self.qgv.new(Dot(Graph("Main_Graph"), show_subgraphs=show_subgraphs))
# Adding nodes with an image as its shape
icon_path = os.path.dirname(os.path.abspath(__file__)) + r"\dbicon.png"
# Build the graph (the layout engine organizes where the nodes and connections are)
self.qgv.build()
# Save it to a file to be loaded by Graphviz if needed
self.qgv.save("test.gv")
self.graphArea = QWidget()
# Add the QGraphViz object to the layout
self.graphArea.setLayout(QVBoxLayout())
self.graphArea.layout().addWidget(self.qgv)
# Add a horizontal layout (a pannel to select what to do)
hpanel = QHBoxLayout()
self.graphArea.layout().addLayout(hpanel)
self.layout.addWidget(self.graphArea)
# Add few buttons to the panel
def manipulate():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Nodes_Move_Mode
def save():
fname = QFileDialog.getSaveFileName(self.qgv, "Save", "", "*.json")
if (fname[0] != ""):
self.qgv.saveAsJson(fname[0])
def new():
self.qgv.engine.graph = Graph("MainGraph")
self.qgv.build()
self.qgv.repaint()
def load():
fname = QFileDialog.getOpenFileName(self.qgv, "Open", "", "*.json")
if (fname[0] != ""):
self.qgv.loadAJson(fname[0])
def add_node():
dlg = QDialog()
dlg.ok = False
dlg.node_name = ""
dlg.node_label = ""
dlg.node_type = "None"
# Layouts
main_layout = QVBoxLayout()
l = QFormLayout()
buttons_layout = QHBoxLayout()
main_layout.addLayout(l)
main_layout.addLayout(buttons_layout)
dlg.setLayout(main_layout)
leNodeName = QLineEdit()
leNodeLabel = QLineEdit()
cbxNodeType = QComboBox()
self.leImagePath = QLineEdit()
leImageBtn = QPushButton(clicked=self.open_file, icon=QIcon('icons/folder.png'))
pbOK = QPushButton()
pbCancel = QPushButton()
cbxNodeType.addItems(["Red Node", "Blue Node", "White Node"])
pbOK.setText("&OK")
pbCancel.setText("&Cancel")
l.setWidget(0, QFormLayout.LabelRole, QLabel("Node Name"))
l.setWidget(0, QFormLayout.FieldRole, leNodeName)
l.setWidget(1, QFormLayout.LabelRole, QLabel("Node Label"))
l.setWidget(1, QFormLayout.FieldRole, leNodeLabel)
l.setWidget(2, QFormLayout.LabelRole, QLabel("Node Type"))
l.setWidget(2, QFormLayout.FieldRole, cbxNodeType)
l.setWidget(3, QFormLayout.LabelRole, QLabel("Node Image"))
l.setWidget(3, QFormLayout.FieldRole, self.leImagePath)
l.addWidget(leImageBtn)
def ok():
dlg.OK = True
dlg.node_name = leNodeName.text()
dlg.node_label = leNodeLabel.text()
if self.leImagePath.text():
dlg.node_type = self.leImagePath.text()
else:
if 'Red Node' in cbxNodeType.currentText():
dlg.node_type = 'icons/red_circle.png'
if 'Blue Node' in cbxNodeType.currentText():
dlg.node_type = 'icons/blue_circle.png'
if 'White Node' in cbxNodeType.currentText():
dlg.node_type = 'icons/white_circle.png'
dlg.close()
def cancel():
self.ok = False
dlg.close()
pbOK.clicked.connect(ok)
pbCancel.clicked.connect(cancel)
buttons_layout.addWidget(pbOK)
buttons_layout.addWidget(pbCancel)
dlg.exec_()
# node_name, okPressed = QInputDialog.getText(wi, "Node name","Node name:", QLineEdit.Normal, "")
#TODO if cancel is selected, we can no longer add nodes
if self.ok:
if dlg.node_name != '' and dlg.node_label!='':
self.qgv.addNode(self.qgv.engine.graph, dlg.node_name, label=dlg.node_label, shape=dlg.node_type)
self.qgv.build()
self.node_added.emit()
else:
QMessageBox.information(self, 'Incorrect Input', 'Node must have a name and a label.')
def remove_node():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Node_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemoveNode.setChecked(True)
def remove_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edge_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemoveEdge.setChecked(True)
def add_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edges_Connect_Mode
for btn in buttons_list:
btn.setChecked(False)
btnAddEdge.setChecked(True)
# Add buttons
btnNew = QPushButton("New")
btnNew.clicked.connect(new)
btnOpen = QPushButton("Open")
btnOpen.clicked.connect(load)
btnSave = QPushButton("Save")
btnSave.clicked.connect(save)
hpanel.addWidget(btnNew)
hpanel.addWidget(btnOpen)
hpanel.addWidget(btnSave)
buttons_list = []
btnManip = QPushButton("Manipulate")
btnManip.setCheckable(True)
btnManip.setChecked(True)
btnManip.clicked.connect(manipulate)
hpanel.addWidget(btnManip)
buttons_list.append(btnManip)
btnAddNode = QPushButton("Add Node")
btnAddNode.clicked.connect(add_node)
hpanel.addWidget(btnAddNode)
buttons_list.append(btnAddNode)
btnRemoveNode = QPushButton("Remove Node")
btnRemoveNode.setCheckable(True)
btnRemoveNode.clicked.connect(remove_node)
hpanel.addWidget(btnRemoveNode)
buttons_list.append(btnRemoveNode)
btnAddEdge = QPushButton("Add Edge")
btnAddEdge.setCheckable(True)
btnAddEdge.clicked.connect(add_edge)
hpanel.addWidget(btnAddEdge)
buttons_list.append(btnAddEdge)
btnRemoveEdge = QPushButton("Remove Edge")
btnRemoveEdge.setCheckable(True)
btnRemoveEdge.clicked.connect(remove_edge)
hpanel.addWidget(btnRemoveEdge)
buttons_list.append(btnRemoveEdge)
btnExport = QPushButton("Export Graph")
btnExport.clicked.connect(self.export_action)
hpanel.addWidget(btnExport)
buttons_list.append(btnExport)
self.setWindowTitle('Graph View')
self.setLayout(self.layout)
# self.addDebugContent()
def add_node_param(self, node_name, node_description, event_type):
img = ''
if 'blue' in event_type:
img = 'icons/blue_circle.png'
if 'red' in event_type:
img = 'icons/red_circle.png'
if 'white' in event_type:
img = 'icons/white_circle.png'
node = self.qgv.core.addNode(self.qgv.engine.graph, node_name=node_name, label=node_description, shape=img)
self.graph_nodes.append(node.name)
self.qgv.build()
def export_action(self):
self.export_window.show()
self.export_window.closeEvent = self.export_image
def export_image(self,event):
print(self.export_window.file_name,self.export_window.extension)
if self.export_window.file_name == '' or self.export_window.extension == '':
QMessageBox.critical(self,"Empty", 'Enter values for both the filename and extension fields')
else:
self.qgv.core.grab().save(self.export_window.file_name + '.' + self.export_window.extension,
self.export_window.extension)
QMessageBox.information(self,'Success', 'File Successfully Exported!!!!')
def addEdge(self):
pass
def mousePressEvent(self, QMouseEvent):
if QMouseEvent.button == Qt.RightButton:
self.contextMenuEvent(QMouseEvent)
def open_file(self):
"""
This method opens a file dialog and allows a user to select the directory containing the files
they want to ingest.
:return:
"""
# Open the file selection dialog
file, options = QFileDialog.getOpenFileName(QFileDialog())
self.leImagePath.setText(str(file))
class GraphWindow(object):
def setupUi(self, graphWindow):
self.graphWin = graphWindow.setObjectName("graphWindow")
graphWindow.setObjectName("graphWindow")
graphWindow.resize(1155, 895)
graphWindow.setDockNestingEnabled(True)
self.centralwidget = QtWidgets.QWidget(graphWindow)
self.centralwidget.setObjectName("centralwidget")
self.gridLayout = QtWidgets.QGridLayout(self.centralwidget)
self.gridLayout.setObjectName("gridLayout")
self.configuration = Configuration.get_instance()
#*-----------------------vector selection objects------------------------------------------------*#
self.groupBoxVectorSelection = QtWidgets.QGroupBox(self.centralwidget)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.groupBoxVectorSelection.sizePolicy().hasHeightForWidth())
self.groupBoxVectorSelection.setSizePolicy(sizePolicy)
self.groupBoxVectorSelection.setObjectName("groupBoxVectorSelection")
self.gridLayout.addWidget(self.groupBoxVectorSelection, 0, 0, 1, 1)
self.formLayoutVector = QtWidgets.QHBoxLayout(
self.groupBoxVectorSelection)
self.formLayoutVector.setObjectName("formLayoutVector")
#vector label
self.labelVector = QtWidgets.QLabel(self.groupBoxVectorSelection)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Minimum)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.labelVector.sizePolicy().hasHeightForWidth())
self.labelVector.setSizePolicy(sizePolicy)
self.labelVector.setObjectName("labelVector")
self.formLayoutVector.addWidget(self.labelVector)
#vector combo box
self.comboBoxVector = QtWidgets.QComboBox(self.groupBoxVectorSelection)
self.comboBoxVector.setMinimumSize(QtCore.QSize(150, 0))
self.comboBoxVector.setMaximumSize(QtCore.QSize(69, 16777215))
self.comboBoxVector.setObjectName("comboBoxVector")
self.formLayoutVector.addWidget(self.comboBoxVector)
self.comboBoxVector.addItem("None")
self.vectorList = Configuration.get_list_of_vector_dicts(
self.configuration)
for vector in self.vectorList:
self.comboBoxVector.addItem(vector["name"])
#vector description label
self.labelVectorDesc = QtWidgets.QLabel(self.groupBoxVectorSelection)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Preferred)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.labelVectorDesc.sizePolicy().hasHeightForWidth())
self.labelVectorDesc.setSizePolicy(sizePolicy)
self.labelVectorDesc.setObjectName("labelVectorDesc")
self.formLayoutVector.addWidget(self.labelVectorDesc)
#vector description line edit
self.lineEditVectorDesc = QtWidgets.QLineEdit(
self.groupBoxVectorSelection)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.lineEditVectorDesc.sizePolicy().hasHeightForWidth())
self.lineEditVectorDesc.setSizePolicy(sizePolicy)
self.lineEditVectorDesc.setReadOnly(True)
self.lineEditVectorDesc.setObjectName("lineEditVectorDesc")
self.formLayoutVector.addWidget(self.lineEditVectorDesc)
self.comboBoxVector.currentTextChanged.connect(
lambda: self.changeVector())
#*----------------------multi-document-area---------------------------------------------------------------*#
self.mdiArea = QtWidgets.QMdiArea(self.centralwidget)
self.mdiArea.setLayoutDirection(QtCore.Qt.LeftToRight)
self.mdiArea.setFrameShape(QtWidgets.QFrame.StyledPanel)
self.mdiArea.setSizeAdjustPolicy(
QtWidgets.QAbstractScrollArea.AdjustIgnored)
self.mdiArea.setActivationOrder(
QtWidgets.QMdiArea.ActivationHistoryOrder)
self.mdiArea.setViewMode(QtWidgets.QMdiArea.SubWindowView)
self.mdiArea.setDocumentMode(False)
self.mdiArea.setTabsClosable(False)
self.mdiArea.setTabsMovable(True)
self.mdiArea.setTabShape(QtWidgets.QTabWidget.Rounded)
self.mdiArea.setTabPosition(QtWidgets.QTabWidget.North)
self.mdiArea.setObjectName("mdiArea")
#*-------------------------Graph Subwindow------------------------------------------------------------------*#
self.subwindow_Graph = QtWidgets.QWidget()
self.subwindow_Graph.setLayoutDirection(QtCore.Qt.LeftToRight)
self.subwindow_Graph.setObjectName("subwindow_Graph")
self.gridLayout_3 = QtWidgets.QGridLayout(self.subwindow_Graph)
self.gridLayout_3.setObjectName("gridLayout_3")
# Interval Label and combo box
self.IntervalLabel = QtWidgets.QLabel(self.subwindow_Graph)
self.IntervalLabel.setObjectName("IntervalLabel")
self.gridLayout_3.addWidget(self.IntervalLabel, 2, 1, 1, 1,
QtCore.Qt.AlignRight)
self.Interval = QtWidgets.QComboBox(self.subwindow_Graph)
self.Interval.setObjectName("Interval")
self.Interval.addItem("") # Select Interval
self.Interval.addItem("") # Seconds
self.Interval.addItem("") # Minutes
self.Interval.addItem("") # Hours
self.gridLayout_3.addWidget(self.Interval, 2, 2, 1, 1)
#Timeline orientation label and combo box
self.TimelineOrientationLabel = QtWidgets.QLabel(self.subwindow_Graph)
self.TimelineOrientationLabel.setObjectName("TimelineOrientationLabel")
self.gridLayout_3.addWidget(self.TimelineOrientationLabel, 2, 3, 1, 1,
QtCore.Qt.AlignRight)
self.TimelineOrientation = QtWidgets.QComboBox(self.subwindow_Graph)
self.TimelineOrientation.setObjectName("TimelineOrientation")
self.TimelineOrientation.addItem("") # Selection Orientation
self.TimelineOrientation.addItem("") # Horizontal
self.TimelineOrientation.addItem("") # Vertical
self.gridLayout_3.addWidget(self.TimelineOrientation, 2, 4, 1, 1)
self.newRelationshipFlag = False
# Events
def node_selected(node):
if (self.qgv.manipulation_mode ==
QGraphVizManipulationMode.Node_remove_Mode):
print("Node {} removed".format(node))
removeNode(node)
else:
print("Node selected {}".format(node))
if (self.newRelationshipFlag == True):
update_json(node)
self.newRelationshipFlag = False
def edge_selected(edge):
if (self.qgv.manipulation_mode ==
QGraphVizManipulationMode.Edge_remove_Mode):
print("Edge {} removed".format(edge))
removeRelationship(edge)
else:
print("Edge selected {}".format(edge))
def node_invoked(node):
print("Node double clicked")
def edge_invoked(node):
print("Edge double clicked")
def node_removed(node):
print("Node removed")
def edge_removed(node):
print("Edge removed")
# Create QGraphViz graph widget
show_subgraphs = True
self.qgv = QGraphViz(show_subgraphs=show_subgraphs,
node_selected_callback=node_selected,
edge_selected_callback=edge_selected,
node_invoked_callback=node_invoked,
edge_invoked_callback=edge_invoked,
node_removed_callback=node_removed,
edge_removed_callback=edge_removed,
hilight_Nodes=True,
hilight_Edges=True)
self.nodeId = 0
self.relationshipId = 0
f = open('ui/windows/graph.json', 'r+')
f.truncate(0)
self.qgv.setStyleSheet("background-color:white;")
# Create A new Graph using Dot layout engine
self.qgv.new(Dot(Graph("Main_Graph"), show_subgraphs=show_subgraphs))
# Build the graph (the layout engine organizes where the nodes and connections are)
self.qgv.build()
# Save it to a file to be loaded by Graphviz if needed
self.qgv.save("test.gv")
# Create a widget to handle the QGraphViz object
graphWidget = QWidget()
graphWidget.setLayout(QVBoxLayout())
self.gridLayout_3.addWidget(graphWidget, 3, 0, 1, 6)
# Add the QGraphViz object to the layout
graphWidget.layout().addWidget(self.qgv)
#Buttons Functionality
def manipulate():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Nodes_Move_Mode
#export as a CSV file
def save():
fname = QFileDialog.getSaveFileName(self.qgv, "Save", "", "*.csv")
if (fname[0] != ""):
self.qgv.save(fname[0])
print(fname[0])
#fname = QFileDialog.getSaveFileName(self.qgv, "Save", "", "*.gv")
#if(fname[0]!=""):
# self.qgv.save(fname[0])
def new():
self.qgv.engine.graph = Graph("MainGraph")
self.qgv.build()
self.qgv.repaint()
#clear all data from nodes table widget
rowCount = self.tableWidgetNodes.rowCount()
for x in range(1, rowCount):
self.tableWidgetNodes.removeRow(x)
self.tableWidgetNodes.removeRow(1)
#clear all data from relationships table widget
rowCount = self.tableWidgetRelationships.rowCount()
for x in range(rowCount):
self.tableWidgetRelationships.removeRow(x)
#update json file
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
def load():
fname = QFileDialog.getOpenFileName(self.qgv, "Open", "", "*.csv")
if (fname[0] != ""):
self.qgv.load_file(fname[0])
#populate json file with new graph
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
#open json file and load data
with open('ui/windows/graph.json') as json_file:
data = json.load(json_file)
#iterate through each node and add it to the nodes table widget
nodesData = data["nodes"]
for node in nodesData:
self.addNewNode(
self.tableWidgetNodes, node["kwargs"]["id"],
node["kwargs"]["label"], node["kwargs"]["timestamp"],
node["kwargs"]["description"],
node["kwargs"]["leReference"], node["kwargs"]["creator"],
node["kwargs"]["shape"], node["kwargs"]["leSource"])
#iterate through each edge and add it to the relationships table widget
relationshipsData = data["edges"]
for edge in relationshipsData:
try:
label = edge["kwargs"]["label"]
except KeyError:
label = ""
self.addTableData(self.tableWidgetRelationships, label,
edge["source"], edge["dest"])
def add_node():
#add node dialog
addNodeDialog = QDialog()
addNodeDialog.ok = False
addNodeDialog.node_name = ""
addNodeDialog.node_label = ""
addNodeDialog.node_timestamp = ""
addNodeDialog.node_description = ""
addNodeDialog.node_logEntryReference = ""
addNodeDialog.node_logCreator = ""
addNodeDialog.node_logEntrySource = ""
addNodeDialog.node_type = "None"
# Layouts
main_layout = QVBoxLayout()
addNodeLayout = QFormLayout()
buttons_layout = QHBoxLayout()
main_layout.addLayout(addNodeLayout)
main_layout.addLayout(buttons_layout)
addNodeDialog.setLayout(main_layout)
#line edits
leNodeName = QLineEdit()
leNodeLabel = QLineEdit()
teNodeDescription = QTextEdit()
dtNodeTimestamp = QDateTimeEdit()
dtNodeTimestamp.setDate(QtCore.QDate(2020, 1, 1))
dtNodeTimestamp.setCalendarPopup(True)
leLogEntryReference = QLineEdit()
cbxLogCreator = QComboBox()
leLogEntrySource = QLineEdit()
cbxNodeType = QComboBox()
cbxImage = QComboBox()
#buttons
pbOK = QPushButton()
pbCancel = QPushButton()
cbxNodeType.addItems(["None", "circle", "box"])
cbxLogCreator.addItems(["None", "Red", "Blue", "White"])
cbxImage.addItem("None")
self.iconList = Configuration.get_list_of_icon_dicts(
self.configuration)
for icon in self.iconList:
cbxImage.addItem(icon["name"])
pbOK.setText("&OK")
pbCancel.setText("&Cancel")
addNodeLayout.setWidget(0, QFormLayout.LabelRole,
QLabel("Node Name"))
addNodeLayout.setWidget(0, QFormLayout.FieldRole, leNodeName)
addNodeLayout.setWidget(1, QFormLayout.LabelRole,
QLabel("Node Description"))
addNodeLayout.setWidget(1, QFormLayout.FieldRole,
teNodeDescription)
addNodeLayout.setWidget(2, QFormLayout.LabelRole,
QLabel("Node Timestamp"))
addNodeLayout.setWidget(2, QFormLayout.FieldRole, dtNodeTimestamp)
addNodeLayout.setWidget(3, QFormLayout.LabelRole,
QLabel("Log Entry Reference"))
addNodeLayout.setWidget(3, QFormLayout.FieldRole,
leLogEntryReference)
addNodeLayout.setWidget(4, QFormLayout.LabelRole,
QLabel("Log Creator"))
addNodeLayout.setWidget(4, QFormLayout.FieldRole, cbxLogCreator)
addNodeLayout.setWidget(5, QFormLayout.LabelRole,
QLabel("Log Entry Source"))
addNodeLayout.setWidget(5, QFormLayout.FieldRole, leLogEntrySource)
addNodeLayout.setWidget(6, QFormLayout.LabelRole,
QLabel("Node Type"))
addNodeLayout.setWidget(6, QFormLayout.FieldRole, cbxNodeType)
addNodeLayout.setWidget(7, QFormLayout.LabelRole,
QLabel("Node Image"))
addNodeLayout.setWidget(7, QFormLayout.FieldRole, cbxImage)
#ok button handler
def ok():
addNodeDialog.OK = True
addNodeDialog.node_name = leNodeName.text()
addNodeDialog.node_label = leNodeName.text()
addNodeDialog.node_timestamp = dtNodeTimestamp.text()
addNodeDialog.node_description = teNodeDescription.toPlainText(
)
addNodeDialog.node_logEntryReference = leLogEntryReference.text(
)
addNodeDialog.node_logCreator = cbxLogCreator.currentText()
addNodeDialog.node_logEntrySource = leLogEntrySource.text()
if (cbxImage.currentText()):
for icon in self.iconList:
if (cbxImage.currentText() == icon["name"]):
addNodeDialog.node_type = icon["source"]
else:
addNodeDialog.node_type = cbxNodeType.currentText()
addNodeDialog.close()
#cancel button handler
def cancel():
addNodeDialog.OK = False
addNodeDialog.close()
pbOK.clicked.connect(ok)
pbOK.clicked.connect(lambda: self.addNewNode(
self.tableWidgetNodes, str(self.nodeId), leNodeName.text(),
dtNodeTimestamp.text(), teNodeDescription.toPlainText(),
leLogEntryReference.text(), cbxLogCreator.currentText(),
addNodeDialog.node_type, leLogEntrySource.text()))
pbCancel.clicked.connect(cancel)
buttons_layout.addWidget(pbOK)
buttons_layout.addWidget(pbCancel)
addNodeDialog.exec_()
if addNodeDialog.OK and addNodeDialog.node_name != '':
self.qgv.addNode(
self.qgv.engine.graph,
addNodeDialog.node_name,
id=str(self.nodeId),
label=addNodeDialog.node_label,
description=addNodeDialog.node_description,
timestamp=addNodeDialog.node_timestamp,
leReference=addNodeDialog.node_logEntryReference,
creator=addNodeDialog.node_logCreator,
leSource=addNodeDialog.node_logEntrySource,
shape=addNodeDialog.node_type)
self.qgv.build()
self.nodeId += 1
#remove node button event
def rem_node():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Node_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemNode.setChecked(True)
#remove edge button event
def rem_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edge_remove_Mode
for btn in buttons_list:
btn.setChecked(False)
btnRemEdge.setChecked(True)
#add edge button event
def add_edge():
self.qgv.manipulation_mode = QGraphVizManipulationMode.Edges_Connect_Mode
for btn in buttons_list:
btn.setChecked(False)
btnAddEdge.setChecked(True)
self.newRelationshipFlag = True
#updates json file with all graph properties (nodes & edges)
def update_json(item):
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
addRelationshipToTable()
#remove node from node table widget
def removeNode(node):
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
self.row = getNodeRowNumber(self.tableWidgetNodes, node.name)
self.tableWidgetNodes.removeRow(self.row)
#remove relationship from relationship table widget
def removeRelationship(edge):
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
self.row = getEdgeRowNumber(self.tableWidgetRelationships,
edge.source.name, edge.dest.name)
self.tableWidgetRelationships.removeRow(self.row)
#return row number for given node
def getNodeRowNumber(widget, nodeName):
rowCount = widget.rowCount()
for x in range(rowCount):
node = widget.item(x, 2).text()
if node == nodeName:
return x
#return row number for given edge
def getEdgeRowNumber(widget, edgeSource, edgeDestination):
rowCount = widget.rowCount()
for x in range(rowCount):
edgeSrc = widget.item(x, 2).text()
edgeDest = widget.item(x, 3).text()
if edgeSrc == edgeSource and edgeDest == edgeDestination:
return x
#add new relationship from graph onto relationship table
def addRelationshipToTable():
with open('ui/windows/graph.json') as json_file:
data = json.load(json_file)
self.index = len(data['edges']) - 1
#new table widget item
self.tableWidgetRelationships.insertRow(0)
item = QtWidgets.QTableWidgetItem()
item.setCheckState(QtCore.Qt.Unchecked)
item.setFlags(QtCore.Qt.ItemIsDragEnabled
| QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
self.tableWidgetRelationships.setItem(0, 0, item)
#write data on table widget in corresponding cells
insert = QtCore.QCoreApplication.translate
item = self.tableWidgetRelationships.item(0, 0)
item.setText(insert("GraphWindow", str(self.relationshipId)))
self.tableWidgetRelationships.setItem(
0, 2,
QtWidgets.QTableWidgetItem(
data["edges"][self.index]["source"]))
self.tableWidgetRelationships.setItem(
0, 3,
QtWidgets.QTableWidgetItem(data["edges"][self.index]["dest"]))
self.relationshipId += 1
# Add two horizontal layouts (pannels to hold buttons)
hpanelTop = QHBoxLayout()
graphWidget.layout().addLayout(hpanelTop)
hpanelBottom = QHBoxLayout()
graphWidget.layout().addLayout(hpanelBottom)
# Add buttons
btnNew = QPushButton("New")
btnNew.clicked.connect(new)
hpanelTop.addWidget(btnNew)
btnOpen = QPushButton("Open")
btnOpen.clicked.connect(load)
hpanelTop.addWidget(btnOpen)
btnSave = QPushButton("Export")
btnSave.clicked.connect(save)
hpanelTop.addWidget(btnSave)
buttons_list = []
btnManip = QPushButton("Manipulate")
btnManip.setCheckable(True)
btnManip.setChecked(True)
btnManip.clicked.connect(manipulate)
hpanelTop.addWidget(btnManip)
buttons_list.append(btnManip)
btnAddNode = QPushButton("Add Node")
btnAddNode.clicked.connect(add_node)
hpanelBottom.addWidget(btnAddNode)
buttons_list.append(btnManip)
btnRemNode = QPushButton("Rem Node")
btnRemNode.setCheckable(True)
btnRemNode.clicked.connect(rem_node)
hpanelBottom.addWidget(btnRemNode)
buttons_list.append(btnRemNode)
btnAddEdge = QPushButton("Add Relationship")
btnAddEdge.setCheckable(True)
btnAddEdge.clicked.connect(add_edge)
hpanelBottom.addWidget(btnAddEdge)
buttons_list.append(btnAddEdge)
btnRemEdge = QPushButton("Rem Relationship")
btnRemEdge.setCheckable(True)
btnRemEdge.clicked.connect(rem_edge)
hpanelBottom.addWidget(btnRemEdge)
buttons_list.append(btnRemEdge)
#*-------------------------Tabular View Subwindow------------------------------------------------------------------*#
self.subwindow_Table = QtWidgets.QWidget()
self.subwindow_Table.setLayoutDirection(QtCore.Qt.RightToLeft)
self.subwindow_Table.setObjectName("subwindow_Table")
self.verticalLayout_3 = QtWidgets.QVBoxLayout(self.subwindow_Table)
self.verticalLayout_3.setObjectName("verticalLayout_3")
#table widget
self.tableWidgetNodes = QtWidgets.QTableWidget(self.subwindow_Table)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Preferred)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.tableWidgetNodes.sizePolicy().hasHeightForWidth())
self.tableWidgetNodes.setSizePolicy(sizePolicy)
self.tableWidgetNodes.setLayoutDirection(QtCore.Qt.LeftToRight)
self.tableWidgetNodes.setFrameShape(QtWidgets.QFrame.Box)
self.tableWidgetNodes.setFrameShadow(QtWidgets.QFrame.Sunken)
self.tableWidgetNodes.setAlternatingRowColors(True)
self.tableWidgetNodes.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
self.tableWidgetNodes.setSelectionBehavior(
QtWidgets.QAbstractItemView.SelectItems)
self.tableWidgetNodes.setTextElideMode(QtCore.Qt.ElideRight)
self.tableWidgetNodes.setVerticalScrollMode(
QtWidgets.QAbstractItemView.ScrollPerPixel)
self.tableWidgetNodes.setHorizontalScrollMode(
QtWidgets.QAbstractItemView.ScrollPerPixel)
self.tableWidgetNodes.setShowGrid(False)
self.tableWidgetNodes.setGridStyle(QtCore.Qt.SolidLine)
self.tableWidgetNodes.setCornerButtonEnabled(False)
#table properties
self.tableWidgetNodes.horizontalHeader().setCascadingSectionResizes(
False)
self.tableWidgetNodes.horizontalHeader().setDefaultSectionSize(142)
self.tableWidgetNodes.horizontalHeader().setHighlightSections(True)
self.tableWidgetNodes.horizontalHeader().setSortIndicatorShown(True)
self.tableWidgetNodes.horizontalHeader().setStretchLastSection(True)
self.tableWidgetNodes.verticalHeader().setVisible(True)
self.tableWidgetNodes.verticalHeader().setMinimumSectionSize(30)
self.tableWidgetNodes.verticalHeader().setSortIndicatorShown(False)
self.tableWidgetNodes.verticalHeader().setStretchLastSection(False)
self.verticalLayout_3.addWidget(self.tableWidgetNodes)
#*-------------------------Relationship Table Subwindow------------------------------------------------------------------*#
#subwindow and layout properties
self.subwindow_Relationship = QtWidgets.QWidget()
self.subwindow_Relationship.setLayoutDirection(QtCore.Qt.RightToLeft)
self.subwindow_Relationship.setObjectName("subwindow_Relationship")
self.verticalLayout = QtWidgets.QVBoxLayout(
self.subwindow_Relationship)
self.verticalLayout.setContentsMargins(-1, -1, 17, -1)
self.verticalLayout.setObjectName("verticalLayout")
#add relationship frame
self.frameRelationship = QtWidgets.QFrame(self.subwindow_Relationship)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.frameRelationship.sizePolicy().hasHeightForWidth())
self.frameRelationship.setSizePolicy(sizePolicy)
self.frameRelationship.setLayoutDirection(QtCore.Qt.RightToLeft)
self.frameRelationship.setFrameShape(QtWidgets.QFrame.StyledPanel)
self.frameRelationship.setFrameShadow(QtWidgets.QFrame.Raised)
self.frameRelationship.setObjectName("frameRelationship")
self.verticalLayout.addWidget(self.frameRelationship)
self.gridLayout_2 = QtWidgets.QGridLayout(self.frameRelationship)
self.gridLayout_2.setObjectName("gridLayout_2")
#relationship label & line edit
self.labelRelationship = QtWidgets.QLabel(self.frameRelationship)
self.labelRelationship.setObjectName("labelRelationship")
self.gridLayout_2.addWidget(self.labelRelationship, 0, 2, 1, 1)
self.lineEditRelationship = QtWidgets.QLineEdit(self.frameRelationship)
self.lineEditRelationship.setObjectName("lineEditRelationship")
self.gridLayout_2.addWidget(self.lineEditRelationship, 0, 1, 1, 1)
#parent node label & line edit
self.labelParent = QtWidgets.QLabel(self.frameRelationship)
self.labelParent.setObjectName("labelParent")
self.gridLayout_2.addWidget(self.labelParent, 1, 2, 1, 1)
self.lineEditParent = QtWidgets.QLineEdit(self.frameRelationship)
self.lineEditParent.setObjectName("lineEditParent")
self.gridLayout_2.addWidget(self.lineEditParent, 1, 1, 1, 1)
#child node label & line edit
self.labelChild = QtWidgets.QLabel(self.frameRelationship)
self.labelChild.setObjectName("labelChild")
self.gridLayout_2.addWidget(self.labelChild, 2, 2, 1, 1)
self.lineEditChild = QtWidgets.QLineEdit(self.frameRelationship)
self.lineEditChild.setObjectName("lineEditChild")
self.gridLayout_2.addWidget(self.lineEditChild, 2, 1, 1, 1)
#add relationship button
self.pushButtonAddRelationship = QtWidgets.QPushButton(
self.frameRelationship)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.pushButtonAddRelationship.sizePolicy().hasHeightForWidth())
self.pushButtonAddRelationship.setSizePolicy(sizePolicy)
self.pushButtonAddRelationship.setObjectName(
"pushButtonAddRelationship")
self.gridLayout_2.addWidget(self.pushButtonAddRelationship, 3, 1, 1, 1)
#relationships table
self.tableWidgetRelationships = QtWidgets.QTableWidget(
self.subwindow_Relationship)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Preferred)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.tableWidgetRelationships.sizePolicy().hasHeightForWidth())
self.tableWidgetRelationships.setSizePolicy(sizePolicy)
self.tableWidgetRelationships.setLayoutDirection(QtCore.Qt.LeftToRight)
self.tableWidgetRelationships.setAlternatingRowColors(True)
self.tableWidgetRelationships.setVerticalScrollMode(
QtWidgets.QAbstractItemView.ScrollPerPixel)
self.tableWidgetRelationships.setHorizontalScrollMode(
QtWidgets.QAbstractItemView.ScrollPerPixel)
self.tableWidgetRelationships.setShowGrid(False)
self.tableWidgetRelationships.setObjectName("tableWidgetRelationships")
self.tableWidgetRelationships.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
# a relationship is the length of a single relationship
relationship = 4
# relationships is the length of a set of relationships
relationships = 0
# setting the relationship table dimensions
self.tableWidgetRelationships.setColumnCount(relationship)
self.tableWidgetRelationships.setRowCount(relationships)
# creating the row header for relationships
for relation in range(relationship):
item = QtWidgets.QTableWidgetItem()
item.setTextAlignment(QtCore.Qt.AlignLeading
| QtCore.Qt.AlignVCenter)
self.tableWidgetRelationships.setHorizontalHeaderItem(
relation, item)
for relation in range(relationships):
item = QtWidgets.QTableWidgetItem()
self.tableWidgetRelationships.setVerticalHeaderItem(relation, item)
# creates the table
for relation in range(relationships):
# creates a checkbox in the column
item = QtWidgets.QTableWidgetItem()
item.setCheckState(QtCore.Qt.Unchecked)
self.tableWidgetRelationships.setItem(relation, 0, item)
for label in range(relationship):
item = QtWidgets.QTableWidgetItem()
self.tableWidgetRelationships.setItem(relation, label + 1,
item)
self.tableWidgetRelationships.horizontalHeader().setVisible(True)
self.tableWidgetRelationships.horizontalHeader().setMinimumSectionSize(
39)
self.tableWidgetRelationships.horizontalHeader().setSortIndicatorShown(
True)
self.tableWidgetRelationships.horizontalHeader().setStretchLastSection(
True)
self.tableWidgetRelationships.verticalHeader().setVisible(True)
self.tableWidgetRelationships.verticalHeader().setDefaultSectionSize(
30)
self.tableWidgetRelationships.verticalHeader().setSortIndicatorShown(
False)
self.verticalLayout.addWidget(self.tableWidgetRelationships)
self.mdiArea.addSubWindow(self.subwindow_Relationship)
self.mdiArea.addSubWindow(self.subwindow_Table)
self.gridLayout.addWidget(self.mdiArea, 1, 0, 1, 1)
self.graphSubWin = self.mdiArea.addSubWindow(self.subwindow_Graph)
#*------------------------------Menu Bar---------------------------------------------------*#
graphWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(graphWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 1155, 21))
self.menubar.setObjectName("menubar")
self.menuFile = QtWidgets.QMenu(self.menubar)
self.menuFile.setObjectName("menuFile")
self.menuOpen_Window = QtWidgets.QMenu(self.menuFile)
self.menuOpen_Window.setObjectName("menuOpen_Window")
self.menuSubwindow_Layout_2 = QtWidgets.QMenu(self.menuFile)
self.menuSubwindow_Layout_2.setObjectName("menuSubwindow_Layout_2")
self.menuMenu = QtWidgets.QMenu(self.menubar)
self.menuMenu.setObjectName("menuMenu")
graphWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(graphWindow)
self.statusbar.setObjectName("statusbar")
graphWindow.setStatusBar(self.statusbar)
self.actionEvent_Configuration = QtWidgets.QAction(graphWindow)
self.actionEvent_Configuration.setObjectName(
"actionEvent_Configuration")
self.actionEnforcement_Action_Report = QtWidgets.QAction(graphWindow)
self.actionEnforcement_Action_Report.setObjectName(
"actionEnforcement_Action_Report")
self.actionManage_Graph = QtWidgets.QAction(graphWindow)
self.actionManage_Graph.setObjectName("actionManage_Graph")
self.actionFilter_Logs = QtWidgets.QAction(graphWindow)
self.actionFilter_Logs.setObjectName("actionFilter_Logs")
self.actionGraphical_View = QtWidgets.QAction(graphWindow)
self.actionGraphical_View.setObjectName("actionGraphical_View")
self.actionTabular_View = QtWidgets.QAction(graphWindow)
self.actionTabular_View.setObjectName("actionTabular_View")
self.actionRelationship_Table = QtWidgets.QAction(graphWindow)
self.actionRelationship_Table.setObjectName("actionRelationship_Table")
self.actionTile = QtWidgets.QAction(graphWindow)
self.actionTile.setObjectName("actionTile")
self.actionCascade = QtWidgets.QAction(graphWindow)
self.actionCascade.setObjectName("actionCascade")
self.actionCascade.triggered.connect(self.mdiArea.cascadeSubWindows)
self.menuOpen_Window.addAction(self.actionGraphical_View)
self.menuOpen_Window.addAction(self.actionTabular_View)
self.menuOpen_Window.addAction(self.actionRelationship_Table)
self.menuSubwindow_Layout_2.addAction(self.actionTile)
self.menuSubwindow_Layout_2.addAction(self.actionCascade)
self.menuFile.addAction(self.menuOpen_Window.menuAction())
self.menuFile.addAction(self.menuSubwindow_Layout_2.menuAction())
self.menuMenu.addAction(self.actionEvent_Configuration)
self.menuMenu.addAction(self.actionEnforcement_Action_Report)
self.menuMenu.addAction(self.actionManage_Graph)
self.menuMenu.addAction(self.actionFilter_Logs)
self.menubar.addAction(self.menuMenu.menuAction())
self.menubar.addAction(self.menuFile.menuAction())
self.mdiArea.tileSubWindows()
self.retranslateUi(graphWindow)
self.actionTile.triggered.connect(self.mdiArea.tileSubWindows)
QtCore.QMetaObject.connectSlotsByName(graphWindow)
def retranslateUi(self, graphWindow):
insert = QtCore.QCoreApplication.translate
graphWindow.setWindowTitle(insert("graphWindow", "MainWindow"))
#*-------------------------Menu Options------------------------------------------------------------------*#
self.menuFile.setTitle(insert("graphWindow", "Window"))
self.menuOpen_Window.setTitle(insert("graphWindow", "Open Window"))
self.menuSubwindow_Layout_2.setTitle(
insert("graphWindow", "Subwindow Layout"))
self.menuMenu.setTitle(insert("graphWindow", "Menu"))
self.actionEvent_Configuration.setText(
insert("graphWindow", "Event Configuration"))
self.actionEnforcement_Action_Report.setText(
insert("graphWindow", "Enforcement Action Report"))
self.actionManage_Graph.setText(insert("graphWindow", "Manage Graph"))
self.actionTile.setText(insert("graphWindow", "Tile"))
self.actionCascade.setText(insert("graphWindow", "Cascade"))
#*-------------------------Vector selection Options------------------------------------------------------------------*#
self.groupBoxVectorSelection.setTitle(
insert("graphWindow", "Vector Selection"))
self.labelVector.setText(insert("graphWindow", "Vector:"))
self.labelVectorDesc.setText(
insert("graphWindow", "Vector Description"))
#*-------------------------Graph Subwindow------------------------------------------------------------------*#
self.actionGraphical_View.setText(
insert("graphWindow", "Graphical View"))
#Interval data
self.IntervalLabel.setText(insert("MainWindow", "Interval:"))
self.Interval.setItemText(0, insert("MainWindow",
"Select an Interval"))
self.Interval.setItemText(1, insert("MainWindow", "Seconds"))
self.Interval.setItemText(2, insert("MainWindow", "Minutes"))
self.Interval.setItemText(3, insert("MainWindow", "Hours"))
#Timeline orientation data
self.TimelineOrientationLabel.setText(
insert("MainWindow", "Timeline Orientation:"))
self.TimelineOrientation.setItemText(
0, insert("MainWindow", "Select Orienatation"))
self.TimelineOrientation.setItemText(
1, insert("MainWindow", "Horizontal"))
self.TimelineOrientation.setItemText(2, insert("MainWindow",
"Vertical"))
#*-------------------------Tabular View Subwindow------------------------------------------------------------------*#
self.actionTabular_View.setText(insert("graphWindow", "Tabular View"))
self.subwindow_Table.setWindowTitle(
insert("graphWindow", "Tabular View"))
__sortingEnabled = self.tableWidgetNodes.isSortingEnabled()
self.tableWidgetNodes.setSortingEnabled(False)
self.tableWidgetNodes.setSortingEnabled(__sortingEnabled)
self.subwindow_Graph.setWindowTitle(
insert("graphWindow", "Graphical View"))
#*-------------------------Relationship Table Subwindow------------------------------------------------------------------*#
self.actionRelationship_Table.setText(
insert("graphWindow", "Relationship Table"))
#add relationship labels
self.subwindow_Relationship.setWindowTitle(
insert("graphWindow", "Relationship Table"))
self.pushButtonAddRelationship.setText(
insert("graphWindow", "Add Relationship"))
self.pushButtonAddRelationship.clicked.connect(
lambda: self.addTableData(
self.tableWidgetRelationships, self.lineEditRelationship.text(
), self.lineEditParent.text(), self.lineEditChild.text()))
self.pushButtonAddRelationship.clicked.connect(
lambda: self.addNewRelationshipToGraph(
self.qgv, self.lineEditRelationship.text(),
self.lineEditParent.text(), self.lineEditChild.text()))
self.pushButtonAddRelationship.clicked.connect(lambda: self.clearData(
self.tableWidgetRelationships, self.lineEditRelationship, self.
lineEditParent, self.lineEditChild))
#self.pushButtonDeleteRelationship.setText(insert("graphWindow", "Delete Relationship"))
#self.pushButtonDeleteRelationship.clicked.connect(lambda: self.deleteRelationship(self.tableWidgetRelationships))
self.labelRelationship.setText(
insert("graphWindow", "Relationship Label:"))
self.labelParent.setText(insert("graphWindow", "Parent Node Name:"))
self.labelChild.setText(insert("graphWindow", "Child Node Name:"))
self.tableWidgetRelationships.setSortingEnabled(True)
item = self.tableWidgetRelationships.horizontalHeaderItem(0)
item.setText(insert("graphWindow", "Relationship ID"))
item = self.tableWidgetRelationships.horizontalHeaderItem(1)
item.setText(insert("graphWindow", "Label"))
item = self.tableWidgetRelationships.horizontalHeaderItem(2)
item.setText(insert("graphWindow", "Parent"))
item = self.tableWidgetRelationships.horizontalHeaderItem(3)
item.setText(insert("graphWindow", "Child"))
__sortingEnabled = self.tableWidgetRelationships.isSortingEnabled()
self.tableWidgetRelationships.setSortingEnabled(False)
self.tableWidgetRelationships.setSortingEnabled(__sortingEnabled)
#adds data to given tree widget with two columns
def addTableData(self, table, label, parent, child):
table.insertRow(0)
item = QtWidgets.QTableWidgetItem()
item.setCheckState(QtCore.Qt.Unchecked)
item.setFlags(QtCore.Qt.ItemIsDragEnabled
| QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
table.setItem(0, 0, item)
insert = QtCore.QCoreApplication.translate
item = self.tableWidgetRelationships.item(0, 0)
item.setText(insert("GraphWindow", str(self.relationshipId)))
table.setItem(0, 1, QtWidgets.QTableWidgetItem(label))
table.setItem(0, 2, QtWidgets.QTableWidgetItem(parent))
table.setItem(0, 3, QtWidgets.QTableWidgetItem(child))
self.relationshipId += 1
#clears data from line/text edits after adding to tree widget
def clearData(self, table, col1, col2, col3):
col1.clear()
col2.clear()
col3.clear()
#removes selected items from given tree widget
def deleteRelationship(self, table):
selectedItems = table.selectionModel().selectedRows()
for item in sorted(selectedItems):
table.removeRow(item.row())
#add new node to tabular view
def addNewNode(self, table, nodeId, name, timestamp, description,
reference, creator, icon, source):
table.insertRow(1)
item = QtWidgets.QTableWidgetItem()
item.setFlags(QtCore.Qt.ItemIsDragEnabled
| QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
item.setCheckState(QtCore.Qt.Checked)
table.setItem(1, 0, item)
table.setItem(1, 1, QtWidgets.QTableWidgetItem(nodeId))
table.setItem(1, 2, QtWidgets.QTableWidgetItem(name))
table.setItem(1, 3, QtWidgets.QTableWidgetItem(timestamp))
table.setItem(1, 4, QtWidgets.QTableWidgetItem(description))
table.setItem(1, 5, QtWidgets.QTableWidgetItem(reference))
table.setItem(1, 6, QtWidgets.QTableWidgetItem(creator))
table.setItem(1, 7, QtWidgets.QTableWidgetItem(icon))
table.setItem(1, 8, QtWidgets.QTableWidgetItem(source))
#add new relationship from relationship table to graph
def addNewRelationshipToGraph(self, graph, relationshipLabel, parent,
child):
#retrive node objects given node names
parentNode = Graph.findNode(self.qgv.engine.graph, parent)
childNode = Graph.findNode(self.qgv.engine.graph, child)
#add edge to graph
self.qgv.addEdge(parentNode, childNode, {"label": relationshipLabel})
self.qgv.build()
#update json file to keep track of all edges
fname = "ui/windows/graph.json"
self.qgv.saveAsJson(fname)
self.mdiArea.setActiveSubWindow(self.graphSubWin)
def changeVector(self):
for vector in self.vectorList:
if (vector["name"] == self.comboBoxVector.currentText()):
self.lineEditVectorDesc.setText(vector["description"])
self.table_items(self.tableWidgetNodes,
self.comboBoxVector.currentText())
def insert_log_entry_table_view(self, vector_name):
self.tableWidgetNodes.clearContents()
selectedVector = None
for vector in self.configuration.vectors:
if vector.name == vector_name:
selectedVector = vector
if selectedVector is None:
return
self.nodes = len(selectedVector.log_entries) + 1
self.node_properties = 9
self.tableWidgetNodes.setObjectName("tableWidgetNodes")
self.tableWidgetNodes.setColumnCount(self.node_properties)
self.tableWidgetNodes.setRowCount(self.nodes)
# creates node table row header
for node in range(self.nodes):
item = QtWidgets.QTableWidgetItem()
self.tableWidgetNodes.setVerticalHeaderItem(node, item)
# creates node table column header
for property in range(self.node_properties):
item = QtWidgets.QTableWidgetItem()
self.tableWidgetNodes.setHorizontalHeaderItem(property, item)
# creates the first column in the first row without a check box (0,0)
item = QtWidgets.QTableWidgetItem()
self.tableWidgetNodes.setItem(0, 0, item)
# creates the rest of the columns in the first row with checkboxes (0,1) to (0,n)
for property in range(8):
item = QtWidgets.QTableWidgetItem()
item.setFlags(QtCore.Qt.ItemIsDragEnabled
| QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
item.setCheckState(QtCore.Qt.Checked)
self.tableWidgetNodes.setItem(0, property + 1, item)
# # creates the rest of the table starting at (1,0)
for node in range(self.nodes):
# creates the checkbox for every row in the first column
item = QtWidgets.QTableWidgetItem()
item.setFlags(QtCore.Qt.ItemIsDragEnabled
| QtCore.Qt.ItemIsUserCheckable
| QtCore.Qt.ItemIsEnabled)
item.setCheckState(QtCore.Qt.Checked)
self.tableWidgetNodes.setItem(node + 1, 0, item)
for property in range(8):
item = QtWidgets.QTableWidgetItem()
self.tableWidgetNodes.setItem(node + 1, property + 1, item)
# add actual data
for i in range(len(selectedVector.log_entries)):
# self.tableWidgetNodes.insertRow(i+1)
item = QtWidgets.QTableWidgetItem()
item.setText(selectedVector.log_entries[i].time)
self.tableWidgetNodes.setItem(i + 1, 3, item)
item = QtWidgets.QTableWidgetItem()
item.setText(selectedVector.log_entries[i].data)
self.tableWidgetNodes.setItem(i + 1, 4, item)
item = QtWidgets.QTableWidgetItem()
item.setText(selectedVector.log_entries[i].source_file)
self.tableWidgetNodes.setItem(i + 1, 5, item)
insert = QtCore.QCoreApplication.translate
item = self.tableWidgetNodes.verticalHeaderItem(0)
item.setText(insert("graphWindow", "Property Visibility"))
item = self.tableWidgetNodes.horizontalHeaderItem(0)
item.setText(insert("graphWindow", "Node Visibility"))
item = self.tableWidgetNodes.horizontalHeaderItem(1)
item.setText(insert("graphWindow", "Node Id"))
item = self.tableWidgetNodes.horizontalHeaderItem(2)
item.setText(insert("graphWindow", "Node Name"))
item = self.tableWidgetNodes.horizontalHeaderItem(3)
item.setText(insert("graphWindow", "Node Timestamp"))
item = self.tableWidgetNodes.horizontalHeaderItem(4)
item.setText(insert("graphWindow", "Node Description"))
item = self.tableWidgetNodes.horizontalHeaderItem(5)
item.setText(insert("graphWindow", "Log Entry Reference"))
item = self.tableWidgetNodes.horizontalHeaderItem(6)
item.setText(insert("graphWindow", "Log Creator"))
item = self.tableWidgetNodes.horizontalHeaderItem(7)
item.setText(insert("graphWindow", "Icon Type"))
item = self.tableWidgetNodes.horizontalHeaderItem(8)
item.setText(insert("graphWindow", "Source"))
def table_items(self, tableWidgetNodes, vector_name):
selected_vector = None
for vector in self.configuration.vectors:
if vector.name == vector_name:
selected_vector = vector
if selected_vector is None:
return
self.insert_log_entry_table_view(vector_name)