some irisasync support

Author: daimor

setup.cfg

@@ -33,6 +33,8 @@ addopts= --tb native -v -r fxX --maxfail=25 -p no:warnings
 
 [db]
 default=iris://_SYSTEM:SYS@localhost:1972/USER
+iris=iris://_SYSTEM:SYS@localhost:1972/USER
+irisasync=iris+irisasync://_SYSTEM:SYS@localhost:1972/USER
 irisemb=iris+emb:///
 sqlite=sqlite:///:memory:
 

setup.py

@@ -8,6 +8,7 @@
         "sqlalchemy.dialects": [
             "iris = sqlalchemy_iris.iris:IRISDialect_iris",
             "iris.emb = sqlalchemy_iris.embedded:IRISDialect_emb",
+            "iris.irisasync = sqlalchemy_iris.irisasync:IRISDialect_irisasync",
         ]
     },
 )

sqlalchemy_iris/__init__.py

@@ -31,6 +31,7 @@
 
 _registry.register("iris.iris", "sqlalchemy_iris.iris", "IRISDialect_iris")
 _registry.register("iris.emb", "sqlalchemy_iris.embedded", "IRISDialect_emb")
+_registry.register("iris.irisasync", "sqlalchemy_iris.irisasync", "IRISDialect_irisasync")
 
 __all__ = [
     "BIGINT",

sqlalchemy_iris/base.py

@@ -884,8 +884,12 @@ def __init__(self, **kwargs):
 
     def _get_server_version_info(self, connection):
         server_version = connection.connection._connection_info._server_version
-        server_version = server_version[server_version.find("Version") + 8:].split(" ")[0].split(".")
-        return tuple([int(''.join(filter(str.isdigit, v))) for v in server_version])
+        server_version = (
+            server_version[server_version.find("Version") + 8 :]
+            .split(" ")[0]
+            .split(".")
+        )
+        return tuple([int("".join(filter(str.isdigit, v))) for v in server_version])
 
     _isolation_lookup = set(
         [
@@ -906,17 +910,23 @@ def on_connect(conn):
                 super_(conn)
 
             if self.embedded:
-                self.supports_vectors = conn.iris.cls("%SYSTEM.License").GetFeature(28) == 1
+                self.supports_vectors = (
+                    conn.iris.cls("%SYSTEM.License").GetFeature(28) == 1
+                )
             else:
                 try:
                     iris = IRISNative.createIRIS(conn)
-                    self.supports_vectors = iris.classMethodBoolean("%SYSTEM.License", "GetFeature", 28)
-                except: # noqa
+                    self.supports_vectors = iris.classMethodBoolean(
+                        "%SYSTEM.License", "GetFeature", 28
+                    )
+                except:  # noqa
                     self.supports_vectors = False
             if self.supports_vectors:
                 with conn.cursor() as cursor:
                     # Distance or similarity
-                    cursor.execute("select vector_cosine(to_vector('1'), to_vector('1'))")
+                    cursor.execute(
+                        "select vector_cosine(to_vector('1'), to_vector('1'))"
+                    )
                     self.vector_cosine_similarity = cursor.fetchone()[0] == 0
 
             self._dictionary_access = False
@@ -1511,9 +1521,9 @@ def get_multi_foreign_keys(
                 table_fkey[rfknm] = fkey = {
                     "name": rfknm,
                     "constrained_columns": [],
-                    "referred_schema": rschema
-                    if rschema != self.default_schema_name
-                    else None,
+                    "referred_schema": (
+                        rschema if rschema != self.default_schema_name else None
+                    ),
                     "referred_table": rtbl,
                     "referred_columns": [],
                     "options": {},

sqlalchemy_iris/irisasync.py

@@ -0,0 +1,17 @@
+from .base import IRISDialect
+
+
+class IRISDialect_irisasync(IRISDialect):
+    driver = "irisasync"
+
+    is_async = True
+    supports_statement_cache = True
+
+    @classmethod
+    def import_dbapi(cls):
+        import intersystems_iris.dbapi._DBAPI as dbapi
+
+        return dbapi
+
+
+dialect = IRISDialect_irisasync

tests/conftest.py

@@ -8,6 +8,9 @@
 
 registry.register("iris.iris", "sqlalchemy_iris.iris", "IRISDialect_iris")
 registry.register("iris.emb", "sqlalchemy_iris.embedded", "IRISDialect_emb")
+registry.register(
+    "iris.irisasync", "sqlalchemy_iris.irisasync", "IRISDialect_irisasync"
+)
 
 pytest.register_assert_rewrite("sqlalchemy.testing.assertions")