sqlglot.typing.duckdb

 1from __future__ import annotations
 2
 3from sqlglot import exp
 4from sqlglot.typing import EXPRESSION_METADATA
 5
 6EXPRESSION_METADATA = {
 7    **EXPRESSION_METADATA,
 8    **{
 9        expr_type: {"returns": exp.DType.BIGINT}
10        for expr_type in {
11            exp.BitLength,
12            exp.DateDiff,
13            exp.Day,
14            exp.DayOfMonth,
15            exp.DayOfWeek,
16            exp.DayOfWeekIso,
17            exp.DayOfYear,
18            exp.Extract,
19            exp.Hour,
20            exp.Length,
21            exp.Minute,
22            exp.Month,
23            exp.Quarter,
24            exp.Second,
25            exp.Week,
26            exp.Year,
27        }
28    },
29    **{
30        expr_type: {"returns": exp.DType.INT128}
31        for expr_type in {
32            exp.CountIf,
33            exp.Factorial,
34        }
35    },
36    **{
37        expr_type: {"returns": exp.DType.DOUBLE}
38        for expr_type in {
39            exp.Atan2,
40            exp.JarowinklerSimilarity,
41            exp.TimeToUnix,
42        }
43    },
44    **{
45        expr_type: {"returns": exp.DType.VARCHAR}
46        for expr_type in {
47            exp.Format,
48            exp.Reverse,
49        }
50    },
51    exp.DateBin: {"annotator": lambda self, e: self._annotate_by_args(e, "expression")},
52    exp.Localtimestamp: {"returns": exp.DType.TIMESTAMP},
53    exp.ToDays: {"returns": exp.DType.INTERVAL},
54    exp.TimeFromParts: {"returns": exp.DType.TIME},
55}