Move add_ports logic to drag_button

b_asic/GUI/drag_button.py

@@ -56,6 +56,8 @@ class DragButton(QPushButton):
         self._mouse_press_pos = None
         self._mouse_move_pos = None
         self._flipped = False
+        self._properties_window = None
+        self.label = None
         super().__init__(parent)
 
     def contextMenuEvent(self, event):
@@ -193,9 +195,7 @@ class DragButton(QPushButton):
         for signal, ports in self._window.signalPortDict.items():
             if any(
                 map(
-                    lambda port: set(port).intersection(
-                        set(self._window.portDict[self])
-                    ),
+                    lambda port: set(port).intersection(set(self.ports)),
                     ports,
                 )
             ):
@@ -234,15 +234,15 @@ class DragButton(QPushButton):
             del self._window.operationDragDict[self.operation]
 
     def add_ports(self):
-        def _determine_port_distance(height, ports):
+        def _determine_port_distance(opheight, ports):
             """
             Determine the distance between each port on the side of an operation.
             The method returns the distance that each port should have from 0.
             """
             return (
-                [(height - PORTHEIGHT) // 2]
+                [(opheight - PORTHEIGHT) // 2]
                 if ports == 1
-                else [(PORTHEIGHT + GAP) * i for i in range(ports)]
+                else [(PORTHEIGHT + GAP) * no for no in range(ports)]
             )
 
         op = self.operation