feat: using SELECT version() to get server_version

Author: dungdm93

sqlalchemy_trino/dialect.py

@@ -1,3 +1,4 @@
+import re
 from textwrap import dedent
 from typing import *
 
@@ -6,7 +7,6 @@
 from sqlalchemy.engine.default import DefaultDialect, DefaultExecutionContext
 from sqlalchemy.engine.url import URL
 from trino.auth import BasicAuthentication
-import re
 
 from . import compiler
 from . import datatype
@@ -60,7 +60,7 @@ class TrinoDialect(DefaultDialect):
     supports_multivalues_insert = True
 
     # Version parser
-    __version_pattern = re.compile('(\d+).*')
+    __version_pattern = re.compile(r'(\d+).*')
 
     @classmethod
     def dbapi(cls):
@@ -262,14 +262,10 @@ def has_sequence(self, connection: Connection,
         return False
 
     def _get_server_version_info(self, connection: Connection) -> Tuple[int, ...]:
-        query = dedent('''
-            SELECT *
-            FROM system.runtime.nodes
-            WHERE coordinator = true AND state = 'active'
-        ''').strip()
-        res = connection.execute(sql.text(query)).first()
-        parsed_version = self.__version_pattern.match(res.node_version)
-        version = int(parsed_version.group(1)) if parsed_version else 0
+        query = 'SELECT version()'
+        res = connection.execute(sql.text(query)).scalar()
+        match = self.__version_pattern.match(res)
+        version = int(match.group(1)) if match else 0
         return tuple([version])
 
     def _get_default_schema_name(self, connection: Connection) -> Optional[str]: