Maj pyqt6

Author: maxDcb

C2Client/C2Client/GraphPanel.py

@@ -189,7 +189,7 @@ def __init__(self, parent, grpcClient):
         self.scene = QGraphicsScene()
 
         self.view = QGraphicsView(self.scene)
-        self.view.setRenderHint(QPainter.Antialiasing)  
+        self.view.setRenderHint(QPainter.RenderHint.Antialiasing)
 
         self.vbox = QVBoxLayout()
         self.vbox.setContentsMargins(0, 0, 0, 0)
@@ -336,8 +336,8 @@ def updateGraph(self):
                             print("[+] add connector listener:", listenerHash, "beacon", beaconHash)
 
         for item in self.listNodeItem:
-            item.setFlag(QGraphicsItem.ItemIsMovable)
-            item.setFlag(QGraphicsItem.ItemIsSelectable)
+            item.setFlag(QGraphicsItem.GraphicsItemFlag.ItemIsMovable)
+            item.setFlag(QGraphicsItem.GraphicsItemFlag.ItemIsSelectable)
 
 
 class GetGraphInfoWorker(QObject):

C2Client/C2Client/ListenerPanel.py

@@ -15,6 +15,7 @@
     QTableWidgetItem,
     QWidget,
     QHeaderView,
+    QAbstractItemView,
 )
 
 from .grpcClient import TeamServerApi_pb2
@@ -79,7 +80,7 @@ def __init__(self, parent, grpcClient):
         # List of sessions
         self.listListener = QTableWidget()
         self.listListener.setShowGrid(False)
-        self.listListener.setSelectionBehavior(QTableView.SelectRows)
+        self.listListener.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
 
         self.listListener.setRowCount(0)
         self.listListener.setColumnCount(4)
@@ -91,7 +92,7 @@ def __init__(self, parent, grpcClient):
         self.listListener.verticalHeader().setVisible(False)
         header = self.listListener.horizontalHeader()      
         for i in range(header.count()):
-            header.setSectionResizeMode(i, QHeaderView.Stretch)
+            header.setSectionResizeMode(i, QHeaderView.ResizeMode.Stretch)
         self.layout.addWidget(self.listListener)
 
         # Thread to get listeners every second
@@ -118,15 +119,15 @@ def showContextMenu(self, position):
             menu = QMenu()
             menu.addAction('Add')
             menu.triggered.connect(self.actionClicked)
-            menu.exec_(self.listListener.viewport().mapToGlobal(position))
+            menu.exec(self.listListener.viewport().mapToGlobal(position))
         else:
             row = index.row()
             self.item = str(self.listListener.item(row, 0).data(0))
 
             menu = QMenu()
             menu.addAction('Stop')
             menu.triggered.connect(self.actionClicked)
-            menu.exec_(self.listListener.viewport().mapToGlobal(position))
+            menu.exec(self.listListener.viewport().mapToGlobal(position))
 
 
     # catch stopListener menu click

C2Client/C2Client/SessionPanel.py

@@ -11,6 +11,7 @@
     QTableWidgetItem,
     QWidget,
     QHeaderView,
+    QAbstractItemView,
 )
 
 from .grpcClient import TeamServerApi_pb2
@@ -60,7 +61,7 @@ def __init__(self, parent, grpcClient):
         # List of sessions
         self.listSession = QTableWidget()
         self.listSession.setShowGrid(False)
-        self.listSession.setSelectionBehavior(QTableView.SelectRows)
+        self.listSession.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
         self.listSession.setRowCount(0)
         self.listSession.setColumnCount(11)
 
@@ -70,7 +71,7 @@ def __init__(self, parent, grpcClient):
         self.listSession.verticalHeader().setVisible(False)
         header = self.listSession.horizontalHeader()      
         for i in range(header.count()):
-            header.setSectionResizeMode(i, QHeaderView.Stretch)        
+            header.setSectionResizeMode(i, QHeaderView.ResizeMode.Stretch)     
         QTimer.singleShot(100, self.switch_to_interactive)
         self.layout.addWidget(self.listSession)
 
@@ -91,14 +92,14 @@ def resizeEvent(self, event):
         self.listSession.verticalHeader().setVisible(False)
         header = self.listSession.horizontalHeader()      
         for i in range(header.count()):
-            header.setSectionResizeMode(i, QHeaderView.Stretch)        
+            header.setSectionResizeMode(i, QHeaderView.ResizeMode.Stretch)
         QTimer.singleShot(100, self.switch_to_interactive)
 
 
     def switch_to_interactive(self):
         header = self.listSession.horizontalHeader()   
         for i in range(header.count()):
-            header.setSectionResizeMode(i, QHeaderView.Interactive)
+            header.setSectionResizeMode(i, QHeaderView.ResizeMode.Interactive)
 
     def __del__(self):
         self.getSessionsWorker.quit()
@@ -119,7 +120,7 @@ def showContextMenu(self, position):
         menu.addAction('Stop')
         menu.addAction('Delete')
         menu.triggered.connect(self.actionClicked)
-        menu.exec_(self.listSession.viewport().mapToGlobal(position))
+        menu.exec(self.listSession.viewport().mapToGlobal(position))
 
 
     # catch Interact and Stop menu click

C2Client/requirements.txt

@@ -1,8 +1,8 @@
 pycryptodome==3.23.0
-grpcio==1.66.1
+grpcio==1.74.0
 PyQt6==6.7.0
 pyqtdarktheme==2.1.0
-protobuf==5.27.0
+protobuf==6.32.0
 gitpython==3.1.45
 requests==2.32.5
 pwn==1.0