🐛 import MySQLConnectionAbstract instead of concrete implementations

Author: techouse

mysql_to_sqlite3/transporter.py

@@ -13,8 +13,8 @@
 
 import mysql.connector
 import typing_extensions as tx
-from mysql.connector import MySQLConnection, errorcode
-from mysql.connector.connection_cext import CMySQLConnection
+from mysql.connector import errorcode
+from mysql.connector.abstracts import MySQLConnectionAbstract
 from mysql.connector.types import ToPythonOutputTypes
 from tqdm import tqdm, trange
 
@@ -124,7 +124,7 @@ def __init__(self, **kwargs: tx.Unpack[MySQLtoSQLiteParams]) -> None:
                 port=self._mysql_port,
                 ssl_disabled=self._mysql_ssl_disabled,
             )
-            if isinstance(_mysql_connection, (MySQLConnection, CMySQLConnection)):
+            if isinstance(_mysql_connection, MySQLConnectionAbstract):
                 self._mysql = _mysql_connection
             else:
                 raise ConnectionError("Unable to connect to MySQL")

mysql_to_sqlite3/types.py

@@ -5,8 +5,7 @@
 from sqlite3 import Connection, Cursor
 
 import typing_extensions as tx
-from mysql.connector import MySQLConnection
-from mysql.connector.connection_cext import CMySQLConnection
+from mysql.connector.abstracts import MySQLConnectionAbstract
 from mysql.connector.cursor import MySQLCursorDict, MySQLCursorPrepared, MySQLCursorRaw
 
 
@@ -46,7 +45,7 @@ class MySQLtoSQLiteAttributes:
     _json_as_text: bool
     _limit_rows: int
     _logger: Logger
-    _mysql: t.Union[MySQLConnection, CMySQLConnection]
+    _mysql: MySQLConnectionAbstract
     _mysql_cur: MySQLCursorRaw
     _mysql_cur_dict: MySQLCursorDict
     _mysql_cur_prepared: MySQLCursorPrepared