Mercurial Repositories > eric / file diff

diff -r ad7a6d699a0d -r 287a3ae95e00 src/eric7/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py
--- a/src/eric7/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py	Sun Jan 07 11:42:41 2024 +0100
+++ b/src/eric7/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py	Sun Jan 07 12:40:00 2024 +0100
@@ -7,6 +7,7 @@
 Module implementing the downloader for GreaseMonkey scripts.
 """
 
+import enum
 import os
 import pathlib
 
@@ -17,6 +18,15 @@
 from eric7.WebBrowser.WebBrowserWindow import WebBrowserWindow
 
 
+class GreaseMonkeyDownloadType(enum.Enum):
+    """
+    Class defining the download types.
+    """
+
+    MainScript = 1
+    RequireScript = 2
+
+
 class GreaseMonkeyDownloader(QObject):
     """
     Class implementing the downloader for GreaseMonkey scripts.
@@ -29,10 +39,6 @@
     finished = pyqtSignal(str)
     error = pyqtSignal()
 
-    # TODO: change this to an enum
-    DownloadMainScript = 1
-    DownloadRequireScript = 2
-
     def __init__(self, url, manager, mode):
         """
         Constructor
@@ -42,14 +48,14 @@
         @param manager reference to the GreaseMonkey manager
         @type GreaseMonkeyManager
         @param mode download mode
-        @type int (one of DownloadMainScript, DownloadRequireScript)
+        @type GreaseMonkeyDownloadType
         """
         super().__init__()
 
         self.__manager = manager
 
         self.__reply = WebBrowserWindow.networkManager().get(QNetworkRequest(url))
-        if mode == GreaseMonkeyDownloader.DownloadMainScript:
+        if mode == GreaseMonkeyDownloadType.MainScript:
             self.__reply.finished.connect(self.__scriptDownloaded)
         else:
             self.__reply.finished.connect(self.__requireDownloaded)