Update to PyQt6

reilleya · Jun 6, 2023 · dd61214 · dd61214
1 parent 1286748
commit dd61214
Show file tree

Hide file tree

Showing 36 changed files with 92 additions and 95 deletions.
diff --git a/app.py b/app.py
@@ -1,7 +1,7 @@
 import sys
 
-from PyQt5.QtWidgets import QApplication, QMessageBox
-from PyQt5.QtGui import QIcon
+from PyQt6.QtWidgets import QApplication, QMessageBox
+from PyQt6.QtGui import QIcon
 
 import motorlib
 from motorlib import simResult
@@ -80,7 +80,7 @@ def outputMessage(self, content, title='openMotor'):
             msg.setWindowIcon(self.icon)
             msg.setText(content)
             msg.setWindowTitle(title)
-            msg.exec_()
+            msg.exec()
 
     def outputException(self, exception, text, title='openMotor - Error'):
         if self.headless:
@@ -93,4 +93,4 @@ def outputException(self, exception, text, title='openMotor - Error'):
             msg.setText(text)
             msg.setInformativeText(str(exception))
             msg.setWindowTitle(title)
-            msg.exec_()
+            msg.exec()
diff --git a/main.py b/main.py
@@ -1,6 +1,6 @@
 import sys
 from app import App
-from PyQt5.QtCore import Qt
+from PyQt6.QtCore import Qt
 
 app = App(sys.argv)
 sys.exit(app.exec())
diff --git a/pyuic.json b/pyuic.json
@@ -5,8 +5,6 @@
             "uilib/views"
         ]
     ],
-    "pyrcc": "pyrcc5",
-    "pyrcc_options": "",
-    "pyuic": "pyuic5",
-    "pyuic_options": "--from-imports"
+    "pyuic": "pyuic6",
+    "pyuic_options": ""
 }
diff --git a/requirements.txt b/requirements.txt
@@ -10,8 +10,8 @@ numpy
 Pillow
 pyparsing
 pyqt-distutils
-PyQt5
-PyQt5-sip
+PyQt6==6.4.2
+PyQt6-sip==13.4.1
 python-dateutil
 PyYAML
 scikit-fmm

diff --git a/uilib/converter.py b/uilib/converter.py
@@ -1,6 +1,6 @@
-from PyQt5.QtCore import QObject
-from PyQt5.QtWidgets import QFileDialog, QApplication
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtCore import QObject
+from PyQt6.QtWidgets import QFileDialog, QApplication
+from PyQt6.QtCore import pyqtSignal
 
 class Converter(QObject):
     def __init__(self, manager, name, description, fileTypes):

diff --git a/uilib/converters/csvExporter.py b/uilib/converters/csvExporter.py
@@ -1,4 +1,4 @@
-from PyQt5.QtWidgets import QDialog, QApplication
+from PyQt6.QtWidgets import QDialog, QApplication
 
 from ..converter import Exporter
 

diff --git a/uilib/converters/engExporter.py b/uilib/converters/engExporter.py
@@ -1,5 +1,5 @@
 import xml.etree.ElementTree as ET
-from PyQt5.QtWidgets import QDialog, QFileDialog, QDialogButtonBox, QApplication
+from PyQt6.QtWidgets import QDialog, QFileDialog, QDialogButtonBox, QApplication
 
 from motorlib.properties import PropertyCollection, FloatProperty, StringProperty, EnumProperty
 import motorlib

diff --git a/uilib/converters/imageExporter.py b/uilib/converters/imageExporter.py
@@ -1,4 +1,4 @@
-from PyQt5.QtWidgets import QDialog, QApplication
+from PyQt6.QtWidgets import QDialog, QApplication
 
 from motorlib.simResult import singleValueChannels, multiValueChannels
 from ..converter import Exporter

diff --git a/uilib/fileIO.py b/uilib/fileIO.py
@@ -2,7 +2,7 @@
 import os
 import platform
 
-from PyQt5.QtWidgets import QApplication
+from PyQt6.QtWidgets import QApplication
 import yaml
 import appdirs
 

diff --git a/uilib/fileManager.py b/uilib/fileManager.py
@@ -1,6 +1,6 @@
-from PyQt5.QtCore import QObject
-from PyQt5.QtWidgets import QFileDialog, QMessageBox
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtCore import QObject
+from PyQt6.QtWidgets import QFileDialog, QMessageBox
+from PyQt6.QtCore import pyqtSignal
 
 import motorlib
 
@@ -157,7 +157,7 @@ def unsavedCheck(self):
         msg.setWindowTitle("Close without saving?")
         msg.setStandardButtons(QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel)
 
-        res = msg.exec_()
+        res = msg.exec()
         if res == QMessageBox.Save:
             self.save()
             return True

