sqlglot.dialects.postgres

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

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