sqlglot.dialects.drill

 1from __future__ import annotations
 2
 3
 4from sqlglot import tokens
 5from sqlglot.dialects.dialect import Dialect
 6from sqlglot.generators.drill import DrillGenerator
 7from sqlglot.parsers.drill import DrillParser
 8
 9
10class Drill(Dialect):
11    NORMALIZE_FUNCTIONS: bool | str = False
12    PRESERVE_ORIGINAL_NAMES = True
13    NULL_ORDERING = "nulls_are_last"
14    DATE_FORMAT = "'yyyy-MM-dd'"
15    DATEINT_FORMAT = "'yyyyMMdd'"
16    TIME_FORMAT = "'yyyy-MM-dd HH:mm:ss'"
17    SUPPORTS_USER_DEFINED_TYPES = False
18    TYPED_DIVISION = True
19    CONCAT_COALESCE = True
20    CONCAT_WS_COALESCE = True
21
22    TIME_MAPPING = {
23        "y": "%Y",
24        "Y": "%Y",
25        "YYYY": "%Y",
26        "yyyy": "%Y",
27        "YY": "%y",
28        "yy": "%y",
29        "MMMM": "%B",
30        "MMM": "%b",
31        "MM": "%m",
32        "M": "%-m",
33        "dd": "%d",
34        "d": "%-d",
35        "HH": "%H",
36        "H": "%-H",
37        "hh": "%I",
38        "h": "%-I",
39        "mm": "%M",
40        "m": "%-M",
41        "ss": "%S",
42        "s": "%-S",
43        "SSSSSS": "%f",
44        "a": "%p",
45        "DD": "%j",
46        "D": "%-j",
47        "E": "%a",
48        "EE": "%a",
49        "EEE": "%a",
50        "EEEE": "%A",
51        "''T''": "T",
52    }
53
54    class Tokenizer(tokens.Tokenizer):
55        IDENTIFIERS = ["`"]
56        STRING_ESCAPES = ["\\"]
57
58        KEYWORDS = tokens.Tokenizer.KEYWORDS.copy()
59        KEYWORDS.pop("/*+")
60
61    Parser = DrillParser
62
63    Generator = DrillGenerator