fix(sql_lab): Add custom timestamp type for literal casting for presto timestamps

superset/db_engine_specs/presto.py

@@ -48,6 +48,8 @@
     Map,
     Row,
     TinyInteger,
+    TimeStamp,
+    Date
 )
 from superset.result_set import destringify
 from superset.sql_parse import ParsedQuery
@@ -911,10 +913,19 @@ def where_latest_partition(  # pylint: disable=too-many-arguments
         if values is None:
             return None
 
-        column_names = {column.get("name") for column in columns or []}
+        column_type_by_name = { 
+            column.get("name") : column.get('type') for column in columns or []
+        }
+
         for col_name, value in zip(col_names, values):
-            if col_name in column_names:
-                query = query.where(Column(col_name) == value)
+            if col_name in column_type_by_name:
+                col_type = column_type_by_name.get(col_name)
+                if col_type == 'TIMESTAMP':
+                    query = query.where(Column(col_name, TimeStamp()) == value)
+                elif col_type == 'DATE':
+                    query = query.where(Column(col_name, Date()) == value)
+                else:
+                    query = query.where(Column(col_name) == value)
         return query
 
     @classmethod

superset/models/sql_types/presto_sql_types.py

@@ -16,7 +16,8 @@
 # under the License.
 from typing import Any, Dict, List, Optional, Type
 
-from sqlalchemy.sql.sqltypes import Integer
+from sqlalchemy import types
+from sqlalchemy.sql.sqltypes import Integer, TIMESTAMP, DATE
 from sqlalchemy.sql.type_api import TypeEngine
 from sqlalchemy.sql.visitors import Visitable
 
@@ -91,3 +92,34 @@ def python_type(self) -> Optional[Type[Any]]:
     @classmethod
     def _compiler_dispatch(cls, _visitor: Visitable, **_kw: Any) -> str:
         return "ROW"
+
+class TimeStamp(types.TypeDecorator):
+    """
+    A type to extend functionality of timestamp data type.
+    """
+
+    impl = TIMESTAMP
+
+    @classmethod
+    def process_bind_param(cls, value, dialect):
+        """
+        Used for in-line rendering of TIMESTAMP data type
+        as Presto does not support automatic casting.
+        """
+        return "TIMESTAMP '%s'" % value
+
+class Date(types.TypeDecorator):
+    """
+    A type to extend functionality of date data type.
+    """
+
+    impl = DATE
+
+    @classmethod
+    def process_bind_param(cls, value, dialect):
+        """
+        Used for in-line rendering of DATE data type
+        as Presto does not support automatic casting.
+        """
+        return "DATE '%s'" % value
+