Update sqlalchemy_presto.py add try_cast

apache · Jan 21, 2025 · e146aa5 · e146aa5
1 parent b0f3c00
commit e146aa5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/pyhive/sqlalchemy_presto.py b/python/pyhive/sqlalchemy_presto.py
@@ -55,6 +55,8 @@ class PrestoCompiler(SQLCompiler):
     def visit_char_length_func(self, fn, **kw):
         return 'length{}'.format(self.function_argspec(fn, **kw))
 
+    def visit_try_cast(self, element, **kw):
+        return f"try_cast({self.process(element.clause, **kw)} as {self.process(element.typeclause, **kw)})"
 
 class PrestoTypeCompiler(compiler.GenericTypeCompiler):
     def visit_CLOB(self, type_, **kw):