Add types to field expressions

svenkreiss · Sep 26, 2021 · b87e861 · b87e861
1 parent b44adf6
commit b87e861
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pysparkling/sql/expressions/fields.py b/pysparkling/sql/expressions/fields.py
@@ -20,6 +20,9 @@ def output_fields(self, schema):
     def args(self):
         return (self.field,)
 
+    def data_type(self, schema):
+        return schema[find_position_in_schema(schema, self.field)].dataType
+
 
 def find_position_in_schema(schema, expr):
     if isinstance(expr, str):