Feat(presto): add IPADDRESS/IPPREFIX data types (#1965)

* Presto: Add IPADDRESS data type * Linter * Presto: Add IPPREFIX data type * Presto: Add IPPREFIX keyword
tobymao · Jul 27, 2023 · 5e641c2 · 5e641c2
1 parent 9787329
commit 5e641c2
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 0 deletions.
diff --git a/sqlglot/dialects/presto.py b/sqlglot/dialects/presto.py
@@ -192,6 +192,8 @@ class Tokenizer(tokens.Tokenizer):
             "START": TokenType.BEGIN,
             "MATCH_RECOGNIZE": TokenType.MATCH_RECOGNIZE,
             "ROW": TokenType.STRUCT,
+            "IPADDRESS": TokenType.IPADDRESS,
+            "IPPREFIX": TokenType.IPPREFIX,
         }
 
     class Parser(parser.Parser):

diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py
@@ -3377,6 +3377,8 @@ class Type(AutoName):
         HSTORE = auto()
         IMAGE = auto()
         INET = auto()
+        IPADDRESS = auto()
+        IPPREFIX = auto()
         INT = auto()
         INT128 = auto()
         INT256 = auto()

diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -185,6 +185,8 @@ class Parser(metaclass=_Parser):
         TokenType.VARIANT,
         TokenType.OBJECT,
         TokenType.INET,
+        TokenType.IPADDRESS,
+        TokenType.IPPREFIX,
         TokenType.ENUM,
         *NESTED_TYPE_TOKENS,
     }

diff --git a/sqlglot/tokens.py b/sqlglot/tokens.py
@@ -147,6 +147,8 @@ class TokenType(AutoName):
     VARIANT = auto()
     OBJECT = auto()
     INET = auto()
+    IPADDRESS = auto()
+    IPPREFIX = auto()
     ENUM = auto()
 
     # keywords

diff --git a/tests/dialects/test_presto.py b/tests/dialects/test_presto.py
@@ -109,6 +109,8 @@ def test_cast(self):
                 "spark": "CAST(x AS TIMESTAMP)",
             },
         )
+        self.validate_identity("CAST(x AS IPADDRESS)")
+        self.validate_identity("CAST(x AS IPPREFIX)")
 
     def test_regex(self):
         self.validate_all(