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
21    TIME_MAPPING = {
22        "y": "%Y",
23        "Y": "%Y",
24        "YYYY": "%Y",
25        "yyyy": "%Y",
26        "YY": "%y",
27        "yy": "%y",
28        "MMMM": "%B",
29        "MMM": "%b",
30        "MM": "%m",
31        "M": "%-m",
32        "dd": "%d",
33        "d": "%-d",
34        "HH": "%H",
35        "H": "%-H",
36        "hh": "%I",
37        "h": "%-I",
38        "mm": "%M",
39        "m": "%-M",
40        "ss": "%S",
41        "s": "%-S",
42        "SSSSSS": "%f",
43        "a": "%p",
44        "DD": "%j",
45        "D": "%-j",
46        "E": "%a",
47        "EE": "%a",
48        "EEE": "%a",
49        "EEEE": "%A",
50        "''T''": "T",
51    }
52
53    class Tokenizer(tokens.Tokenizer):
54        IDENTIFIERS = ["`"]
55        STRING_ESCAPES = ["\\"]
56
57        KEYWORDS = tokens.Tokenizer.KEYWORDS.copy()
58        KEYWORDS.pop("/*+")
59
60    Parser = DrillParser
61
62    Generator = DrillGenerator