sqlglot.dialects.postgres

 12class Postgres(Dialect):
 13    EXPRESSION_METADATA = EXPRESSION_METADATA.copy()
 14    INDEX_OFFSET = 1
 15    ASCII_ONLY_NORMALIZATION = True
 16    # Normalizing `x IS NOT NULL` to `NOT x IS NULL` is unsafe due to row values,
 17    # e.g. `ROW(1, NULL) IS NOT NULL` is false whereas `NOT ROW(1, NULL) IS NULL` is true
 18    NORMALIZE_NOT_NULL = False
 19    TYPED_DIVISION = True
 20    CONCAT_COALESCE = True
 21    CONCAT_WS_COALESCE = True
 22    NULL_ORDERING = "nulls_are_large"
 23    SUPPORTS_LIMIT_ALL = True
 24    TIME_FORMAT = "'YYYY-MM-DD HH24:MI:SS'"
 25    TABLESAMPLE_SIZE_IS_PERCENT = True
 26    TABLES_REFERENCEABLE_AS_COLUMNS = True
 27
 28    DEFAULT_FUNCTIONS_COLUMN_NAMES = {
 29        exp.ExplodingGenerateSeries: "generate_series",
 30    }
 31
 32    TIME_MAPPING = {
 33        "d": "%u",  # 1-based day of week
 34        "D": "%u",  # 1-based day of week
 35        "dd": "%d",  # day of month
 36        "DD": "%d",  # day of month
 37        "ddd": "%j",  # zero padded day of year
 38        "DDD": "%j",  # zero padded day of year
 39        "FMDD": "%-d",  # - is no leading zero for Python; same for FM in postgres
 40        "FMDDD": "%-j",  # day of year
 41        "FMHH12": "%-I",  # 9
 42        "FMHH24": "%-H",  # 9
 43        "FMMI": "%-M",  # Minute
 44        "FMMM": "%-m",  # 1
 45        "FMSS": "%-S",  # Second
 46        "HH12": "%I",  # 09
 47        "HH24": "%H",  # 09
 48        "mi": "%M",  # zero padded minute
 49        "MI": "%M",  # zero padded minute
 50        "mm": "%m",  # 01
 51        "MM": "%m",  # 01
 52        "OF": "%z",  # utc offset
 53        "ss": "%S",  # zero padded second
 54        "SS": "%S",  # zero padded second
 55        "TMDay": "%A",  # TM is locale dependent
 56        "TMDy": "%a",
 57        "TMMon": "%b",  # Sep
 58        "TMMonth": "%B",  # September
 59        "day": "%Aenlower",  # tuesday
 60        "dy": "%aenlower",  # tue
 61        "TZ": "%Z",  # uppercase timezone name
 62        "US": "%f",  # zero padded microsecond
 63        "ww": "%U",  # 1-based week of year
 64        "WW": "%U",  # 1-based week of year
 65        "yy": "%y",  # 15
 66        "YY": "%y",  # 15
 67        "yyy": "%Ythree",  # 015
 68        "YYY": "%Ythree",  # 015
 69        "yyyy": "%Y",  # 2015
 70        "YYYY": "%Y",  # 2015
 71    }
 72
 73    class Tokenizer(tokens.Tokenizer):
 74        BIT_STRINGS = [("b'", "'"), ("B'", "'")]
 75        HEX_STRINGS = [("x'", "'"), ("X'", "'")]
 76        BYTE_STRINGS = [("e'", "'"), ("E'", "'")]
 77        UNICODE_STRINGS = [("U&'", "'"), ("u&'", "'")]
 78        BYTE_STRING_ESCAPES = ["'", "\\"]
 79        HEREDOC_STRINGS = ["$"]
 80
 81        HEREDOC_TAG_IS_IDENTIFIER = True
 82        HEREDOC_STRING_ALTERNATIVE = TokenType.PARAMETER
 83
 84        COMMANDS = {*tokens.Tokenizer.COMMANDS, TokenType.LOCK}
 85
 86        KEYWORDS = {
 87            **tokens.Tokenizer.KEYWORDS,
 88            "~": TokenType.RLIKE,
 89            "@@": TokenType.DAT,
 90            "@?": TokenType.AT_QMARK,
 91            "@>": TokenType.AT_GT,
 92            "<@": TokenType.LT_AT,
 93            "?&": TokenType.QMARK_AMP,
 94            "?|": TokenType.QMARK_PIPE,
 95            "#-": TokenType.HASH_DASH,
 96            "|/": TokenType.PIPE_SLASH,
 97            "||/": TokenType.DPIPE_SLASH,
 98            "BEGIN": TokenType.BEGIN,
 99            "BIGSERIAL": TokenType.BIGSERIAL,
100            "CSTRING": TokenType.PSEUDO_TYPE,
101            "DECLARE": TokenType.COMMAND,
102            "DO": TokenType.COMMAND,
103            "EXEC": TokenType.COMMAND,
104            "HSTORE": TokenType.HSTORE,
105            "INT8": TokenType.BIGINT,
106            "MONEY": TokenType.MONEY,
107            "NAME": TokenType.NAME,
108            "OID": TokenType.OBJECT_IDENTIFIER,
109            "ONLY": TokenType.ONLY,
110            "POINT": TokenType.POINT,
111            "REFRESH": TokenType.COMMAND,
112            "REINDEX": TokenType.COMMAND,
113            "RESET": TokenType.COMMAND,
114            "SERIAL": TokenType.SERIAL,
115            "SMALLSERIAL": TokenType.SMALLSERIAL,
116            "TEMP": TokenType.TEMPORARY,
117            "TYPE": TokenType.TYPE,
118            "REGCLASS": TokenType.OBJECT_IDENTIFIER,
119            "REGCOLLATION": TokenType.OBJECT_IDENTIFIER,
120            "REGCONFIG": TokenType.OBJECT_IDENTIFIER,
121            "REGDICTIONARY": TokenType.OBJECT_IDENTIFIER,
122            "REGNAMESPACE": TokenType.OBJECT_IDENTIFIER,
123            "REGOPER": TokenType.OBJECT_IDENTIFIER,
124            "REGOPERATOR": TokenType.OBJECT_IDENTIFIER,
125            "REGPROC": TokenType.OBJECT_IDENTIFIER,
126            "REGPROCEDURE": TokenType.OBJECT_IDENTIFIER,
127            "REGROLE": TokenType.OBJECT_IDENTIFIER,
128            "REGTYPE": TokenType.OBJECT_IDENTIFIER,
129            "FLOAT": TokenType.DOUBLE,
130            "XML": TokenType.XML,
131            "VARIADIC": TokenType.VARIADIC,
132            "INOUT": TokenType.INOUT,
133        }
134        KEYWORDS.pop("/*+")
135        KEYWORDS.pop("DIV")
136
137        SINGLE_TOKENS = {
138            **tokens.Tokenizer.SINGLE_TOKENS,
139            "$": TokenType.HEREDOC_STRING,
140        }
141
142        VAR_SINGLE_TOKENS = {"$"}
143
144    Parser = PostgresParser
145
146    Generator = PostgresGenerator

 73    class Tokenizer(tokens.Tokenizer):
 74        BIT_STRINGS = [("b'", "'"), ("B'", "'")]
 75        HEX_STRINGS = [("x'", "'"), ("X'", "'")]
 76        BYTE_STRINGS = [("e'", "'"), ("E'", "'")]
 77        UNICODE_STRINGS = [("U&'", "'"), ("u&'", "'")]
 78        BYTE_STRING_ESCAPES = ["'", "\\"]
 79        HEREDOC_STRINGS = ["$"]
 80
 81        HEREDOC_TAG_IS_IDENTIFIER = True
 82        HEREDOC_STRING_ALTERNATIVE = TokenType.PARAMETER
 83
 84        COMMANDS = {*tokens.Tokenizer.COMMANDS, TokenType.LOCK}
 85
 86        KEYWORDS = {
 87            **tokens.Tokenizer.KEYWORDS,
 88            "~": TokenType.RLIKE,
 89            "@@": TokenType.DAT,
 90            "@?": TokenType.AT_QMARK,
 91            "@>": TokenType.AT_GT,
 92            "<@": TokenType.LT_AT,
 93            "?&": TokenType.QMARK_AMP,
 94            "?|": TokenType.QMARK_PIPE,
 95            "#-": TokenType.HASH_DASH,
 96            "|/": TokenType.PIPE_SLASH,
 97            "||/": TokenType.DPIPE_SLASH,
 98            "BEGIN": TokenType.BEGIN,
 99            "BIGSERIAL": TokenType.BIGSERIAL,
100            "CSTRING": TokenType.PSEUDO_TYPE,
101            "DECLARE": TokenType.COMMAND,
102            "DO": TokenType.COMMAND,
103            "EXEC": TokenType.COMMAND,
104            "HSTORE": TokenType.HSTORE,
105            "INT8": TokenType.BIGINT,
106            "MONEY": TokenType.MONEY,
107            "NAME": TokenType.NAME,
108            "OID": TokenType.OBJECT_IDENTIFIER,
109            "ONLY": TokenType.ONLY,
110            "POINT": TokenType.POINT,
111            "REFRESH": TokenType.COMMAND,
112            "REINDEX": TokenType.COMMAND,
113            "RESET": TokenType.COMMAND,
114            "SERIAL": TokenType.SERIAL,
115            "SMALLSERIAL": TokenType.SMALLSERIAL,
116            "TEMP": TokenType.TEMPORARY,
117            "TYPE": TokenType.TYPE,
118            "REGCLASS": TokenType.OBJECT_IDENTIFIER,
119            "REGCOLLATION": TokenType.OBJECT_IDENTIFIER,
120            "REGCONFIG": TokenType.OBJECT_IDENTIFIER,
121            "REGDICTIONARY": TokenType.OBJECT_IDENTIFIER,
122            "REGNAMESPACE": TokenType.OBJECT_IDENTIFIER,
123            "REGOPER": TokenType.OBJECT_IDENTIFIER,
124            "REGOPERATOR": TokenType.OBJECT_IDENTIFIER,
125            "REGPROC": TokenType.OBJECT_IDENTIFIER,
126            "REGPROCEDURE": TokenType.OBJECT_IDENTIFIER,
127            "REGROLE": TokenType.OBJECT_IDENTIFIER,
128            "REGTYPE": TokenType.OBJECT_IDENTIFIER,
129            "FLOAT": TokenType.DOUBLE,
130            "XML": TokenType.XML,
131            "VARIADIC": TokenType.VARIADIC,
132            "INOUT": TokenType.INOUT,
133        }
134        KEYWORDS.pop("/*+")
135        KEYWORDS.pop("DIV")
136
137        SINGLE_TOKENS = {
138            **tokens.Tokenizer.SINGLE_TOKENS,
139            "$": TokenType.HEREDOC_STRING,
140        }
141
142        VAR_SINGLE_TOKENS = {"$"}