diff --git a/uilib/helpers.py b/uilib/helpers.py
@@ -1,3 +1,3 @@
-from PyQt5.QtCore import Qt
+from PyQt6.QtCore import Qt
 
-FLAGS_NO_ICON = Qt.Dialog | Qt.CustomizeWindowHint | Qt.WindowTitleHint | Qt.WindowCloseButtonHint
+FLAGS_NO_ICON = Qt.WindowType.Dialog | Qt.WindowType.CustomizeWindowHint | Qt.WindowType.WindowTitleHint | Qt.WindowType.WindowCloseButtonHint
diff --git a/uilib/importExportManager.py b/uilib/importExportManager.py
@@ -1,6 +1,5 @@
-from PyQt5.QtCore import QObject
-from PyQt5.QtWidgets import QAction
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtCore import pyqtSignal, QObject
+from PyQt6.QtGui import QAction
 
 from .converter import Importer
 from .converters import BurnSimImporter, BurnSimExporter, EngExporter, CsvExporter, ImageExporter

diff --git a/uilib/preferencesManager.py b/uilib/preferencesManager.py
@@ -1,4 +1,4 @@
-from PyQt5.QtCore import QObject, pyqtSignal
+from PyQt6.QtCore import QObject, pyqtSignal
 
 from motorlib.properties import PropertyCollection, FloatProperty, IntProperty, EnumProperty
 from motorlib.units import unitLabels, getAllConversions

diff --git a/uilib/propellantManager.py b/uilib/propellantManager.py
@@ -1,4 +1,4 @@
-from PyQt5.QtCore import QObject, pyqtSignal
+from PyQt6.QtCore import QObject, pyqtSignal
 
 import motorlib
 

diff --git a/uilib/simulationManager.py b/uilib/simulationManager.py
@@ -1,7 +1,7 @@
 from threading import Thread
 
-from PyQt5.QtCore import QObject
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtCore import QObject
+from PyQt6.QtCore import pyqtSignal
 
 from .widgets.simulationAlertsDialog import SimulationAlertsDialog
 from .widgets.simulationProgressDialog import SimulationProgressDialog

diff --git a/uilib/tool.py b/uilib/tool.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QDialog, QVBoxLayout, QLabel, QApplication
-from PyQt5.QtGui import QIcon
+from PyQt6.QtWidgets import QDialog, QVBoxLayout, QLabel, QApplication
+from PyQt6.QtGui import QIcon
 
 import motorlib
 

diff --git a/uilib/toolManager.py b/uilib/toolManager.py
@@ -1,6 +1,5 @@
-from PyQt5.QtCore import QObject
-from PyQt5.QtWidgets import QAction
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtCore import QObject, pyqtSignal
+from PyQt6.QtGui import QAction
 
 from .tools import ChangeDiameterTool, InitialKNTool, MaxKNTool
 from .tools import ExpansionTool

diff --git a/uilib/widgets/aboutDialog.py b/uilib/widgets/aboutDialog.py
@@ -1,4 +1,4 @@
-from PyQt5.QtWidgets import QDialog, QApplication
+from PyQt6.QtWidgets import QDialog, QApplication
 from ..views.AboutDialog_ui import Ui_AboutDialog
 
 

diff --git a/uilib/widgets/channelSelector.py b/uilib/widgets/channelSelector.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QGroupBox, QCheckBox, QRadioButton, QVBoxLayout
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QGroupBox, QCheckBox, QRadioButton, QVBoxLayout
+from PyQt6.QtCore import pyqtSignal, Qt
 
 import motorlib
 
@@ -27,7 +27,7 @@ def setupChecks(self, multiselect, disabled=[], default=None, exclude=[]):
                 if default is not None:
                     if multiselect:
                         if channel in default:
-                            self.checks[channel].setCheckState(2)
+                            self.checks[channel].setCheckState(Qt.CheckState.Checked)
                     else:
                         self.checks[channel].setChecked(channel == default)
                 self.checks[channel].toggled.connect(self.checksChanged.emit)
@@ -56,7 +56,7 @@ def resetChecks(self):
     def unselect(self, channels):
         for channel in channels:
             if channel in self.checks.keys():
-                self.checks[channel].setCheckState(0)
+                self.checks[channel].setCheckState(Qt.CheckState.Unchecked)
 
     def toggleEnabled(self, channels, enabled):
         for channel in channels:

