1414from . import qt_compat
1515from .qt_compat import (
1616 QtCore , QtGui , QtWidgets , __version__ , QT_API ,
17- _enum , _to_int , _isdeleted , _maybe_allow_interrupt
17+ _to_int , _isdeleted , _maybe_allow_interrupt
1818)
1919
2020
2121# SPECIAL_KEYS are Qt::Key that do *not* return their Unicode name
2222# instead they have manually specified names.
2323SPECIAL_KEYS = {
24- _to_int (getattr (_enum ( " QtCore.Qt.Key" ) , k )): v for k , v in [
24+ _to_int (getattr (QtCore .Qt .Key , k )): v for k , v in [
2525 ("Key_Escape" , "escape" ),
2626 ("Key_Tab" , "tab" ),
2727 ("Key_Backspace" , "backspace" ),
6666# Elements are (Qt::KeyboardModifiers, Qt::Key) tuples.
6767# Order determines the modifier order (ctrl+alt+...) reported by Matplotlib.
6868_MODIFIER_KEYS = [
69- (_to_int (getattr (_enum ( " QtCore.Qt.KeyboardModifier" ) , mod )),
70- _to_int (getattr (_enum ( " QtCore.Qt.Key" ) , key )))
69+ (_to_int (getattr (QtCore .Qt .KeyboardModifier , mod )),
70+ _to_int (getattr (QtCore .Qt .Key , key )))
7171 for mod , key in [
7272 ("ControlModifier" , "Key_Control" ),
7373 ("AltModifier" , "Key_Alt" ),
7676 ]
7777]
7878cursord = {
79- k : getattr (_enum ( " QtCore.Qt.CursorShape" ) , v ) for k , v in [
79+ k : getattr (QtCore .Qt .CursorShape , v ) for k , v in [
8080 (cursors .MOVE , "SizeAllCursor" ),
8181 (cursors .HAND , "PointingHandCursor" ),
8282 (cursors .POINTER , "ArrowCursor" ),
@@ -132,8 +132,8 @@ def _create_qApp():
132132 break
133133 try :
134134 QtWidgets .QApplication .setAttribute (
135- QtCore .Qt .AA_EnableHighDpiScaling )
136- except AttributeError : # Only for Qt>=5.6, <6.
135+ QtCore .Qt .ApplicationAttribute . AA_EnableHighDpiScaling )
136+ except AttributeError :
137137 pass
138138 try :
139139 QtWidgets .QApplication .setHighDpiScaleFactorRoundingPolicy (
@@ -147,9 +147,9 @@ def _create_qApp():
147147 app .setWindowIcon (icon )
148148 app .lastWindowClosed .connect (app .quit )
149149 cbook ._setup_new_guiapp ()
150-
151150 try :
152- app .setAttribute (QtCore .Qt .AA_UseHighDpiPixmaps ) # Only for Qt<6.
151+ # Only for Qt<6.
152+ app .setAttribute (QtCore .Qt .ApplicationAttribute .AA_UseHighDpiPixmaps )
153153 except AttributeError :
154154 pass
155155
@@ -191,7 +191,7 @@ class FigureCanvasQT(FigureCanvasBase, QtWidgets.QWidget):
191191 manager_class = _api .classproperty (lambda cls : FigureManagerQT )
192192
193193 buttond = {
194- getattr (_enum ( " QtCore.Qt.MouseButton" ) , k ): v for k , v in [
194+ getattr (QtCore .Qt .MouseButton , k ): v for k , v in [
195195 ("LeftButton" , MouseButton .LEFT ),
196196 ("RightButton" , MouseButton .RIGHT ),
197197 ("MiddleButton" , MouseButton .MIDDLE ),
@@ -209,8 +209,7 @@ def __init__(self, figure=None):
209209 self ._draw_rect_callback = lambda painter : None
210210 self ._in_resize_event = False
211211
212- self .setAttribute (
213- _enum ("QtCore.Qt.WidgetAttribute" ).WA_OpaquePaintEvent )
212+ self .setAttribute (QtCore .Qt .WidgetAttribute .WA_OpaquePaintEvent )
214213 self .setMouseTracking (True )
215214 self .resize (* self .get_width_height ())
216215
@@ -564,7 +563,7 @@ def __init__(self, canvas, num):
564563 # StrongFocus accepts both tab and click to focus and will enable the
565564 # canvas to process event without clicking.
566565 # https://doc.qt.io/qt-5/qt.html#FocusPolicy-enum
567- self .canvas .setFocusPolicy (_enum ( " QtCore.Qt.FocusPolicy" ) .StrongFocus )
566+ self .canvas .setFocusPolicy (QtCore .Qt .FocusPolicy .StrongFocus )
568567 self .canvas .setFocus ()
569568
570569 self .window .raise_ ()
@@ -642,9 +641,8 @@ def __init__(self, canvas, parent=None, coordinates=True):
642641 """coordinates: should we show the coordinates on the right?"""
643642 QtWidgets .QToolBar .__init__ (self , parent )
644643 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
645- _to_int (_enum ("QtCore.Qt.ToolBarArea" ).TopToolBarArea ) |
646- _to_int (_enum ("QtCore.Qt.ToolBarArea" ).BottomToolBarArea )))
647-
644+ _to_int (QtCore .Qt .ToolBarArea .TopToolBarArea ) |
645+ _to_int (QtCore .Qt .ToolBarArea .BottomToolBarArea )))
648646 self .coordinates = coordinates
649647 self ._actions = {} # mapping of toolitem method names to QActions.
650648 self ._subplot_dialog = None
@@ -667,11 +665,12 @@ def __init__(self, canvas, parent=None, coordinates=True):
667665 if self .coordinates :
668666 self .locLabel = QtWidgets .QLabel ("" , self )
669667 self .locLabel .setAlignment (QtCore .Qt .AlignmentFlag (
670- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignRight ) |
671- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignVCenter )))
668+ _to_int (QtCore .Qt .AlignmentFlag .AlignRight ) |
669+ _to_int (QtCore .Qt .AlignmentFlag .AlignVCenter )))
670+
672671 self .locLabel .setSizePolicy (QtWidgets .QSizePolicy (
673- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
674- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
672+ QtWidgets .QSizePolicy .Policy .Expanding ,
673+ QtWidgets .QSizePolicy .Policy .Ignored ,
675674 ))
676675 labelAction = self .addWidget (self .locLabel )
677676 labelAction .setVisible (True )
@@ -697,7 +696,7 @@ def _icon(self, name):
697696 icon_color = self .palette ().color (self .foregroundRole ())
698697 mask = pm .createMaskFromColor (
699698 QtGui .QColor ('black' ),
700- _enum ( " QtCore.Qt.MaskMode" ) .MaskOutColor )
699+ QtCore .Qt .MaskMode .MaskOutColor )
701700 pm .fill (icon_color )
702701 pm .setMask (mask )
703702 return QtGui .QIcon (pm )
@@ -801,8 +800,8 @@ def save_figure(self, *args):
801800 except Exception as e :
802801 QtWidgets .QMessageBox .critical (
803802 self , "Error saving file" , str (e ),
804- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .Ok ,
805- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .NoButton )
803+ QtWidgets .QMessageBox .StandardButton .Ok ,
804+ QtWidgets .QMessageBox .StandardButton .NoButton )
806805
807806 def set_history_buttons (self ):
808807 can_backward = self ._nav_stack ._pos > 0
@@ -916,15 +915,15 @@ def __init__(self, toolmanager, parent=None):
916915 ToolContainerBase .__init__ (self , toolmanager )
917916 QtWidgets .QToolBar .__init__ (self , parent )
918917 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
919- _to_int (_enum ( " QtCore.Qt.ToolBarArea" ) .TopToolBarArea ) |
920- _to_int (_enum ( " QtCore.Qt.ToolBarArea" ) .BottomToolBarArea )))
918+ _to_int (QtCore .Qt .ToolBarArea .TopToolBarArea ) |
919+ _to_int (QtCore .Qt .ToolBarArea .BottomToolBarArea )))
921920 message_label = QtWidgets .QLabel ("" )
922921 message_label .setAlignment (QtCore .Qt .AlignmentFlag (
923- _to_int (_enum ( " QtCore.Qt.AlignmentFlag" ) .AlignRight ) |
924- _to_int (_enum ( " QtCore.Qt.AlignmentFlag" ) .AlignVCenter )))
922+ _to_int (QtCore .Qt .AlignmentFlag .AlignRight ) |
923+ _to_int (QtCore .Qt .AlignmentFlag .AlignVCenter )))
925924 message_label .setSizePolicy (QtWidgets .QSizePolicy (
926- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
927- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
925+ QtWidgets .QSizePolicy .Policy .Expanding ,
926+ QtWidgets .QSizePolicy .Policy .Ignored ,
928927 ))
929928 self ._message_action = self .addWidget (message_label )
930929 self ._toolitems = {}
0 commit comments