Mercurial Repositories > eric / file diff

--- a/src/eric7/QScintilla/MarkupProviders/ImageMarkupDialog.py	Fri Jan 05 16:04:03 2024 +0100
+++ b/src/eric7/QScintilla/MarkupProviders/ImageMarkupDialog.py	Sat Jan 06 15:21:02 2024 +0100
@@ -8,6 +8,7 @@
 """
 
 import contextlib
+import enum
 
 from PyQt6.QtCore import QSize, pyqtSlot
 from PyQt6.QtGui import QImage, QImageReader
@@ -18,16 +19,21 @@
 from .Ui_ImageMarkupDialog import Ui_ImageMarkupDialog
 
 
+class ImageMarkupMode(enum.Enum):
+    """
+    Class defining the image markup modes.
+    """
+
+    Html = 0
+    MarkDown = 1
+    Rest = 2
+
+
 class ImageMarkupDialog(QDialog, Ui_ImageMarkupDialog):
     """
     Class implementing a dialog to enter data for an image markup.
     """
 
-    # TODO: change this to an enum
-    HtmlMode = 0
-    MarkDownMode = 1
-    RestMode = 2
-
     def __init__(self, mode, parent=None):
         """
         Constructor
@@ -40,12 +46,12 @@
         super().__init__(parent)
         self.setupUi(self)
 
-        if mode == ImageMarkupDialog.MarkDownMode:
+        if mode == ImageMarkupMode.MarkDown:
             self.sizeCheckBox.setEnabled(False)
             self.aspectRatioCheckBox.setEnabled(False)
             self.widthSpinBox.setEnabled(False)
             self.heightSpinBox.setEnabled(False)
-        elif mode == ImageMarkupDialog.RestMode:
+        elif mode == ImageMarkupMode.Rest:
             self.titleEdit.setEnabled(False)
 
         self.__mode = mode
@@ -105,7 +111,7 @@
         Private slot to set the state of the OK button.
         """
         enable = bool(self.imagePicker.text())
-        if self.__mode == ImageMarkupDialog.MarkDownMode:
+        if self.__mode == ImageMarkupMode.MarkDown:
             enable = enable and bool(self.altTextEdit.text())
 
         self.buttonBox.button(QDialogButtonBox.StandardButton.Ok).setEnabled(enable)