sqlglot.typing.spark

 1from __future__ import annotations
 2
 3from sqlglot import exp
 4from sqlglot.typing.spark2 import EXPRESSION_METADATA
 5
 6
 7EXPRESSION_METADATA = {
 8    **EXPRESSION_METADATA,
 9    **{
10        exp_type: {"returns": exp.DType.DOUBLE}
11        for exp_type in {
12            exp.Sec,
13        }
14    },
15    **{
16        exp_type: {"returns": exp.DType.INT}
17        for exp_type in {
18            exp.ArraySize,
19        }
20    },
21    **{
22        exp_type: {"returns": exp.DType.VARCHAR}
23        for exp_type in {
24            exp.Collation,
25            exp.CurrentTimezone,
26            exp.Randstr,
27        }
28    },
29    **{
30        exp_type: {"annotator": lambda self, e: self._annotate_by_args(e, "this")}
31        for exp_type in {
32            exp.ArrayCompact,
33            exp.ArrayInsert,
34            exp.BitwiseAndAgg,
35            exp.BitwiseOrAgg,
36            exp.BitwiseXorAgg,
37            exp.Overlay,
38        }
39    },
40    exp.BitmapCount: {"returns": exp.DType.BIGINT},
41    exp.Localtimestamp: {"returns": exp.DType.TIMESTAMPNTZ},
42    exp.ToBinary: {"returns": exp.DType.BINARY},
43    exp.DateFromUnixDate: {"returns": exp.DType.DATE},
44}