Merge pull request #6 from cccs-tom/master

Handle the Trino node_version being non-numeric

Author: dungdm93

sqlalchemy_trino/dialect.py

@@ -6,6 +6,7 @@
 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
@@ -58,6 +59,9 @@ class TrinoDialect(DefaultDialect):
     supports_empty_insert = False
     supports_multivalues_insert = True
 
+    # Version parser
+    __version_pattern = re.compile('(\d+).*')
+
     @classmethod
     def dbapi(cls):
         """
@@ -264,7 +268,8 @@ def _get_server_version_info(self, connection: Connection) -> Tuple[int, ...]:
             WHERE coordinator = true AND state = 'active'
         ''').strip()
         res = connection.execute(sql.text(query)).first()
-        version = int(res.node_version)
+        parsed_version = self.__version_pattern.match(res.node_version)
+        version = int(parsed_version.group(1)) if parsed_version else 0
         return tuple([version])
 
     def _get_default_schema_name(self, connection: Connection) -> Optional[str]: