Return None when failed to get server version in SQLAlchemy

This is the same approach as pymssql.

Author: ebyhr

trino/sqlalchemy/dialect.py

@@ -17,11 +17,14 @@
 from sqlalchemy.engine.default import DefaultDialect, DefaultExecutionContext
 from sqlalchemy.engine.url import URL
 
-from trino import dbapi as trino_dbapi
+from trino import dbapi as trino_dbapi, logging
 from trino.auth import BasicAuthentication
 from trino.dbapi import Cursor
+from trino.exceptions import TrinoUserError
 from trino.sqlalchemy import compiler, datatype, error
 
+logger = logging.get_logger(__name__)
+
 
 class TrinoDialect(DefaultDialect):
     name = "trino"
@@ -266,11 +269,15 @@ def has_sequence(self, connection: Connection, sequence_name: str, schema: str =
         """Trino has no support for sequence. Returns False indicate that given sequence does not exists."""
         return False
 
-    def _get_server_version_info(self, connection: Connection) -> Tuple[int, ...]:
+    def _get_server_version_info(self, connection: Connection) -> Any:
         query = "SELECT version()"
-        res = connection.execute(sql.text(query))
-        version = res.scalar()
-        return tuple([version])
+        try:
+            res = connection.execute(sql.text(query))
+            version = res.scalar()
+            return tuple([version])
+        except TrinoUserError as e:
+            logger.debug(f"Failed to get server version: {e.message}")
+            return None
 
     def _get_default_schema_name(self, connection: Connection) -> Optional[str]:
         dbapi_connection: trino_dbapi.Connection = connection.connection