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" ),
@@ -142,7 +142,6 @@ def _create_qApp():
142142 app .setWindowIcon (icon )
143143 app .lastWindowClosed .connect (app .quit )
144144 cbook ._setup_new_guiapp ()
145-
146145 if qt_version == 5 :
147146 app .setAttribute (QtCore .Qt .AA_UseHighDpiPixmaps )
148147
@@ -184,7 +183,7 @@ class FigureCanvasQT(FigureCanvasBase, QtWidgets.QWidget):
184183 manager_class = _api .classproperty (lambda cls : FigureManagerQT )
185184
186185 buttond = {
187- getattr (_enum ( " QtCore.Qt.MouseButton" ) , k ): v for k , v in [
186+ getattr (QtCore .Qt .MouseButton , k ): v for k , v in [
188187 ("LeftButton" , MouseButton .LEFT ),
189188 ("RightButton" , MouseButton .RIGHT ),
190189 ("MiddleButton" , MouseButton .MIDDLE ),
@@ -202,8 +201,7 @@ def __init__(self, figure=None):
202201 self ._draw_rect_callback = lambda painter : None
203202 self ._in_resize_event = False
204203
205- self .setAttribute (
206- _enum ("QtCore.Qt.WidgetAttribute" ).WA_OpaquePaintEvent )
204+ self .setAttribute (QtCore .Qt .WidgetAttribute .WA_OpaquePaintEvent )
207205 self .setMouseTracking (True )
208206 self .resize (* self .get_width_height ())
209207
@@ -554,7 +552,7 @@ def __init__(self, canvas, num):
554552 # StrongFocus accepts both tab and click to focus and will enable the
555553 # canvas to process event without clicking.
556554 # https://doc.qt.io/qt-5/qt.html#FocusPolicy-enum
557- self .canvas .setFocusPolicy (_enum ( " QtCore.Qt.FocusPolicy" ) .StrongFocus )
555+ self .canvas .setFocusPolicy (QtCore .Qt .FocusPolicy .StrongFocus )
558556 self .canvas .setFocus ()
559557
560558 self .window .raise_ ()
@@ -634,9 +632,8 @@ def __init__(self, canvas, parent=None, coordinates=True):
634632 """coordinates: should we show the coordinates on the right?"""
635633 QtWidgets .QToolBar .__init__ (self , parent )
636634 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
637- _to_int (_enum ("QtCore.Qt.ToolBarArea" ).TopToolBarArea ) |
638- _to_int (_enum ("QtCore.Qt.ToolBarArea" ).BottomToolBarArea )))
639-
635+ _to_int (QtCore .Qt .ToolBarArea .TopToolBarArea ) |
636+ _to_int (QtCore .Qt .ToolBarArea .BottomToolBarArea )))
640637 self .coordinates = coordinates
641638 self ._actions = {} # mapping of toolitem method names to QActions.
642639 self ._subplot_dialog = None
@@ -659,11 +656,12 @@ def __init__(self, canvas, parent=None, coordinates=True):
659656 if self .coordinates :
660657 self .locLabel = QtWidgets .QLabel ("" , self )
661658 self .locLabel .setAlignment (QtCore .Qt .AlignmentFlag (
662- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignRight ) |
663- _to_int (_enum ("QtCore.Qt.AlignmentFlag" ).AlignVCenter )))
659+ _to_int (QtCore .Qt .AlignmentFlag .AlignRight ) |
660+ _to_int (QtCore .Qt .AlignmentFlag .AlignVCenter )))
661+
664662 self .locLabel .setSizePolicy (QtWidgets .QSizePolicy (
665- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
666- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
663+ QtWidgets .QSizePolicy .Policy .Expanding ,
664+ QtWidgets .QSizePolicy .Policy .Ignored ,
667665 ))
668666 labelAction = self .addWidget (self .locLabel )
669667 labelAction .setVisible (True )
@@ -689,7 +687,7 @@ def _icon(self, name):
689687 icon_color = self .palette ().color (self .foregroundRole ())
690688 mask = pm .createMaskFromColor (
691689 QtGui .QColor ('black' ),
692- _enum ( " QtCore.Qt.MaskMode" ) .MaskOutColor )
690+ QtCore .Qt .MaskMode .MaskOutColor )
693691 pm .fill (icon_color )
694692 pm .setMask (mask )
695693 return QtGui .QIcon (pm )
@@ -793,8 +791,8 @@ def save_figure(self, *args):
793791 except Exception as e :
794792 QtWidgets .QMessageBox .critical (
795793 self , "Error saving file" , str (e ),
796- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .Ok ,
797- _enum ( " QtWidgets.QMessageBox.StandardButton" ) .NoButton )
794+ QtWidgets .QMessageBox .StandardButton .Ok ,
795+ QtWidgets .QMessageBox .StandardButton .NoButton )
798796
799797 def set_history_buttons (self ):
800798 can_backward = self ._nav_stack ._pos > 0
@@ -908,15 +906,15 @@ def __init__(self, toolmanager, parent=None):
908906 ToolContainerBase .__init__ (self , toolmanager )
909907 QtWidgets .QToolBar .__init__ (self , parent )
910908 self .setAllowedAreas (QtCore .Qt .ToolBarArea (
911- _to_int (_enum ( " QtCore.Qt.ToolBarArea" ) .TopToolBarArea ) |
912- _to_int (_enum ( " QtCore.Qt.ToolBarArea" ) .BottomToolBarArea )))
909+ _to_int (QtCore .Qt .ToolBarArea .TopToolBarArea ) |
910+ _to_int (QtCore .Qt .ToolBarArea .BottomToolBarArea )))
913911 message_label = QtWidgets .QLabel ("" )
914912 message_label .setAlignment (QtCore .Qt .AlignmentFlag (
915- _to_int (_enum ( " QtCore.Qt.AlignmentFlag" ) .AlignRight ) |
916- _to_int (_enum ( " QtCore.Qt.AlignmentFlag" ) .AlignVCenter )))
913+ _to_int (QtCore .Qt .AlignmentFlag .AlignRight ) |
914+ _to_int (QtCore .Qt .AlignmentFlag .AlignVCenter )))
917915 message_label .setSizePolicy (QtWidgets .QSizePolicy (
918- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Expanding ,
919- _enum ( " QtWidgets.QSizePolicy.Policy" ) .Ignored ,
916+ QtWidgets .QSizePolicy .Policy .Expanding ,
917+ QtWidgets .QSizePolicy .Policy .Ignored ,
920918 ))
921919 self ._message_action = self .addWidget (message_label )
922920 self ._toolitems = {}
0 commit comments