Remove unwanted whitespace

Author: mrkickling

mal_gui/AssociationTableView.py

@@ -10,7 +10,6 @@ def __init__(self, parent=None):
         self.mainWindow = parent
 
         self.tableAssociationView = QTableView(self)
-
         self.associationInfoModel = QStandardItemModel()
 
         #headers for the columns
@@ -40,5 +39,3 @@ def __init__(self, parent=None):
 
     def getAssociationInfo(self):
         return self.associationInfo
-
-

mal_gui/ConnectionDialog.py

@@ -13,7 +13,7 @@
 class ConnectionDialog(QDialog):
     def filterItems(self, text):
         pass
-    
+
     def OkButtonClicked(self):
         pass
 
@@ -25,7 +25,7 @@ def __init__(self, startItem, endItem,langGraph, lcs,model,parent=None):
         self.langGraph = langGraph
         self.lcs = lcs
         self.model = model
-        
+
         self.setWindowTitle("Select Association Type")
         self.setMinimumWidth(300)
 
@@ -121,7 +121,7 @@ def __init__(self, startItem, endItem,langGraph, lcs,model,parent=None):
 
         # Select the first item by default
         self.associationListWidget.setCurrentRow(0)
-    
+
     def filterItems(self, text):
         for i in range(self.associationListWidget.count()):
             item = self.associationListWidget.item(i)
@@ -132,7 +132,7 @@ def OkButtonClicked(self):
         if selectedItem:
             selectedAssociationText = selectedItem.text()
             # QMessageBox.information(self, "Selected Item", f"You selected: {selectedAssociationText}")
-            
+
             (assoc, leftAsset, rightAsset) = self._str_to_assoc[selectedAssociationText]
             # TODO: Create association based on its full name instead in order
             # to avoid conflicts when multiple associations with the same name
@@ -144,7 +144,7 @@ def OkButtonClicked(self):
             selectedItem.association = association
             # self.model.add_association(association)
         self.accept()
-        
+
 class EntrypointConnectionDialog(ConnectionDialog):
     def __init__(self, attackerItem, assetItem, langGraph, lcs, model, parent=None):
         super().__init__(parent)
@@ -193,7 +193,6 @@ def __init__(self, attackerItem, assetItem, langGraph, lcs, model, parent=None):
             self.layout.addWidget(self.filterEdit)
             self.layout.addWidget(self.attackStepListWidget)
 
-
         buttonLayout = QHBoxLayout()
         self.okButton = QPushButton("OK")
         self.okButton.clicked.connect(self.OkButtonClicked)

mal_gui/ConnectionItem.py

@@ -13,14 +13,14 @@ def createLabel(self, text):
 
     def updatePath(self):
         pass
-    
+
     def removeLabels(self):
         pass
-    
+
     def restoreLabels(self):
         pass
-    
-    
+
+
 class AssociationConnectionItem(IConnectionItem):
     def __init__(
         self,
@@ -31,23 +31,23 @@ def __init__(
         parent = None
     ):
         super().__init__(parent)
-        
+
         pen = QPen(QColor(0, 255, 0), 2)  # Green color with 2-pixel thickness
         self.setPen(pen)
-        
+
         self.setZValue(0)  # Ensure connection items are behind rect items
-        
+
         self.showAssociationFlag = False
-        
+
         self.startItem = startItem
         self.endItem = endItem
         self.scene = scene
-        
+
         self.startItem.addConnection(self)
         self.endItem.addConnection(self)
-        
+
         if self.startItem.assetType != 'Attacker' and self.endItem.assetType != 'Attacker':
-        
+
             self.associationDetails = selectedAssociationText.split("-->")
             assocLeftField = self.associationDetails[0]
             assocMiddleName = self.associationDetails[1]
@@ -58,9 +58,9 @@ def __init__(
             self.labelAssocLeftField = self.createLabel(assocLeftField.split(".")[1])
             self.labelAssocMiddleName = self.createLabel(assocMiddleName)
             self.labelAssocRightField = self.createLabel(assocRightField.split(".")[1])
-        
+
         else:
-            
+
             #Need to check who is attacker and get the name of target and attackStep Name
             # Assumption is Both are not 'Attacker'
             if self.startItem.assetType == 'Attacker':
@@ -69,15 +69,15 @@ def __init__(
             else:
                 attacker = self.endItem.attackerAttachment
                 target = str(self.startItem.assetName)
-            
+
             #selectedAssociationText is representing 'AttackStep' name
             attacker.entry_points.append(target + ' -> ' + selectedAssociationText)
             self.labelAssocLeftField = self.createLabel("")
             self.labelAssocMiddleName = self.createLabel(selectedAssociationText)
             self.labelAssocRightField = self.createLabel("")
-        
-        
-        
+
+
+
         self.updatePath()
 
     def createLabel(self, text):
@@ -104,7 +104,7 @@ def updatePath(self):
         self.startPos = self.startItem.sceneBoundingRect().center()
         self.endPos = self.endItem.sceneBoundingRect().center()
         self.setLine(QLineF(self.startPos, self.endPos))
-        
+
         labelAssocLeftFieldPos = self.line().pointAt(0.2)
         self.labelAssocLeftField.setPos(labelAssocLeftFieldPos - QPointF(self.labelAssocLeftField.boundingRect().width() / 2, self.labelAssocLeftField.boundingRect().height() / 2))
 
@@ -113,9 +113,9 @@ def updatePath(self):
 
         labelAssocRightFieldPos = self.line().pointAt(0.8)
         self.labelAssocRightField.setPos(labelAssocRightFieldPos - QPointF(self.labelAssocRightField.boundingRect().width() / 2, self.labelAssocRightField.boundingRect().height() / 2))
-        
+
         # print("isAssociationVisibilityChecked = "+ str(self.isAssociationVisibilityChecked))
-        
+
         self.labelAssocLeftField.setVisible(self.scene.getShowAssociationCheckBoxStatus())
         self.labelAssocRightField.setVisible(self.scene.getShowAssociationCheckBoxStatus())
 
@@ -125,24 +125,24 @@ def calculateOffset(self, rect, label_pos, angle):
         """
         offset_distance = 10  # Distance to move the label outside the rectangle
         offset = QPointF()
-        
+
         if angle < 90 or angle > 270:
             offset.setX(rect.width() / 2 + offset_distance)
         else:
             offset.setX(-(rect.width() / 2 + offset_distance))
-        
+
         if angle < 180:
             offset.setY(rect.height() / 2 + offset_distance)
         else:
             offset.setY(-(rect.height() / 2 + offset_distance))
-        
+
         return offset
-        
+
     def removeLabels(self):
         self.scene.removeItem(self.labelAssocLeftField)
         self.scene.removeItem(self.labelAssocMiddleName)
         self.scene.removeItem(self.labelAssocRightField)
-    
+
     def restoreLabels(self):
         self.scene.addItem(self.labelAssocLeftField)
         self.scene.addItem(self.labelAssocMiddleName)
@@ -151,7 +151,6 @@ def restoreLabels(self):
     def delete(self):
         self.removeLabels()
         self.scene.removeItem(self)
-        
 
 
 class EntrypointConnectionItem(IConnectionItem):
@@ -193,8 +192,8 @@ def createLabel(self, text):
         labelGroup = self.scene.createItemGroup([labelBackground, label])
         labelGroup.setZValue(1)  # Ensure labels are above the line
 
-        return labelGroup 
-    
+        return labelGroup
+
     def updatePath(self):
         """
         Draws a straight line from the start to end items and updates label positions.