Add visit_try_cast to sqlalchemy compiler

Author: dwolfeu

tests/unit/sqlalchemy/test_compiler.py

@@ -151,3 +151,14 @@ def test_ignore_nulls(dialect, function, element):
     assert str(query) == \
            f'SELECT {function}("table".id) OVER (PARTITION BY "table".name) AS window ' \
            f'\nFROM "table"'
+
+
+@pytest.mark.skipif(
+    sqlalchemy_version() < "2.0",
+    reason="ImportError: cannot import name 'try_cast' from 'sqlalchemy'"
+)
+def test_try_cast(dialect):
+    from sqlalchemy import try_cast
+    statement = select(try_cast(table_without_catalog.c.id, String))
+    query = statement.compile(dialect=dialect)
+    assert str(query) == 'SELECT try_cast("table".id as VARCHAR) AS id \nFROM "table"'

trino/sqlalchemy/compiler.py

@@ -175,6 +175,9 @@ def compile_ignore_nulls(element, compiler, **kwargs):
             compiled += ' IGNORE NULLS'
         return compiled
 
+    def visit_try_cast(self, element, **kw):
+        return f"try_cast({self.process(element.clause, **kw)} as {self.process(element.typeclause, **kw)})"
+
 
 class TrinoDDLCompiler(compiler.DDLCompiler):
     pass