diff --git a/uilib/widgets/collectionEditor.py b/uilib/widgets/collectionEditor.py
@@ -1,7 +1,7 @@
-from PyQt5.QtWidgets import QWidget, QFormLayout, QVBoxLayout, QHBoxLayout
-from PyQt5.QtWidgets import QLabel, QPushButton
-from PyQt5.QtWidgets import QSpacerItem, QSizePolicy
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QWidget, QFormLayout, QVBoxLayout, QHBoxLayout
+from PyQt6.QtWidgets import QLabel, QPushButton
+from PyQt6.QtWidgets import QSpacerItem, QSizePolicy
+from PyQt6.QtCore import pyqtSignal
 
 from .propertyEditor import PropertyEditor
 
@@ -27,7 +27,7 @@ def __init__(self, parent, buttons=False):
         self.layout().addLayout(self.stats)
 
 
-        self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum, QSizePolicy.Expanding)
+        self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Policy.Minimum, QSizePolicy.Policy.Expanding)
         self.layout().addItem(self.verticalSpacer)
 
         self.buttons = buttons
@@ -71,7 +71,7 @@ def loadProperties(self, obj):
             self.propertyEditors[prop] = PropertyEditor(self, obj.props[prop], self.preferences)
             self.propertyEditors[prop].valueChanged.connect(self.propertyUpdate)
             label = QLabel('{}:'.format(obj.props[prop].dispName))
-            label.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Expanding)
+            label.setSizePolicy(QSizePolicy.Policy.Minimum, QSizePolicy.Policy.Expanding)
             self.form.addRow(label, self.propertyEditors[prop])
         if self.buttons:
             self.applyButton.show()

diff --git a/uilib/widgets/grainImageWidget.py b/uilib/widgets/grainImageWidget.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QLabel
-from PyQt5.QtGui import QPixmap, QImage
+from PyQt6.QtWidgets import QLabel
+from PyQt6.QtGui import QPixmap, QImage
 import numpy as np
 
 class GrainImageWidget(QLabel):
@@ -8,6 +8,6 @@ def showImage(self, image):
         image = np.logical_not(image.filled())
         image = image.astype(np.uint8) * 255
         height, width = image.shape
-        qImg = QImage(image.data, width, height, QImage.Format_Grayscale8)
+        qImg = QImage(image.data, width, height, QImage.Format.Format_Grayscale8)
         pixmap = QPixmap(qImg)
         self.setPixmap(pixmap)
diff --git a/uilib/widgets/grainPreviewWidget.py b/uilib/widgets/grainPreviewWidget.py
@@ -1,7 +1,7 @@
 from threading import Thread
 
-from PyQt5.QtWidgets import QWidget
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QWidget
+from PyQt6.QtCore import pyqtSignal
 
 import motorlib
 

diff --git a/uilib/widgets/grainSelector.py b/uilib/widgets/grainSelector.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QGroupBox, QCheckBox, QRadioButton, QVBoxLayout
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QGroupBox, QCheckBox, QRadioButton, QVBoxLayout
+from PyQt6.QtCore import pyqtSignal, Qt
 
 class GrainSelector(QGroupBox):
 
@@ -23,7 +23,7 @@ def setupChecks(self, simRes, multiselect):
             checkTitle = "Grain " + str(gid + 1)
             if multiselect:
                 check = QCheckBox(checkTitle)
-                check.setCheckState(2)
+                check.setCheckState(Qt.CheckState.Checked)
             else:
                 check = QRadioButton(checkTitle)
             self.layout().addWidget(check)
@@ -46,6 +46,6 @@ def getUnselectedGrains(self):
 
     def setChecks(self, checks):
         for check in self.checks:
-            check.setCheckState(0)
+            check.setCheckState(Qt.CheckState.Unchecked)
         for check in checks:
-            self.checks[check].setCheckState(2)
+            self.checks[check].setCheckState(Qt.CheckState.Checked)
diff --git a/uilib/widgets/mainWindow.py b/uilib/widgets/mainWindow.py
@@ -1,7 +1,7 @@
 import sys
 
-from PyQt5.QtWidgets import QWidget, QMainWindow, QTableWidgetItem, QHeaderView, QTableWidget
-from PyQt5.QtCore import Qt
+from PyQt6.QtWidgets import QWidget, QMainWindow, QTableWidgetItem, QHeaderView, QTableWidget
+from PyQt6.QtCore import Qt
 
 import motorlib
 import uilib.widgets.aboutDialog
@@ -132,8 +132,8 @@ def setupGrainTable(self):
         self.ui.tableWidgetGrainList.clearContents()
 
         header = self.ui.tableWidgetGrainList.horizontalHeader()
-        header.setSectionResizeMode(0, QHeaderView.ResizeToContents)
-        header.setSectionResizeMode(1, QHeaderView.Stretch)
+        header.setSectionResizeMode(0, QHeaderView.ResizeMode.ResizeToContents)
+        header.setSectionResizeMode(1, QHeaderView.ResizeMode.Stretch)
 
         self.updateGrainTable()
 
