chore: disable flake8 warnings for special imports

Author: karenc-bq

aws_advanced_python_wrapper/mysql_driver_dialect.py

@@ -23,30 +23,30 @@
 from concurrent.futures import Executor, ThreadPoolExecutor, TimeoutError
 from inspect import signature
 
+from aws_advanced_python_wrapper.driver_dialect import DriverDialect
+from aws_advanced_python_wrapper.driver_dialect_codes import DriverDialectCodes
+from aws_advanced_python_wrapper.errors import UnsupportedOperationError
+from aws_advanced_python_wrapper.utils.decorators import timeout
+from aws_advanced_python_wrapper.utils.messages import Messages
+from aws_advanced_python_wrapper.utils.properties import (Properties,
+                                                          PropertiesUtils,
+                                                          WrapperProperties)
+
 CMYSQL_ENABLED = False
 
-from mysql.connector import MySQLConnection
-from mysql.connector.cursor import MySQLCursor
+from mysql.connector import MySQLConnection  # noqa: E402
+from mysql.connector.cursor import MySQLCursor  # noqa: E402
 
 try:
-    from mysql.connector import CMySQLConnection
-    from mysql.connector.cursor_cext import CMySQLCursor
+    from mysql.connector import CMySQLConnection  # noqa: E402
+    from mysql.connector.cursor_cext import CMySQLCursor  # noqa: E402
 
     CMYSQL_ENABLED = True
 
 except ImportError:
     # Do nothing
     pass
 
-from aws_advanced_python_wrapper.driver_dialect import DriverDialect
-from aws_advanced_python_wrapper.driver_dialect_codes import DriverDialectCodes
-from aws_advanced_python_wrapper.errors import UnsupportedOperationError
-from aws_advanced_python_wrapper.utils.decorators import timeout
-from aws_advanced_python_wrapper.utils.messages import Messages
-from aws_advanced_python_wrapper.utils.properties import (Properties,
-                                                          PropertiesUtils,
-                                                          WrapperProperties)
-
 
 class MySQLDriverDialect(DriverDialect):
     _driver_name = "MySQL Connector Python"
@@ -94,7 +94,7 @@ def is_closed(self, conn: Connection) -> bool:
             if self.can_execute_query(conn):
                 socket_timeout = WrapperProperties.SOCKET_TIMEOUT_SEC.get_float(self._props)
                 timeout_sec = socket_timeout if socket_timeout > 0 else MySQLDriverDialect.IS_CLOSED_TIMEOUT_SEC
-                is_connected_with_timeout = timeout(MySQLDriverDialect._executor, timeout_sec)(conn.is_connected) # type: ignore
+                is_connected_with_timeout = timeout(MySQLDriverDialect._executor, timeout_sec)(conn.is_connected)  # type: ignore
 
                 try:
                     return not is_connected_with_timeout()
@@ -106,14 +106,14 @@ def is_closed(self, conn: Connection) -> bool:
 
     def get_autocommit(self, conn: Connection) -> bool:
         if MySQLDriverDialect._is_mysql_connection(conn):
-            return conn.autocommit # type: ignore
+            return conn.autocommit  # type: ignore
 
         raise UnsupportedOperationError(
             Messages.get_formatted("DriverDialect.UnsupportedOperationError", self._driver_name, "autocommit"))
 
     def set_autocommit(self, conn: Connection, autocommit: bool):
         if MySQLDriverDialect._is_mysql_connection(conn):
-            conn.autocommit = autocommit # type: ignore
+            conn.autocommit = autocommit  # type: ignore
             return
 
         raise UnsupportedOperationError(
@@ -132,13 +132,13 @@ def abort_connection(self, conn: Connection):
 
     def can_execute_query(self, conn: Connection) -> bool:
         if MySQLDriverDialect._is_mysql_connection(conn):
-            if conn.unread_result: # type: ignore
-                return conn.can_consume_results # type: ignore
+            if conn.unread_result:  # type: ignore
+                return conn.can_consume_results  # type: ignore
         return True
 
     def is_in_transaction(self, conn: Connection) -> bool:
         if MySQLDriverDialect._is_mysql_connection(conn):
-            return bool(conn.in_transaction) # type: ignore
+            return bool(conn.in_transaction)  # type: ignore
 
         raise UnsupportedOperationError(
             Messages.get_formatted("DriverDialect.UnsupportedOperationError", self._driver_name,
@@ -176,7 +176,7 @@ def get_connection_from_obj(self, obj: object) -> Any:
 
     def transfer_session_state(self, from_conn: Connection, to_conn: Connection):
         if MySQLDriverDialect._is_mysql_connection(from_conn) and MySQLDriverDialect._is_mysql_connection(to_conn):
-            to_conn.autocommit = from_conn.autocommit # type: ignore
+            to_conn.autocommit = from_conn.autocommit  # type: ignore
 
     def ping(self, conn: Connection) -> bool:
         return not self.is_closed(conn)