Add typing, renaming, convenience methods/properties

b_asic/GUI/drag_button.py

@@ -51,7 +51,7 @@ class DragButton(QPushButton):
         parent=None,
     ):
         self.name = operation.graph_id
-        self.ports: List[PortButton] = []
+        self._ports: List[PortButton] = []
         self.show_name = show_name
         self._window = window
         self.operation = operation
@@ -99,6 +99,10 @@ class DragButton(QPushButton):
 
         super().mousePressEvent(event)
 
+    @property
+    def port_list(self) -> List[PortButton]:
+        return self._ports
+
     def mouseMoveEvent(self, event):
         if event.buttons() == Qt.MouseButton.LeftButton and self._m_press:
             self._m_drag = True
@@ -138,7 +142,7 @@ class DragButton(QPushButton):
 
     def _flip(self, event=None):
         self._flipped = not self._flipped
-        for pb in self.ports:
+        for pb in self._ports:
             if isinstance(pb.port, InputPort):
                 newx = MINBUTTONSIZE - PORTWIDTH if self._flipped else 0
             else:
@@ -196,7 +200,7 @@ class DragButton(QPushButton):
         for signal, ports in self._window._arrow_ports.items():
             if any(
                 map(
-                    lambda port: set(port).intersection(set(self.ports)),
+                    lambda port: set(port).intersection(set(self._ports)),
                     ports,
                 )
             ):
@@ -256,11 +260,11 @@ class DragButton(QPushButton):
             port.setFixedSize(PORTWIDTH, PORTHEIGHT)
             port.move(0, dist)
             port.show()
-            self.ports.append(port)
+            self._ports.append(port)
 
         for i, dist in enumerate(output_ports_dist):
             port = PortButton(">", self, op.output(i))
             port.setFixedSize(PORTWIDTH, PORTHEIGHT)
             port.move(MINBUTTONSIZE - PORTWIDTH, dist)
             port.show()
-            self.ports.append(port)
+            self._ports.append(port)