sqlglot.dialects.postgres

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

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