Fix: error handling (#765)

Author: karenc-bq

.github/workflows/draft_release.yml

@@ -59,6 +59,7 @@ jobs:
         with:
           repository-url: https://test.pypi.org/legacy/
       - name: 'Create a Draft Release'
+        if: always()
         uses: ncipollo/release-action@v1
         with:
           draft: true

aws_advanced_python_wrapper/utils/mysql_exception_handler.py

@@ -14,7 +14,7 @@
 
 from typing import List, Optional
 
-from mysql.connector import InterfaceError, OperationalError
+from mysql.connector import DatabaseError, InterfaceError
 
 from aws_advanced_python_wrapper.errors import QueryTimeoutError
 from aws_advanced_python_wrapper.exception_handling import ExceptionHandler
@@ -52,7 +52,7 @@ def is_network_exception(self, error: Optional[Exception] = None, sql_state: Opt
             # e.g. 2013 (HY000): Lost connection to MySQL server during query
             return True
 
-        if isinstance(error, OperationalError):
+        if isinstance(error, DatabaseError):
             if error.errno in self._NETWORK_ERRORS:
                 return True
             if error.msg is not None and self._UNAVAILABLE_CONNECTION in error.msg: