Update version number and add ConnectionClosedError exception

AresJef · Dec 29, 2023 · efd3747 · efd3747
1 parent 6b5ce80
commit efd3747
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = aselenium
-version = 0.1.12.7
+version = 0.1.12.8
 author = Jiefu Chen
 author_email = keppa1991@163.com
 description = Asyncronous implementation of web browser automation for Python.

diff --git a/src/aselenium/__init__.py b/src/aselenium/__init__.py
@@ -111,6 +111,7 @@
     WebDriverError,
     WebDriverTimeoutError,
     WebdriverNotFoundError,
+    ConnectionClosedError,
     InternetDisconnectedError,
     InvalidValueError,
     InvalidArgumentError,
@@ -209,7 +210,7 @@
     "ServiceError", "ServiceExecutableNotFoundError", "ServiceStartError", "ServiceStopError",
     "ServiceSocketError", "ServiceProcessError", "ServiceTimeoutError",
     # . webdriver
-    "WebDriverError", "WebDriverTimeoutError", "WebdriverNotFoundError", "InternetDisconnectedError",
+    "WebDriverError", "WebDriverTimeoutError", "WebdriverNotFoundError", "ConnectionClosedError", "InternetDisconnectedError",
     "InvalidValueError", "InvalidArgumentError", "InvalidMethodError", "InvalidRectValueError",
     "InvalidResponseError", "InvalidExtensionError", "UnknownMethodError", "SessionError",
     "SessionClientError", "InvalidSessionError", "IncompatibleWebdriverError", "SessionDataError",

diff --git a/src/aselenium/errors.py b/src/aselenium/errors.py
@@ -204,14 +204,14 @@ class WebdriverNotFoundError(WebDriverError):
     """Exception raised when target not found."""
 
 
-class InternetDisconnectedError(WebDriverError):
-    """Exception raised when internet disconnected."""
-
-
 class ConnectionClosedError(WebDriverError):
     """Exception raised when connection closed."""
 
 
+class InternetDisconnectedError(WebDriverError):
+    """Exception raised when internet disconnected."""
+
+
 # . Inavlid value error
 class InvalidValueError(WebDriverError, AseleniumInvalidValueError):
     """Base exception class for all invalid value errors."""
@@ -779,10 +779,10 @@ def webdriver_error_handler(res: dict[str, Any]) -> None:
     if error is UnknownError:
         if ErrorCode.FAILED_TO_CHANGE_WINDOW_STATE in message:
             error = ChangeWindowStateError
-        elif ErrorCode.INTERNET_DISCONNECTED in message:
-            error = InternetDisconnectedError
         elif ErrorCode.CONNECTION_CLOSED in message:
             error = ConnectionClosedError
+        elif ErrorCode.INTERNET_DISCONNECTED in message:
+            error = InternetDisconnectedError
     elif error is InvalidSessionError:
         if INCOMPATIBLE_DRIVER_PATTERN.search(message) is not None:
             error = IncompatibleWebdriverError