@@ -381,6 +381,6 @@ def applyPreferences(self, prefDict):
         self.setupGraph()
 
     def keyPressEvent(self, event):
-        if event.key() == Qt.Key_Delete or event.key() == Qt.Key_Backspace:
+        if event.key() == Qt.Key.Key_Delete or event.key() == Qt.Key.Key_Backspace:
             if len(self.ui.tableWidgetGrainList.selectedItems()) != 0:
                 self.deleteGrain()
diff --git a/uilib/widgets/motorEditor.py b/uilib/widgets/motorEditor.py
@@ -1,4 +1,4 @@
-from PyQt5.QtWidgets import QLabel
+from PyQt6.QtWidgets import QLabel
 
 import motorlib.grain
 import motorlib.nozzle

diff --git a/uilib/widgets/nozzlePreviewWidget.py b/uilib/widgets/nozzlePreviewWidget.py
@@ -1,8 +1,8 @@
 from math import radians, tan
 
-from PyQt5.QtWidgets import QWidget, QApplication, QGraphicsScene, QGraphicsPolygonItem
-from PyQt5.QtGui import QPolygonF, QBrush
-from PyQt5.QtCore import QPointF, Qt
+from PyQt6.QtWidgets import QWidget, QApplication, QGraphicsScene, QGraphicsPolygonItem
+from PyQt6.QtGui import QPolygonF, QBrush
+from PyQt6.QtCore import QPointF, Qt
 
 import motorlib
 from ..views.NozzlePreview_ui import Ui_NozzlePreview
@@ -14,7 +14,8 @@ def __init__(self):
         self.ui.setupUi(self)
 
         self.brush = QBrush()
-        self.brush.setStyle(1)
+
+        self.brush.setStyle(Qt.BrushStyle.SolidPattern)
         self.scene = QGraphicsScene(self)
         self.upper = QGraphicsPolygonItem()
         self.lower = QGraphicsPolygonItem()
@@ -82,4 +83,4 @@ def loadNozzle(self, nozzle):
 
     def rescale(self):
         self.scene.setSceneRect(self.scene.itemsBoundingRect())
-        self.ui.tabCrossSection.fitInView(self.scene.sceneRect(), Qt.KeepAspectRatio)
+        self.ui.tabCrossSection.fitInView(self.scene.sceneRect(), Qt.AspectRatioMode.KeepAspectRatio)
diff --git a/uilib/widgets/polygonEditor.py b/uilib/widgets/polygonEditor.py
@@ -1,8 +1,8 @@
 import math
 import itertools
 
-from PyQt5.QtWidgets import QWidget, QPushButton, QHBoxLayout, QFileDialog, QApplication
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QWidget, QPushButton, QHBoxLayout, QFileDialog, QApplication
+from PyQt6.QtCore import pyqtSignal
 
 import ezdxf
 import motorlib

diff --git a/uilib/widgets/preferencesMenu.py b/uilib/widgets/preferencesMenu.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QDialog, QApplication
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QDialog, QApplication
+from PyQt6.QtCore import pyqtSignal
 
 from ..views.Preferences_ui import Ui_PreferencesDialog
 

diff --git a/uilib/widgets/propellantMenu.py b/uilib/widgets/propellantMenu.py
@@ -1,7 +1,7 @@
-from PyQt5.QtWidgets import QDialog, QMessageBox, QApplication
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QDialog, QMessageBox, QApplication
+from PyQt6.QtCore import pyqtSignal
 from ..helpers import FLAGS_NO_ICON
-from PyQt5.QtCore import Qt
+from PyQt6.QtCore import Qt
 
 import motorlib.propellant
 
@@ -128,13 +128,13 @@ def unsavedCheck(self):
         msg.setWindowTitle("Close without saving?")
         msg.setStandardButtons(QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel)
 
-        res = msg.exec_()
+        res = msg.exec()
         if res == QMessageBox.Save:
             self.propEdited(self.ui.propEditor.getProperties())
             return True
         return res == QMessageBox.Discard
 
     def keyPressEvent(self, event):
-        if event.key() == Qt.Key_Delete or event.key() == Qt.Key_Backspace:
+        if event.key() == Qt.Key.Key_Delete or event.key() == Qt.Key.Key_Backspace:
             if len(self.ui.listWidgetPropellants.selectedItems()) != 0:
                 self.deleteProp()
diff --git a/uilib/widgets/propellantPreviewWidget.py b/uilib/widgets/propellantPreviewWidget.py
@@ -1,5 +1,5 @@
-from PyQt5.QtWidgets import QWidget
-from PyQt5.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QWidget
+from PyQt6.QtCore import pyqtSignal
 
 import motorlib