Handle QUndoStack compatibility between Qt5 and Qt6

Author: dalthviz

qtpy/QtGui.py

@@ -61,6 +61,7 @@ def __getattr__(name):
         QFileSystemModel,
         QShortcut,
         QUndoCommand,
+        QUndoStack,
     )
 
 elif PYQT6:
@@ -116,6 +117,7 @@ def __getattr__(name):
         QFileSystemModel,
         QShortcut,
         QUndoCommand,
+        QUndoStack,
     )
 
     if hasattr(QFontMetrics, "horizontalAdvance"):

qtpy/QtWidgets.py

@@ -42,6 +42,7 @@ def __getattr__(name):
         QFileSystemModel,
         QShortcut,
         QUndoCommand,
+        QundoStack,
     )
 
     if parse(_qt_version) < parse("6.4"):
@@ -118,7 +119,7 @@ def __getattr__(name):
 elif PYSIDE2:
     from PySide2.QtWidgets import *
 elif PYSIDE6:
-    from PySide6.QtGui import QActionGroup, QShortcut, QUndoCommand
+    from PySide6.QtGui import QActionGroup, QShortcut, QUndoCommand, QUndoStack
 
     if parse(_qt_version) < parse("6.4"):
         # Make `QAction.setShortcut` and `QAction.setShortcuts` compatible with Qt>=6.4

qtpy/tests/test_qtgui.py

@@ -61,6 +61,7 @@ def test_what_moved_to_qtgui_in_qt6():
     assert QtGui.QFileSystemModel is not None
     assert QtGui.QShortcut is not None
     assert QtGui.QUndoCommand is not None
+    assert QtGui.QUndoStack is not None
 
 
 def test_qtextdocument_functions(pdf_writer):

qtpy/tests/test_qtwidgets.py

@@ -44,6 +44,7 @@ def test_what_moved_to_qtgui_in_qt6():
     assert QtWidgets.QFileSystemModel is not None
     assert QtWidgets.QShortcut is not None
     assert QtWidgets.QUndoCommand is not None
+    assert QtWidgets.QUndoStack is not None
 
 
 def test_qplaintextedit_functions(qtbot, pdf_writer):