sqlglot.dialects.singlestore

 8class SingleStore(MySQL):
 9    SUPPORTS_ORDER_BY_ALL = True
10
11    MYSQL_INVERSE_TIME_MAPPING = MySQL.INVERSE_TIME_MAPPING
12    MYSQL_INVERSE_TIME_TRIE = MySQL.INVERSE_TIME_TRIE
13    CAST_TO_TIME6 = staticmethod(cast_to_time6)
14
15    TIME_MAPPING: dict[str, str] = {
16        "D": "%u",  # Day of week (1-7)
17        "DD": "%d",  # day of month (01-31)
18        "DY": "%a",  # abbreviated name of day
19        "HH": "%I",  # Hour of day (01-12)
20        "HH12": "%I",  # alias for HH
21        "HH24": "%H",  # Hour of day (00-23)
22        "MI": "%M",  # Minute (00-59)
23        "MM": "%m",  # Month (01-12; January = 01)
24        "MON": "%b",  # Abbreviated name of month
25        "MONTH": "%B",  # Name of month
26        "SS": "%S",  # Second (00-59)
27        "RR": "%y",  # 15
28        "YY": "%y",  # 15
29        "YYYY": "%Y",  # 2015
30        "FF6": "%f",  # only 6 digits are supported in python formats
31    }
32
33    VECTOR_TYPE_ALIASES = {
34        "I8": "TINYINT",
35        "I16": "SMALLINT",
36        "I32": "INT",
37        "I64": "BIGINT",
38        "F32": "FLOAT",
39        "F64": "DOUBLE",
40    }
41
42    INVERSE_VECTOR_TYPE_ALIASES = {v: k for k, v in VECTOR_TYPE_ALIASES.items()}
43
44    class Tokenizer(MySQL.Tokenizer):
45        BYTE_STRINGS = [("e'", "'"), ("E'", "'")]
46
47        KEYWORDS = {
48            **MySQL.Tokenizer.KEYWORDS,
49            "BSON": TokenType.JSONB,
50            "GEOGRAPHYPOINT": TokenType.GEOGRAPHYPOINT,
51            "TIMESTAMP": TokenType.TIMESTAMP,
52            "UTC_DATE": TokenType.UTC_DATE,
53            "UTC_TIME": TokenType.UTC_TIME,
54            "UTC_TIMESTAMP": TokenType.UTC_TIMESTAMP,
55            ":>": TokenType.COLON_GT,
56            "!:>": TokenType.NCOLON_GT,
57            "::$": TokenType.DCOLONDOLLAR,
58            "::%": TokenType.DCOLONPERCENT,
59            "::?": TokenType.DCOLONQMARK,
60            "RECORD": TokenType.STRUCT,
61        }
62
63    Parser = SingleStoreParser
64
65    Generator = SingleStoreGenerator

44    class Tokenizer(MySQL.Tokenizer):
45        BYTE_STRINGS = [("e'", "'"), ("E'", "'")]
46
47        KEYWORDS = {
48            **MySQL.Tokenizer.KEYWORDS,
49            "BSON": TokenType.JSONB,
50            "GEOGRAPHYPOINT": TokenType.GEOGRAPHYPOINT,
51            "TIMESTAMP": TokenType.TIMESTAMP,
52            "UTC_DATE": TokenType.UTC_DATE,
53            "UTC_TIME": TokenType.UTC_TIME,
54            "UTC_TIMESTAMP": TokenType.UTC_TIMESTAMP,
55            ":>": TokenType.COLON_GT,
56            "!:>": TokenType.NCOLON_GT,
57            "::$": TokenType.DCOLONDOLLAR,
58            "::%": TokenType.DCOLONPERCENT,
59            "::?": TokenType.DCOLONQMARK,
60            "RECORD": TokenType.STRUCT,
61        }