Migrate to class based dbapi

Author: LuckySting

test_dbapi/conftest.py

@@ -5,6 +5,6 @@
 
 @pytest.fixture(scope="module")
 def connection():
-    conn = dbapi.connect(host="localhost", port="2136", database="/local")
+    conn = dbapi.YdbDBApi().connect(host="localhost", port="2136", database="/local")
     yield conn
     conn.close()

test_dbapi/test_dbapi.py

@@ -69,7 +69,7 @@ def test_cursor_raw_query(connection):
 
 def test_errors(connection):
     with pytest.raises(dbapi.InterfaceError):
-        dbapi.connect("localhost:2136", database="/local666")
+        dbapi.YdbDBApi().connect("localhost:2136", database="/local666")
 
     cur = connection.cursor()
 

ydb_sqlalchemy/dbapi/__init__.py

@@ -13,25 +13,27 @@
     NotSupportedError,
 )
 
-apilevel = "1.0"
+class YdbDBApi:
+    def __init__(self):
+        self.paramstyle = "pyformat"
+        self.threadsafety = 0
+        self.apilevel = "1.0"
+        self._init_dbapi_attributes()
+    
+    def _init_dbapi_attributes(self):
+        for name, value in {
+            "Warning": Warning,
+            "Error": Error,
+            "InterfaceError": InterfaceError,
+            "DatabaseError": DatabaseError,
+            "DataError": DataError,
+            "OperationalError": OperationalError,
+            "IntegrityError": IntegrityError,
+            "InternalError": InternalError,
+            "ProgrammingError": ProgrammingError,
+            "NotSupportedError": NotSupportedError,
+        }.items():
+            setattr(self, name, value)
 
-threadsafety = 0
-
-paramstyle = "pyformat"
-
-errors = (
-    Warning,
-    Error,
-    InterfaceError,
-    DatabaseError,
-    DataError,
-    OperationalError,
-    IntegrityError,
-    InternalError,
-    ProgrammingError,
-    NotSupportedError,
-)
-
-
-def connect(*args, **kwargs):
-    return Connection(*args, **kwargs)
+    def connect(self, *args, **kwargs):
+        return Connection(*args, **kwargs)

ydb_sqlalchemy/sqlalchemy/__init__.py

@@ -542,7 +542,7 @@ class YqlDialect(StrCompileDialect):
 
     @classmethod
     def import_dbapi(cls: Any):
-        return dbapi
+        return dbapi.YdbDBApi()
 
     def _describe_table(self, connection, table_name, schema=None):
         if schema is not None: