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" ),
@@ -130,11 +130,8 @@ def _create_qApp():
130130 'versions may not work as expected.'
131131 )
132132 break
133- try :
134- QtWidgets .QApplication .setAttribute (
135- QtCore .Qt .AA_EnableHighDpiScaling )
136- except AttributeError : # Only for Qt>=5.6, <6.
137- pass
133+ QtWidgets .QApplication .setAttribute (
134+ QtCore .Qt .ApplicationAttribute .AA_EnableHighDpiScaling )
138135 try :
139136 QtWidgets .QApplication .setHighDpiScaleFactorRoundingPolicy (
140137 QtCore .Qt .HighDpiScaleFactorRoundingPolicy .PassThrough )
@@ -148,10 +145,7 @@ def _create_qApp():
148145 app .lastWindowClosed .connect (app .quit )
149146 cbook ._setup_new_guiapp ()
150147
151- try :
152- app .setAttribute (QtCore .Qt .AA_UseHighDpiPixmaps ) # Only for Qt<6.
153- except AttributeError :
154- pass
148+ app .setAttribute (QtCore .Qt .ApplicationAttribute .AA_UseHighDpiPixmaps )
155149
156150 return app
157151
@@ -191,7 +185,7 @@ class FigureCanvasQT(FigureCanvasBase, QtWidgets.QWidget):
191185 manager_class = _api .classproperty (lambda cls : FigureManagerQT )
192186
193187 buttond = {
194- getattr (_enum ( " QtCore.Qt.MouseButton" ) , k ): v for k , v in [
188+ getattr (QtCore .Qt .MouseButton , k ): v for k , v in [
195189 ("LeftButton" , MouseButton .LEFT ),
196190 ("RightButton" , MouseButton .RIGHT ),
197191 ("MiddleButton" , MouseButton .MIDDLE ),
@@ -209,8 +203,7 @@ def __init__(self, figure=None):
209203 self ._draw_rect_callback = lambda painter : None
210204 self ._in_resize_event = False
211205
212- self .setAttribute (
213- _enum ("QtCore.Qt.WidgetAttribute" ).WA_OpaquePaintEvent )
206+ self .setAttribute (QtCore .Qt .WidgetAttribute .WA_OpaquePaintEvent )
214207 self .setMouseTracking (True )
215208 self .resize (* self .get_width_height ())
216209
@@ -564,7 +557,7 @@ def __init__(self, canvas, num):
564557 # StrongFocus accepts both tab and click to focus and will enable the
565558 # canvas to process event without clicking.
566559 # https://doc.qt.io/qt-5/qt.html#FocusPolicy-enum
567- self .canvas .setFocusPolicy (_enum ( " QtCore.Qt.FocusPolicy" ) .StrongFocus )
560+ self .canvas .setFocusPolicy (QtCore .Qt .FocusPolicy .StrongFocus )
568561 self .canvas .setFocus ()
569562
570563 self .window .raise_ ()
@@ -642,8 +635,8 @@ def __init__(self, canvas, parent=None, coordinates=True):
642635 """coordinates: should we show the coordinates on the right?"""
643636 QtWidgets .QToolBar .__init__ (self , parent )
644637 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
645- _to_int ( _enum ( " QtCore.Qt.ToolBarArea" ) .TopToolBarArea ) |
646- _to_int ( _enum ( " QtCore.Qt.ToolBarArea" ) .BottomToolBarArea ) ))
638+ QtCore .Qt .ToolBarArea .TopToolBarArea |
639+ QtCore .Qt .ToolBarArea .BottomToolBarArea ))
647640
648641 self .coordinates = coordinates
649642 self ._actions = {} # mapping of toolitem method names to QActions.
@@ -667,11 +660,12 @@ def __init__(self, canvas, parent=None, coordinates=True):
667660 if self .coordinates :
668661 self .locLabel = QtWidgets .QLabel ("" , self )
669662 self .locLabel .setAlignment (QtCore .Qt .AlignmentFlag (
670- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignRight ) |
671- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignVCenter )))
663+ QtCore .Qt .AlignmentFlag .AlignRight |
664+ QtCore .Qt .AlignmentFlag .AlignVCenter
665+ ))
672666 self .locLabel .setSizePolicy (QtWidgets .QSizePolicy (
673- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
674- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
667+ QtWidgets .QSizePolicy .Policy .Expanding ,
668+ QtWidgets .QSizePolicy .Policy .Ignored ,
675669 ))
676670 labelAction = self .addWidget (self .locLabel )
677671 labelAction .setVisible (True )
@@ -697,7 +691,7 @@ def _icon(self, name):
697691 icon_color = self .palette ().color (self .foregroundRole ())
698692 mask = pm .createMaskFromColor (
699693 QtGui .QColor ('black' ),
700- _enum ( " QtCore.Qt.MaskMode" ) .MaskOutColor )
694+ QtCore .Qt .MaskMode .MaskOutColor )
701695 pm .fill (icon_color )
702696 pm .setMask (mask )
703697 return QtGui .QIcon (pm )
@@ -801,8 +795,8 @@ def save_figure(self, *args):
801795 except Exception as e :
802796 QtWidgets .QMessageBox .critical (
803797 self , "Error saving file" , str (e ),
804- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .Ok ,
805- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .NoButton )
798+ QtWidgets .QMessageBox .StandardButton .Ok ,
799+ QtWidgets .QMessageBox .StandardButton .NoButton )
806800
807801 def set_history_buttons (self ):
808802 can_backward = self ._nav_stack ._pos > 0
@@ -916,15 +910,15 @@ def __init__(self, toolmanager, parent=None):
916910 ToolContainerBase .__init__ (self , toolmanager )
917911 QtWidgets .QToolBar .__init__ (self , parent )
918912 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
919- _to_int ( _enum ( " QtCore.Qt.ToolBarArea" ) .TopToolBarArea ) |
920- _to_int ( _enum ( " QtCore.Qt.ToolBarArea" ) .BottomToolBarArea ) ))
913+ QtCore .Qt .ToolBarArea .TopToolBarArea |
914+ QtCore .Qt .ToolBarArea .BottomToolBarArea ))
921915 message_label = QtWidgets .QLabel ("" )
922916 message_label .setAlignment (QtCore .Qt .AlignmentFlag (
923- _to_int ( _enum ( " QtCore.Qt.AlignmentFlag" ) .AlignRight ) |
924- _to_int ( _enum ( " QtCore.Qt.AlignmentFlag" ) .AlignVCenter ) ))
917+ QtCore .Qt .AlignmentFlag .AlignRight |
918+ QtCore .Qt .AlignmentFlag .AlignVCenter ))
925919 message_label .setSizePolicy (QtWidgets .QSizePolicy (
926- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
927- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
920+ QtWidgets .QSizePolicy .Policy .Expanding ,
921+ QtWidgets .QSizePolicy .Policy .Ignored ,
928922 ))
929923 self ._message_action = self .addWidget (message_label )
930924 self ._toolitems = {}
0 commit comments